cascade-ai 0.2.0 → 0.2.1

package/dist/index.cjs

@@ -164,7 +164,7 @@ var require_keytar2 = __commonJS({
 });
 
 // src/constants.ts
-var CASCADE_VERSION = "0.1.2";
+var CASCADE_VERSION = "0.2.0";
 var CASCADE_CONFIG_DIR = ".cascade";
 var CASCADE_MD_FILE = "CASCADE.md";
 var CASCADE_IGNORE_FILE = ".cascadeignore";
@@ -1284,6 +1284,15 @@ var ModelSelector = class {
   markProviderUnavailable(provider) {
     this.availableProviders.delete(provider);
   }
+  /**
+   * Re-add a provider to the available set after it has recovered (e.g. after
+   * a failover timeout expires or a successful call confirms recovery). Only
+   * re-enables providers that were originally configured — callers should
+   * guard against enabling providers that were never configured.
+   */
+  markProviderAvailable(provider) {
+    this.availableProviders.add(provider);
+  }
   resolveDynamicModel(overrideModelId) {
     let providerStr = null;
     let actualId = overrideModelId;
@@ -1353,10 +1362,23 @@ var FailoverManager = class {
     if (!failure) return true;
     if (Date.now() - failure.failedAt >= failure.retryAfterMs) {
       this.failures.delete(provider);
+      this.selector.markProviderAvailable(provider);
       return true;
     }
     return false;
   }
+  /**
+   * Call after a successful generation to immediately re-enable a provider
+   * that had previously been marked unavailable. This allows fast recovery
+   * when a transient rate-limit clears before the backoff window expires,
+   * preventing unnecessary routing to more expensive fallback models.
+   */
+  recordSuccess(provider) {
+    if (this.failures.has(provider)) {
+      this.failures.delete(provider);
+      this.selector.markProviderAvailable(provider);
+    }
+  }
   getFallbackModel(currentModel, tier) {
     return this.selector.getNextFallback(currentModel.id, tier);
   }
@@ -1373,6 +1395,7 @@ var FailoverManager = class {
   }
   clearFailure(provider) {
     this.failures.delete(provider);
+    this.selector.markProviderAvailable(provider);
   }
 };
 
@@ -1585,6 +1608,7 @@ var CascadeRouter = class _CascadeRouter extends EventEmitter__default.default {
         throw new Error(`Provider ${model.provider}:${model.id} returned an invalid generation result.`);
       }
       this.recordStats(tier, model, result.usage);
+      this.failover.recordSuccess(model.provider);
       return result;
     } catch (err) {
       const errMsg = err instanceof Error ? err.message : String(err);