@mariozechner/pi-coding-agent 0.50.1 → 0.50.2

package/dist/core/agent-session.js

@@ -177,6 +177,18 @@ export class AgentSession {
             // Track assistant message for auto-compaction (checked on agent_end)
             if (event.message.role === "assistant") {
                 this._lastAssistantMessage = event.message;
+                // Reset retry counter immediately on successful assistant response
+                // This prevents accumulation across multiple LLM calls within a turn
+                const assistantMsg = event.message;
+                if (assistantMsg.stopReason !== "error" && this._retryAttempt > 0) {
+                    this._emit({
+                        type: "auto_retry_end",
+                        success: true,
+                        attempt: this._retryAttempt,
+                    });
+                    this._retryAttempt = 0;
+                    this._resolveRetry();
+                }
             }
         }
         // Check auto-retry and auto-compaction after agent completes
@@ -189,17 +201,6 @@ export class AgentSession {
                 if (didRetry)
                     return; // Retry was initiated, don't proceed to compaction
             }
-            else if (this._retryAttempt > 0) {
-                // Previous retry succeeded - emit success event and reset counter
-                this._emit({
-                    type: "auto_retry_end",
-                    success: true,
-                    attempt: this._retryAttempt,
-                });
-                this._retryAttempt = 0;
-                // Resolve the retry promise so waitForRetry() completes
-                this._resolveRetry();
-            }
             await this._checkCompaction(msg);
         }
     };
@@ -477,7 +478,7 @@ export class AgentSession {
         // Validate model
         if (!this.model) {
             throw new Error("No model selected.\n\n" +
-                `Use /login or set an API key environment variable. See ${join(getDocsPath(), "authentication.md")}\n\n` +
+                `Use /login or set an API key environment variable. See ${join(getDocsPath(), "providers.md")}\n\n` +
                 "Then use /model to select a model.");
         }
         // Validate API key
@@ -490,7 +491,7 @@ export class AgentSession {
                     `Run '/login ${this.model.provider}' to re-authenticate.`);
             }
             throw new Error(`No API key found for ${this.model.provider}.\n\n` +
-                `Use /login or set an API key environment variable. See ${join(getDocsPath(), "authentication.md")}`);
+                `Use /login or set an API key environment variable. See ${join(getDocsPath(), "providers.md")}`);
         }
         // Check if we need to compact before sending (catches aborted responses)
         const lastAssistant = this._findLastAssistantMessage();