@openeryc/pi-coding-agent 0.75.54 → 0.75.56

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,
@@ -1148,6 +1130,7 @@ export class AgentSession {
         this.agent.state.model = model;
         this.sessionManager.appendModelChange(model.provider, model.id);
         this.settingsManager.setDefaultModelAndProvider(model.provider, model.id);
+        this.settingsManager.pushRecentModel(model.provider, model.id);
         // Re-clamp thinking level for new model's capabilities
         this.setThinkingLevel(thinkingLevel);
         await this._emitModelSelect(model, previousModel, "set");
@@ -1180,6 +1163,7 @@ export class AgentSession {
         this.agent.state.model = next.model;
         this.sessionManager.appendModelChange(next.model.provider, next.model.id);
         this.settingsManager.setDefaultModelAndProvider(next.model.provider, next.model.id);
+        this.settingsManager.pushRecentModel(next.model.provider, next.model.id);
         // Apply thinking level.
         // - Explicit scoped model thinking level overrides current session level
         // - Undefined scoped model thinking level inherits the current session preference
@@ -1203,6 +1187,7 @@ export class AgentSession {
         this.agent.state.model = nextModel;
         this.sessionManager.appendModelChange(nextModel.provider, nextModel.id);
         this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
+        this.settingsManager.pushRecentModel(nextModel.provider, nextModel.id);
         // Re-clamp thinking level for new model's capabilities
         this.setThinkingLevel(thinkingLevel);
         await this._emitModelSelect(nextModel, currentModel, "cycle");