@openeryc/pi-coding-agent 0.75.54 → 0.75.55

package/dist/core/agent-session.js

@@ -659,32 +659,14 @@ export class AgentSession {
         const disabledSkills = new Set(this.settingsManager.getDisabledSkills());
         const activeSkills = loadedSkills.filter((s) => !disabledSkills.has(s.name) && !s.disableModelInvocation);
         const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
-        // Collect MCP tools grouped by server name
-        // Uses the label field (mcp.<server>.<tool>) for reliable server name extraction
-        const mcpTools = [];
-        const mcpServerMap = new Map();
-        for (const name of validToolNames) {
-            if (!name.startsWith("mcp_"))
-                continue;
-            const def = this._toolDefinitions.get(name);
-            if (!def)
-                continue;
-            // Extract server name from label (mcp.<server>.<tool>)
-            const label = def.definition.label ?? "";
-            const labelParts = label.split(".");
-            const serverName = labelParts.length >= 3 ? labelParts.slice(1, -1).join(".") : label;
-            const desc = def.definition.description || `Tool from MCP server "${serverName}"`;
-            const existing = mcpServerMap.get(serverName);
-            if (existing) {
-                existing.push({ name, description: desc });
-            }
-            else {
-                mcpServerMap.set(serverName, [{ name, description: desc }]);
-            }
-        }
-        for (const [serverName, tools] of mcpServerMap) {
-            mcpTools.push({ serverName, tools });
-        }
+        // Collect MCP tool descriptions grouped by server
+        const mcpToolNames = new Set(validToolNames.filter((n) => n.startsWith("mcp_")));
+        const mcpTools = (this._mcpManager?.getMcpToolDescriptions() ?? [])
+            .map((server) => ({
+            ...server,
+            tools: server.tools.filter((t) => mcpToolNames.has(t.name)),
+        }))
+            .filter((server) => server.tools.length > 0);
         this._baseSystemPromptOptions = {
             cwd: this._cwd,
             skills: activeSkills,