🐛 fix: sync hub tool cache and map

src/main/mcpServers/hub/index.ts

@@ -4,7 +4,7 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js'
 import { CallToolRequestSchema, ErrorCode, ListToolsRequestSchema, McpError } from '@modelcontextprotocol/sdk/types.js'
 
 import { generateToolFunction } from './generator'
-import { callMcpTool, listAllTools } from './mcp-bridge'
+import { callMcpTool, clearToolMap, listAllTools, syncToolMapFromGeneratedTools } from './mcp-bridge'
 import { Runtime } from './runtime'
 import { searchTools } from './search'
 import type { ExecInput, GeneratedTool, SearchQuery } from './types'
@@ -124,6 +124,7 @@ export class HubServer {
     const cached = CacheService.get<GeneratedTool[]>(TOOLS_CACHE_KEY)
     if (cached) {
       logger.debug('Returning cached tools')
+      syncToolMapFromGeneratedTools(cached)
       return cached
     }
 
@@ -132,11 +133,13 @@ export class HubServer {
     const existingNames = new Set<string>()
     const tools = allTools.map((tool) => generateToolFunction(tool, existingNames, callMcpTool))
     CacheService.set(TOOLS_CACHE_KEY, tools, TOOLS_CACHE_TTL)
+    syncToolMapFromGeneratedTools(tools)
     return tools
   }
 
   invalidateCache(): void {
     CacheService.remove(TOOLS_CACHE_KEY)
+    clearToolMap()
     logger.debug('Tools cache invalidated')
   }
 

src/main/mcpServers/hub/mcp-bridge.ts

@@ -4,6 +4,9 @@
  */
 import mcpService from '@main/services/MCPService'
 import { generateMcpToolFunctionName } from '@shared/mcp'
+import type { MCPTool } from '@types'
+
+import type { GeneratedTool } from './types'
 
 export const listAllTools = () => mcpService.listAllActiveServerTools()
 
@@ -11,6 +14,10 @@ const toolFunctionNameToIdMap = new Map<string, { serverId: string; toolName: st
 
 export async function refreshToolMap(): Promise<void> {
   const tools = await listAllTools()
+  syncToolMapFromTools(tools)
+}
+
+export function syncToolMapFromTools(tools: MCPTool[]): void {
   toolFunctionNameToIdMap.clear()
   const existingNames = new Set<string>()
   for (const tool of tools) {
@@ -19,6 +26,17 @@ export async function refreshToolMap(): Promise<void> {
   }
 }
 
+export function syncToolMapFromGeneratedTools(tools: GeneratedTool[]): void {
+  toolFunctionNameToIdMap.clear()
+  for (const tool of tools) {
+    toolFunctionNameToIdMap.set(tool.functionName, { serverId: tool.serverId, toolName: tool.toolName })
+  }
+}
+
+export function clearToolMap(): void {
+  toolFunctionNameToIdMap.clear()
+}
+
 export const callMcpTool = async (functionName: string, params: unknown, callId?: string): Promise<unknown> => {
   const toolInfo = toolFunctionNameToIdMap.get(functionName)
   if (!toolInfo) {