@monotykamary/pi-retry 0.3.6 → 0.3.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@monotykamary/pi-retry",
-  "version": "0.3.6",
+  "version": "0.3.8",
   "description": "Extension suite for pi coding agent that handles 400/413 errors and connection errors with automatic retry",
   "type": "module",
   "author": "Tom X Nguyen",

package/retry.ts

@@ -6,6 +6,7 @@ import {
   hasConnectionError,
   hasRetryableError,
   isNonRetryableError,
+  isSilencedError,
   hasMaxTokensStop,
   isAssistantMessage,
   getLastAssistantMessage,
@@ -191,7 +192,8 @@ export default function (pi: ExtensionAPI) {
     }
 
     // Log non-retryable errors so the user knows why we didn't retry
-    if (isNonRetryableError(lastAssistant)) {
+    // (silenced errors are neither retried nor shown)
+    if (isNonRetryableError(lastAssistant) && !isSilencedError(lastAssistant)) {
       const errorMsg = lastAssistant.errorMessage || "Unknown error";
       ctx.ui.notify(`Non-retryable error (not retried): ${errorMsg.substring(0, 100)}`, "error");
     }
@@ -355,10 +357,16 @@ export default function (pi: ExtensionAPI) {
       // finishRun() after agent_end listeners return).
       await _agent.waitForIdle();
 
-      // Fire-and-forget with .catch() as final safety net.
-      // The continue() monkey-patch means the session can't collide
-      // with us even if it tries — but .catch() is belt-and-suspenders.
-      _agent.prompt([]).catch(() => {});
+      try {
+        // Await so _continueInProgress stays true for the full retry.
+        // The session's continue() is blocked (monkey-patch) and the
+        // session's _runAgentPrompt stays alive, keeping the UI
+        // "Working…" until the agent is actually done.
+        await _agent.prompt([]);
+      } catch {
+        // Ignore — if prompt throws, something else is driving.
+        // The session will handle it or report the error.
+      }
     } finally {
       _continueInProgress = false;
     }

package/src/error-patterns.ts

@@ -75,6 +75,12 @@ const NON_RETRYABLE_PATTERNS = [
   /no\s*such\s*model/i,
   /model\s*does\s*not\s*exist/i,
   /unsupported\s*model/i,
+  /cannot continue from message role/i,
+];
+
+// Errors that are non-retryable AND should be silently ignored (no notification)
+const SILENCED_PATTERNS = [
+  /cannot continue from message role/i,
 ];
 
 // ── Type guard ──
@@ -124,6 +130,17 @@ export function isNonRetryableError(message: AgentMessage): boolean {
   return NON_RETRYABLE_PATTERNS.some(p => p.test(message.errorMessage));
 }
 
+/**
+ * Returns true for errors that are non-retryable and should be silently
+ * ignored (no UI notification). These are provider-level refusals that
+ * the user cannot act on and that would only add noise.
+ */
+export function isSilencedError(message: AgentMessage): boolean {
+  if (!isAssistantMessage(message)) return false;
+  if (message.stopReason !== "error" || !message.errorMessage) return false;
+  return SILENCED_PATTERNS.some(p => p.test(message.errorMessage));
+}
+
 // ── Categorisation (for UI messages) ──
 
 export function getErrorCategory(errorMessage: string): '400-413' | 'credit' | 'connection' | 'builtin' | 'other' {