35 lines
1.7 KiB
TypeScript
35 lines
1.7 KiB
TypeScript
import type { TeacherToolActivity } from '../../shared/coding-teacher';
|
|
|
|
export interface ToolActivityUpdate {
|
|
id: string;
|
|
name?: string;
|
|
status: TeacherToolActivity['status'];
|
|
}
|
|
|
|
function object(value: unknown): Record<string, unknown> | undefined {
|
|
return value && typeof value === 'object' && !Array.isArray(value) ? value as Record<string, unknown> : undefined;
|
|
}
|
|
|
|
/** Project Yuxi's typed tool events; arguments, outputs and internal errors never enter the UI. */
|
|
export function cloudToolActivity(chunk: Record<string, unknown>, requestId: string): ToolActivityUpdate | undefined {
|
|
const event = object(chunk.stream_event);
|
|
if (event?.type === 'tool_call' || event?.type === 'tool_call_delta') {
|
|
if (typeof event.tool_call_id !== 'string' || !event.tool_call_id || typeof event.name !== 'string' || !event.name) return;
|
|
return { id: requestId + ':' + event.tool_call_id, name: event.name, status: 'running' };
|
|
}
|
|
const custom = object(chunk.event);
|
|
if (chunk.status !== 'stream_event' || custom?.method !== 'tools') return;
|
|
const data = object(custom.data);
|
|
if (!data || typeof data.tool_call_id !== 'string' || !data.tool_call_id) return;
|
|
// A raw tool-error can be a checkpoint interrupt. Only a tool result or the
|
|
// question's terminal state establishes failure, as in Yuxi's tool audit.
|
|
if (data.event !== 'tool-started' && data.event !== 'tool-finished') return;
|
|
const output = object(data.output);
|
|
return {
|
|
id: requestId + ':' + data.tool_call_id,
|
|
...(typeof data.tool_name === 'string' && data.tool_name ? { name: data.tool_name } : {}),
|
|
status: data.event === 'tool-started' ? 'running'
|
|
: data.error || output?.status === 'error' || output?.status === 'failed' ? 'failed' : 'completed',
|
|
};
|
|
}
|