@narumitw/pi-usage 0.57.0 → 0.58.0

package/dist/index.ts

@@ -520,10 +520,163 @@ 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 = asObject3(payload.quota_snapshots);
-  const premium = asObject3(snapshots?.premium_interactions);
+  const snapshots = asObject4(payload.quota_snapshots);
+  const premium = asObject4(snapshots?.premium_interactions);
   const metrics = [];
   let semanticsLabel;
   let bucket;
@@ -564,8 +717,8 @@ function normalizeGitHubCopilotUsagePayload(payload, capturedAt) {
       };
     }
   } else {
-    const limited = asObject3(payload.limited_user_quotas);
-    const monthly = asObject3(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) {
@@ -604,7 +757,7 @@ function resetTimestamp(payload) {
   const milliseconds = Date.parse(raw);
   return Number.isNaN(milliseconds) ? {} : { resetsAt: Math.floor(milliseconds / 1e3) };
 }
-function asObject3(value) {
+function asObject4(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -627,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 = asObject4(payload);
+  const root = asObject5(payload);
   if (!root) throw new Error("Kimi Coding usage response was not an object.");
   const candidates = [];
   let omittedWindow = false;
@@ -636,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 = asObject4(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));
@@ -673,7 +826,7 @@ function normalizeKimiCodingUsagePayload(payload, capturedAt) {
   };
 }
 function parseUsageRow(value, windowMinutes, label) {
-  const row = asObject4(value);
+  const row = asObject5(value);
   if (!row) return void 0;
   const used = asNonnegativeInteger2(row.used);
   const limit = asNonnegativeInteger2(row.limit);
@@ -691,7 +844,7 @@ function parseUsageRow(value, windowMinutes, label) {
   };
 }
 function parseWindowMinutes(value) {
-  const window = asObject4(value);
+  const window = asObject5(value);
   if (!window) return void 0;
   const duration = asPositiveInteger(window.duration);
   if (duration === void 0) return void 0;
@@ -701,8 +854,8 @@ function parseWindowMinutes(value) {
   return Number.isSafeInteger(minutes) ? minutes : void 0;
 }
 function parseBoosterWallet(value) {
-  const wallet = asObject4(value);
-  const balance = asObject4(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 [];
@@ -753,7 +906,7 @@ function parseBoosterWallet(value) {
   return metrics;
 }
 function parseMoney(value) {
-  const money = asObject4(value);
+  const money = asObject5(value);
   if (!money) return void 0;
   const cents = asNonnegativeInteger2(money.priceInCents);
   if (cents === void 0) return void 0;
@@ -767,7 +920,7 @@ function fixedPointToMajor(value) {
   const major = roundedCents / 100;
   return Number.isSafeInteger(roundedCents) && Number.isFinite(major) ? major : void 0;
 }
-function asObject4(value) {
+function asObject5(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -841,12 +994,12 @@ var ZEN_WINDOWS = [
   { key: "monthly", label: "Monthly" }
 ];
 function normalizeOpenCodeZenPayload(payload, capturedAt) {
-  const usage = asObject5(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 = asObject5(usage[window.key]);
+    const raw = asObject6(usage[window.key]);
     if (!raw) continue;
     const status = asString3(raw.status);
     if (status !== "ok" && status !== "rate-limited") {
@@ -883,7 +1036,7 @@ function normalizeOpenCodeZenPayload(payload, capturedAt) {
     ...notes.length > 0 ? { notes } : {}
   };
 }
-function asObject5(value) {
+function asObject6(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -907,7 +1060,7 @@ function clampPercent2(value) {
 
 // src/providers/openrouter.ts
 function normalizeOpenRouterKeyPayload(payload, capturedAt) {
-  const data = asObject6(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);
@@ -953,7 +1106,7 @@ function addUsageMetric(metrics, id, label, value) {
   if (amount === void 0) return;
   metrics.push({ id, label, value: amount, unit: "usd" });
 }
-function asObject6(value) {
+function asObject7(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -1131,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 = asObject7(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 = asObject7(raw);
+    const limit = asObject8(raw);
     if (!limit) continue;
     const type = asString5(limit.type);
     const unit = asNonnegativeNumber4(limit.unit);
@@ -1209,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 = asObject7(raw);
+    const detail = asObject8(raw);
     if (!detail) continue;
     const label = asString5(detail.modelCode);
     const usage = asNonnegativeNumber4(detail.usage);
@@ -1217,7 +1370,7 @@ function addUsageDetailMetrics(metrics, value) {
     metrics.push({ id: `mcp-${kebabCase(label)}`, label, value: usage, unit: "count" });
   }
 }
-function asObject7(value) {
+function asObject8(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -1244,6 +1397,9 @@ 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";
@@ -1331,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",
@@ -1535,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));
@@ -1678,7 +1863,7 @@ function resolveXaiUsageAuth(auth, model, salt, candidates) {
   const matches = [];
   for (const candidate of candidates) {
     try {
-      const credential = asObject8(candidate);
+      const credential = asObject9(candidate);
       if (credential?.type !== "oauth") continue;
       sawOAuth = true;
       if (credential.access !== resolvedAccess) continue;
@@ -1732,7 +1917,7 @@ function resolveGitHubCopilotUsageAuth(auth, model, salt, candidates, standalone
   const matches = /* @__PURE__ */ new Map();
   for (const candidate of candidates) {
     try {
-      const credential = asObject8(candidate);
+      const credential = asObject9(candidate);
       if (credential?.type !== "oauth") continue;
       sawOAuth = true;
       const storedAccess2 = typeof credential.access === "string" && credential.access ? credential.access : void 0;
@@ -1794,7 +1979,7 @@ function bearerToken(authorization) {
   const match = /^Bearer\s+(.+)$/iu.exec(authorization ?? "");
   return match?.[1];
 }
-function asObject8(value) {
+function asObject9(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -1814,6 +1999,7 @@ function hasOfficialUrlOrigin(value, providerId) {
     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";
@@ -1843,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.");
@@ -2001,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(asObject9).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) {
@@ -2016,7 +2278,7 @@ function selectCodexResetCredential(candidates, resolvedAccess, resolvedAccountI
   const matches = /* @__PURE__ */ new Map();
   for (const candidate of candidates) {
     try {
-      const credential = asObject9(candidate);
+      const credential = asObject10(candidate);
       if (credential?.type !== "oauth") continue;
       sawOAuth = true;
       const storedAccess = asNonemptyString(credential.access);
@@ -2055,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 = asObject9(asObject9(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;
@@ -2087,7 +2349,7 @@ function normalizeResetOption(credit) {
 function isCodexResetOutcomeCode(value) {
   return value === "reset" || value === "nothing_to_reset" || value === "no_credit" || value === "already_redeemed";
 }
-function asObject9(value) {
+function asObject10(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) return void 0;
   return value;
 }
@@ -2128,12 +2390,13 @@ var BAR_SEGMENTS = 20;
 var VALUE_COLUMN = 29;
 function formatUsageReport(report, displayState) {
   const stateLabel = displayState === "current" ? "Current" : "Configured";
-  const title = report.providerId === "deepseek" ? "DeepSeek API Balance" : `${report.providerName} Usage`;
+  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);
@@ -2152,6 +2415,7 @@ function formatUsageStatusline(report, model, now = Date.now(), showCodexResetCo
     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");
@@ -2222,6 +2486,29 @@ function formatDeepSeekStatusline(report) {
   });
   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) {
@@ -2583,9 +2870,13 @@ function normalizeUsageSettings(value) {
   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,
-    codexStatusResetCountdown: typeof value.codexStatusResetCountdown === "boolean" ? value.codexStatusResetCountdown : DEFAULT_USAGE_SETTINGS.codexStatusResetCountdown
+    codexStatusResetCountdown: typeof value.codexStatusResetCountdown === "boolean" ? value.codexStatusResetCountdown : DEFAULT_USAGE_SETTINGS.codexStatusResetCountdown,
+    ...isFireworksAccountId(value.fireworksAccountId) ? { fireworksAccountId: value.fireworksAccountId } : {}
   };
 }
 async function loadUsageSettings(path = usageSettingsPath(), signal) {
@@ -2669,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);
@@ -2715,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();
@@ -2771,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,
@@ -2834,6 +3134,7 @@ function registerCodexFastMode(pi, settingsRuntime, refreshStatus) {
     await settingsRuntime.flush();
   });
   return {
+    prepareSession,
     availability(model) {
       return codexFastAvailability(model, settingsRuntime.get().settings.codexFastMode);
     },
@@ -2912,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) {
@@ -2919,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]);
@@ -2927,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",
@@ -2941,6 +3261,13 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
         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();
@@ -2950,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,
@@ -2958,35 +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") {
-            settingsList.updateValue(id, displayValue(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({ [settingId]: requested }, signal);
-          } catch (error) {
-            if (signal.aborted || !isCurrent()) return;
-            settingsList.updateValue(id, displayValue(previous));
-            ctx.ui.notify(`Could not save pi-usage.json: ${errorMessage(error)}`, "error");
-            tui.requestRender();
-            return;
-          }
-          if (previous !== requested) {
-            changed = true;
-            onApplied(settingId);
+        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(requested));
-          tui.requestRender();
-        });
+          return;
+        }
+        const settingId = id;
+        queueUpdate(settingId, value !== OFF, value);
       },
       cancel
     );
@@ -3008,8 +3347,41 @@ async function showUsageSettings(ctx, settingsRuntime, parentSignal, isCurrent,
     };
   });
 }
-function displayValue(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
@@ -3152,6 +3524,8 @@ function usageExtension(pi, dependencies = {}) {
     const expectedSessionGeneration = sessionGeneration;
     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(
@@ -3175,8 +3549,8 @@ function usageExtension(pi, dependencies = {}) {
         }
       };
     }
-    const requiresRequestBoundaryGuard = adapter.id === "deepseek" || adapter.id === "xai";
-    const requestContextChanged = () => expectedSessionGeneration !== sessionGeneration || ctx.sessionManager.getSessionId() !== expectedSessionId || modelIdentity(ctx.model) !== expectedModelIdentity;
+    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") {
@@ -3193,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: {
@@ -3209,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 {
@@ -3247,10 +3622,17 @@ function usageExtension(pi, dependencies = {}) {
           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 {
@@ -3880,7 +4262,7 @@ function usageExtension(pi, dependencies = {}) {
       }
     }
   });
-  pi.on("session_start", (_event, ctx) => {
+  pi.on("session_start", async (_event, ctx) => {
     sessionGeneration += 1;
     statusGeneration += 1;
     clearStatusTimers();
@@ -3888,7 +4270,13 @@ function usageExtension(pi, dependencies = {}) {
     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);
@@ -3916,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 {
@@ -3949,6 +4338,8 @@ export {
   normalizeCodexBackendPayload,
   normalizeCodexResetCreditsPayload,
   normalizeDeepSeekBalancePayload,
+  normalizeFireworksAccountsPayload,
+  normalizeFireworksBillingSummaryPayload,
   normalizeGitHubCopilotUsagePayload,
   normalizeKimiCodingUsagePayload,
   normalizeOpenCodeZenPayload,