diff options
| author | Wanlin Du <[email protected]> | 2025-07-16 14:32:34 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-07-16 21:32:34 +0000 |
| commit | f6ee0d182bcb6c9c6e3c9d34174d8d984bbddb37 (patch) | |
| tree | cdfe052e9124b21a3567d3b495882dec57664f11 /packages/core/src/tools/mcp-client.ts | |
| parent | 21eb44b242e045cad957d21b049d00b55abf9489 (diff) | |
fix: update google/genai to v1.9.0 and switch to parametersJsonSchema for MCP related tools (#4176)
Co-authored-by: Jack Wotherspoon <[email protected]>
Diffstat (limited to 'packages/core/src/tools/mcp-client.ts')
| -rw-r--r-- | packages/core/src/tools/mcp-client.ts | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/packages/core/src/tools/mcp-client.ts b/packages/core/src/tools/mcp-client.ts index beb70549..24c975c3 100644 --- a/packages/core/src/tools/mcp-client.ts +++ b/packages/core/src/tools/mcp-client.ts @@ -18,8 +18,9 @@ import { import { parse } from 'shell-quote'; import { MCPServerConfig } from '../config/config.js'; import { DiscoveredMCPTool } from './mcp-tool.js'; -import { FunctionDeclaration, Type, mcpToTool } from '@google/genai'; -import { sanitizeParameters, ToolRegistry } from './tool-registry.js'; + +import { FunctionDeclaration, mcpToTool } from '@google/genai'; +import { ToolRegistry } from './tool-registry.js'; import { ActiveFileNotificationSchema, IDE_SERVER_NAME, @@ -275,15 +276,13 @@ export async function discoverTools( const toolNameForModel = generateValidName(funcDecl, mcpServerName); - sanitizeParameters(funcDecl.parameters); - discoveredTools.push( new DiscoveredMCPTool( mcpCallableTool, mcpServerName, toolNameForModel, funcDecl.description ?? '', - funcDecl.parameters ?? { type: Type.OBJECT, properties: {} }, + funcDecl.parametersJsonSchema ?? { type: 'object', properties: {} }, funcDecl.name!, mcpServerConfig.timeout ?? MCP_DEFAULT_TIMEOUT_MSEC, mcpServerConfig.trust, |