@narumitw/pi-usage 0.54.0 → 0.58.0

package/dist/index.ts

@@ -449,10 +449,234 @@ function clampPercent(value) {
   return Math.min(100, Math.max(0, value));
 }
 
+// src/providers/deepseek.ts
+var CURRENCIES = ["CNY", "USD"];
+var BALANCE_FIELDS = [
+  ["total", "Total balance", "total_balance"],
+  ["granted", "Granted balance", "granted_balance"],
+  ["topped-up", "Topped-up balance", "topped_up_balance"]
+];
+function normalizeDeepSeekBalancePayload(payload, capturedAt) {
+  if (typeof payload.is_available !== "boolean") {
+    throw new Error("DeepSeek API balance response availability was not a boolean.");
+  }
+  if (!Array.isArray(payload.balance_infos) || payload.balance_infos.length === 0) {
+    throw new Error("DeepSeek API balance response returned no balance information.");
+  }
+  const balances = /* @__PURE__ */ new Map();
+  for (const raw of payload.balance_infos) {
+    const balance = asObject2(raw);
+    if (!balance) throw new Error("DeepSeek API balance row was not an object.");
+    const currency = deepSeekCurrency(balance.currency);
+    if (!currency) throw new Error("DeepSeek API balance row returned an unsupported currency.");
+    if (balances.has(currency)) {
+      throw new Error(`DeepSeek API balance response repeated ${currency}.`);
+    }
+    for (const [, label, field] of BALANCE_FIELDS) {
+      if (!decimalAmount(balance[field])) {
+        throw new Error(`DeepSeek API balance ${label.toLowerCase()} was not a valid amount.`);
+      }
+    }
+    balances.set(currency, balance);
+  }
+  const metrics = [
+    {
+      id: "api-availability",
+      label: "API calls",
+      value: payload.is_available ? "available" : "unavailable"
+    }
+  ];
+  for (const currency of CURRENCIES) {
+    const balance = balances.get(currency);
+    if (!balance) continue;
+    for (const [id, label, field] of BALANCE_FIELDS) {
+      metrics.push({
+        id: `${currency.toLowerCase()}-${id}`,
+        label,
+        value: balance[field],
+        unit: "currency",
+        currency
+      });
+    }
+  }
+  return {
+    providerId: "deepseek",
+    providerName: "DeepSeek",
+    capturedAt,
+    source: "deepseek-balance",
+    semantics: { kind: "api-key", label: "DeepSeek API balance" },
+    buckets: [],
+    metrics
+  };
+}
+function asObject2(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
+  return value;
+}
+function deepSeekCurrency(value) {
+  return CURRENCIES.find((currency) => currency === value);
+}
+function decimalAmount(value) {
+  return typeof value === "string" && value.length <= 64 && /^(?:0|[1-9]\d*)(?:\.\d+)?$/u.test(value);
+}
+
+// src/providers/fireworks.ts
+var NANOS_PER_UNIT = 1000000000n;
+var CURRENCY_PATTERN = /^[A-Z]{3}$/u;
+var ACCOUNT_ID_PATTERN = /^[A-Za-z0-9][A-Za-z0-9._~-]{0,127}$/u;
+var INTEGER_PATTERN = /^-?\d+$/u;
+var INT64_MIN = -(2n ** 63n);
+var INT64_MAX = 2n ** 63n - 1n;
+var MAX_UNITS_CHARS = 20;
+var MAX_NANOS_CHARS = 11;
+var SERIES_KEYS = ["serverless", "dedicated", "training", "other"];
+var SERIES_LABELS = {
+  serverless: "Serverless",
+  dedicated: "Dedicated deployments",
+  training: "Training",
+  other: "Other"
+};
+function isFireworksAccountId(value) {
+  return typeof value === "string" && ACCOUNT_ID_PATTERN.test(value);
+}
+function normalizeFireworksAccountsPayload(payload) {
+  if (!Array.isArray(payload.accounts)) {
+    throw new Error("Fireworks accounts response did not contain an accounts array.");
+  }
+  const accounts = [];
+  for (const raw of payload.accounts) {
+    const account = asObject3(raw);
+    if (!account) throw new Error("Fireworks accounts response row was not an object.");
+    if (typeof account.name !== "string") {
+      throw new Error("Fireworks accounts response omitted the account resource name.");
+    }
+    const match = /^accounts\/([^/]+)$/u.exec(account.name);
+    if (!match || !isFireworksAccountId(match[1])) {
+      throw new Error("Fireworks accounts response returned an unsafe account resource name.");
+    }
+    const accountId = match[1];
+    if (accounts.includes(accountId)) {
+      throw new Error(`Fireworks accounts response repeated ${accountId}.`);
+    }
+    accounts.push(accountId);
+  }
+  return accounts;
+}
+function normalizeFireworksBillingSummaryPayload(payload, accountId, capturedAt) {
+  if (!isFireworksAccountId(accountId)) {
+    throw new Error("Fireworks billing summary received an unsafe account identifier.");
+  }
+  if (payload.lineItems !== void 0 && !Array.isArray(payload.lineItems)) {
+    throw new Error("Fireworks billing summary lineItems was not an array.");
+  }
+  const totals = /* @__PURE__ */ new Map();
+  for (const raw of payload.lineItems ?? []) {
+    const lineItem = asObject3(raw);
+    if (!lineItem) throw new Error("Fireworks billing line item was not an object.");
+    const cost = moneyAmount(lineItem.totalCost, "line item total cost");
+    const series = seriesKey(lineItem.series);
+    let amounts = totals.get(cost.currency);
+    if (!amounts) {
+      amounts = /* @__PURE__ */ new Map();
+      totals.set(cost.currency, amounts);
+    }
+    amounts.set(series, (amounts.get(series) ?? 0n) + cost.amount);
+  }
+  const metrics = [];
+  for (const [currency, amounts] of totals) {
+    metrics.push({
+      id: `${currency.toLowerCase()}-total`,
+      label: "Total spend",
+      value: formatMoneyAmount(sumSeries(amounts)),
+      unit: "currency",
+      currency
+    });
+    for (const series of SERIES_KEYS) {
+      const amount = amounts.get(series);
+      if (amount === void 0) continue;
+      metrics.push({
+        id: `${currency.toLowerCase()}-${series}`,
+        label: SERIES_LABELS[series],
+        value: formatMoneyAmount(amount),
+        unit: "currency",
+        currency
+      });
+    }
+  }
+  const notes = [
+    "Rated line items may differ from the final invoice once credits or adjustments are applied."
+  ];
+  if (metrics.length === 0) {
+    notes.push("Fireworks returned no rated line items for the last 30 days.");
+  }
+  return {
+    providerId: "fireworks",
+    providerName: "Fireworks",
+    capturedAt,
+    source: "fireworks-billing-summary",
+    semantics: { kind: "api-key", label: "Fireworks API spend" },
+    accountLabel: sanitizeDisplayText(accountId, 80),
+    buckets: [],
+    metrics,
+    notes
+  };
+}
+function moneyAmount(value, description) {
+  const money = asObject3(value);
+  if (!money) throw new Error(`Fireworks billing ${description} was not a money object.`);
+  const currency = typeof money.currencyCode === "string" ? money.currencyCode : void 0;
+  if (!currency || !CURRENCY_PATTERN.test(currency)) {
+    throw new Error(`Fireworks billing ${description} currency was not an ISO 4217 code.`);
+  }
+  const units = money.units === void 0 ? 0n : integerComponent(money.units, description, "whole units", MAX_UNITS_CHARS);
+  if (units < INT64_MIN || units > INT64_MAX) {
+    throw new Error(`Fireworks billing ${description} whole units exceeded the int64 range.`);
+  }
+  const nanos = money.nanos === void 0 ? 0n : integerComponent(money.nanos, description, "nano units", MAX_NANOS_CHARS);
+  if (nanos <= -NANOS_PER_UNIT || nanos >= NANOS_PER_UNIT) {
+    throw new Error(`Fireworks billing ${description} nano units exceeded the Money range.`);
+  }
+  if (units > 0n && nanos < 0n || units < 0n && nanos > 0n) {
+    throw new Error(`Fireworks billing ${description} mixed unit and nano signs.`);
+  }
+  return { currency, amount: units * NANOS_PER_UNIT + nanos };
+}
+function integerComponent(value, description, component, maxChars) {
+  const text = typeof value === "number" && Number.isSafeInteger(value) ? String(value) : typeof value === "string" && INTEGER_PATTERN.test(value) ? value : void 0;
+  if (text === void 0 || text.length > maxChars) {
+    throw new Error(`Fireworks billing ${description} ${component} was not a bounded integer.`);
+  }
+  return BigInt(text);
+}
+function seriesKey(value) {
+  if (value === void 0 || value === null) return "other";
+  if (typeof value !== "string") throw new Error("Fireworks billing line item series was invalid.");
+  if (value === "SERVERLESS") return "serverless";
+  if (value === "DEDICATED_DEPLOYMENT") return "dedicated";
+  if (value === "TRAINING") return "training";
+  return "other";
+}
+function sumSeries(amounts) {
+  let total = 0n;
+  for (const amount of amounts.values()) total += amount;
+  return total;
+}
+function formatMoneyAmount(amount) {
+  const negative = amount < 0n;
+  const magnitude = negative ? -amount : amount;
+  const units = magnitude / NANOS_PER_UNIT;
+  const nanos = (magnitude % NANOS_PER_UNIT).toString().padStart(9, "0").replace(/0+$/u, "");
+  return `${negative ? "-" : ""}${units.toString()}${nanos ? `.${nanos}` : ""}`;
+}
+function asObject3(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
+  return value;
+}
+
 // src/providers/github-copilot.ts
 function normalizeGitHubCopilotUsagePayload(payload, capturedAt) {
-  const snapshots = asObject2(payload.quota_snapshots);
-  const premium = asObject2(snapshots?.premium_interactions);
+  const snapshots = asObject4(payload.quota_snapshots);
+  const premium = asObject4(snapshots?.premium_interactions);
   const metrics = [];
   let semanticsLabel;
   let bucket;
@@ -493,8 +717,8 @@ function normalizeGitHubCopilotUsagePayload(payload, capturedAt) {
       };
     }
   } else {
-    const limited = asObject2(payload.limited_user_quotas);
-    const monthly = asObject2(payload.monthly_quotas);
+    const limited = asObject4(payload.limited_user_quotas);
+    const monthly = asObject4(payload.monthly_quotas);
     const remaining = asNonnegativeNumber(limited?.chat);
     const entitlement = asNonnegativeNumber(monthly?.chat);
     if (remaining === void 0 || entitlement === void 0) {
@@ -533,7 +757,7 @@ function resetTimestamp(payload) {
   const milliseconds = Date.parse(raw);
   return Number.isNaN(milliseconds) ? {} : { resetsAt: Math.floor(milliseconds / 1e3) };
 }
-function asObject2(value) {
+function asObject4(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -556,7 +780,7 @@ var DAILY_WINDOW_MINUTES = 1440;
 var WEEKLY_WINDOW_MINUTES = 10080;
 var FIXED_POINT_UNITS_PER_CENT = 1e6;
 function normalizeKimiCodingUsagePayload(payload, capturedAt) {
-  const root = asObject3(payload);
+  const root = asObject5(payload);
   if (!root) throw new Error("Kimi Coding usage response was not an object.");
   const candidates = [];
   let omittedWindow = false;
@@ -565,7 +789,7 @@ function normalizeKimiCodingUsagePayload(payload, capturedAt) {
   else if (root.usage !== void 0) omittedWindow = true;
   if (Array.isArray(root.limits)) {
     for (const raw of root.limits) {
-      const item = asObject3(raw);
+      const item = asObject5(raw);
       const windowMinutes = parseWindowMinutes(item?.window);
       const label = sanitizedLabel(item?.name);
       const bucket = windowMinutes === void 0 ? void 0 : parseUsageRow(item?.detail, windowMinutes, label ?? defaultWindowLabel(windowMinutes));
@@ -602,7 +826,7 @@ function normalizeKimiCodingUsagePayload(payload, capturedAt) {
   };
 }
 function parseUsageRow(value, windowMinutes, label) {
-  const row = asObject3(value);
+  const row = asObject5(value);
   if (!row) return void 0;
   const used = asNonnegativeInteger2(row.used);
   const limit = asNonnegativeInteger2(row.limit);
@@ -620,7 +844,7 @@ function parseUsageRow(value, windowMinutes, label) {
   };
 }
 function parseWindowMinutes(value) {
-  const window = asObject3(value);
+  const window = asObject5(value);
   if (!window) return void 0;
   const duration = asPositiveInteger(window.duration);
   if (duration === void 0) return void 0;
@@ -630,8 +854,8 @@ function parseWindowMinutes(value) {
   return Number.isSafeInteger(minutes) ? minutes : void 0;
 }
 function parseBoosterWallet(value) {
-  const wallet = asObject3(value);
-  const balance = asObject3(wallet?.balance);
+  const wallet = asObject5(value);
+  const balance = asObject5(wallet?.balance);
   if (!wallet || !balance || balance.type !== "BOOSTER") return [];
   const totalRaw = asPositiveInteger(balance.amount);
   if (totalRaw === void 0) return [];
@@ -682,7 +906,7 @@ function parseBoosterWallet(value) {
   return metrics;
 }
 function parseMoney(value) {
-  const money = asObject3(value);
+  const money = asObject5(value);
   if (!money) return void 0;
   const cents = asNonnegativeInteger2(money.priceInCents);
   if (cents === void 0) return void 0;
@@ -696,7 +920,7 @@ function fixedPointToMajor(value) {
   const major = roundedCents / 100;
   return Number.isSafeInteger(roundedCents) && Number.isFinite(major) ? major : void 0;
 }
-function asObject3(value) {
+function asObject5(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -770,12 +994,12 @@ var ZEN_WINDOWS = [
   { key: "monthly", label: "Monthly" }
 ];
 function normalizeOpenCodeZenPayload(payload, capturedAt) {
-  const usage = asObject4(payload.usage);
+  const usage = asObject6(payload.usage);
   if (!usage) throw new Error("OpenCode Zen usage response was not an object.");
   const buckets = [];
   const notes = [];
   for (const window of ZEN_WINDOWS) {
-    const raw = asObject4(usage[window.key]);
+    const raw = asObject6(usage[window.key]);
     if (!raw) continue;
     const status = asString3(raw.status);
     if (status !== "ok" && status !== "rate-limited") {
@@ -812,7 +1036,7 @@ function normalizeOpenCodeZenPayload(payload, capturedAt) {
     ...notes.length > 0 ? { notes } : {}
   };
 }
-function asObject4(value) {
+function asObject6(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -836,7 +1060,7 @@ function clampPercent2(value) {
 
 // src/providers/openrouter.ts
 function normalizeOpenRouterKeyPayload(payload, capturedAt) {
-  const data = asObject5(payload.data);
+  const data = asObject7(payload.data);
   if (!data) throw new Error("OpenRouter key response data was not an object.");
   const limit = asNonnegativeNumber3(data.limit);
   const remaining = asNonnegativeNumber3(data.limit_remaining);
@@ -882,7 +1106,7 @@ function addUsageMetric(metrics, id, label, value) {
   if (amount === void 0) return;
   metrics.push({ id, label, value: amount, unit: "usd" });
 }
-function asObject5(value) {
+function asObject7(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -1060,13 +1284,13 @@ function isRecord2(value) {
 var FIVE_HOUR_WINDOW_MINUTES2 = 300;
 var WEEKLY_WINDOW_MINUTES2 = 10080;
 function normalizeZaiQuotaPayload(providerId, providerName, payload, capturedAt) {
-  const data = asObject6(payload.data);
+  const data = asObject8(payload.data);
   if (!data) throw new Error("Z.AI quota response data was not an object.");
   const limits = Array.isArray(data.limits) ? data.limits : [];
   const buckets = [];
   const metrics = [];
   for (const raw of limits) {
-    const limit = asObject6(raw);
+    const limit = asObject8(raw);
     if (!limit) continue;
     const type = asString5(limit.type);
     const unit = asNonnegativeNumber4(limit.unit);
@@ -1138,7 +1362,7 @@ function addCountBucket(buckets, limit, id, label, windowMinutes) {
 function addUsageDetailMetrics(metrics, value) {
   if (!Array.isArray(value)) return;
   for (const raw of value) {
-    const detail = asObject6(raw);
+    const detail = asObject8(raw);
     if (!detail) continue;
     const label = asString5(detail.modelCode);
     const usage = asNonnegativeNumber4(detail.usage);
@@ -1146,7 +1370,7 @@ function addUsageDetailMetrics(metrics, value) {
     metrics.push({ id: `mcp-${kebabCase(label)}`, label, value: usage, unit: "count" });
   }
 }
-function asObject6(value) {
+function asObject8(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -1172,6 +1396,10 @@ function clampPercent3(value) {
 
 // src/query.ts
 var CODEX_USAGE_URL = "https://chatgpt.com/backend-api/wham/usage";
+var DEEPSEEK_BALANCE_URL = "https://api.deepseek.com/user/balance";
+var FIREWORKS_BILLING_SUMMARY_ORIGIN = "https://api.fireworks.ai";
+var FIREWORKS_SPEND_WINDOW_DAYS = 30;
+var FIREWORKS_MAX_ACCOUNT_PAGES = 5;
 var GITHUB_COPILOT_USAGE_URL = "https://api.github.com/copilot_internal/user";
 var OPENROUTER_KEY_URL = "https://openrouter.ai/api/v1/key";
 var OPENCODE_GO_USAGE_URL = "https://opencode.ai/zen/go/v1/usage";
@@ -1205,6 +1433,27 @@ var SUPPORTED_ADAPTERS = [
       return normalizeCodexBackendPayload(payload, Date.now());
     }
   },
+  {
+    id: "deepseek",
+    displayName: "DeepSeek",
+    semantics: { kind: "api-key", label: "DeepSeek API balance" },
+    async query(auth, signal, timeoutMs, guard) {
+      if (!guard) throw new Error("DeepSeek API balance requires request-boundary revalidation.");
+      const startedAt = Date.now();
+      await guard();
+      const remainingMs = timeoutMs - (Date.now() - startedAt);
+      if (remainingMs <= 0) throw new Error("Timed out while revalidating DeepSeek runtime auth.");
+      const payload = await fetchProviderJson(
+        DEEPSEEK_BALANCE_URL,
+        auth,
+        signal,
+        remainingMs,
+        "DeepSeek API balance endpoint",
+        { redirect: "error" }
+      );
+      return normalizeDeepSeekBalancePayload(payload, Date.now());
+    }
+  },
   {
     id: "github-copilot",
     displayName: "GitHub Copilot",
@@ -1238,6 +1487,35 @@ var SUPPORTED_ADAPTERS = [
       return normalizeOpenRouterKeyPayload(payload, Date.now());
     }
   },
+  {
+    id: "fireworks",
+    displayName: "Fireworks",
+    semantics: { kind: "api-key", label: "Fireworks API spend" },
+    async query(auth, signal, timeoutMs, guard, settings) {
+      if (!guard) throw new Error("Fireworks API spend requires request-boundary revalidation.");
+      const startedAt = Date.now();
+      await guard();
+      const accountId = await resolveFireworksAccountId(
+        auth,
+        signal,
+        remainingTimeout(timeoutMs, startedAt, "resolving the Fireworks account"),
+        guard,
+        settings?.fireworksAccountId
+      );
+      await guard();
+      const billingWindowAt = Date.now();
+      const payload = await fetchProviderJson(
+        fireworksBillingSummaryUrl(accountId, billingWindowAt),
+        auth,
+        signal,
+        remainingTimeout(timeoutMs, startedAt, "fetching Fireworks rated spend"),
+        "Fireworks billing summary endpoint",
+        { redirect: "error" }
+      );
+      await guard();
+      return normalizeFireworksBillingSummaryPayload(payload, accountId, Date.now());
+    }
+  },
   {
     id: "opencode-go",
     displayName: "OpenCode Go",
@@ -1273,7 +1551,6 @@ var SUPPORTED_ADAPTERS = [
     id: "zai",
     displayName: "Z.AI",
     semantics: { kind: "consumer-subscription", label: "GLM Coding Plan usage" },
-    publishesStatusline: false,
     async query(auth, signal, timeoutMs) {
       const payload = await fetchProviderJson(
         zaiMonitorUrl(auth.model.baseUrl),
@@ -1289,7 +1566,6 @@ var SUPPORTED_ADAPTERS = [
     id: "zai-coding-cn",
     displayName: "Z.AI Coding CN",
     semantics: { kind: "consumer-subscription", label: "GLM Coding Plan usage" },
-    publishesStatusline: false,
     async query(auth, signal, timeoutMs) {
       const payload = await fetchProviderJson(
         zaiMonitorUrl(auth.model.baseUrl),
@@ -1351,11 +1627,11 @@ var XAI_ADAPTER = {
     return normalizeXaiBillingPayload(billingPayload, userPayload.subscriptionTier, Date.now());
   }
 };
-function usageAdapters(xaiUsage = true) {
-  return xaiUsage ? [...SUPPORTED_ADAPTERS, XAI_ADAPTER] : SUPPORTED_ADAPTERS;
+function usageAdapters() {
+  return [...SUPPORTED_ADAPTERS, XAI_ADAPTER];
 }
-function adapterForProvider(providerId, xaiUsage = true) {
-  return usageAdapters(xaiUsage).find((adapter) => adapter.id === providerId);
+function adapterForProvider(providerId) {
+  return usageAdapters().find((adapter) => adapter.id === providerId);
 }
 function isStaleExtensionContextError(error) {
   return error instanceof Error && error.message.includes("This extension ctx is stale after session replacement or reload");
@@ -1372,11 +1648,14 @@ async function resolveUsageAuth(ctx, adapter, salt = AUTH_FINGERPRINT_SALT, cred
   if (!model) return void 0;
   const registry = ctx.modelRegistry;
   let modelAuth;
-  if (ctx.model?.provider === adapter.id && typeof registry.getApiKeyAndHeaders === "function") {
-    const result = await registry.getApiKeyAndHeaders(ctx.model);
+  const currentModel = ctx.model?.provider === adapter.id ? ctx.model : void 0;
+  const resolveCurrentModelAuth = async () => {
+    if (!currentModel || typeof registry.getApiKeyAndHeaders !== "function") return void 0;
+    const result = await registry.getApiKeyAndHeaders(currentModel);
     if (!result.ok) throw new Error(redactUsageError(result.error));
-    if (authorizationFrom(result)) modelAuth = result;
-  }
+    return authorizationFrom(result) ? result : void 0;
+  };
+  if (adapter.id !== "deepseek") modelAuth = await resolveCurrentModelAuth();
   if (typeof registry.getProviderAuth !== "function") {
     throw new Error("pi-usage requires Pi 0.81.0 or newer to validate resolved provider auth.");
   }
@@ -1386,6 +1665,7 @@ async function resolveUsageAuth(ctx, adapter, salt = AUTH_FINGERPRINT_SALT, cred
       `${adapter.displayName} usage cannot send a proxy-resolved credential to the official usage endpoint.`
     );
   }
+  if (adapter.id === "deepseek") modelAuth = await resolveCurrentModelAuth();
   const auth = modelAuth ?? providerResult?.auth;
   if (!auth) return void 0;
   if (adapter.id === "github-copilot") {
@@ -1406,6 +1686,26 @@ async function resolveUsageAuth(ctx, adapter, salt = AUTH_FINGERPRINT_SALT, cred
     if (!offered.ok) throw new Error("xAI OAuth credential discovery failed closed.");
     return resolveXaiUsageAuth(auth, model, salt, offered.candidates);
   }
+  if (adapter.id === "deepseek") {
+    const resolvedAuthorization = authorizationFrom(auth);
+    const access = bearerToken(resolvedAuthorization);
+    if (!access) throw new Error("DeepSeek API balance requires Bearer authentication.");
+    const authorization2 = `Bearer ${access}`;
+    const headers2 = { Authorization: authorization2 };
+    return {
+      apiKey: access,
+      headers: headers2,
+      fingerprint: fingerprintResolvedAuth({ headers: headers2 }, salt),
+      secrets: [
+        access,
+        auth.apiKey,
+        headerValue(auth.headers, "Authorization"),
+        resolvedAuthorization,
+        authorization2
+      ].filter((value) => Boolean(value)),
+      model
+    };
+  }
   const authorization = authorizationFrom(auth);
   if (!authorization) return void 0;
   const headers = { Authorization: authorization };
@@ -1420,9 +1720,9 @@ async function resolveUsageAuth(ctx, adapter, salt = AUTH_FINGERPRINT_SALT, cred
     model
   };
 }
-async function queryProviderUsage(adapter, auth, signal, timeoutMs, guard) {
+async function queryProviderUsage(adapter, auth, signal, timeoutMs, guard, settings) {
   try {
-    return await adapter.query(auth, signal, timeoutMs, guard);
+    return await adapter.query(auth, signal, timeoutMs, guard, settings);
   } catch (error) {
     if (isStaleExtensionContextError(error) || isAbortError(error)) throw error;
     throw new Error(redactUsageError(errorMessage(error), auth.secrets));
@@ -1563,7 +1863,7 @@ function resolveXaiUsageAuth(auth, model, salt, candidates) {
   const matches = [];
   for (const candidate of candidates) {
     try {
-      const credential = asObject7(candidate);
+      const credential = asObject9(candidate);
       if (credential?.type !== "oauth") continue;
       sawOAuth = true;
       if (credential.access !== resolvedAccess) continue;
@@ -1617,7 +1917,7 @@ function resolveGitHubCopilotUsageAuth(auth, model, salt, candidates, standalone
   const matches = /* @__PURE__ */ new Map();
   for (const candidate of candidates) {
     try {
-      const credential = asObject7(candidate);
+      const credential = asObject9(candidate);
       if (credential?.type !== "oauth") continue;
       sawOAuth = true;
       const storedAccess2 = typeof credential.access === "string" && credential.access ? credential.access : void 0;
@@ -1679,7 +1979,7 @@ function bearerToken(authorization) {
   const match = /^Bearer\s+(.+)$/iu.exec(authorization ?? "");
   return match?.[1];
 }
-function asObject7(value) {
+function asObject9(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -1698,6 +1998,8 @@ function hasOfficialUrlOrigin(value, providerId) {
   try {
     const url = new URL(value);
     if (providerId === "openai-codex") return url.origin === "https://chatgpt.com";
+    if (providerId === "deepseek") return url.origin === "https://api.deepseek.com";
+    if (providerId === "fireworks") return url.origin === FIREWORKS_BILLING_SUMMARY_ORIGIN;
     if (providerId === "openrouter") return url.origin === "https://openrouter.ai";
     if (providerId === "opencode-go") return url.origin === "https://opencode.ai";
     if (providerId === "kimi-coding") return url.origin === "https://api.kimi.com";
@@ -1727,11 +2029,87 @@ function validatedXaiUserId(value) {
   }
   return value;
 }
-function remainingTimeout(timeoutMs, startedAt) {
+function remainingTimeout(timeoutMs, startedAt, description = "fetching xAI consumer usage") {
   const remaining = timeoutMs - (Date.now() - startedAt);
-  if (remaining <= 0) throw new Error("Timed out while fetching xAI consumer usage.");
+  if (remaining <= 0) throw new Error(`Timed out while ${description}.`);
   return remaining;
 }
+async function resolveFireworksAccountId(auth, signal, timeoutMs, guard, configuredAccountId) {
+  if (configuredAccountId !== void 0 && !isFireworksAccountId(configuredAccountId)) {
+    throw new Error("The Fireworks account setting was not a safe account slug.");
+  }
+  const startedAt = Date.now();
+  const accounts = [];
+  let pageToken;
+  for (let page = 0; page < FIREWORKS_MAX_ACCOUNT_PAGES; page += 1) {
+    await guard();
+    const payload = await fetchProviderJson(
+      fireworksAccountsUrl(pageToken),
+      auth,
+      signal,
+      remainingTimeout(timeoutMs, startedAt, "fetching Fireworks accounts"),
+      "Fireworks accounts endpoint",
+      { redirect: "error" }
+    );
+    for (const accountId of normalizeFireworksAccountsPayload(
+      payload
+    )) {
+      if (accounts.includes(accountId)) {
+        throw new Error(`Fireworks accounts listing repeated ${accountId}.`);
+      }
+      accounts.push(accountId);
+      if (configuredAccountId === accountId) return accountId;
+    }
+    pageToken = fireworksNextPageToken(payload.nextPageToken);
+    if (!pageToken) break;
+  }
+  if (pageToken) {
+    throw new Error(
+      configuredAccountId ? `The configured Fireworks account was not found within the first ${FIREWORKS_MAX_ACCOUNT_PAGES} listing pages.` : `Fireworks account listing exceeded ${FIREWORKS_MAX_ACCOUNT_PAGES} pages; set fireworksAccountId in pi-usage.json to an account returned in those pages.`
+    );
+  }
+  if (accounts.length === 0) {
+    throw new Error("Fireworks account discovery returned no accounts for this API key.");
+  }
+  if (configuredAccountId) {
+    throw new Error(
+      "The configured Fireworks account does not match an account visible to this API key."
+    );
+  }
+  if (accounts.length === 1) return accounts[0];
+  const preview = accounts.slice(0, 8).join(", ");
+  const suffix = accounts.length > 8 ? ` \u2026and ${accounts.length - 8} more` : "";
+  throw new Error(
+    `The Fireworks key can see ${accounts.length} accounts (${preview}${suffix}); set fireworksAccountId in pi-usage.json to one of them.`
+  );
+}
+function fireworksAccountsUrl(pageToken) {
+  const url = new URL("/v1/accounts", FIREWORKS_BILLING_SUMMARY_ORIGIN);
+  url.searchParams.set("pageSize", "200");
+  if (pageToken !== void 0) url.searchParams.set("pageToken", pageToken);
+  return url.toString();
+}
+function fireworksNextPageToken(value) {
+  if (value === void 0 || value === null) return void 0;
+  if (typeof value !== "string" || !value || value.length > 512) {
+    throw new Error("Fireworks accounts listing returned an invalid page token.");
+  }
+  return value;
+}
+function fireworksBillingSummaryUrl(accountId, startedAt) {
+  const dayMs = 24 * 60 * 60 * 1e3;
+  const dayFloor = (time) => `${new Date(time).toISOString().slice(0, 10)}T00:00:00Z`;
+  const url = new URL(
+    `/v1/accounts/${accountId}/billing/summary`,
+    FIREWORKS_BILLING_SUMMARY_ORIGIN
+  );
+  url.searchParams.set(
+    "startTime",
+    dayFloor(startedAt - (FIREWORKS_SPEND_WINDOW_DAYS - 1) * dayMs)
+  );
+  url.searchParams.set("endTime", dayFloor(startedAt + dayMs));
+  return url.toString();
+}
 function zaiMonitorUrl(baseUrl) {
   const base = baseUrl?.trim();
   if (!base) throw new Error("Z.AI model base URL is unavailable.");
@@ -1885,7 +2263,7 @@ function normalizeCodexResetCreditsPayload(payload) {
   if (rawCredits !== void 0 && !Array.isArray(rawCredits)) {
     throw new Error("Codex reset credits response returned invalid credits.");
   }
-  const options = (rawCredits ?? []).map(asObject8).filter((credit) => Boolean(credit)).filter((credit) => credit.status === "available" && credit.reset_type === "codex_rate_limits").map(normalizeResetOption).sort(
+  const options = (rawCredits ?? []).map(asObject10).filter((credit) => Boolean(credit)).filter((credit) => credit.status === "available" && credit.reset_type === "codex_rate_limits").map(normalizeResetOption).sort(
     (left, right) => (left.expiresAt ?? Number.MAX_SAFE_INTEGER) - (right.expiresAt ?? Number.MAX_SAFE_INTEGER)
   ).slice(0, Math.min(availableCount, MAX_RESET_OPTIONS));
   if (availableCount > 0 && options.length === 0) {
@@ -1900,7 +2278,7 @@ function selectCodexResetCredential(candidates, resolvedAccess, resolvedAccountI
   const matches = /* @__PURE__ */ new Map();
   for (const candidate of candidates) {
     try {
-      const credential = asObject8(candidate);
+      const credential = asObject10(candidate);
       if (credential?.type !== "oauth") continue;
       sawOAuth = true;
       const storedAccess = asNonemptyString(credential.access);
@@ -1939,7 +2317,7 @@ function codexAccountIdFromAccessToken(access) {
     const parts = access.split(".");
     if (parts.length !== 3 || !parts[1]) return void 0;
     const payload = JSON.parse(Buffer.from(parts[1], "base64url").toString("utf8"));
-    const claims = asObject8(asObject8(payload)?.["https://api.openai.com/auth"]);
+    const claims = asObject10(asObject10(payload)?.["https://api.openai.com/auth"]);
     return validHeaderValue(claims?.chatgpt_account_id);
   } catch {
     return void 0;
@@ -1971,7 +2349,7 @@ function normalizeResetOption(credit) {
 function isCodexResetOutcomeCode(value) {
   return value === "reset" || value === "nothing_to_reset" || value === "no_credit" || value === "already_redeemed";
 }
-function asObject8(value) {
+function asObject10(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -2012,10 +2390,13 @@ var BAR_SEGMENTS = 20;
 var VALUE_COLUMN = 29;
 function formatUsageReport(report, displayState) {
   const stateLabel = displayState === "current" ? "Current" : "Configured";
-  const lines = [`${report.providerName} Usage \xB7 ${stateLabel}`];
+  const title = report.providerId === "deepseek" ? "DeepSeek API Balance" : report.providerId === "fireworks" ? "Fireworks API Spend" : `${report.providerName} Usage`;
+  const lines = [`${title} \xB7 ${stateLabel}`];
   if (report.accountLabel) lines.push(`Account: ${report.accountLabel}`);
   lines.push(`Semantics: ${report.semantics.label}`, "");
   if (report.providerId === "openai-codex") formatCodexReport(lines, report);
+  else if (report.providerId === "deepseek") formatDeepSeekReport(lines, report);
+  else if (report.providerId === "fireworks") formatFireworksReport(lines, report);
   else if (report.providerId === "github-copilot") formatGitHubCopilotReport(lines, report);
   else if (report.providerId === "openrouter") formatOpenRouterReport(lines, report);
   else if (report.providerId === "opencode-go") formatOpenCodeZenReport(lines, report);
@@ -2029,8 +2410,12 @@ function formatUsageReport(report, displayState) {
   }
   return lines.join("\n").trimEnd();
 }
-function formatUsageStatusline(report, model) {
-  if (report.providerId === "openai-codex") return formatCodexStatusline(report, model);
+function formatUsageStatusline(report, model, now = Date.now(), showCodexResetCountdown = true) {
+  if (report.providerId === "openai-codex") {
+    return formatCodexStatusline(report, model, now, showCodexResetCountdown);
+  }
+  if (report.providerId === "deepseek") return formatDeepSeekStatusline(report);
+  if (report.providerId === "fireworks") return formatFireworksStatusline(report);
   if (report.providerId === "github-copilot") return formatGitHubCopilotStatusline(report);
   if (report.providerId === "openrouter") {
     const limit = report.buckets.find((bucket) => bucket.id === "key-limit");
@@ -2040,6 +2425,9 @@ function formatUsageStatusline(report, model) {
   }
   if (report.providerId === "opencode-go") return formatOpenCodeZenStatusline(report);
   if (report.providerId === "kimi-coding") return formatKimiCodingStatusline(report);
+  if (report.providerId === "zai" || report.providerId === "zai-coding-cn") {
+    return formatZaiStatusline(report);
+  }
   return void 0;
 }
 function formatProviderStates(states) {
@@ -2073,6 +2461,54 @@ function formatCodexReport(lines, report) {
     }
   }
 }
+function formatDeepSeekReport(lines, report) {
+  const availability = report.metrics.find((metric) => metric.id === "api-availability");
+  lines.push(
+    `${"API calls:".padEnd(VALUE_COLUMN)}${availability?.value === "available" ? "Available" : "Unavailable"}`
+  );
+  for (const currency of ["CNY", "USD"]) {
+    const metrics = report.metrics.filter((metric) => metric.currency === currency);
+    if (metrics.length === 0) continue;
+    lines.push("", `${currency} balance:`);
+    for (const metric of metrics) {
+      lines.push(`${`${metric.label}:`.padEnd(VALUE_COLUMN)}${currency} ${metric.value}`);
+    }
+  }
+}
+function formatDeepSeekStatusline(report) {
+  const availability = report.metrics.find((metric) => metric.id === "api-availability");
+  if (availability?.value !== "available") return "deepseek API unavailable";
+  const totals = ["CNY", "USD"].flatMap((currency) => {
+    const metric = report.metrics.find(
+      (candidate) => candidate.id === `${currency.toLowerCase()}-total`
+    );
+    return metric ? [`${currency} ${metric.value}`] : [];
+  });
+  return totals.length > 0 ? `deepseek ${totals.join(" \xB7 ")}` : "deepseek balance unavailable";
+}
+function formatFireworksReport(lines, report) {
+  lines.push(`${"Spend window:".padEnd(VALUE_COLUMN)}Last 30 days (rated)`);
+  for (const currency of fireworksCurrencies(report)) {
+    lines.push("", `${currency} rated spend:`);
+    for (const metric of report.metrics) {
+      if (metric.currency !== currency) continue;
+      lines.push(`${`${metric.label}:`.padEnd(VALUE_COLUMN)}${currency} ${metric.value}`);
+    }
+  }
+}
+function formatFireworksStatusline(report) {
+  const totals = report.metrics.filter((metric) => metric.id.endsWith("-total"));
+  if (totals.length === 0) return "fireworks no rated usage";
+  return `fireworks ${totals.map((metric) => `${metric.currency} ${metric.value}`).join(" \xB7 ")}`;
+}
+function fireworksCurrencies(report) {
+  const currencies = [];
+  for (const metric of report.metrics) {
+    if (!metric.currency || currencies.includes(metric.currency)) continue;
+    currencies.push(metric.currency);
+  }
+  return currencies;
+}
 function formatGitHubCopilotReport(lines, report) {
   const quota = findGitHubCopilotQuota(report);
   if (!quota || quota.limit === void 0 || quota.remaining === void 0) {
@@ -2187,6 +2623,21 @@ function formatKimiCodingStatusline(report) {
   }
   return parts.length > 1 ? parts.join(" ") : void 0;
 }
+function formatZaiStatusline(report) {
+  const selected = [
+    report.buckets.find((bucket) => bucket.id === "five-hour"),
+    report.buckets.find((bucket) => bucket.id === "weekly")
+  ];
+  const parts = ["zai"];
+  for (const bucket of selected) {
+    if (!bucket?.limit || bucket.remaining === void 0) continue;
+    const fallback = bucket.id === "weekly" ? "weekly" : "5h";
+    parts.push(
+      `${percentRemaining(bucket)}% ${formatWindowLabel(bucket.windowMinutes, fallback, true)}`
+    );
+  }
+  return parts.length > 1 ? parts.join(" ") : void 0;
+}
 function formatCurrencyMetric(metric) {
   if (typeof metric.value !== "number") return String(metric.value);
   if (!metric.currency) return "unavailable";
@@ -2258,7 +2709,7 @@ function formatGenericReport(lines, report) {
     );
   }
 }
-function formatCodexStatusline(report, model) {
+function formatCodexStatusline(report, model, now = Date.now(), showResetCountdown = true) {
   const group = selectCodexGroup(report, model);
   if (!group) return formatCodexCreditsStatus(report);
   const buckets = report.buckets.filter((bucket) => (bucket.groupId ?? bucket.id) === group);
@@ -2268,10 +2719,15 @@ function formatCodexStatusline(report, model) {
   ];
   for (const bucket of buckets) {
     if (bucket.remaining === void 0) continue;
+    const percent = `${clampPercent4(bucket.remaining).toFixed(0)}%`;
     const fallback = bucket.id.endsWith(":secondary") ? "weekly" : "5h";
-    parts.push(
-      `${clampPercent4(bucket.remaining).toFixed(0)}% ${formatWindowLabel(bucket.windowMinutes, fallback, true)}`
-    );
+    const window = formatWindowLabel(bucket.windowMinutes, fallback, true);
+    if (!showResetCountdown) {
+      parts.push(`${percent} ${window}`);
+      continue;
+    }
+    const reset = formatResetCountdown(bucket.resetsAt, now);
+    parts.push(`${percent} ${reset ? `\u21BB ${reset}` : window}`);
   }
   return parts.length > 1 ? parts.join(" ") : formatCodexCreditsStatus(report);
 }
@@ -2352,6 +2808,23 @@ function formatWindowLabel(minutes, fallback, compact) {
   if (minutes % 60 === 0) return `${minutes / 60}h`;
   return `${minutes}m`;
 }
+function formatResetCountdown(resetsAt, now) {
+  if (resetsAt === void 0 || !Number.isFinite(resetsAt) || !Number.isFinite(now))
+    return void 0;
+  const totalMinutes = Math.max(0, Math.ceil((resetsAt * 1e3 - now) / 6e4));
+  const days = Math.floor(totalMinutes / 1440);
+  const hours = Math.floor(totalMinutes % 1440 / 60);
+  const minutes = totalMinutes % 60;
+  if (days > 0) {
+    return [
+      `${String(days)}d`,
+      hours > 0 ? `${String(hours)}h` : minutes > 0 ? `${String(minutes)}m` : ""
+    ].filter(Boolean).join("");
+  }
+  if (hours > 0)
+    return [`${String(hours)}h`, minutes > 0 ? `${String(minutes)}m` : ""].filter(Boolean).join("");
+  return `${String(minutes)}m`;
+}
 function formatMetricValue(value, unit) {
   if (unit === "usd" && typeof value === "number") return formatUsd(value);
   return String(value);
@@ -2384,7 +2857,7 @@ var USAGE_SETTINGS_FILE = "pi-usage.json";
 var MAX_USAGE_SETTINGS_BYTES = 64 * 1024;
 var DEFAULT_USAGE_SETTINGS = Object.freeze({
   codexFastMode: false,
-  xaiUsage: true
+  codexStatusResetCountdown: true
 });
 function usageSettingsPath() {
   return join(getAgentDir(), USAGE_SETTINGS_FILE);
@@ -2394,12 +2867,16 @@ function normalizeUsageSettings(value) {
   if (Object.hasOwn(value, "codexFastMode") && typeof value.codexFastMode !== "boolean") {
     return void 0;
   }
-  if (Object.hasOwn(value, "xaiUsage") && typeof value.xaiUsage !== "boolean") {
+  if (Object.hasOwn(value, "codexStatusResetCountdown") && typeof value.codexStatusResetCountdown !== "boolean") {
+    return void 0;
+  }
+  if (Object.hasOwn(value, "fireworksAccountId") && !isFireworksAccountId(value.fireworksAccountId)) {
     return void 0;
   }
   return {
     codexFastMode: typeof value.codexFastMode === "boolean" ? value.codexFastMode : DEFAULT_USAGE_SETTINGS.codexFastMode,
-    xaiUsage: typeof value.xaiUsage === "boolean" ? value.xaiUsage : DEFAULT_USAGE_SETTINGS.xaiUsage
+    codexStatusResetCountdown: typeof value.codexStatusResetCountdown === "boolean" ? value.codexStatusResetCountdown : DEFAULT_USAGE_SETTINGS.codexStatusResetCountdown,
+    ...isFireworksAccountId(value.fireworksAccountId) ? { fireworksAccountId: value.fireworksAccountId } : {}
   };
 }
 async function loadUsageSettings(path = usageSettingsPath(), signal) {
@@ -2483,7 +2960,11 @@ async function saveUsageSettingsPatch(path, patch, operations, signal) {
   if (latest.kind === "invalid") {
     throw new Error("Cannot overwrite an invalid pi-usage.json; repair it and reload first");
   }
-  const document = { ...latest.document, ...patch };
+  const document = { ...latest.document };
+  for (const [key, value] of Object.entries(patch)) {
+    if (value === void 0) delete document[key];
+    else document[key] = value;
+  }
   const settings = normalizeUsageSettings(document);
   if (!settings) throw new Error("Refusing to save invalid pi-usage settings");
   const directory = dirname(path);
@@ -2529,7 +3010,7 @@ import { randomUUID as randomUUID2 } from "node:crypto";
 // src/codex-fast-runtime.ts
 var NO_FAST_REQUEST = /* @__PURE__ */ Symbol("no-fast-request");
 var FAST_USAGE_WARNING = "Fast is about 1.5\xD7 faster and uses more of your plan allowance.";
-function registerCodexFastMode(pi, settingsRuntime, refreshStatus) {
+function registerCodexFastMode(pi, settingsRuntime, refreshStatus, options = {}) {
   let sessionController = new AbortController();
   let generation = 0;
   const pendingFastRequests = /* @__PURE__ */ new Map();
@@ -2585,37 +3066,42 @@ function registerCodexFastMode(pi, settingsRuntime, refreshStatus) {
       await toggle(ctx, !availability.enabled);
     }
   });
-  pi.on("session_start", async (_event, ctx) => {
+  const prepareSession = (ctx) => {
+    const sessionId = ctx.sessionManager.getSessionId();
     generation += 1;
     sessionController.abort();
     pendingFastRequests.clear();
     sessionController = new AbortController();
     const ownerGeneration = generation;
-    const sessionId = ctx.sessionManager.getSessionId();
-    let state;
-    try {
-      state = await settingsRuntime.reload(sessionController.signal);
-    } catch (error) {
-      if (sessionController.signal.aborted || ownerGeneration !== generation) return;
-      if (ctx.hasUI) {
+    return (async () => {
+      let state;
+      try {
+        state = await settingsRuntime.reload(sessionController.signal);
+      } catch (error) {
+        if (sessionController.signal.aborted || ownerGeneration !== generation) return;
+        if (ctx.hasUI) {
+          ctx.ui.notify(
+            `Could not load pi-usage.json; using defaults. ${errorMessage(error)}`,
+            "warning"
+          );
+        }
+        return;
+      }
+      if (sessionController.signal.aborted || ownerGeneration !== generation || ctx.sessionManager.getSessionId() !== sessionId) {
+        return;
+      }
+      if (ctx.hasUI && state.kind === "invalid") {
         ctx.ui.notify(
-          `Could not load pi-usage.json; using defaults. ${errorMessage(error)}`,
+          `Invalid pi-usage.json; using defaults without overwriting it. ${state.issue}`,
           "warning"
         );
       }
-      return;
-    }
-    if (sessionController.signal.aborted || ownerGeneration !== generation || ctx.sessionManager.getSessionId() !== sessionId) {
-      return;
-    }
-    if (ctx.hasUI && state.kind === "invalid") {
-      ctx.ui.notify(
-        `Invalid pi-usage.json; using defaults without overwriting it. ${state.issue}`,
-        "warning"
-      );
-    }
-    refreshStatus(ctx);
-  });
+      refreshStatus(ctx);
+    })();
+  };
+  if (options.registerSessionStart !== false) {
+    pi.on("session_start", async (_event, ctx) => prepareSession(ctx));
+  }
   pi.on("before_provider_request", (event, ctx) => {
     const rewritten = rewriteCodexFastPayload(
       event.payload,
@@ -2648,6 +3134,7 @@ function registerCodexFastMode(pi, settingsRuntime, refreshStatus) {
     await settingsRuntime.flush();
   });
   return {
+    prepareSession,
     availability(model) {
       return codexFastAvailability(model, settingsRuntime.get().settings.codexFastMode);
     },
@@ -2684,8 +3171,8 @@ function isAbortError2(error) {
 }
 
 // src/usage-helpers.ts
-function configuredAdapters(ctx, xaiUsage = true) {
-  return usageAdapters(xaiUsage).filter(
+function configuredAdapters(ctx) {
+  return usageAdapters().filter(
     (adapter) => adapter.id === ctx.model?.provider || providerIsConfigured(ctx, adapter.id)
   );
 }
@@ -2726,6 +3213,8 @@ import {
   SettingsList,
   Text
 } from "@earendil-works/pi-tui";
+var AUTO = "Auto";
+var EDIT = "Edit\u2026";
 var OFF = "Off";
 var ON = "On";
 async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent, onApplied) {
@@ -2733,7 +3222,23 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
     if (ctx.hasUI) ctx.ui.notify(`Edit settings manually: ${settingsRuntime.get().path}`, "info");
     return false;
   }
-  if (parentSignal.aborted || !isCurrent()) return false;
+  let changed = false;
+  while (!parentSignal.aborted && isCurrent()) {
+    const result = await showSettingsList(ctx, settingsRuntime, parentSignal, isCurrent, onApplied);
+    if (!result) return changed;
+    changed ||= result.changed;
+    if (!result.editFireworksAccount) return changed;
+    changed ||= await editFireworksAccount(
+      ctx,
+      settingsRuntime,
+      parentSignal,
+      isCurrent,
+      onApplied
+    );
+  }
+  return changed;
+}
+async function showSettingsList(ctx, settingsRuntime, parentSignal, isCurrent, onApplied) {
   return ctx.ui.custom((tui, theme, _keybindings, done) => {
     const localController = new AbortController();
     const signal = AbortSignal.any([parentSignal, localController.signal]);
@@ -2741,6 +3246,7 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
     let closing = false;
     let saveQueue = Promise.resolve();
     const state = settingsRuntime.get();
+    const fireworksValue = state.settings.fireworksAccountId ?? AUTO;
     const items = [
       {
         id: "codexFastMode",
@@ -2750,11 +3256,18 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
         values: [OFF, ON]
       },
       {
-        id: "xaiUsage",
-        label: "xAI usage",
-        description: "Report OAuth subscription allowance and credits.",
-        currentValue: state.kind !== "invalid" && state.settings.xaiUsage ? ON : OFF,
+        id: "codexStatusResetCountdown",
+        label: "Codex reset countdown",
+        description: "Show time remaining until each Codex usage limit resets.",
+        currentValue: state.settings.codexStatusResetCountdown ? ON : OFF,
         values: [OFF, ON]
+      },
+      {
+        id: "fireworksAccountId",
+        label: "Fireworks account",
+        description: "Select Edit to enter a visible account slug, or submit blank to clear it.",
+        currentValue: fireworksValue,
+        values: state.settings.fireworksAccountId ? [state.settings.fireworksAccountId, EDIT] : [AUTO, EDIT]
       }
     ];
     const container = new Container();
@@ -2764,7 +3277,36 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
       if (closing) return;
       closing = true;
       localController.abort();
-      done(changed);
+      done({ changed, editFireworksAccount: false });
+    };
+    const queueUpdate = (id, requested, display) => {
+      saveQueue = saveQueue.then(async () => {
+        const previous = settingsRuntime.get().settings[id];
+        if (settingsRuntime.get().kind === "invalid") {
+          settingsList.updateValue(id, displaySetting(id, previous));
+          if (!signal.aborted && isCurrent()) {
+            ctx.ui.notify("Repair pi-usage.json and reload before changing settings.", "error");
+            tui.requestRender();
+          }
+          return;
+        }
+        try {
+          await settingsRuntime.update({ [id]: requested }, signal);
+        } catch (error) {
+          if (signal.aborted || !isCurrent()) return;
+          settingsList.updateValue(id, displaySetting(id, previous));
+          ctx.ui.notify(`Could not save pi-usage.json: ${errorMessage(error)}`, "error");
+          tui.requestRender();
+          return;
+        }
+        if (previous !== requested) {
+          changed = true;
+          onApplied(id);
+        }
+        if (signal.aborted || !isCurrent()) return;
+        settingsList.updateValue(id, display);
+        tui.requestRender();
+      });
     };
     settingsList = new SettingsList(
       items,
@@ -2772,36 +3314,18 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
       getSettingsListTheme(),
       (id, value) => {
         if (closing || signal.aborted || !isCurrent()) return;
-        const settingId = id;
-        const requested = value !== OFF;
-        saveQueue = saveQueue.then(async () => {
-          const previous = settingsRuntime.get().settings[settingId];
-          if (settingsRuntime.get().kind === "invalid") {
-            const effectivePrevious = settingId === "xaiUsage" ? false : previous;
-            settingsList.updateValue(id, displayValue(settingId, effectivePrevious));
-            if (!signal.aborted && isCurrent()) {
-              ctx.ui.notify("Repair pi-usage.json and reload before changing settings.", "error");
-              tui.requestRender();
-            }
-            return;
-          }
-          try {
-            await settingsRuntime.update({ [settingId]: requested }, signal);
-          } catch (error) {
-            if (signal.aborted || !isCurrent()) return;
-            settingsList.updateValue(id, displayValue(settingId, previous));
-            ctx.ui.notify(`Could not save pi-usage.json: ${errorMessage(error)}`, "error");
-            tui.requestRender();
-            return;
-          }
-          if (previous !== requested) {
-            changed = true;
-            onApplied(settingId, previous, requested);
+        if (id === "fireworksAccountId") {
+          if (value === EDIT) {
+            saveQueue = saveQueue.then(() => {
+              if (closing || signal.aborted || !isCurrent()) return;
+              closing = true;
+              done({ changed, editFireworksAccount: true });
+            });
           }
-          if (signal.aborted || !isCurrent()) return;
-          settingsList.updateValue(id, displayValue(settingId, requested));
-          tui.requestRender();
-        });
+          return;
+        }
+        const settingId = id;
+        queueUpdate(settingId, value !== OFF, value);
       },
       cancel
     );
@@ -2823,12 +3347,46 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
     };
   });
 }
-function displayValue(_id, enabled) {
-  return enabled ? ON : OFF;
+async function editFireworksAccount(ctx, settingsRuntime, signal, isCurrent, onApplied) {
+  while (!signal.aborted && isCurrent()) {
+    const state = settingsRuntime.get();
+    if (state.kind === "invalid") {
+      ctx.ui.notify("Repair pi-usage.json and reload before changing settings.", "error");
+      return false;
+    }
+    const entered = await ctx.ui.input(
+      "Fireworks account slug \xB7 submit blank for Auto",
+      state.settings.fireworksAccountId ?? "Example: acme",
+      { signal }
+    );
+    if (signal.aborted || !isCurrent() || entered === void 0) return false;
+    const normalized = entered.trim();
+    const requested = normalized || void 0;
+    if (requested !== void 0 && !isFireworksAccountId(requested)) {
+      ctx.ui.notify("Enter a URL-safe Fireworks account slug.", "warning");
+      continue;
+    }
+    if (requested === state.settings.fireworksAccountId) return false;
+    try {
+      await settingsRuntime.update({ fireworksAccountId: requested }, signal);
+    } catch (error) {
+      if (signal.aborted || !isCurrent()) return false;
+      ctx.ui.notify(`Could not save pi-usage.json: ${errorMessage(error)}`, "error");
+      return false;
+    }
+    onApplied("fireworksAccountId");
+    return true;
+  }
+  return false;
+}
+function displaySetting(id, value) {
+  if (id === "fireworksAccountId") return typeof value === "string" ? value : AUTO;
+  return value ? ON : OFF;
 }
 
 // src/usage.ts
 var CACHE_TTL_MS = 5 * 60 * 1e3;
+var STATUS_COUNTDOWN_REFRESH_MS = 60 * 1e3;
 var DEFAULT_TIMEOUT_MS = 15e3;
 var ALL_PROVIDER_CONCURRENCY = 2;
 var FAILURE_BACKOFF_MS = 3e4;
@@ -2854,19 +3412,22 @@ function usageExtension(pi, dependencies = {}) {
   let sessionActive = false;
   let statusGeneration = 0;
   let sessionGeneration = 0;
-  let xaiSettingsGeneration = 0;
   let statusRefreshTimer;
+  let statusCountdownTimer;
   let statusController;
   let fastRuntime;
-  const xaiUsageEnabled = () => {
-    const state = settingsRuntime.get();
-    return state.kind !== "invalid" && state.settings.xaiUsage;
-  };
-  const activeAdapterForProvider = (providerId) => adapterForProvider(providerId, xaiUsageEnabled());
-  const clearStatusTimer = () => {
+  const clearStatusRefreshTimer = () => {
     if (statusRefreshTimer) clearTimeout(statusRefreshTimer);
     statusRefreshTimer = void 0;
   };
+  const clearStatusCountdownTimer = () => {
+    if (statusCountdownTimer) clearTimeout(statusCountdownTimer);
+    statusCountdownTimer = void 0;
+  };
+  const clearStatusTimers = () => {
+    clearStatusRefreshTimer();
+    clearStatusCountdownTimer();
+  };
   const safeSetStatus = (ctx, value) => {
     try {
       ctx.ui.setStatus(STATUS_KEY, value);
@@ -2880,11 +3441,11 @@ function usageExtension(pi, dependencies = {}) {
     statusGeneration += 1;
     statusController?.abort();
     statusController = void 0;
-    clearStatusTimer();
+    clearStatusTimers();
     safeSetStatus(ctx, void 0);
   };
   const scheduleStatusRefresh = (ctx, model) => {
-    clearStatusTimer();
+    clearStatusRefreshTimer();
     const generation = statusGeneration;
     statusRefreshTimer = setTimeout(() => {
       statusRefreshTimer = void 0;
@@ -2894,13 +3455,14 @@ function usageExtension(pi, dependencies = {}) {
     statusRefreshTimer.unref?.();
   };
   const publishStatus = (ctx, outcome, model, shouldSchedule) => {
-    if (activeAdapterForProvider(model.provider)?.publishesStatusline === false) {
-      clearStatusTimer();
+    clearStatusCountdownTimer();
+    if (adapterForProvider(model.provider)?.publishesStatusline === false) {
+      clearStatusRefreshTimer();
       safeSetStatus(ctx, void 0);
       return;
     }
     if (outcome.state.status === "unsupported") {
-      clearStatusTimer();
+      clearStatusRefreshTimer();
       safeSetStatus(ctx, void 0);
       return;
     }
@@ -2913,10 +3475,30 @@ function usageExtension(pi, dependencies = {}) {
       }
       return;
     }
-    const rawValue = formatUsageStatusline(outcome.state.report, model);
+    const showCodexResetCountdown = outcome.state.report.providerId === "openai-codex" && settingsRuntime.get().settings.codexStatusResetCountdown;
+    const now = Date.now();
+    const rawValue = formatUsageStatusline(
+      outcome.state.report,
+      model,
+      now,
+      showCodexResetCountdown
+    );
     const value = rawValue ? fastRuntime.decorateStatus(model, rawValue) : void 0;
     if (!safeSetStatus(ctx, value)) return;
     if (shouldSchedule && sessionActive) scheduleStatusRefresh(ctx, model);
+    if (sessionActive && showCodexResetCountdown && outcome.state.report.buckets.some(
+      (bucket) => bucket.resetsAt !== void 0 && Number.isFinite(bucket.resetsAt) && bucket.resetsAt * 1e3 > now
+    )) {
+      const generation = statusGeneration;
+      statusCountdownTimer = setTimeout(() => {
+        statusCountdownTimer = void 0;
+        if (!sessionActive || generation !== statusGeneration || modelIdentity(ctx.model) !== modelIdentity(model)) {
+          return;
+        }
+        publishStatus(ctx, outcome, model, false);
+      }, STATUS_COUNTDOWN_REFRESH_MS);
+      statusCountdownTimer.unref?.();
+    }
   };
   const invalidateProviderState = (providerId) => {
     cache.clearProvider(providerId);
@@ -2938,18 +3520,18 @@ function usageExtension(pi, dependencies = {}) {
     }
     activeCurrentIdentity = nextIdentity;
   };
-  const queryAdapterState = async (ctx, adapter, displayState, force, signal) => {
-    const startedAt = Date.now();
+  const queryAdapterState = async (ctx, adapter, displayState, force, signal, authRetry = 0, deadlineAt = Date.now() + DEFAULT_TIMEOUT_MS) => {
     const expectedSessionGeneration = sessionGeneration;
-    const expectedXaiSettingsGeneration = xaiSettingsGeneration;
     const expectedSessionId = ctx.sessionManager.getSessionId();
     const expectedModelIdentity = modelIdentity(ctx.model);
+    const expectedFireworksAccountId = adapter.id === "fireworks" ? settingsRuntime.get().settings.fireworksAccountId : void 0;
+    const querySettings = adapter.id === "fireworks" ? { fireworksAccountId: expectedFireworksAccountId } : void 0;
     let auth;
     try {
       auth = await awaitWithDeadline(
         resolveUsageAuth(ctx, adapter, void 0, credentialReader, credentialCandidates),
         signal,
-        DEFAULT_TIMEOUT_MS,
+        Math.max(1, deadlineAt - Date.now()),
         `resolving ${adapter.displayName} runtime auth`
       );
     } catch (error) {
@@ -2967,9 +3549,9 @@ function usageExtension(pi, dependencies = {}) {
         }
       };
     }
-    if (adapter.id === "xai" && (expectedSessionGeneration !== sessionGeneration || expectedXaiSettingsGeneration !== xaiSettingsGeneration || !xaiUsageEnabled() || ctx.sessionManager.getSessionId() !== expectedSessionId || modelIdentity(ctx.model) !== expectedModelIdentity)) {
-      throw abortError();
-    }
+    const requiresRequestBoundaryGuard = ["deepseek", "fireworks", "xai"].includes(adapter.id);
+    const requestContextChanged = () => expectedSessionGeneration !== sessionGeneration || ctx.sessionManager.getSessionId() !== expectedSessionId || modelIdentity(ctx.model) !== expectedModelIdentity || adapter.id === "fireworks" && settingsRuntime.get().settings.fireworksAccountId !== expectedFireworksAccountId;
+    if (requiresRequestBoundaryGuard && requestContextChanged()) throw abortError();
     if (!auth) {
       if (displayState === "current") {
         transitionCurrentIdentity(`${adapter.id}:unavailable`, adapter.id);
@@ -2985,10 +3567,11 @@ function usageExtension(pi, dependencies = {}) {
         authState: "unavailable"
       };
     }
+    const queryFingerprint = adapter.id === "fireworks" ? `${auth.fingerprint}:account:${expectedFireworksAccountId ?? "auto"}` : auth.fingerprint;
     if (displayState === "current") {
-      transitionCurrentIdentity(`${adapter.id}:${auth.fingerprint}`, adapter.id);
+      transitionCurrentIdentity(`${adapter.id}:${queryFingerprint}`, adapter.id);
     }
-    const cached = !force ? cache.get(adapter.id, auth.fingerprint) : void 0;
+    const cached = !force ? cache.get(adapter.id, queryFingerprint) : void 0;
     if (cached) {
       return {
         state: {
@@ -3001,7 +3584,7 @@ function usageExtension(pi, dependencies = {}) {
         fingerprint: auth.fingerprint
       };
     }
-    const failureKey = `${adapter.id}:${auth.fingerprint}`;
+    const failureKey = `${adapter.id}:${queryFingerprint}`;
     const previousFailure = failureBackoff.get(failureKey);
     if (!force && previousFailure && previousFailure.until > Date.now()) {
       return {
@@ -3019,26 +3602,37 @@ function usageExtension(pi, dependencies = {}) {
     querySequence += 1;
     const queryId = querySequence;
     setBoundedMap(latestQueries, failureKey, queryId, MAX_ACCOUNT_STATES);
+    let deepSeekAuthChanged = false;
     try {
-      const remainingMs = Math.max(1, DEFAULT_TIMEOUT_MS - (Date.now() - startedAt));
-      const guard = adapter.id === "xai" ? async () => {
-        if (signal.aborted || expectedSessionGeneration !== sessionGeneration || expectedXaiSettingsGeneration !== xaiSettingsGeneration || !xaiUsageEnabled() || ctx.sessionManager.getSessionId() !== expectedSessionId || modelIdentity(ctx.model) !== expectedModelIdentity) {
-          throw abortError();
-        }
+      const remainingMs = Math.max(1, deadlineAt - Date.now());
+      const guard = requiresRequestBoundaryGuard ? async () => {
+        if (signal.aborted || requestContextChanged()) throw abortError();
         const revalidated = await awaitWithDeadline(
           resolveUsageAuth(ctx, adapter, void 0, credentialReader, credentialCandidates),
           signal,
-          Math.max(1, DEFAULT_TIMEOUT_MS - (Date.now() - startedAt)),
-          "revalidating xAI runtime auth"
+          Math.max(1, deadlineAt - Date.now()),
+          `revalidating ${adapter.displayName} runtime auth`
         );
-        if (signal.aborted || expectedSessionGeneration !== sessionGeneration || expectedXaiSettingsGeneration !== xaiSettingsGeneration || !xaiUsageEnabled() || ctx.sessionManager.getSessionId() !== expectedSessionId || modelIdentity(ctx.model) !== expectedModelIdentity || revalidated?.fingerprint !== auth.fingerprint) {
+        if (signal.aborted || requestContextChanged()) throw abortError();
+        if (revalidated?.fingerprint !== auth.fingerprint) {
+          if (adapter.id === "deepseek") {
+            deepSeekAuthChanged = true;
+            throw new Error("DeepSeek runtime credential changed during the balance query.");
+          }
           throw abortError();
         }
       } : void 0;
-      const report = await queryProviderUsage(adapter, auth, signal, remainingMs, guard);
+      const report = await queryProviderUsage(
+        adapter,
+        auth,
+        signal,
+        remainingMs,
+        guard,
+        querySettings
+      );
       if (guard) await guard();
       if (latestQueries.get(failureKey) === queryId) {
-        cache.set(adapter.id, auth.fingerprint, report);
+        cache.set(adapter.id, queryFingerprint, report);
         failureBackoff.delete(failureKey);
       }
       return {
@@ -3053,6 +3647,18 @@ function usageExtension(pi, dependencies = {}) {
       };
     } catch (error) {
       if (isStaleExtensionContextError(error) || isAbortError3(error)) throw error;
+      if (deepSeekAuthChanged && authRetry === 0 && !signal.aborted && !requestContextChanged() && Date.now() < deadlineAt) {
+        if (latestQueries.get(failureKey) === queryId) latestQueries.delete(failureKey);
+        return queryAdapterState(
+          ctx,
+          adapter,
+          displayState,
+          true,
+          signal,
+          authRetry + 1,
+          deadlineAt
+        );
+      }
       const message = errorMessage(error);
       const now = Date.now();
       for (const [key, failure] of failureBackoff) {
@@ -3079,7 +3685,7 @@ function usageExtension(pi, dependencies = {}) {
     }
   };
   const queryCurrentState = async (ctx, model, force, signal) => {
-    const adapter = activeAdapterForProvider(model?.provider);
+    const adapter = adapterForProvider(model?.provider);
     if (!adapter) {
       const providerId = model?.provider ?? "none";
       transitionCurrentIdentity(`unsupported:${providerId}`, providerId);
@@ -3089,14 +3695,14 @@ function usageExtension(pi, dependencies = {}) {
           providerName: providerDisplayName(ctx, providerId),
           displayState: "current",
           status: "unsupported",
-          message: providerId === "xai" && !xaiUsageEnabled() ? "xAI usage is disabled. Open Settings to enable it." : model ? `Usage reporting is not supported for ${providerDisplayName(ctx, providerId)}.` : "No model is selected."
+          message: model ? `Usage reporting is not supported for ${providerDisplayName(ctx, providerId)}.` : "No model is selected."
         }
       };
     }
     return queryAdapterState(ctx, adapter, "current", force, signal);
   };
   const refreshCurrentStatus = async (ctx, model, force) => {
-    const adapter = activeAdapterForProvider(model?.provider);
+    const adapter = adapterForProvider(model?.provider);
     if (!adapter || !model) {
       const providerId = model?.provider ?? "none";
       transitionCurrentIdentity(`unsupported:${providerId}`, providerId);
@@ -3109,6 +3715,7 @@ function usageExtension(pi, dependencies = {}) {
     }
     statusGeneration += 1;
     const generation = statusGeneration;
+    clearStatusCountdownTimer();
     statusController?.abort();
     const controller = new AbortController();
     statusController = controller;
@@ -3159,7 +3766,7 @@ function usageExtension(pi, dependencies = {}) {
     if (generation !== statusGeneration || modelIdentity(ctx.model) !== modelIdentity(model)) {
       return false;
     }
-    const adapter = activeAdapterForProvider(model?.provider);
+    const adapter = adapterForProvider(model?.provider);
     if (outcome.authState === "unavailable") {
       if (!adapter) return false;
       try {
@@ -3218,7 +3825,7 @@ function usageExtension(pi, dependencies = {}) {
     const menuGeneration = statusGeneration;
     statusController?.abort();
     statusController = void 0;
-    clearStatusTimer();
+    clearStatusTimers();
     const controller = new AbortController();
     activeControllers.add(controller);
     try {
@@ -3283,7 +3890,7 @@ function usageExtension(pi, dependencies = {}) {
           providers: () => ({
             kind: "actions",
             title: "Select a configured provider",
-            items: configuredAdapters(ctx, xaiUsageEnabled()).filter((adapter) => adapter.id !== ctx.model?.provider).map((adapter) => ({
+            items: configuredAdapters(ctx).filter((adapter) => adapter.id !== ctx.model?.provider).map((adapter) => ({
               id: adapter.id,
               label: adapter.displayName,
               action: "provider"
@@ -3353,14 +3960,9 @@ function usageExtension(pi, dependencies = {}) {
               settingsRuntime,
               controller.signal,
               () => statusGeneration === menuGeneration && !controller.signal.aborted,
-              (id, _previous, next) => {
-                if (id !== "xaiUsage") return;
-                xaiSettingsGeneration += 1;
-                invalidateProviderState("xai");
-                if (!next) {
-                  for (const active of activeControllers) {
-                    if (active !== controller) active.abort();
-                  }
+              (id) => {
+                if (id === "codexStatusResetCountdown" && stableCurrent && statusGeneration === menuGeneration && !controller.signal.aborted) {
+                  publishStableCurrent(ctx, stableCurrent);
                 }
               }
             );
@@ -3546,7 +4148,7 @@ function usageExtension(pi, dependencies = {}) {
             return { kind: "stay" };
           },
           another: async () => {
-            const others = configuredAdapters(ctx, xaiUsageEnabled()).filter(
+            const others = configuredAdapters(ctx).filter(
               (adapter) => adapter.id !== ctx.model?.provider
             );
             if (others.length === 0) {
@@ -3556,7 +4158,7 @@ function usageExtension(pi, dependencies = {}) {
             return { kind: "to", screen: "providers" };
           },
           provider: async ({ itemId }) => {
-            const adapter = configuredAdapters(ctx, xaiUsageEnabled()).find(
+            const adapter = configuredAdapters(ctx).find(
               (candidate) => candidate.id === itemId && candidate.id !== ctx.model?.provider
             );
             if (!adapter) return { kind: "back" };
@@ -3582,7 +4184,7 @@ function usageExtension(pi, dependencies = {}) {
             return { kind: "back" };
           },
           all: async () => {
-            const adapters = configuredAdapters(ctx, xaiUsageEnabled());
+            const adapters = configuredAdapters(ctx);
             const currentProviderId = ctx.model?.provider;
             const settled = await runMenuOperation(
               ctx,
@@ -3660,16 +4262,21 @@ function usageExtension(pi, dependencies = {}) {
       }
     }
   });
-  pi.on("session_start", (_event, ctx) => {
+  pi.on("session_start", async (_event, ctx) => {
     sessionGeneration += 1;
-    xaiSettingsGeneration += 1;
     statusGeneration += 1;
-    clearStatusTimer();
+    clearStatusTimers();
     for (const controller of activeControllers) controller.abort();
     activeControllers.clear();
     statusController = void 0;
     sessionActive = true;
-    startStatusRefresh(ctx, ctx.model, false);
+    const ownerGeneration = sessionGeneration;
+    try {
+      await fastRuntime.prepareSession(ctx);
+    } catch (error) {
+      if (isStaleExtensionContextError(error) || ownerGeneration !== sessionGeneration) return;
+      throw error;
+    }
   });
   pi.on("session_tree", (_event, ctx) => {
     startStatusRefresh(ctx, ctx.model, false);
@@ -3683,9 +4290,8 @@ function usageExtension(pi, dependencies = {}) {
   pi.on("session_shutdown", (_event, ctx) => {
     sessionActive = false;
     sessionGeneration += 1;
-    xaiSettingsGeneration += 1;
     statusGeneration += 1;
-    clearStatusTimer();
+    clearStatusTimers();
     for (const controller of activeControllers) controller.abort();
     activeControllers.clear();
     statusController = void 0;
@@ -3698,7 +4304,8 @@ function usageExtension(pi, dependencies = {}) {
   fastRuntime = registerCodexFastMode(
     pi,
     settingsRuntime,
-    (ctx) => startStatusRefresh(ctx, ctx.model, false)
+    (ctx) => startStatusRefresh(ctx, ctx.model, false),
+    { registerSessionStart: false }
   );
 }
 export {
@@ -3730,6 +4337,9 @@ export {
   loadUsageSettings,
   normalizeCodexBackendPayload,
   normalizeCodexResetCreditsPayload,
+  normalizeDeepSeekBalancePayload,
+  normalizeFireworksAccountsPayload,
+  normalizeFireworksBillingSummaryPayload,
   normalizeGitHubCopilotUsagePayload,
   normalizeKimiCodingUsagePayload,
   normalizeOpenCodeZenPayload,