@dyyz1993/pi-coding-agent 0.74.10 → 0.74.12

package/dist/core/agent-session.js

@@ -121,6 +121,7 @@ export class AgentSession {
     _fileSnapshotManager = null;
     _mcpManager;
     _mcpToolDefinitions = new Map();
+    _mcpServerScopes = new Map();
     _noMcp;
     _tierModels = {};
     // Tool registry for extension getTools/setTools
@@ -182,10 +183,16 @@ export class AgentSession {
             this._mcpManager.dispose().catch(() => { });
             this._mcpManager = undefined;
         }
+        this._mcpServerScopes.clear();
+        const globalServers = this.settingsManager.getMcpServers("global");
+        const projectServers = this.settingsManager.getMcpServers("project");
         const settings = this.settingsManager.getMergedSettings();
         const servers = settings?.mcp?.servers;
         if (!servers || Object.keys(servers).length === 0)
             return;
+        for (const name of Object.keys(servers)) {
+            this._mcpServerScopes.set(name, projectServers[name] ? "project" : "global");
+        }
         this._mcpManager = new McpManager({
             onConnectionChange: (conn) => {
                 this._emit({
@@ -521,6 +528,7 @@ export class AgentSession {
                 toolCallId: event.toolCallId,
                 toolName: event.toolName,
                 args: event.args,
+                timestamp: event.timestamp,
             };
             await this._extensionRunner.emit(extensionEvent);
         }
@@ -541,6 +549,8 @@ export class AgentSession {
                 toolName: event.toolName,
                 result: event.result,
                 isError: event.isError,
+                timestamp: event.timestamp,
+                durationMs: event.durationMs,
             };
             await this._extensionRunner.emit(extensionEvent);
         }
@@ -603,8 +613,24 @@ export class AgentSession {
                 fullName: t.fullName,
                 description: t.description,
             })),
+            scope: this._mcpServerScopes.get(c.name) ?? "global",
+            disabled: c.config.disabled,
         }));
     }
+    async toggleMcpServer(name, enabled) {
+        if (!this._mcpManager)
+            throw new Error("MCP is not initialized");
+        const scope = this._mcpServerScopes.get(name);
+        if (!scope)
+            throw new Error(`MCP server "${name}" not found`);
+        await this._mcpManager.setServerEnabled(name, enabled);
+        this.settingsManager.setMcpServerDisabled(name, !enabled, scope);
+    }
+    async restartMcpServer(name) {
+        if (!this._mcpManager)
+            throw new Error("MCP is not initialized");
+        await this._mcpManager.restartServer(name);
+    }
     // =========================================================================
     // Read-only State Access
     // =========================================================================