@dyyz1993/pi-coding-agent 0.74.8 → 0.74.10

package/dist/core/agent-session.js

@@ -34,6 +34,7 @@ import { emitSessionShutdownEvent } from "./extensions/runner.js";
 import { FileSnapshotManager } from "./file-store/file-snapshot-manager.js";
 import { InternalGit } from "./file-store/internal-git.js";
 import { McpManager } from "./mcp/mcp-manager.js";
+import { createMcpToolDefinition } from "./mcp/tool-converter.js";
 import { resolveModelAlias } from "./model-resolver.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
@@ -119,6 +120,8 @@ export class AgentSession {
     _modelRegistry;
     _fileSnapshotManager = null;
     _mcpManager;
+    _mcpToolDefinitions = new Map();
+    _noMcp;
     _tierModels = {};
     // Tool registry for extension getTools/setTools
     _toolRegistry = new Map();
@@ -142,6 +145,7 @@ export class AgentSession {
         this._allowedToolNames = config.allowedToolNames ? new Set(config.allowedToolNames) : undefined;
         this._baseToolsOverride = config.baseToolsOverride;
         this._sessionStartEvent = config.sessionStartEvent ?? { type: "session_start", reason: "startup" };
+        this._noMcp = config.noMcp ?? false;
         // Always subscribe to agent events for internal handling
         // (session persistence, extensions, auto-compaction, retry logic)
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
@@ -172,6 +176,8 @@ export class AgentSession {
         }
     }
     _initMcpServers() {
+        if (this._noMcp)
+            return;
         if (this._mcpManager) {
             this._mcpManager.dispose().catch(() => { });
             this._mcpManager = undefined;
@@ -193,6 +199,21 @@ export class AgentSession {
                         description: t.description,
                     })),
                 });
+                if (conn.status === "connected" && this._mcpManager) {
+                    this._mcpToolDefinitions.clear();
+                    const allTools = this._mcpManager.getAllTools();
+                    for (const tool of allTools) {
+                        this._mcpToolDefinitions.set(tool.fullName, createMcpToolDefinition(tool, this._mcpManager));
+                    }
+                    this._refreshToolRegistry();
+                }
+                else if (conn.status === "error" || conn.status === "disconnected") {
+                    const staleKeys = [...this._mcpToolDefinitions.keys()].filter((key) => key.startsWith(`mcp__${conn.name}__`));
+                    for (const key of staleKeys) {
+                        this._mcpToolDefinitions.delete(key);
+                    }
+                    this._refreshToolRegistry();
+                }
             },
         });
         this._mcpManager.connectAll(servers).catch(() => { });
@@ -1901,12 +1922,19 @@ export class AgentSession {
         const allowedToolNames = this._allowedToolNames;
         const isAllowedTool = (name) => !allowedToolNames || allowedToolNames.has(name);
         const registeredTools = this._extensionRunner.getAllRegisteredTools();
+        const mcpTools = [...this._mcpToolDefinitions.values()]
+            .filter((def) => isAllowedTool(def.name))
+            .map((definition) => ({
+            definition,
+            sourceInfo: createSyntheticSourceInfo(`<mcp:${definition.name}>`, { source: "mcp" }),
+        }));
         const allCustomTools = [
             ...registeredTools,
             ...this._customTools.map((definition) => ({
                 definition,
                 sourceInfo: createSyntheticSourceInfo(`<sdk:${definition.name}>`, { source: "sdk" }),
             })),
+            ...mcpTools,
         ].filter((tool) => isAllowedTool(tool.definition.name));
         const definitionRegistry = new Map(Array.from(this._baseToolDefinitions.entries())
             .filter(([name]) => isAllowedTool(name))