@mariozechner/pi-coding-agent 0.58.4 → 0.60.0

package/dist/core/agent-session.js

@@ -1643,6 +1643,17 @@ export class AgentSession {
             ? runner.onError(this._extensionErrorListener)
             : undefined;
     }
+    _refreshCurrentModelFromRegistry() {
+        const currentModel = this.model;
+        if (!currentModel) {
+            return;
+        }
+        const refreshedModel = this._modelRegistry.find(currentModel.provider, currentModel.id);
+        if (!refreshedModel || refreshedModel === currentModel) {
+            return;
+        }
+        this.agent.setModel(refreshedModel);
+    }
     _bindExtensionCore(runner) {
         const normalizeLocation = (source) => {
             if (source === "user" || source === "project" || source === "path") {
@@ -1744,6 +1755,15 @@ export class AgentSession {
                 })();
             },
             getSystemPrompt: () => this.systemPrompt,
+        }, {
+            registerProvider: (name, config) => {
+                this._modelRegistry.registerProvider(name, config);
+                this._refreshCurrentModelFromRegistry();
+            },
+            unregisterProvider: (name) => {
+                this._modelRegistry.unregisterProvider(name);
+                this._refreshCurrentModelFromRegistry();
+            },
         });
     }
     _refreshToolRegistry(options) {
@@ -1756,7 +1776,7 @@ export class AgentSession {
         ];
         this._toolPromptSnippets = new Map(allCustomTools
             .map((registeredTool) => {
-            const snippet = this._normalizePromptSnippet(registeredTool.definition.promptSnippet ?? registeredTool.definition.description);
+            const snippet = this._normalizePromptSnippet(registeredTool.definition.promptSnippet);
             return snippet ? [registeredTool.definition.name, snippet] : undefined;
         })
             .filter((entry) => entry !== undefined));
@@ -1864,8 +1884,8 @@ export class AgentSession {
         if (isContextOverflow(message, contextWindow))
             return false;
         const err = message.errorMessage;
-        // Match: overloaded_error, rate limit, 429, 500, 502, 503, 504, service unavailable, connection errors, fetch failed, terminated, retry delay exceeded
-        return /overloaded|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server.?error|internal.?error|connection.?error|connection.?refused|other side closed|fetch failed|upstream.?connect|reset before headers|terminated|retry delay/i.test(err);
+        // Match: overloaded_error, provider returned error, rate limit, 429, 500, 502, 503, 504, service unavailable, network/connection errors, fetch failed, terminated, retry delay exceeded
+        return /overloaded|provider.?returned.?error|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server.?error|internal.?error|network.?error|connection.?error|connection.?refused|other side closed|fetch failed|upstream.?connect|reset before headers|socket hang up|timed? out|timeout|terminated|retry delay/i.test(err);
     }
     /**
      * Handle retryable errors with exponential backoff.