diff options
| author | joshualitt <[email protected]> | 2025-08-06 10:50:02 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-08-06 17:50:02 +0000 |
| commit | 6133bea388a2de69c71a6be6f1450707f2ce4dfb (patch) | |
| tree | 367de1d618069ea80e47d7e86c4fb8f82ad032a7 /packages/core/src/telemetry/uiTelemetry.test.ts | |
| parent | 882a97aff998b2f19731e9966d135f1db5a59914 (diff) | |
feat(core): Introduce `DeclarativeTool` and `ToolInvocation`. (#5613)
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: { |