@openeryc/pi-coding-agent 0.75.49 → 0.75.51

package/dist/core/agent-session.js

@@ -1701,8 +1701,19 @@ export class AgentSession {
             this._mcpManager = null;
         }
         this._mcpManager = new MCPManager();
+        this._mcpManager.onStatusChange = (_serverName, status) => {
+            if (status === "connected") {
+                // A server reconnected — refresh the tool registry
+                this._refreshMcpTools();
+            }
+        };
         await this._mcpManager.start(mcpServers);
-        const mcpTools = this._mcpManager.getToolDefinitions();
+        this._refreshMcpTools();
+    }
+    _refreshMcpTools() {
+        // Remove old MCP tool defs from _customTools
+        this._customTools = this._customTools.filter((t) => !t.name.startsWith("mcp_"));
+        const mcpTools = this._mcpManager?.getToolDefinitions() ?? [];
         if (mcpTools.length > 0) {
             this._customTools.push(...mcpTools);
             this._refreshToolRegistry();
@@ -1712,6 +1723,14 @@ export class AgentSession {
     async reloadMcp() {
         await this._initMcp();
     }
+    /** Get MCP connection statuses (for UI display) */
+    getMcpStatuses() {
+        return this._mcpManager?.getAllStatuses() ?? [];
+    }
+    /** Manually trigger reconnection for a specific MCP server */
+    async reconnectMcp(serverName) {
+        await this._mcpManager?.reconnect(serverName);
+    }
     async extendResourcesFromExtensions(reason) {
         if (!this._extensionRunner.hasHandlers("resources_discover")) {
             return;