@dyyz1993/pi-coding-agent 0.70.2 → 0.70.3

package/dist/core/agent-session.js

@@ -1656,6 +1656,7 @@ export class AgentSession {
         this._applyExtensionBindings(this._extensionRunner);
         await this._extensionRunner.emit(this._sessionStartEvent);
         await this.extendResourcesFromExtensions(this._sessionStartEvent.reason === "reload" ? "reload" : "startup");
+        await this._initMcpServers();
     }
     async extendResourcesFromExtensions(reason) {
         if (!this._extensionRunner.hasHandlers("resources_discover")) {
@@ -1697,6 +1698,48 @@ export class AgentSession {
         const name = base.replace(/\.(ts|js)$/, "");
         return `extension:${name}`;
     }
+    _mcpManager;
+    _createMcpToolDef;
+    async _initMcpServers() {
+        const settings = this.settingsManager.getProjectSettings();
+        const servers = settings?.mcp?.servers;
+        if (!servers || Object.keys(servers).length === 0)
+            return;
+        const { McpManager } = await import("./mcp/mcp-manager.js");
+        const { createMcpToolDefinition } = await import("./mcp/tool-converter.js");
+        this._createMcpToolDef = createMcpToolDefinition;
+        const mcpOptions = settings?.mcp?.options;
+        const manager = new McpManager({
+            ...mcpOptions,
+            onConnectionChange: () => {
+                this._syncMcpTools();
+            },
+        });
+        this._mcpManager = manager;
+        try {
+            await manager.connectAll(servers);
+            this._syncMcpTools();
+        }
+        catch (e) {
+            console.error("[mcp] Failed to initialize:", e);
+        }
+        this.sessionSignal.addEventListener("abort", () => {
+            this._mcpManager?.dispose().catch(() => { });
+            this._mcpManager = undefined;
+        });
+    }
+    _syncMcpTools() {
+        this._customTools = this._customTools.filter((t) => !t.name.startsWith("mcp__"));
+        if (!this._createMcpToolDef || !this._mcpManager)
+            return;
+        const tools = this._mcpManager.getAllTools();
+        for (const tool of tools) {
+            this._customTools.push(this._createMcpToolDef(tool, this._mcpManager));
+        }
+        if (tools.length > 0) {
+            this._refreshToolRegistry();
+        }
+    }
     _applyExtensionBindings(runner) {
         runner.setUIContext(this._extensionUIContext);
         runner.bindCommandContext(this._extensionCommandContextActions);