diff options
Diffstat (limited to 'packages/core/src/test-utils/tools.ts')
| -rw-r--r-- | packages/core/src/test-utils/tools.ts | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/packages/core/src/test-utils/tools.ts b/packages/core/src/test-utils/tools.ts index 0e3e6b86..7c45f24e 100644 --- a/packages/core/src/test-utils/tools.ts +++ b/packages/core/src/test-utils/tools.ts @@ -13,7 +13,6 @@ import { ToolResult, Kind, } from '../tools/tools.js'; -import { Schema, Type } from '@google/genai'; import { ModifiableDeclarativeTool, ModifyContext, @@ -74,9 +73,9 @@ export class MockTool extends BaseDeclarativeTool< name = 'mock-tool', displayName?: string, description = 'A mock tool for testing.', - params: Schema = { - type: Type.OBJECT, - properties: { param: { type: Type.STRING } }, + params = { + type: 'object', + properties: { param: { type: 'string' } }, }, ) { super(name, displayName ?? name, description, Kind.Other, params); |