@averyyy/pi-coding-agent 0.84.4-piclient.1 → 0.85.1-piclient.1

package/dist/core/agent-session.js

@@ -102,6 +102,16 @@ function getPiServerFailurePhase(message) {
     }
     return undefined;
 }
+function getPiServerFailureRetryable(message) {
+    for (const diagnostic of message.diagnostics ?? []) {
+        if (diagnostic.type !== "pi_server_failure")
+            continue;
+        const retryable = diagnostic.details?.retryable;
+        if (typeof retryable === "boolean")
+            return retryable;
+    }
+    return undefined;
+}
 function toolTextContent(content) {
     return content
         .filter((part) => part.type === "text")
@@ -488,7 +498,7 @@ export class AgentSession {
         return this._idleWaitPromise;
     }
     _resolveIdleWaitIfIdle() {
-        if (this._isAgentRunActive || !this._resolveIdleWait) {
+        if (!this.isIdle || !this._resolveIdleWait) {
             return;
         }
         const resolve = this._resolveIdleWait;
@@ -824,9 +834,9 @@ export class AgentSession {
     get isStreaming() {
         return this._isAgentRunActive;
     }
-    /** Whether the session has no active agent run, retry, auto-compaction, or queued continuation. */
+    /** Whether the session has no active agent run, compaction, branch summary, retry, or queued continuation. */
     get isIdle() {
-        return !this._isAgentRunActive;
+        return !this._isAgentRunActive && !this.isCompacting;
     }
     /** Current effective system prompt (includes any per-turn extension modifications) */
     get systemPrompt() {
@@ -1518,6 +1528,8 @@ export class AgentSession {
      */
     async abort() {
         this.abortRetry();
+        this.abortCompaction();
+        this.abortBranchSummary();
         this.agent.abort();
         await this.waitForIdle();
     }
@@ -1774,6 +1786,10 @@ export class AgentSession {
         }
         return compactionResult;
     }
+    _clearManualCompactionState() {
+        this._compactionAbortController = undefined;
+        this._resolveIdleWaitIfIdle();
+    }
     /**
      * Manually compact the session context.
      * Aborts current agent operation first.
@@ -1897,7 +1913,7 @@ export class AgentSession {
                 details,
             };
             // compaction_end listeners may submit queued prompts, so expose idle state before notifying them.
-            this._compactionAbortController = undefined;
+            this._clearManualCompactionState();
             this._emit({
                 type: "compaction_end",
                 reason: "manual",
@@ -1911,18 +1927,20 @@ export class AgentSession {
             const message = error instanceof Error ? error.message : String(error);
             const aborted = message === "Compaction cancelled" || (error instanceof Error && error.name === "AbortError");
             this._compactionAbortController = undefined;
+            const errorMessage = aborted ? undefined : `Compaction failed: ${message}`;
+            this._clearManualCompactionState();
             this._emit({
                 type: "compaction_end",
                 reason: "manual",
                 result: undefined,
                 aborted,
                 willRetry: false,
-                errorMessage: aborted ? undefined : `Compaction failed: ${message}`,
+                errorMessage,
             });
             throw error;
         }
         finally {
-            this._compactionAbortController = undefined;
+            this._clearManualCompactionState();
         }
     }
     /**
@@ -2229,6 +2247,7 @@ export class AgentSession {
         }
         finally {
             this._autoCompactionAbortController = undefined;
+            this._resolveIdleWaitIfIdle();
         }
     }
     /**
@@ -2637,6 +2656,9 @@ export class AgentSession {
      * Check if an error is retryable. Provider error text is retryable by default,
      * except usage/quota/balance limits and session or context failures.
      * Context overflow errors are NOT retryable (handled by compaction instead).
+     * Pi-server diagnostics may mark recovery/history failures as non-retryable;
+     * transient session_init/tree_sync outages stay retryable so the next attempt
+     * re-runs the whole stream path.
      */
     _isRetryableError(message) {
         // Context overflow is handled by compaction, not retry.
@@ -2644,9 +2666,14 @@ export class AgentSession {
             return false;
         if (this._isPostAssistantPiServerSyncFailure(message))
             return false;
-        const piServerPhase = getPiServerFailurePhase(message);
-        if (piServerPhase && piServerPhase !== "provider_stream")
+        const piServerRetryable = getPiServerFailureRetryable(message);
+        if (piServerRetryable === false)
             return false;
+        if (piServerRetryable !== true) {
+            const piServerPhase = getPiServerFailurePhase(message);
+            if (piServerPhase && piServerPhase !== "provider_stream")
+                return false;
+        }
         return isRetryableAssistantError(message);
     }
     /**
@@ -3042,6 +3069,7 @@ export class AgentSession {
         }
         finally {
             this._branchSummaryAbortController = undefined;
+            this._resolveIdleWaitIfIdle();
         }
     }
     /**