From b9cece767d1abccd06fb95cab759afd06cc2c1e1 Mon Sep 17 00:00:00 2001 From: joshualitt Date: Tue, 19 Aug 2025 13:55:06 -0700 Subject: feat(core): Cleanup after migrating tools. (#6199) Co-authored-by: Jacob Richman --- packages/core/src/test-utils/tools.ts | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'packages/core/src/test-utils') 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); -- cgit v1.2.3