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

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")
@@ -2637,6 +2647,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 +2657,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);
     }
     /**