diff options
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, |