@renjfk/opencode-model-fallback 0.1.1 → 0.2.0

package/README.md

@@ -55,7 +55,6 @@ If you want to set plugin options, use the tuple form:
 - `retry_on_errors`: retryable HTTP status codes. Defaults to `429`.
 - `retryable_error_patterns`: retryable error message patterns. Defaults to `["rate.?limit"]`.
 - `cooldown_seconds`: how long a failed original model remains on fallback. Defaults to `3600`.
-- `timeout_seconds`: abort and retry if a response is inactive for this long. Defaults to `30`.
 - `notify_on_fallback`: show fallback/recovery toasts. Defaults to `true`.
 
 ## How it works
@@ -227,7 +226,7 @@ at the local repo path, not the npm package name:
 
 ```json
 {
-  "plugin": ["/Users/your-user/opencode-model-fallback"]
+  "plugin": ["/Users/your-user/opencode-model-fallback/index.js"]
 }
 ```
 

package/lib/options.js

@@ -3,7 +3,6 @@ const DEFAULT_OPTIONS = {
   retry_on_errors: [429],
   retryable_error_patterns: ["rate.?limit"],
   cooldown_seconds: 3600,
-  timeout_seconds: 30,
   notify_on_fallback: true,
 };
 

package/lib/router.js

@@ -13,9 +13,9 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
   );
   const store = createStateStore();
   const retrying = new Set();
-  const timers = new Map();
   const selfAbortAt = new Map();
   const activeOriginals = new Map();
+  const activeTargets = new Map();
   let agentConfigs;
 
   function hasMapping(model) {
@@ -41,13 +41,6 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
     return cooldown ? fallback : original;
   }
 
-  function timedOriginal(requested) {
-    if (!hasMapping(requested)) return undefined;
-    const cooldown = store.getModelCooldown(requested);
-    if (cooldown) return undefined;
-    return requested;
-  }
-
   function shouldRoute(requested) {
     return hasMapping(requested) || !!fallbackToOriginal[requested];
   }
@@ -65,31 +58,6 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
     return !cooldown;
   }
 
-  function clearTimer(sessionID) {
-    const timer = timers.get(sessionID);
-    if (timer) clearTimeout(timer);
-    timers.delete(sessionID);
-  }
-
-  function scheduleTimeout(sessionID, original, agent) {
-    clearTimer(sessionID);
-    if (options.timeout_seconds <= 0 || !hasMapping(original)) return;
-    timers.set(
-      sessionID,
-      setTimeout(async () => {
-        timers.delete(sessionID);
-        if (retrying.has(sessionID) || !timedOriginal(original)) return;
-        retrying.add(sessionID);
-        try {
-          await abortCurrentSession(sessionID);
-          await retryWithFallback(sessionID, original, agent, "timeout");
-        } finally {
-          retrying.delete(sessionID);
-        }
-      }, options.timeout_seconds * 1000),
-    );
-  }
-
   async function abortCurrentSession(sessionID) {
     const aborted = await abortSession(ctx.client, sessionID);
     if (aborted) selfAbortAt.set(sessionID, Date.now());
@@ -104,7 +72,6 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
     store.setModelCooldown(original, reason, failedAt, cooldownUntil);
     const parts = await getReplayParts(ctx.client, ctx.directory, sessionID);
     if (parts.length === 0) return;
-    clearTimer(sessionID);
     try {
       await new Promise((resolve) => setTimeout(resolve, POST_ABORT_DELAY_MS));
       await ctx.client.session.promptAsync({
@@ -123,6 +90,19 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
       .catch(() => {});
   }
 
+  async function toastRouteChange(sessionID, requested, target, original) {
+    const previous = activeTargets.get(sessionID);
+    activeTargets.set(sessionID, target);
+    if (previous === target) return;
+    if (!previous && requested === target) return;
+    const isFallback = target !== original;
+    await toast(
+      isFallback ? "Model Fallback" : "Model Recovered",
+      previous ? `${previous} -> ${target}` : `Using ${target} instead of ${requested}`,
+      isFallback ? "warning" : "info",
+    );
+  }
+
   async function handleError(sessionID, error, model, agent, source) {
     if (!sessionID || retrying.has(sessionID)) return;
     const name = extractErrorName(error);
@@ -138,7 +118,6 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
     }
     if (!shouldFallbackFromError(failed, original)) return;
     retrying.add(sessionID);
-    clearTimer(sessionID);
     try {
       await retryWithFallback(sessionID, original, agent, source);
     } finally {
@@ -166,7 +145,6 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
     }
     if (!shouldFallbackFromError(failed, original)) return;
     retrying.add(sessionID);
-    clearTimer(sessionID);
     try {
       await abortCurrentSession(sessionID);
       await retryWithFallback(sessionID, original, agent, "session.status");
@@ -197,16 +175,7 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
       if (original) activeOriginals.set(sessionID, original);
       const model = modelObject(target);
       if (model && output.message) output.message.model = model;
-      if (requested !== target) {
-        const isFallback = hasMapping(requested);
-        await toast(
-          isFallback ? "Model Fallback" : "Model Recovered",
-          `Using ${target} instead of ${requested}`,
-          isFallback ? "warning" : "info",
-        );
-      }
-      if (hasMapping(target)) scheduleTimeout(sessionID, target, input.agent);
-      else clearTimer(sessionID);
+      await toastRouteChange(sessionID, requested, target, original);
     },
 
     event: async ({ event }) => {
@@ -216,7 +185,7 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
         if (id) {
           retrying.delete(id);
           activeOriginals.delete(id);
-          clearTimer(id);
+          activeTargets.delete(id);
         }
         return;
       }
@@ -237,11 +206,8 @@ export function createMappedFallbackRouter(ctx, rawOptions) {
       if (event.type === "message.updated") {
         const info = props?.info;
         if (info?.role !== "assistant") return;
+        if (!info?.error) return;
         const sessionID = info?.sessionID;
-        if (!info?.error) {
-          if (sessionID) clearTimer(sessionID);
-          return;
-        }
         const model =
           info?.model ??
           (typeof info?.providerID === "string" && typeof info?.modelID === "string"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@renjfk/opencode-model-fallback",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "description": "Mapped model fallback router for OpenCode. Routes retryable model failures to configured fallback models and recovers after cooldown.",
   "keywords": [
     "fallback",
@@ -39,6 +39,6 @@
   "devDependencies": {
     "oxfmt": "0.50.0",
     "oxlint": "1.65.0",
-    "vitest": "latest"
+    "vitest": "4.1.6"
   }
 }