@1930dev/opencode-usage 0.3.3 → 0.3.4

package/README.md

@@ -124,7 +124,11 @@ Normalized percentage of budget consumed per provider, from these sources (in pr
    - `opencode-go` (Zen): rolling 5h / weekly / monthly % (binding window)
    - `openrouter`: credits used / total credits
    - `orcarouter`: spend since top-up; a per-key credit cap turns into a %
-   - `zai`: coding plan quota (needs an active GLM coding plan)
+   - `zai`: coding plan quota when the key has an active GLM plan; otherwise the
+     wallet balance (fallback)
+   - `snowflake-cortex`: tokens in the last 30d from the account usage views
+     (SQL API); the percentage comes from a `snowflake-cortex` line in
+     `budgets.json`
 
 2. **Documented limits** — published quotas, used when the provider reports none.
    Each one is read against the period it resets on: a daily limit against today,
@@ -142,8 +146,8 @@ Normalized percentage of budget consumed per provider, from these sources (in pr
    - `cloudflare-workers-ai`: 10k neurons/day (free tier, resets 00:00 UTC)
    - `nvidia`: credit allowance retired — the trial is rate-limited per model
    - `opencode`: no spend API for API keys
-   - `snowflake-cortex`: 100 credits/month, billed per token
-   - `zai`: no coding plan on this key
+   - `zai`: a key without a coding plan shows its wallet balance (live), and the
+     percentage comes from a `zai` line in `budgets.json`
 
 3. **budgets.json** — your monthly USD per provider, read against what the
    provider cost so far this calendar month:

package/dist/cli.js

@@ -284,12 +284,39 @@ function parseZai(data) {
     raw: data
   };
 }
+function parseZaiWallet(data) {
+  const failed = data.success === false || data.code !== undefined && data.code !== 0 && data.code !== 200;
+  if (failed) {
+    return { provider: "zai", ok: false, detail: data.msg ?? "wallet endpoint error", windows: [] };
+  }
+  const balance = data.data?.availableBalance ?? data.data?.rechargeAmount ?? 0;
+  const spend = data.data?.totalSpendAmount ?? 0;
+  const windows = [
+    { label: "balance", percentUsed: 0, detail: `$${balance.toFixed(2)}` },
+    { label: "spend", percentUsed: 0, detail: `$${spend.toFixed(2)}` }
+  ];
+  const today = data.data?.todaySpendAmount;
+  if (typeof today === "number") {
+    windows.push({ label: "today", percentUsed: 0, detail: `$${today.toFixed(2)}` });
+  }
+  return {
+    provider: "zai",
+    ok: true,
+    detail: `balance $${balance.toFixed(2)}`,
+    windows,
+    raw: data
+  };
+}
+var QUOTA_URL = "https://api.z.ai/api/monitor/usage/quota/limit";
+var WALLET_URL = "https://api.z.ai/api/biz/account/query-customer-account-report";
 async function fetchZai(key) {
   try {
-    const data = await getJson("https://api.z.ai/api/monitor/usage/quota/limit", {
-      Authorization: `Bearer ${key}`,
-      Accept: "application/json"
-    });
+    const auth = { Authorization: `Bearer ${key}`, Accept: "application/json" };
+    const data = await getJson(QUOTA_URL, auth);
+    if (data.code !== undefined && data.code !== 0) {
+      const wallet = await getJson(WALLET_URL, auth);
+      return parseZaiWallet(wallet);
+    }
     return parseZai(data);
   } catch (err) {
     return { provider: "zai", ok: false, detail: err.message, windows: [] };
@@ -338,11 +365,13 @@ function parseOrcaRouter(usage, subscription) {
   const cap = hard > 0 && hard < NO_CAP ? hard : soft > 0 && soft < NO_CAP ? soft : 0;
   const capped = cap > 0;
   const pct = capped ? spent / cap * 100 : 0;
+  const hint = capped ? "" : " (no cap set)";
+  const detail = `$${spent.toFixed(2)} spent${hint}`;
   return {
     provider: "orcarouter",
     ok: true,
-    detail: `$${spent.toFixed(2)} spent`,
-    windows: [{ label: "spend", percentUsed: pct, detail: `$${spent.toFixed(2)}` }],
+    detail,
+    windows: [{ label: "spend", percentUsed: pct, detail: `$${spent.toFixed(2)}${hint}` }],
     budget: capped ? { percentUsed: pct, label: `$${cap}` } : undefined,
     raw: { usage, subscription }
   };
@@ -358,6 +387,56 @@ async function fetchOrcaRouter(key) {
   }
 }
 
+// packages/core/src/quota/snowflake.ts
+var SQL_API = (account) => `https://${account}.snowflakecomputing.com/api/v2/statements`;
+var STATEMENT = `SELECT
+  SUM(CASE WHEN START_TIME >= DATE_TRUNC('day', CURRENT_TIMESTAMP())
+      AND START_TIME < DATEADD('day', 1, DATE_TRUNC('day', CURRENT_TIMESTAMP()))
+      THEN TOKENS ELSE 0 END) AS today_tokens,
+  SUM(TOKENS) AS month_tokens
+FROM SNOWFLAKE.ACCOUNT_USAGE.CORTEX_REST_API_USAGE_HISTORY
+WHERE START_TIME >= DATEADD('day', -30, CURRENT_TIMESTAMP())`;
+var fmtTokens = (n) => n.toLocaleString("en-US");
+function parseSnowflake(row) {
+  const today = Number(row?.[0] ?? 0);
+  const month = Number(row?.[1] ?? 0);
+  return {
+    provider: "snowflake-cortex",
+    ok: true,
+    detail: `${fmtTokens(month)} tokens/30d`,
+    windows: [
+      { label: "tokens/mo", percentUsed: 0, detail: `${fmtTokens(month)}` },
+      { label: "today", percentUsed: 0, detail: `${fmtTokens(today)}` }
+    ],
+    raw: { row }
+  };
+}
+async function fetchSnowflake(key, metadata) {
+  const account = String(metadata?.account ?? "").toUpperCase();
+  if (!account) {
+    return { provider: "snowflake-cortex", ok: false, detail: "metadata.account is required", windows: [] };
+  }
+  try {
+    const res = await fetch(SQL_API(account), {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${key}`,
+        "Content-Type": "application/json",
+        Accept: "application/json"
+      },
+      body: JSON.stringify({ statement: STATEMENT, timeout: 30 })
+    });
+    const body = await res.json().catch(() => null);
+    if (!res.ok) {
+      return { provider: "snowflake-cortex", ok: false, detail: body?.message ?? `HTTP ${res.status}`, windows: [] };
+    }
+    const rows = Array.isArray(body?.data) ? body.data : [];
+    return parseSnowflake(rows[0]);
+  } catch (err) {
+    return { provider: "snowflake-cortex", ok: false, detail: err.message, windows: [] };
+  }
+}
+
 // packages/core/src/providers.ts
 var TTL_MS = 15 * 60 * 1000;
 async function readCache() {
@@ -381,7 +460,8 @@ var FETCHERS = {
   "github-copilot": fetchCopilot,
   zai: fetchZai,
   amd: fetchAmd,
-  orcarouter: fetchOrcaRouter
+  orcarouter: fetchOrcaRouter,
+  "snowflake-cortex": fetchSnowflake
 };
 async function providerStatuses(opts) {
   const auth = await readAuth();
@@ -402,7 +482,7 @@ async function providerStatuses(opts) {
       const secret = authSecret(auth[p]);
       if (!fetcher || !secret)
         return null;
-      return fetcher(secret);
+      return fetcher(secret, auth[p].metadata);
     }));
     for (const q of live) {
       if (q)
@@ -425,7 +505,7 @@ async function providerStatuses(opts) {
   }));
 }
 // packages/core/src/report.ts
-function fmtTokens(n) {
+function fmtTokens2(n) {
   if (n >= 1e9)
     return `${(n / 1e9).toFixed(1)}B`;
   if (n >= 1e6)
@@ -467,8 +547,8 @@ function usageTable(rows, totals, sinceLabel, groupBy, pct) {
     const row = [
       r.group,
       String(r.messages),
-      fmtTokens(r.tokensInput),
-      fmtTokens(r.tokensOutput),
+      fmtTokens2(r.tokensInput),
+      fmtTokens2(r.tokensOutput),
       fmtCost(r.cost)
     ];
     if (pct) {
@@ -485,7 +565,7 @@ function usageTable(rows, totals, sinceLabel, groupBy, pct) {
 [sources: ${[...pct.values()].map((p) => `${p.source}${p.label ? `(${p.label})` : ""}`).filter((v, i, a) => a.indexOf(v) === i).join(", ")}]` : "";
   return `${t}
 
-TOTAL since ${sinceLabel}: ${totals.messages} msgs, ${fmtTokens(totals.tokensInput)} in / ${fmtTokens(totals.tokensOutput)} out, ${fmtCost(totals.cost)} est.${sourceNote}`;
+TOTAL since ${sinceLabel}: ${totals.messages} msgs, ${fmtTokens2(totals.tokensInput)} in / ${fmtTokens2(totals.tokensOutput)} out, ${fmtCost(totals.cost)} est.${sourceNote}`;
 }
 function providersTable(statuses, local) {
   const headers = ["PROVIDER", "QUOTA", "USAGE 7D (LOCAL)", "LAST USED"];
@@ -735,7 +815,7 @@ var PROVIDER_LIMITS = [
     limit: 100,
     tier: "paid",
     source: "https://docs.snowflake.com/en/sql-reference/account-usage/cortex_ai_functions_usage_history",
-    note: "Cortex AI Functions bill per token (credits per million tokens, Service Consumption Table). Usage lives in ACCOUNT_USAGE views, which the API key alone cannot read."
+    note: "Billed per token (Service Consumption Table). Live tokens come from ACCOUNT_USAGE.CORTEX_REST_API_USAGE_HISTORY through the SQL API with the stored JWT; the percentage needs a snowflake-cortex line in budgets.json."
   },
   {
     provider: "zai",

package/dist/tui.js

@@ -274,12 +274,39 @@ function parseZai(data) {
     raw: data
   };
 }
+function parseZaiWallet(data) {
+  const failed = data.success === false || data.code !== undefined && data.code !== 0 && data.code !== 200;
+  if (failed) {
+    return { provider: "zai", ok: false, detail: data.msg ?? "wallet endpoint error", windows: [] };
+  }
+  const balance = data.data?.availableBalance ?? data.data?.rechargeAmount ?? 0;
+  const spend = data.data?.totalSpendAmount ?? 0;
+  const windows = [
+    { label: "balance", percentUsed: 0, detail: `$${balance.toFixed(2)}` },
+    { label: "spend", percentUsed: 0, detail: `$${spend.toFixed(2)}` }
+  ];
+  const today = data.data?.todaySpendAmount;
+  if (typeof today === "number") {
+    windows.push({ label: "today", percentUsed: 0, detail: `$${today.toFixed(2)}` });
+  }
+  return {
+    provider: "zai",
+    ok: true,
+    detail: `balance $${balance.toFixed(2)}`,
+    windows,
+    raw: data
+  };
+}
+var QUOTA_URL = "https://api.z.ai/api/monitor/usage/quota/limit";
+var WALLET_URL = "https://api.z.ai/api/biz/account/query-customer-account-report";
 async function fetchZai(key) {
   try {
-    const data = await getJson("https://api.z.ai/api/monitor/usage/quota/limit", {
-      Authorization: `Bearer ${key}`,
-      Accept: "application/json"
-    });
+    const auth = { Authorization: `Bearer ${key}`, Accept: "application/json" };
+    const data = await getJson(QUOTA_URL, auth);
+    if (data.code !== undefined && data.code !== 0) {
+      const wallet = await getJson(WALLET_URL, auth);
+      return parseZaiWallet(wallet);
+    }
     return parseZai(data);
   } catch (err) {
     return { provider: "zai", ok: false, detail: err.message, windows: [] };
@@ -328,11 +355,13 @@ function parseOrcaRouter(usage, subscription) {
   const cap = hard > 0 && hard < NO_CAP ? hard : soft > 0 && soft < NO_CAP ? soft : 0;
   const capped = cap > 0;
   const pct = capped ? spent / cap * 100 : 0;
+  const hint = capped ? "" : " (no cap set)";
+  const detail = `$${spent.toFixed(2)} spent${hint}`;
   return {
     provider: "orcarouter",
     ok: true,
-    detail: `$${spent.toFixed(2)} spent`,
-    windows: [{ label: "spend", percentUsed: pct, detail: `$${spent.toFixed(2)}` }],
+    detail,
+    windows: [{ label: "spend", percentUsed: pct, detail: `$${spent.toFixed(2)}${hint}` }],
     budget: capped ? { percentUsed: pct, label: `$${cap}` } : undefined,
     raw: { usage, subscription }
   };
@@ -348,6 +377,56 @@ async function fetchOrcaRouter(key) {
   }
 }
 
+// packages/core/src/quota/snowflake.ts
+var SQL_API = (account) => `https://${account}.snowflakecomputing.com/api/v2/statements`;
+var STATEMENT = `SELECT
+  SUM(CASE WHEN START_TIME >= DATE_TRUNC('day', CURRENT_TIMESTAMP())
+      AND START_TIME < DATEADD('day', 1, DATE_TRUNC('day', CURRENT_TIMESTAMP()))
+      THEN TOKENS ELSE 0 END) AS today_tokens,
+  SUM(TOKENS) AS month_tokens
+FROM SNOWFLAKE.ACCOUNT_USAGE.CORTEX_REST_API_USAGE_HISTORY
+WHERE START_TIME >= DATEADD('day', -30, CURRENT_TIMESTAMP())`;
+var fmtTokens = (n) => n.toLocaleString("en-US");
+function parseSnowflake(row) {
+  const today = Number(row?.[0] ?? 0);
+  const month = Number(row?.[1] ?? 0);
+  return {
+    provider: "snowflake-cortex",
+    ok: true,
+    detail: `${fmtTokens(month)} tokens/30d`,
+    windows: [
+      { label: "tokens/mo", percentUsed: 0, detail: `${fmtTokens(month)}` },
+      { label: "today", percentUsed: 0, detail: `${fmtTokens(today)}` }
+    ],
+    raw: { row }
+  };
+}
+async function fetchSnowflake(key, metadata) {
+  const account = String(metadata?.account ?? "").toUpperCase();
+  if (!account) {
+    return { provider: "snowflake-cortex", ok: false, detail: "metadata.account is required", windows: [] };
+  }
+  try {
+    const res = await fetch(SQL_API(account), {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${key}`,
+        "Content-Type": "application/json",
+        Accept: "application/json"
+      },
+      body: JSON.stringify({ statement: STATEMENT, timeout: 30 })
+    });
+    const body = await res.json().catch(() => null);
+    if (!res.ok) {
+      return { provider: "snowflake-cortex", ok: false, detail: body?.message ?? `HTTP ${res.status}`, windows: [] };
+    }
+    const rows = Array.isArray(body?.data) ? body.data : [];
+    return parseSnowflake(rows[0]);
+  } catch (err) {
+    return { provider: "snowflake-cortex", ok: false, detail: err.message, windows: [] };
+  }
+}
+
 // packages/core/src/providers.ts
 var TTL_MS = 15 * 60 * 1000;
 async function readCache() {
@@ -371,7 +450,8 @@ var FETCHERS = {
   "github-copilot": fetchCopilot,
   zai: fetchZai,
   amd: fetchAmd,
-  orcarouter: fetchOrcaRouter
+  orcarouter: fetchOrcaRouter,
+  "snowflake-cortex": fetchSnowflake
 };
 async function providerStatuses(opts) {
   const auth = await readAuth();
@@ -392,7 +472,7 @@ async function providerStatuses(opts) {
       const secret = authSecret(auth[p]);
       if (!fetcher || !secret)
         return null;
-      return fetcher(secret);
+      return fetcher(secret, auth[p].metadata);
     }));
     for (const q of live) {
       if (q)
@@ -500,7 +580,7 @@ var PROVIDER_LIMITS = [
     limit: 100,
     tier: "paid",
     source: "https://docs.snowflake.com/en/sql-reference/account-usage/cortex_ai_functions_usage_history",
-    note: "Cortex AI Functions bill per token (credits per million tokens, Service Consumption Table). Usage lives in ACCOUNT_USAGE views, which the API key alone cannot read."
+    note: "Billed per token (Service Consumption Table). Live tokens come from ACCOUNT_USAGE.CORTEX_REST_API_USAGE_HISTORY through the SQL API with the stored JWT; the percentage needs a snowflake-cortex line in budgets.json."
   },
   {
     provider: "zai",
@@ -655,7 +735,7 @@ function columnsFor(inner) {
   const budget = Math.max(BUDGET_MIN, inner - used);
   return { cols: [...base, { title: "BUDGET", width: budget, align: "left" }], bar: wide ? WIDE_BAR : COMPACT_BAR };
 }
-function fmtTokens(n) {
+function fmtTokens2(n) {
   if (n >= 1e6)
     return `${(n / 1e6).toFixed(1)}M`;
   if (n >= 1000)
@@ -821,8 +901,8 @@ function leadCells(cols, r) {
   return row(cols, [
     r.provider,
     String(r.messages),
-    fmtTokens(r.tokensInput),
-    fmtTokens(r.tokensOutput),
+    fmtTokens2(r.tokensInput),
+    fmtTokens2(r.tokensOutput),
     `$${r.cost.toFixed(2)}`
   ]).trimEnd().padEnd(upToBudget) + " ";
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@1930dev/opencode-usage",
-  "version": "0.3.3",
+  "version": "0.3.4",
   "description": "Usage tracking, budget percentages and model ranking for opencode — CLI plus a /usage TUI plugin",
   "type": "module",
   "bin": {