opencode-auth-load-balancer 0.9.1 → 0.10.0

package/dist/v1.js

@@ -13054,6 +13054,18 @@ function normalizeWindow(w) {
     win.resetAt = 0;
   return win;
 }
+function normalizeCredits(c) {
+  if (!isPlainObject2(c))
+    return;
+  const credits = c;
+  if (typeof credits.hasCredits !== "boolean" || typeof credits.unlimited !== "boolean")
+    return;
+  if (!isFiniteNumber(credits.balance))
+    delete credits.balance;
+  if (credits.limitReached !== true)
+    delete credits.limitReached;
+  return credits;
+}
 function normalizeAccounts(rows) {
   let accounts = null;
   for (let i = 0;i < rows.length; i++) {
@@ -13073,7 +13085,13 @@ function normalizeAccounts(rows) {
         row.usage.capturedAt = 0;
       row.usage.hourly = normalizeWindow(row.usage.hourly);
       row.usage.weekly = normalizeWindow(row.usage.weekly);
+      delete row.usage.credits;
     }
+    const credits = normalizeCredits(row.credits);
+    if (credits)
+      row.credits = credits;
+    else
+      delete row.credits;
     if (!Number.isFinite(row.cooldownUntil))
       row.cooldownUntil = 0;
     if (row.cooldownKind !== undefined && row.cooldownKind !== "quota" && row.cooldownKind !== "auth" && row.cooldownKind !== "transient")
@@ -13326,14 +13344,19 @@ function maxUtil(account, now) {
 function isExhausted(account, cfg, now) {
   return maxUtil(account, now) >= cfg.exhaustedAt;
 }
+function hasCreditHeadroom(account) {
+  const credits = account.credits;
+  return !!credits && !credits.limitReached && (credits.unlimited || credits.hasCredits);
+}
+function hasHeadroom(account, cfg, now) {
+  return !isExhausted(account, cfg, now) || hasCreditHeadroom(account);
+}
 function isAvailable(account, cfg, now) {
   if (account.disabledReason)
     return false;
   if (account.cooldownUntil > now)
     return false;
-  if (isExhausted(account, cfg, now))
-    return false;
-  return true;
+  return hasHeadroom(account, cfg, now);
 }
 function weeklyUrgency(account, cfg, now) {
   const weekly = account.usage.weekly;
@@ -13364,9 +13387,11 @@ function scoreAccount(account, cfg, now) {
 function compareRanked(x, y) {
   if (x.available !== y.available)
     return x.available ? -1 : 1;
-  if (x.available)
-    return y.score - x.score;
-  return x.weeklyUtil - y.weeklyUtil;
+  if (!x.available)
+    return x.weeklyUtil - y.weeklyUtil;
+  if (x.onCredits !== y.onCredits)
+    return x.onCredits ? 1 : -1;
+  return y.score - x.score;
 }
 var DAY_MS, WEEK_MS2, HOUR_MS, HOURLY_WINDOW_MS, RESET_CUSHION_DAYS = 0.25, SCORE_DEFAULTS;
 var init_score_core = __esm(() => {
@@ -13388,7 +13413,7 @@ var init_score_core = __esm(() => {
 function reconcileQuotaCooldown(account, cfg, now) {
   if (account.cooldownKind === "auth" || account.cooldownKind === "transient")
     return;
-  if (account.cooldownUntil > now + MAX_TRANSIENT_COOLDOWN_MS && !isExhausted(account, cfg, now)) {
+  if (account.cooldownUntil > now + MAX_TRANSIENT_COOLDOWN_MS && hasHeadroom(account, cfg, now)) {
     account.cooldownUntil = 0;
     delete account.cooldownKind;
   }
@@ -13459,6 +13484,8 @@ function clearLostLogin(row, login) {
 function mergeLoginUsage(row, usage) {
   if (!usage)
     return;
+  if (usage.credits)
+    row.credits = usage.credits;
   row.usage = {
     hourly: usage.hourly ?? row.usage.hourly,
     weekly: usage.weekly ? preserveWeeklyAnchor(usage.weekly, row.usage.weekly, usage.capturedAt) : row.usage.weekly,
@@ -13584,7 +13611,8 @@ function classifyProviderRecovery(accounts, providerID, cfg, now, pollMs) {
       continue;
     const windows = exhaustedWindows(account, cfg, now);
     const cooldownActive = account.cooldownUntil > now;
-    if (!cooldownActive && windows.length === 0)
+    const blocking = hasCreditHeadroom(account) ? [] : windows;
+    if (!cooldownActive && blocking.length === 0)
       return { state: "available" };
     if (cooldownActive && (account.cooldownKind === "auth" || account.cooldownKind === "transient"))
       continue;
@@ -13593,7 +13621,7 @@ function classifyProviderRecovery(accounts, providerID, cfg, now, pollMs) {
     hasQuotaCandidate = true;
     const constraints = [];
     let accountHasUnknownReset = false;
-    for (const window of windows) {
+    for (const window of blocking) {
       if (window.resetAt > now)
         constraints.push(window.resetAt);
       else
@@ -13906,22 +13934,27 @@ function weeklyUtil(account) {
 function selectAccount(accounts, providerID, now, cfg = DEFAULT_CONFIG, exclude = NO_EXCLUSIONS, incumbentId = null) {
   let bestAvail = null;
   let bestAvailScore = Number.NEGATIVE_INFINITY;
+  let bestAvailOnCredits = true;
   let bestFallback = null;
   let lowestWeeklyUtil = Number.POSITIVE_INFINITY;
   let incumbentAccount = null;
   let incumbentScore = Number.NEGATIVE_INFINITY;
+  let incumbentOnCredits = false;
   for (const account of accounts) {
     if (account.providerID !== providerID || account.disabledReason || exclude.has(account.id))
       continue;
     if (isAvailable(account, cfg, now)) {
       const score = scoreAccount(account, cfg, now);
-      if (score > bestAvailScore) {
+      const onCredits = isExhausted(account, cfg, now);
+      if (onCredits === bestAvailOnCredits ? score > bestAvailScore : bestAvailOnCredits) {
         bestAvailScore = score;
         bestAvail = account;
+        bestAvailOnCredits = onCredits;
       }
       if (account.id === incumbentId) {
         incumbentAccount = account;
         incumbentScore = score;
+        incumbentOnCredits = onCredits;
       }
     } else {
       const util = weeklyUtil(account);
@@ -13932,7 +13965,7 @@ function selectAccount(accounts, providerID, now, cfg = DEFAULT_CONFIG, exclude
     }
   }
   if (bestAvail) {
-    if (incumbentAccount && incumbentScore >= bestAvailScore * (1 - INCUMBENT_HYSTERESIS))
+    if (incumbentAccount && incumbentOnCredits === bestAvailOnCredits && incumbentScore >= bestAvailScore * (1 - INCUMBENT_HYSTERESIS))
       return { account: incumbentAccount, degraded: false, sticky: false };
     return { account: bestAvail, degraded: false, sticky: false };
   }
@@ -13971,7 +14004,7 @@ function selectForSession(pool, providerID, sessionKey, now, cfg = DEFAULT_CONFI
     const cheap = (!pinnedImminent || cfg.drainMigrate) && isCheapMoment(typeof requestBytes === "function" ? requestBytes() : requestBytes, cfg);
     if (pinnedOverSoft && (pinnedImminent || cheap) || cfg.drainMigrate && cheap) {
       const alt = selectAccount(pool.accounts, providerID, now, cfg, withExcluded(exclude, pinned.id), null);
-      if (alt && !alt.degraded) {
+      if (alt && !alt.degraded && !isExhausted(alt.account, cfg, now)) {
         if (pinnedOverSoft) {
           const altUtil = maxUtil(alt.account, now);
           const proactiveBetter = pinnedImminent ? altUtil < pinnedUtil : pinnedUtil - altUtil >= PROACTIVE_MIGRATE_MIN_DELTA && weeklyUrgency(alt.account, cfg, now) >= weeklyUrgency(pinned, cfg, now);
@@ -15569,10 +15602,31 @@ var init_transform2 = __esm(() => {
 function isWeeklyMinutes(minutes) {
   return minutes >= WEEK_MINUTES * 0.95 && minutes <= WEEK_MINUTES * 1.05;
 }
+function creditsFrom(hasCredits, unlimited, balance, limitReached) {
+  if (typeof hasCredits !== "boolean" || typeof unlimited !== "boolean")
+    return;
+  const amount = typeof balance === "string" && balance.trim() !== "" ? Number(balance) : Number.NaN;
+  return {
+    hasCredits,
+    unlimited,
+    ...Number.isFinite(amount) ? { balance: amount } : {},
+    ...limitReached === true ? { limitReached } : {}
+  };
+}
+function headerBool(raw) {
+  const value = raw?.toLowerCase();
+  if (value === "true" || value === "1")
+    return true;
+  if (value === "false" || value === "0")
+    return false;
+  return;
+}
 function parseUsageHeaders2(headers) {
   const primaryWindowMinutes = Number(headers.get(PRIMARY_WINDOW_MINUTES_HEADER));
   const spec = isWeeklyMinutes(primaryWindowMinutes) ? WEEKLY_PRIMARY_HEADER_SPEC : HEADER_SPEC2;
-  return parseWindowPairHeaders(headers, spec);
+  const windows = parseWindowPairHeaders(headers, spec);
+  const credits = creditsFrom(headerBool(headers.get("x-codex-credits-has-credits")), headerBool(headers.get("x-codex-credits-unlimited")), headers.get("x-codex-credits-balance"));
+  return credits ? { ...windows, credits } : windows;
 }
 function endpointWindow2(w) {
   return endpointWindowFrom(w, (win) => win.used_percent, 100, (win) => secondsToMs(Number(win.reset_at ?? 0)));
@@ -15592,10 +15646,13 @@ async function fetchUsage2(account, now) {
     return null;
   const { primary_window: primary, secondary_window: secondary } = rateLimit;
   const primaryIsWeekly = isWeeklyMinutes(Math.ceil(Number(primary?.limit_window_seconds) / 60));
+  const reported = json?.credits;
+  const credits = isPlainObject2(reported) ? creditsFrom(reported.has_credits, reported.unlimited, reported.balance, reported.overage_limit_reached) : undefined;
   return {
     hourly: endpointWindow2(primaryIsWeekly ? secondary : primary),
     weekly: endpointWindow2(primaryIsWeekly ? primary : secondary),
-    capturedAt: now
+    capturedAt: now,
+    ...credits ? { credits } : {}
   };
 }
 var WEEK_MINUTES, PRIMARY_WINDOW_MINUTES_HEADER = "x-codex-primary-window-minutes", HEADER_SPEC2, WEEKLY_PRIMARY_HEADER_SPEC;
@@ -15642,6 +15699,7 @@ var init_adapter3 = __esm(() => {
     },
     parseUsageHeaders: parseUsageHeaders2,
     fetchUsage: fetchUsage2,
+    reportsCredits: true,
     classifyError: classifyHttpStatus
   };
 });
@@ -15675,11 +15733,14 @@ async function refreshUsage(adapters, now, poolSnapshot) {
     aliveIds?.add(account.id);
     if (account.disabledReason && account.disabledReason !== MANUAL_DISABLED_REASON)
       continue;
-    if (account.usage.capturedAt !== 0 && now - account.usage.capturedAt < USAGE_REFRESH_TTL_MS)
+    const fresh = account.usage.capturedAt !== 0 && now - account.usage.capturedAt < USAGE_REFRESH_TTL_MS;
+    if (fresh && account.credits)
       continue;
     const adapter = adapterFor(adapters, account.providerID);
     if (!adapter)
       continue;
+    if (fresh && !adapter.reportsCredits)
+      continue;
     if ((lastPoll.get(account.id) ?? 0) > now - USAGE_REFRESH_TTL_MS)
       continue;
     lastPoll.set(account.id, now);
@@ -15711,6 +15772,8 @@ async function refreshUsage(adapters, now, poolSnapshot) {
         await mutatePool((p) => {
           const stored = findAccount(p, account.id);
           if (stored) {
+            if (snapshot.credits)
+              stored.credits = snapshot.credits;
             stored.usage = {
               hourly: snapshot.hourly ?? stored.usage.hourly,
               weekly: snapshot.weekly === null ? stored.usage.weekly : preserveWeeklyAnchor(snapshot.weekly, stored.usage.weekly, now),
@@ -15763,6 +15826,8 @@ async function bestEffort(what, op) {
 function applyUsagePartial(account, partial, now) {
   if (partial.hourly !== undefined)
     account.usage.hourly = partial.hourly;
+  if (partial.credits !== undefined)
+    account.credits = partial.credits;
   if (partial.weekly !== undefined) {
     account.usage.weekly = preserveWeeklyAnchor(partial.weekly, account.usage.weekly, now);
     account.usage.capturedAt = now;
@@ -15831,7 +15896,7 @@ async function recordSuccess(adapter, accountId, partial, sessionKey, now, cfg,
     if (account) {
       if (partial)
         applyUsagePartial(account, partial, now);
-      if (partial && account.cooldownUntil > now + ACCOUNT_COOLDOWN_MS && !isExhausted(account, cfg, now)) {
+      if (partial && account.cooldownUntil > now + ACCOUNT_COOLDOWN_MS && hasHeadroom(account, cfg, now)) {
         account.cooldownUntil = 0;
         delete account.cooldownKind;
       }
@@ -16175,7 +16240,18 @@ function byProviderId(a, b) {
 function isExpired(window, now) {
   return !!window && window.resetAt > 0 && window.resetAt <= now;
 }
-function toStatus(account, now, current, available, rank) {
+function creditsText(credits) {
+  if (!credits)
+    return "";
+  if (credits.limitReached)
+    return "credit limit reached";
+  if (credits.unlimited)
+    return "credits unlimited";
+  if (!credits.hasCredits)
+    return "no credits";
+  return credits.balance === undefined ? "credits available" : `credits ${Math.round(credits.balance * 100) / 100}`;
+}
+function toStatus(account, now, current, available, exhausted, rank) {
   const weekly = account.usage.weekly;
   const weeklyExpired = isExpired(weekly, now);
   return {
@@ -16191,6 +16267,7 @@ function toStatus(account, now, current, available, rank) {
     login: credentialTag(account),
     warnings: loginWarnings(account, now),
     lostNotes: lostLoginNotes(account, now),
+    credits: exhausted ? creditsText(account.credits) : "",
     current,
     rank
   };
@@ -16209,14 +16286,17 @@ function buildStatus(pool, now, cfg = DEFAULT_CONFIG) {
     const currentAccountId = pool.lastSelected[providerID] ?? null;
     const ranked = providerAccounts.map((a) => {
       const available = isAvailable(a, cfg, now);
+      const exhausted = isExhausted(a, cfg, now);
       return {
         a,
         available,
+        exhausted,
+        onCredits: available && exhausted,
         score: available ? scoreAccount(a, cfg, now) : Number.NEGATIVE_INFINITY,
         weeklyUtil: a.usage.weekly?.utilization ?? 0
       };
     }).sort(compareRanked);
-    const accounts = ranked.map(({ a, available }, i) => toStatus(a, now, a.id === currentAccountId, available, i + 1));
+    const accounts = ranked.map(({ a, available, exhausted }, i) => toStatus(a, now, a.id === currentAccountId, available, exhausted, i + 1));
     return { providerID, accounts };
   });
 }
@@ -16256,12 +16336,10 @@ function stateOf(a, now) {
   if (a.cooldownUntil > now)
     return `cooldown ${relTime(a.cooldownUntil, now)}`;
   if (!a.available)
-    return "exhausted";
+    return a.credits ? `exhausted · ${a.credits}` : "exhausted";
   const base = a.current ? "in use" : "ready";
   const tiers = Object.entries(a.modelCooldownsUntil).filter(([, until]) => until > now).sort(([x], [y]) => compareAscii(x, y)).map(([tier, until]) => `${tier} ${relTime(until, now)}`);
-  if (tiers.length > 0)
-    return `${base} · ${tiers.join(" · ")}`;
-  return base;
+  return [base, a.credits, ...tiers].filter(Boolean).join(" · ");
 }
 function providerName(providerID) {
   return PROVIDER_NAMES[providerID] ?? providerID;
@@ -16737,7 +16815,7 @@ class PendingCoordinator {
     await this.notifyPending(ref, recovery);
     try {
       for (;; ) {
-        await this.dependencies.sleep(Math.max(0, recovery.nextCheckAt - this.dependencies.now()), controller.signal);
+        await this.dependencies.sleep(Math.max(0, Math.min(recovery.nextCheckAt - this.dependencies.now(), USAGE_REFRESH_TTL_MS)), controller.signal);
         const now = this.dependencies.now();
         await this.dependencies.refreshUsage(now).catch(ignore);
         let next;

package/dist/v2-entry.js

@@ -41812,6 +41812,18 @@ function normalizeWindow(w) {
     win.resetAt = 0;
   return win;
 }
+function normalizeCredits(c) {
+  if (!isPlainObject4(c))
+    return;
+  const credits = c;
+  if (typeof credits.hasCredits !== "boolean" || typeof credits.unlimited !== "boolean")
+    return;
+  if (!isFiniteNumber(credits.balance))
+    delete credits.balance;
+  if (credits.limitReached !== true)
+    delete credits.limitReached;
+  return credits;
+}
 function normalizeAccounts(rows) {
   let accounts = null;
   for (let i = 0;i < rows.length; i++) {
@@ -41831,7 +41843,13 @@ function normalizeAccounts(rows) {
         row.usage.capturedAt = 0;
       row.usage.hourly = normalizeWindow(row.usage.hourly);
       row.usage.weekly = normalizeWindow(row.usage.weekly);
+      delete row.usage.credits;
     }
+    const credits = normalizeCredits(row.credits);
+    if (credits)
+      row.credits = credits;
+    else
+      delete row.credits;
     if (!Number.isFinite(row.cooldownUntil))
       row.cooldownUntil = 0;
     if (row.cooldownKind !== undefined && row.cooldownKind !== "quota" && row.cooldownKind !== "auth" && row.cooldownKind !== "transient")
@@ -42084,14 +42102,19 @@ function maxUtil(account, now) {
 function isExhausted(account, cfg, now) {
   return maxUtil(account, now) >= cfg.exhaustedAt;
 }
+function hasCreditHeadroom(account) {
+  const credits = account.credits;
+  return !!credits && !credits.limitReached && (credits.unlimited || credits.hasCredits);
+}
+function hasHeadroom(account, cfg, now) {
+  return !isExhausted(account, cfg, now) || hasCreditHeadroom(account);
+}
 function isAvailable(account, cfg, now) {
   if (account.disabledReason)
     return false;
   if (account.cooldownUntil > now)
     return false;
-  if (isExhausted(account, cfg, now))
-    return false;
-  return true;
+  return hasHeadroom(account, cfg, now);
 }
 function weeklyUrgency(account, cfg, now) {
   const weekly = account.usage.weekly;
@@ -42122,9 +42145,11 @@ function scoreAccount(account, cfg, now) {
 function compareRanked(x, y) {
   if (x.available !== y.available)
     return x.available ? -1 : 1;
-  if (x.available)
-    return y.score - x.score;
-  return x.weeklyUtil - y.weeklyUtil;
+  if (!x.available)
+    return x.weeklyUtil - y.weeklyUtil;
+  if (x.onCredits !== y.onCredits)
+    return x.onCredits ? 1 : -1;
+  return y.score - x.score;
 }
 var DAY_MS, WEEK_MS2, HOUR_MS, HOURLY_WINDOW_MS, RESET_CUSHION_DAYS = 0.25, SCORE_DEFAULTS;
 var init_score_core = __esm(() => {
@@ -42146,7 +42171,7 @@ var init_score_core = __esm(() => {
 function reconcileQuotaCooldown(account, cfg, now) {
   if (account.cooldownKind === "auth" || account.cooldownKind === "transient")
     return;
-  if (account.cooldownUntil > now + MAX_TRANSIENT_COOLDOWN_MS && !isExhausted(account, cfg, now)) {
+  if (account.cooldownUntil > now + MAX_TRANSIENT_COOLDOWN_MS && hasHeadroom(account, cfg, now)) {
     account.cooldownUntil = 0;
     delete account.cooldownKind;
   }
@@ -42217,6 +42242,8 @@ function clearLostLogin(row, login) {
 function mergeLoginUsage(row, usage) {
   if (!usage)
     return;
+  if (usage.credits)
+    row.credits = usage.credits;
   row.usage = {
     hourly: usage.hourly ?? row.usage.hourly,
     weekly: usage.weekly ? preserveWeeklyAnchor(usage.weekly, row.usage.weekly, usage.capturedAt) : row.usage.weekly,
@@ -42342,7 +42369,8 @@ function classifyProviderRecovery(accounts, providerID, cfg, now, pollMs) {
       continue;
     const windows = exhaustedWindows(account, cfg, now);
     const cooldownActive = account.cooldownUntil > now;
-    if (!cooldownActive && windows.length === 0)
+    const blocking = hasCreditHeadroom(account) ? [] : windows;
+    if (!cooldownActive && blocking.length === 0)
       return { state: "available" };
     if (cooldownActive && (account.cooldownKind === "auth" || account.cooldownKind === "transient"))
       continue;
@@ -42351,7 +42379,7 @@ function classifyProviderRecovery(accounts, providerID, cfg, now, pollMs) {
     hasQuotaCandidate = true;
     const constraints = [];
     let accountHasUnknownReset = false;
-    for (const window of windows) {
+    for (const window of blocking) {
       if (window.resetAt > now)
         constraints.push(window.resetAt);
       else
@@ -42664,22 +42692,27 @@ function weeklyUtil(account) {
 function selectAccount(accounts, providerID, now, cfg = DEFAULT_CONFIG, exclude = NO_EXCLUSIONS, incumbentId = null) {
   let bestAvail = null;
   let bestAvailScore = Number.NEGATIVE_INFINITY;
+  let bestAvailOnCredits = true;
   let bestFallback = null;
   let lowestWeeklyUtil = Number.POSITIVE_INFINITY;
   let incumbentAccount = null;
   let incumbentScore = Number.NEGATIVE_INFINITY;
+  let incumbentOnCredits = false;
   for (const account of accounts) {
     if (account.providerID !== providerID || account.disabledReason || exclude.has(account.id))
       continue;
     if (isAvailable(account, cfg, now)) {
       const score = scoreAccount(account, cfg, now);
-      if (score > bestAvailScore) {
+      const onCredits = isExhausted(account, cfg, now);
+      if (onCredits === bestAvailOnCredits ? score > bestAvailScore : bestAvailOnCredits) {
         bestAvailScore = score;
         bestAvail = account;
+        bestAvailOnCredits = onCredits;
       }
       if (account.id === incumbentId) {
         incumbentAccount = account;
         incumbentScore = score;
+        incumbentOnCredits = onCredits;
       }
     } else {
       const util = weeklyUtil(account);
@@ -42690,7 +42723,7 @@ function selectAccount(accounts, providerID, now, cfg = DEFAULT_CONFIG, exclude
     }
   }
   if (bestAvail) {
-    if (incumbentAccount && incumbentScore >= bestAvailScore * (1 - INCUMBENT_HYSTERESIS))
+    if (incumbentAccount && incumbentOnCredits === bestAvailOnCredits && incumbentScore >= bestAvailScore * (1 - INCUMBENT_HYSTERESIS))
       return { account: incumbentAccount, degraded: false, sticky: false };
     return { account: bestAvail, degraded: false, sticky: false };
   }
@@ -42729,7 +42762,7 @@ function selectForSession(pool, providerID, sessionKey, now, cfg = DEFAULT_CONFI
     const cheap = (!pinnedImminent || cfg.drainMigrate) && isCheapMoment(typeof requestBytes === "function" ? requestBytes() : requestBytes, cfg);
     if (pinnedOverSoft && (pinnedImminent || cheap) || cfg.drainMigrate && cheap) {
       const alt = selectAccount(pool.accounts, providerID, now, cfg, withExcluded(exclude, pinned.id), null);
-      if (alt && !alt.degraded) {
+      if (alt && !alt.degraded && !isExhausted(alt.account, cfg, now)) {
         if (pinnedOverSoft) {
           const altUtil = maxUtil(alt.account, now);
           const proactiveBetter = pinnedImminent ? altUtil < pinnedUtil : pinnedUtil - altUtil >= PROACTIVE_MIGRATE_MIN_DELTA && weeklyUrgency(alt.account, cfg, now) >= weeklyUrgency(pinned, cfg, now);
@@ -44327,10 +44360,31 @@ var init_transform2 = __esm(() => {
 function isWeeklyMinutes(minutes) {
   return minutes >= WEEK_MINUTES * 0.95 && minutes <= WEEK_MINUTES * 1.05;
 }
+function creditsFrom(hasCredits, unlimited, balance, limitReached) {
+  if (typeof hasCredits !== "boolean" || typeof unlimited !== "boolean")
+    return;
+  const amount = typeof balance === "string" && balance.trim() !== "" ? Number(balance) : Number.NaN;
+  return {
+    hasCredits,
+    unlimited,
+    ...Number.isFinite(amount) ? { balance: amount } : {},
+    ...limitReached === true ? { limitReached } : {}
+  };
+}
+function headerBool(raw) {
+  const value = raw?.toLowerCase();
+  if (value === "true" || value === "1")
+    return true;
+  if (value === "false" || value === "0")
+    return false;
+  return;
+}
 function parseUsageHeaders2(headers) {
   const primaryWindowMinutes = Number(headers.get(PRIMARY_WINDOW_MINUTES_HEADER));
   const spec = isWeeklyMinutes(primaryWindowMinutes) ? WEEKLY_PRIMARY_HEADER_SPEC : HEADER_SPEC2;
-  return parseWindowPairHeaders(headers, spec);
+  const windows = parseWindowPairHeaders(headers, spec);
+  const credits = creditsFrom(headerBool(headers.get("x-codex-credits-has-credits")), headerBool(headers.get("x-codex-credits-unlimited")), headers.get("x-codex-credits-balance"));
+  return credits ? { ...windows, credits } : windows;
 }
 function endpointWindow2(w) {
   return endpointWindowFrom(w, (win) => win.used_percent, 100, (win) => secondsToMs(Number(win.reset_at ?? 0)));
@@ -44350,10 +44404,13 @@ async function fetchUsage2(account, now) {
     return null;
   const { primary_window: primary, secondary_window: secondary } = rateLimit;
   const primaryIsWeekly = isWeeklyMinutes(Math.ceil(Number(primary?.limit_window_seconds) / 60));
+  const reported = json?.credits;
+  const credits = isPlainObject4(reported) ? creditsFrom(reported.has_credits, reported.unlimited, reported.balance, reported.overage_limit_reached) : undefined;
   return {
     hourly: endpointWindow2(primaryIsWeekly ? secondary : primary),
     weekly: endpointWindow2(primaryIsWeekly ? primary : secondary),
-    capturedAt: now
+    capturedAt: now,
+    ...credits ? { credits } : {}
   };
 }
 var WEEK_MINUTES, PRIMARY_WINDOW_MINUTES_HEADER = "x-codex-primary-window-minutes", HEADER_SPEC2, WEEKLY_PRIMARY_HEADER_SPEC;
@@ -44400,6 +44457,7 @@ var init_adapter3 = __esm(() => {
     },
     parseUsageHeaders: parseUsageHeaders2,
     fetchUsage: fetchUsage2,
+    reportsCredits: true,
     classifyError: classifyHttpStatus
   };
 });
@@ -44433,11 +44491,14 @@ async function refreshUsage(adapters, now, poolSnapshot) {
     aliveIds?.add(account.id);
     if (account.disabledReason && account.disabledReason !== MANUAL_DISABLED_REASON)
       continue;
-    if (account.usage.capturedAt !== 0 && now - account.usage.capturedAt < USAGE_REFRESH_TTL_MS)
+    const fresh = account.usage.capturedAt !== 0 && now - account.usage.capturedAt < USAGE_REFRESH_TTL_MS;
+    if (fresh && account.credits)
       continue;
     const adapter = adapterFor(adapters, account.providerID);
     if (!adapter)
       continue;
+    if (fresh && !adapter.reportsCredits)
+      continue;
     if ((lastPoll.get(account.id) ?? 0) > now - USAGE_REFRESH_TTL_MS)
       continue;
     lastPoll.set(account.id, now);
@@ -44469,6 +44530,8 @@ async function refreshUsage(adapters, now, poolSnapshot) {
         await mutatePool((p) => {
           const stored = findAccount(p, account.id);
           if (stored) {
+            if (snapshot.credits)
+              stored.credits = snapshot.credits;
             stored.usage = {
               hourly: snapshot.hourly ?? stored.usage.hourly,
               weekly: snapshot.weekly === null ? stored.usage.weekly : preserveWeeklyAnchor(snapshot.weekly, stored.usage.weekly, now),
@@ -44521,6 +44584,8 @@ async function bestEffort(what, op) {
 function applyUsagePartial(account, partial, now) {
   if (partial.hourly !== undefined)
     account.usage.hourly = partial.hourly;
+  if (partial.credits !== undefined)
+    account.credits = partial.credits;
   if (partial.weekly !== undefined) {
     account.usage.weekly = preserveWeeklyAnchor(partial.weekly, account.usage.weekly, now);
     account.usage.capturedAt = now;
@@ -44589,7 +44654,7 @@ async function recordSuccess(adapter, accountId, partial, sessionKey, now, cfg,
     if (account) {
       if (partial)
         applyUsagePartial(account, partial, now);
-      if (partial && account.cooldownUntil > now + ACCOUNT_COOLDOWN_MS && !isExhausted(account, cfg, now)) {
+      if (partial && account.cooldownUntil > now + ACCOUNT_COOLDOWN_MS && hasHeadroom(account, cfg, now)) {
         account.cooldownUntil = 0;
         delete account.cooldownKind;
       }
@@ -44933,7 +44998,18 @@ function byProviderId(a, b) {
 function isExpired(window, now) {
   return !!window && window.resetAt > 0 && window.resetAt <= now;
 }
-function toStatus(account, now, current, available, rank) {
+function creditsText(credits) {
+  if (!credits)
+    return "";
+  if (credits.limitReached)
+    return "credit limit reached";
+  if (credits.unlimited)
+    return "credits unlimited";
+  if (!credits.hasCredits)
+    return "no credits";
+  return credits.balance === undefined ? "credits available" : `credits ${Math.round(credits.balance * 100) / 100}`;
+}
+function toStatus(account, now, current, available, exhausted, rank) {
   const weekly = account.usage.weekly;
   const weeklyExpired = isExpired(weekly, now);
   return {
@@ -44949,6 +45025,7 @@ function toStatus(account, now, current, available, rank) {
     login: credentialTag(account),
     warnings: loginWarnings(account, now),
     lostNotes: lostLoginNotes(account, now),
+    credits: exhausted ? creditsText(account.credits) : "",
     current,
     rank
   };
@@ -44967,14 +45044,17 @@ function buildStatus(pool, now, cfg = DEFAULT_CONFIG) {
     const currentAccountId = pool.lastSelected[providerID] ?? null;
     const ranked = providerAccounts.map((a) => {
       const available = isAvailable(a, cfg, now);
+      const exhausted = isExhausted(a, cfg, now);
       return {
         a,
         available,
+        exhausted,
+        onCredits: available && exhausted,
         score: available ? scoreAccount(a, cfg, now) : Number.NEGATIVE_INFINITY,
         weeklyUtil: a.usage.weekly?.utilization ?? 0
       };
     }).sort(compareRanked);
-    const accounts = ranked.map(({ a, available }, i) => toStatus(a, now, a.id === currentAccountId, available, i + 1));
+    const accounts = ranked.map(({ a, available, exhausted }, i) => toStatus(a, now, a.id === currentAccountId, available, exhausted, i + 1));
     return { providerID, accounts };
   });
 }
@@ -45014,12 +45094,10 @@ function stateOf(a, now) {
   if (a.cooldownUntil > now)
     return `cooldown ${relTime(a.cooldownUntil, now)}`;
   if (!a.available)
-    return "exhausted";
+    return a.credits ? `exhausted · ${a.credits}` : "exhausted";
   const base = a.current ? "in use" : "ready";
   const tiers = Object.entries(a.modelCooldownsUntil).filter(([, until]) => until > now).sort(([x], [y]) => compareAscii(x, y)).map(([tier, until]) => `${tier} ${relTime(until, now)}`);
-  if (tiers.length > 0)
-    return `${base} · ${tiers.join(" · ")}`;
-  return base;
+  return [base, a.credits, ...tiers].filter(Boolean).join(" · ");
 }
 function providerName(providerID) {
   return PROVIDER_NAMES[providerID] ?? providerID;
@@ -45495,7 +45573,7 @@ class PendingCoordinator {
     await this.notifyPending(ref, recovery);
     try {
       for (;; ) {
-        await this.dependencies.sleep(Math.max(0, recovery.nextCheckAt - this.dependencies.now()), controller.signal);
+        await this.dependencies.sleep(Math.max(0, Math.min(recovery.nextCheckAt - this.dependencies.now(), USAGE_REFRESH_TTL_MS)), controller.signal);
         const now = this.dependencies.now();
         await this.dependencies.refreshUsage(now).catch(ignore);
         let next;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-auth-load-balancer",
-  "version": "0.9.1",
+  "version": "0.10.0",
   "description": "Load-balance opencode requests across multiple Anthropic (Claude) and OpenAI (Codex) OAuth accounts and Kimi Code subscriptions, weighted primarily by weekly usage with drain-before-reset scheduling.",
   "type": "module",
   "license": "MIT",