diff options
Diffstat (limited to 'packages/core/src/telemetry/uiTelemetry.test.ts')
| -rw-r--r-- | packages/core/src/telemetry/uiTelemetry.test.ts | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/packages/core/src/telemetry/uiTelemetry.test.ts b/packages/core/src/telemetry/uiTelemetry.test.ts index 221804d2..ac9727f1 100644 --- a/packages/core/src/telemetry/uiTelemetry.test.ts +++ b/packages/core/src/telemetry/uiTelemetry.test.ts @@ -23,7 +23,8 @@ import { SuccessfulToolCall, } from '../core/coreToolScheduler.js'; import { ToolErrorType } from '../tools/tool-error.js'; -import { Tool, ToolConfirmationOutcome } from '../tools/tools.js'; +import { ToolConfirmationOutcome } from '../tools/tools.js'; +import { MockTool } from '../test-utils/tools.js'; const createFakeCompletedToolCall = ( name: string, @@ -39,12 +40,14 @@ const createFakeCompletedToolCall = ( isClientInitiated: false, prompt_id: 'prompt-id-1', }; + const tool = new MockTool(name); if (success) { return { status: 'success', request, - tool: { name } as Tool, // Mock tool + tool, + invocation: tool.build({}), response: { callId: request.callId, responseParts: { @@ -65,6 +68,7 @@ const createFakeCompletedToolCall = ( return { status: 'error', request, + tool, response: { callId: request.callId, responseParts: { |