swayrouter 1.0.8 → 1.0.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "swayrouter",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "Self-hosted AI gateway with a clean local dashboard",
   "license": "MIT",
   "repository": {

package/src/config/errorConfig.js

@@ -59,8 +59,22 @@ export function isNonRetryableRequestStatus(status) {
 export function isCancellationStatus(status) {
   return Number(status) === 499;
 }
-
-const T1_BACKOFF_BASE = 30 * 1000;
+
+const QUOTA_EXHAUSTION_PATTERNS = [
+  /\binsufficient[_\s-]+quota\b/i,
+  /\b(?:quota|resource)[\s_-]+(?:is[\s_-]+)?(?:exhausted|depleted)\b/i,
+  /\b(?:quota|resource)[\s_-]+(?:has[\s_-]+been[\s_-]+)?(?:exceeded|reached)\b/i,
+  /\b(?:exceeded|exhausted|depleted)\s+(?:(?:your|the)\s+)?(?:current\s+)?(?:quota|resource)\b/i,
+  /\b(?:daily|weekly|monthly)\s+(?:usage\s+)?(?:quota|limit)\s+(?:is\s+)?(?:exceeded|exhausted|reached)\b/i,
+  /\bno\s+(?:remaining\s+)?quota\b/i,
+];
+
+export function isQuotaExhaustionError(errorText) {
+  const text = typeof errorText === "string" ? errorText : JSON.stringify(errorText ?? "");
+  return QUOTA_EXHAUSTION_PATTERNS.some((pattern) => pattern.test(text));
+}
+
+const T1_BACKOFF_BASE = 30 * 1000;
 const T1_BACKOFF_CAP = 5 * 60 * 1000;
 
 const T3_DEPRIORITIZE_MS = 10 * 60 * 1000;
@@ -93,22 +107,30 @@ export const COOLDOWN_MS = {
   requestNotAllowed: COOLDOWN.short,
 };
 
-export function classifyError(status, errorText, retryCount = 0, opts = {}) {
-  const lowerError = errorText
-    ? (typeof errorText === "string" ? errorText : JSON.stringify(errorText)).toLowerCase()
-    : "";
-  const rng = typeof opts.rng === "function" ? opts.rng : null;
-
+export function classifyError(status, errorText, retryCount = 0, opts = {}) {
+  const lowerError = errorText
+    ? (typeof errorText === "string" ? errorText : JSON.stringify(errorText)).toLowerCase()
+    : "";
+  const rng = typeof opts.rng === "function" ? opts.rng : null;
+
+  const quotaExhausted = isQuotaExhaustionError(lowerError);
+  if (quotaExhausted || Number(status) === 429) {
+    const exp = T1_BACKOFF_BASE * Math.pow(2, retryCount);
+    const base = Math.min(exp, T1_BACKOFF_CAP);
+    const jitter = rng ? Math.floor(rng() * base * JITTER_FRACTION) : 0;
+    return { tier: ERROR_TIERS.T1, action: "cooldown", cooldownMs: base + jitter };
+  }
+
   if (isNonRetryableRequestStatus(status) || isCancellationStatus(status)) {
-    return {
-      tier: ERROR_TIERS.T3,
-      action: "final",
-      cooldownMs: 0,
+    return {
+      tier: ERROR_TIERS.T3,
+      action: "final",
+      cooldownMs: 0,
     };
   }
 
   const t1Texts = ["rate limit", "too many requests", "quota exceeded", "capacity", "overloaded"];
-  if (t1Texts.some((t) => lowerError.includes(t)) || Number(status) === 429) {
+  if (t1Texts.some((t) => lowerError.includes(t))) {
     const exp = T1_BACKOFF_BASE * Math.pow(2, retryCount);
     const base = Math.min(exp, T1_BACKOFF_CAP);
     const jitter = rng ? Math.floor(rng() * base * JITTER_FRACTION) : 0;

package/src/routes/api/providers/[id]/route.js

@@ -1,10 +1,11 @@
 import { NextResponse } from "next/server";
-import {
-  getProviderConnectionById,
-  getProxyPoolById,
-  updateProviderConnection,
-  deleteProviderConnection,
-} from "@/models";
+import {
+  getProviderConnectionById,
+  getProxyPoolById,
+  updateProviderConnection,
+  deleteProviderConnection,
+} from "@/models";
+import { clearAllLocksForConnection } from "@/lib/db/repos/accountModelLocksRepo.js";
 
 function normalizeProxyConfig(body = {}) {
   const hasAnyProxyField =
@@ -118,7 +119,12 @@ export async function PUT(request, { params }) {
     if (priority !== undefined) updateData.priority = priority;
     if (globalPriority !== undefined) updateData.globalPriority = globalPriority;
     if (defaultModel !== undefined) updateData.defaultModel = defaultModel;
-    if (isActive !== undefined) updateData.isActive = isActive;
+    if (isActive !== undefined) {
+      updateData.isActive = isActive;
+      if (isActive === true && existing.isActive === false) {
+        await clearAllLocksForConnection(id);
+      }
+    }
     if (apiKey && existing.authType === "apikey") {
       updateData.apiKey = typeof apiKey === "string" ? apiKey.trim() : apiKey;
     }

package/src/services/accountFallback.js

@@ -7,6 +7,7 @@ import {
   isDeprioitized,
   isCancellationStatus,
   isNonRetryableRequestStatus,
+  isQuotaExhaustionError,
 } from "../config/errorConfig.js";
 
 import { fnv1a } from "@/lib/security/payloadCapture.js";
@@ -18,6 +19,11 @@ export function getQuotaCooldown(backoffLevel = 0) {
 }
 
 export function checkFallbackError(status, errorText, backoffLevel = 0) {
+  if (isQuotaExhaustionError(errorText)) {
+    const newLevel = Math.min(backoffLevel + 1, BACKOFF_CONFIG.maxLevel);
+    return { shouldFallback: true, cooldownMs: getQuotaCooldown(newLevel), newBackoffLevel: newLevel };
+  }
+
   if (isNonRetryableRequestStatus(status) || isCancellationStatus(status)) {
     return { shouldFallback: false, cooldownMs: 0, action: "final", newBackoffLevel: backoffLevel };
   }

package/src/sse/services/auth.js

@@ -1,13 +1,13 @@
 import { getProviderConnectionById, getProviderConnectionsForRouting, validateApiKey, updateProviderConnection, getSettings, getProxyPools } from "@/lib/localDb";
 import { resolveConnectionProxyConfig, pickProxyPoolId } from "@/lib/network/connectionProxy";
 import {
-  formatRetryAfter, checkFallbackError, isModelLockActive, buildModelLockUpdate,
-  getEarliestModelLockUntil,
-
-  isDeprioritized, sortWithDeprioritization, pickCacheAffine,
-  DEPRIORITIZE_FIELD,
-} from "../../services/accountFallback.js";
-import { MAX_RATE_LIMIT_COOLDOWN_MS, classifyError, ERROR_TIERS } from "../../config/errorConfig.js";
+  formatRetryAfter, checkFallbackError, isModelLockActive, buildModelLockUpdate,
+  getEarliestModelLockUntil,
+
+  isDeprioritized, sortWithDeprioritization, pickCacheAffine,
+  DEPRIORITIZE_FIELD,
+} from "../../services/accountFallback.js";
+import { MAX_RATE_LIMIT_COOLDOWN_MS, classifyError, ERROR_TIERS, isQuotaExhaustionError } from "../../config/errorConfig.js";
 import { HTTP_STATUS } from "../../config/runtimeConfig.js";
 import { recordLock, clearLock, clearAllLocksForConnection } from "@/lib/db/repos/accountModelLocksRepo.js";
 import { accountFallbackDecisionsTotal, cacheAffineSelectionsTotal } from "@/observability/metrics.js";
@@ -397,15 +397,20 @@ export async function markAccountUnavailable(connectionId, status, errorText, pr
   }
   if (!shouldFallback) return { shouldFallback: false, cooldownMs: 0 };
 
-  const reason = typeof errorText === "string" ? errorText.slice(0, 100) : "Provider error";
-  const lockUpdate = buildModelLockUpdate(githubResetAtMs ? null : model, cooldownMs);
+  const reason = typeof errorText === "string" ? errorText.slice(0, 100) : "Provider error";
+  const providerId = resolveProviderId(provider);
+  const codeBuddyQuotaExhausted = ["codebuddy-cn", "codebuddy-intl"].includes(providerId)
+    && isQuotaExhaustionError(errorText);
+  const lockModel = githubResetAtMs || codeBuddyQuotaExhausted ? null : model;
+  const lockUpdate = buildModelLockUpdate(lockModel, cooldownMs);
 
   const deprioUpdate = deprioitizeUntilMs ? { [DEPRIORITIZE_FIELD]: deprioitizeUntilMs } : {};
 
-  await updateProviderConnection(connectionId, {
-    ...lockUpdate,
-    ...deprioUpdate,
-    testStatus: "unavailable",
+  await updateProviderConnection(connectionId, {
+    ...lockUpdate,
+    ...deprioUpdate,
+    ...(codeBuddyQuotaExhausted ? { isActive: false } : {}),
+    testStatus: "unavailable",
     lastError: reason,
     errorCode: status,
     lastErrorAt: new Date().toISOString(),
@@ -413,7 +418,7 @@ export async function markAccountUnavailable(connectionId, status, errorText, pr
   });
   invalidateProviderAccountPool(provider);
 
-  await recordLock(connectionId, githubResetAtMs ? null : model, {
+  await recordLock(connectionId, lockModel, {
     tier, reason, expiresAt: new Date(Date.now() + cooldownMs).toISOString(),
   }).catch(() => {});
 

package/src/utils/error.js

@@ -45,7 +45,7 @@ export async function parseUpstreamError(response, executor = null) {
       if (parsed && typeof parsed === "object") {
         return {
           statusCode: parsed.status || response.status,
-          message: executor?.provider === "codebuddy-intl" && parsed.message
+          message: ["codebuddy-cn", "codebuddy-intl"].includes(executor?.provider) && parsed.message
             ? parsed.message
             : DEFAULT_ERROR_MESSAGES[response.status] || "Upstream provider request failed",
           resetsAtMs: parsed.resetsAtMs,