@caupulican/pi-adaptative 0.75.6 → 0.76.1

package/dist/core/agent-session.js

@@ -132,6 +132,7 @@ export class AgentSession {
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
         this._installAgentToolHooks();
         this._installAgentContextTransform();
+        this._installAgentTurnRefresh();
         this._buildRuntime({
             activeToolNames: this._initialActiveToolNames,
             includeAllExtensionTools: true,
@@ -202,6 +203,30 @@ export class AgentSession {
             }
         };
     }
+    _installAgentTurnRefresh() {
+        const previousPrepareNextTurn = this.agent.prepareNextTurn?.bind(this.agent);
+        this.agent.prepareNextTurn = async (signal) => {
+            const previous = previousPrepareNextTurn ? await previousPrepareNextTurn(signal) : undefined;
+            const snapshot = this._createAgentContextSnapshot();
+            return {
+                ...previous,
+                context: {
+                    ...(previous?.context ?? snapshot),
+                    systemPrompt: snapshot.systemPrompt,
+                    tools: snapshot.tools,
+                },
+                model: previous?.model ?? this.agent.state.model,
+                thinkingLevel: previous?.thinkingLevel ?? this.agent.state.thinkingLevel,
+            };
+        };
+    }
+    _createAgentContextSnapshot() {
+        return {
+            systemPrompt: this.agent.state.systemPrompt,
+            messages: this.agent.state.messages.slice(),
+            tools: this.agent.state.tools.slice(),
+        };
+    }
     _estimateCurrentContextTokens(messages) {
         const estimate = estimateContextTokens(messages);
         const compactionEntry = getLatestCompactionEntry(this.sessionManager.getBranch());
@@ -1877,12 +1902,9 @@ export class AgentSession {
                 })();
             },
             reload: () => {
-                if (this.isStreaming) {
-                    throw new Error("Cannot reload while the agent is streaming.");
-                }
                 const actions = this._extensionCommandContextActions;
-                if (!actions) {
-                    throw new Error("Reload is unavailable before extension command context is bound.");
+                if (this.isStreaming || !actions) {
+                    return this.reload();
                 }
                 return actions.reload();
             },
@@ -2032,6 +2054,9 @@ export class AgentSession {
     // =========================================================================
     // Auto-Retry
     // =========================================================================
+    _isNonRetryableProviderLimitError(errorMessage) {
+        return /GoUsageLimitError|FreeUsageLimitError|Monthly usage limit reached|available balance|insufficient_quota|out of budget|quota exceeded|billing/i.test(errorMessage);
+    }
     /**
      * Check if an error is retryable (overloaded, rate limit, server errors).
      * Context overflow errors are NOT retryable (handled by compaction instead).
@@ -2044,6 +2069,8 @@ export class AgentSession {
         if (isContextOverflow(message, contextWindow))
             return false;
         const err = message.errorMessage;
+        if (this._isNonRetryableProviderLimitError(err))
+            return false;
         // Match: overloaded_error, provider returned error, rate limit, 429, 500, 502, 503, 504, service unavailable, network/connection errors (including connection lost), WebSocket transport closes/errors, fetch failed, premature stream endings, HTTP/2 closed before response, 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|connection.?lost|websocket.?closed|websocket.?error|other side closed|fetch failed|upstream.?connect|reset before headers|socket hang up|ended without|stream ended before message_stop|http2 request did not get a response|timed? out|timeout|terminated|retry delay/i.test(err);
     }