@kenkaiiii/gg-core 5.22.1 → 5.22.3

package/dist/index.cjs

@@ -2192,9 +2192,10 @@ function getNextThinkingLevel(provider, model, current) {
 
 // src/provider-usage.ts
 var SubscriptionUsageError = class extends Error {
-  constructor(message, status) {
+  constructor(message, status, retryAfterMs2) {
     super(message);
     this.status = status;
+    this.retryAfterMs = retryAfterMs2;
     this.name = "SubscriptionUsageError";
   }
 };
@@ -2244,12 +2245,20 @@ function normalizedCodexWindow(window, fallbackKind, now) {
     resetsAt: codexResetAt(window, now)
   };
 }
-async function readUsageResponse(response) {
+function retryAfterMs(response, now) {
+  const value = response.headers.get("retry-after")?.trim();
+  if (!value) return void 0;
+  if (/^\d+(?:\.\d+)?$/.test(value)) return Math.ceil(Number(value) * 1e3);
+  const retryAt = Date.parse(value);
+  return Number.isFinite(retryAt) ? Math.max(0, retryAt - now) : void 0;
+}
+async function readUsageResponse(response, now) {
   const text = await response.text();
   if (!response.ok) {
     throw new SubscriptionUsageError(
       `Subscription usage request failed with HTTP ${response.status}`,
-      response.status
+      response.status,
+      retryAfterMs(response, now)
     );
   }
   try {
@@ -2270,7 +2279,7 @@ async function fetchAnthropicUsage(credentials, fetchFn, signal, now) {
       "User-Agent": "ggcoder"
     }
   });
-  const data = await readUsageResponse(response);
+  const data = await readUsageResponse(response, now());
   const windows = [];
   const currentPercent = clampPercent(data.five_hour?.utilization);
   if (currentPercent !== void 0) {
@@ -2306,7 +2315,7 @@ async function fetchCodexUsage(credentials, fetchFn, signal, now) {
   };
   if (credentials.accountId) headers["ChatGPT-Account-Id"] = credentials.accountId;
   const response = await fetchFn(CODEX_USAGE_URL, { method: "GET", signal, headers });
-  const data = await readUsageResponse(response);
+  const data = await readUsageResponse(response, now());
   const windows = [
     normalizedCodexWindow(data.rate_limit?.primary_window, "current", now()),
     normalizedCodexWindow(data.rate_limit?.secondary_window, "weekly", now())
@@ -2363,7 +2372,7 @@ async function fetchKimiUsage(credentials, fetchFn, signal, now) {
       ...kimiCodingHeaders()
     }
   });
-  const data = await readUsageResponse(response);
+  const data = await readUsageResponse(response, now());
   const windows = [];
   const weeklyPercent = kimiUsagePercent(data.usage);
   if (weeklyPercent !== void 0) {