opencode-immune 1.0.3 → 1.0.5

package/dist/plugin.js

@@ -210,6 +210,33 @@ async function setSessionFallbackModel(state, sessionID, model) {
     });
     await writeManagedSessionsCache(state);
 }
+async function forceRetryManagedSession(state, sessionID, reason) {
+    const managedSession = state.managedUltraworkSessions.get(sessionID);
+    if (!managedSession)
+        return;
+    const retryAgent = managedSession.agent || ULTRAWORK_AGENT;
+    const fallbackModel = managedSession.fallbackModel;
+    const retryText = retryAgent === ULTRAWORK_AGENT
+        ? "[SYSTEM: Previous API call failed with a transient error. Re-read memory-bank/tasks.md, check the Phase Status block, and continue the pipeline. Use the exact neutral prompt from your Step 5 table for the next router call. Do NOT analyze or evaluate file contents.]"
+        : `[SYSTEM: Previous API call failed with a transient error. Resume the current task in your current role as ${retryAgent}. Continue from the existing session state. Do not restart from scratch unless the current session state is missing.]`;
+    await state.input.client.session.promptAsync({
+        body: {
+            ...(fallbackModel ? { model: fallbackModel } : {}),
+            agent: retryAgent,
+            parts: [
+                {
+                    type: "text",
+                    text: retryText,
+                },
+            ],
+        },
+        path: { id: sessionID },
+    });
+    console.log(`[opencode-immune] Forced retry sent to session ${sessionID} (${reason})` +
+        (fallbackModel
+            ? ` using fallback model ${fallbackModel.providerID}/${fallbackModel.modelID}`
+            : ""));
+}
 // ═══════════════════════════════════════════════════════════════════════════════
 // UTILITY: ERROR BOUNDARY
 // ═══════════════════════════════════════════════════════════════════════════════
@@ -747,6 +774,7 @@ const MAX_CYCLES = 10;
 const PRE_COMMIT_MARKER = "0-ULTRAWORK: PRE_COMMIT";
 const CYCLE_COMPLETE_MARKER = "0-ULTRAWORK: CYCLE_COMPLETE";
 const NEXT_TASK_PATTERN = /Next task:\s*(.+)/;
+const RATE_LIMIT_MESSAGE_PATTERN = /too many requests|rate_limit|rate limit/i;
 /**
  * chat.message part: scans assistant messages for PRE_COMMIT and CYCLE_COMPLETE markers.
  *
@@ -756,8 +784,6 @@ const NEXT_TASK_PATTERN = /Next task:\s*(.+)/;
 function createMultiCycleHandler(state) {
     return async (input, output) => {
         const sessionID = input.sessionID;
-        if (!isManagedRootUltraworkSession(state, sessionID))
-            return;
         // Extract text content from parts
         const parts = output.parts ?? [];
         let messageContent = "";
@@ -769,6 +795,28 @@ function createMultiCycleHandler(state) {
         messageContent = messageContent.trim();
         if (!messageContent)
             return;
+        if (sessionID && RATE_LIMIT_MESSAGE_PATTERN.test(messageContent)) {
+            const managedSession = getManagedSession(state, sessionID);
+            if (managedSession && !managedSession.fallbackModel) {
+                await setSessionFallbackModel(state, sessionID, RATE_LIMIT_FALLBACK_MODEL);
+                console.log(`[opencode-immune] Rate limit message detected in chat output for session ${sessionID}. ` +
+                    `Fallback model pinned to ${RATE_LIMIT_FALLBACK_MODEL.providerID}/${RATE_LIMIT_FALLBACK_MODEL.modelID}.`);
+            }
+            if (managedSession && !state.retryTimers.has(sessionID)) {
+                const timer = setTimeout(async () => {
+                    state.retryTimers.delete(sessionID);
+                    try {
+                        await forceRetryManagedSession(state, sessionID, "rate-limit message fallback");
+                    }
+                    catch (err) {
+                        console.error(`[opencode-immune] Forced retry after rate-limit message failed for session ${sessionID}:`, err);
+                    }
+                }, 1_000);
+                state.retryTimers.set(sessionID, timer);
+            }
+        }
+        if (!isManagedRootUltraworkSession(state, sessionID))
+            return;
         // ── PRE_COMMIT: execute /commit ──
         if (messageContent.includes(PRE_COMMIT_MARKER) && !state.commitPending) {
             state.commitPending = true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-immune",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "OpenCode plugin: session recovery, auto-retry, multi-cycle automation, context monitoring",
   "exports": {
     "./server": "./dist/plugin.js"