fix: update google/genai to v1.9.0 and switch to parametersJsonSchema for MCP related tools (#4176)

Co-authored-by: Jack Wotherspoon <jackwoth@google.com>
This commit is contained in:
Wanlin Du
2025-07-16 14:32:34 -07:00
committed by GitHub
parent 21eb44b242
commit f6ee0d182b
7 changed files with 40 additions and 20 deletions

View File

@@ -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,