summaryrefslogtreecommitdiff
path: root/packages/core/src/test-utils/tools.ts
diff options
context:
space:
mode:
authorjoshualitt <[email protected]>2025-08-19 13:55:06 -0700
committerGitHub <[email protected]>2025-08-19 20:55:06 +0000
commitb9cece767d1abccd06fb95cab759afd06cc2c1e1 (patch)
tree8c29a49bbaedd0a706bd2dba82eb4c587797ba7e /packages/core/src/test-utils/tools.ts
parent2143731f6efdf1aafff38ec249caf01a8bcd163e (diff)
feat(core): Cleanup after migrating tools. (#6199)
Co-authored-by: Jacob Richman <[email protected]>
Diffstat (limited to 'packages/core/src/test-utils/tools.ts')
-rw-r--r--packages/core/src/test-utils/tools.ts7
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);