@1930dev/opencode-usage 0.3.4 → 0.3.5

package/README.md

@@ -58,6 +58,10 @@ opencode-usage usage --pct              # add the % BUDGET column (--by provider
 opencode-usage providers
 opencode-usage providers --no-net       # cached only
 
+# What the providers themselves answer about their budget
+opencode-usage probe                    # one free-model call per provider
+opencode-usage probe --json
+
 # Model ranking by intelligence per blended dollar
 opencode-usage top
 opencode-usage top --limit 10 --json
@@ -161,6 +165,23 @@ Normalized percentage of budget consumed per provider, from these sources (in pr
 
 Providers without any source show `—`.
 
+## Free-model probe (`probe`)
+
+`probe` sends one minimal request against a free model per connected provider
+and reports what the answer reveals about the budget: a rate-limit header, a
+"no credits" error, or the meter itself (`usage.neurons` for Cloudflare). It
+reads the key from opencode's auth store and never prints it. Providers covered:
+
+- `cloudflare-workers-ai` — the response carries neurons per request (the free
+  tier's unit), so the probe answers whether the account can still run
+- `nvidia` — an HTTP 200 means the free per-model allowance still works
+- `orcarouter` — the `free` model answers `rate_limit_error` with a
+  `retry-after`, and credits raise the cap
+- `zai` — an HTTP 429 with `code 1113` means the balance is empty
+
+`opencode` exposes no chat API for keys and `snowflake-cortex` is read through
+its SQL quota, so both are skipped.
+
 ## Ranking (`top`)
 
 Models ranked by intelligence index per blended dollar:

package/dist/cli.js

@@ -526,6 +526,15 @@ function fmtAgo(ms, now = Date.now()) {
     return `${h}h ago`;
   return `${Math.floor(h / 24)}d ago`;
 }
+function fmtRetry(seconds) {
+  if (seconds < 60)
+    return `${seconds}s`;
+  if (seconds < 3600)
+    return `${Math.round(seconds / 60)}m`;
+  if (seconds < 86400)
+    return `${Math.ceil(seconds / 3600)}h`;
+  return `${Math.ceil(seconds / 86400)}d`;
+}
 function pad(s, n) {
   return s.length >= n ? s : s + " ".repeat(n - s.length);
 }
@@ -578,6 +587,18 @@ function providersTable(statuses, local) {
   });
   return table(headers, rows);
 }
+function probeTable(results) {
+  const headers = ["PROVIDER", "TAUGHT", "SIGNAL", "RETRY"];
+  const rows = results.map((r) => {
+    if (r.status === 0)
+      return [r.provider, "error", r.message ?? "no signal", "\u2014"];
+    const taught = r.ok ? "yes" : "no";
+    const signal = r.ok ? r.usage && Object.keys(r.usage).length > 0 ? Object.entries(r.usage).map(([k, v]) => `${k}=${v}`).join(", ") : "http ok" : r.message || (r.code ? `code ${r.code}` : `HTTP ${r.status}`);
+    const retry = r.retryAfterSeconds !== undefined ? fmtRetry(r.retryAfterSeconds) : "\u2014";
+    return [r.provider, taught, signal, retry];
+  });
+  return table(headers, rows);
+}
 // packages/core/src/ranking/fetch.ts
 import { mkdir as mkdir2, readFile as readFile3, writeFile as writeFile2 } from "fs/promises";
 import path3 from "path";
@@ -875,6 +896,89 @@ function resolvePct(provider, sources) {
   }
   return { pct: 0, source: "none" };
 }
+// packages/core/src/probe.ts
+function numericUsage(u) {
+  if (typeof u !== "object" || u === null)
+    return;
+  const out = {};
+  for (const [k, v] of Object.entries(u)) {
+    if (typeof v === "number")
+      out[k] = v;
+  }
+  return Object.keys(out).length > 0 ? out : undefined;
+}
+function redact(text, key) {
+  return key.length > 6 ? text.split(key).join("[redacted]") : text;
+}
+var chatBody = (model) => JSON.stringify({ model, messages: [{ role: "user", content: "ping" }], max_tokens: 4 });
+async function probeChat(p, key) {
+  let res;
+  try {
+    res = await fetch(p.url, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Accept: "application/json",
+        Authorization: `Bearer ${key}`
+      },
+      body: p.body
+    });
+  } catch (err) {
+    return { provider: p.name, status: 0, ok: false, message: redact(err.message, key) };
+  }
+  const text = await res.text();
+  let parsed = null;
+  try {
+    parsed = JSON.parse(text);
+  } catch {
+    parsed = null;
+  }
+  const obj = parsed ?? {};
+  const err = obj.error;
+  const firstErr = Array.isArray(obj.errors) ? obj.errors[0] : undefined;
+  const message = redact(err?.message ?? err?.detail ?? firstErr?.message ?? "", key);
+  const rawRetry = res.headers.get("retry-after");
+  const headerRetry = rawRetry !== null && rawRetry !== "" ? Number(rawRetry) : Number.NaN;
+  const retry = Number.isFinite(headerRetry) ? headerRetry : err?.metadata?.retry_after_seconds;
+  const usage = p.extractUsage ? p.extractUsage(obj) : numericUsage(obj.usage);
+  return {
+    provider: p.name,
+    status: res.status,
+    ok: res.status >= 200 && res.status < 300,
+    ...err?.code ?? firstErr?.code ? { code: err?.code ?? firstErr?.code } : {},
+    ...message ? { message } : {},
+    ...Number.isFinite(retry) ? { retryAfterSeconds: retry } : {},
+    ...usage ? { usage } : {}
+  };
+}
+var PROBERS = {
+  zai: (key) => probeChat({ name: "zai", url: "https://api.z.ai/api/paas/v4/chat/completions", body: chatBody("glm-4.5-air") }, key),
+  nvidia: (key) => probeChat({ name: "nvidia", url: "https://integrate.api.nvidia.com/v1/chat/completions", body: chatBody("meta/llama-3.2-11b-vision-instruct") }, key),
+  orcarouter: (key) => probeChat({ name: "orcarouter", url: "https://api.orcarouter.ai/v1/chat/completions", body: chatBody("orcarouter/free") }, key),
+  "cloudflare-workers-ai": (key, meta) => {
+    const account = String(meta?.accountId ?? "");
+    if (!account) {
+      return Promise.resolve({ provider: "cloudflare-workers-ai", status: 0, ok: false, message: "metadata.accountId is required" });
+    }
+    return probeChat({
+      name: "cloudflare-workers-ai",
+      url: `https://api.cloudflare.com/client/v4/accounts/${account}/ai/run/@cf/meta/llama-3.1-8b-fast-v2`,
+      body: JSON.stringify({ prompt: "ping", max_tokens: 4 }),
+      extractUsage: (parsed) => numericUsage(typeof parsed === "object" && parsed !== null && "result" in parsed ? parsed.result.usage : undefined)
+    }, key);
+  }
+};
+async function runProbes() {
+  const auth = await readAuth();
+  const results = await Promise.all(Object.entries(auth).sort().map(async ([name, entry]) => {
+    const prober = PROBERS[name];
+    const key = authSecret(entry);
+    if (!prober || !key)
+      return null;
+    return prober(key, entry.metadata);
+  }));
+  return results.filter((r) => r !== null);
+}
 // packages/core/src/snapshot.ts
 async function withConnectedProviders(rows) {
   let connected;
@@ -948,6 +1052,7 @@ var USAGE = `opencode-usage \u2014 usage tracking and model ranking for opencode
 USAGE
   opencode-usage usage  [--since 7d] [--by provider|model|day|project|agent] [--today] [--pct] [--json]
   opencode-usage providers [--no-net] [--json]
+  opencode-usage probe [--json]
   opencode-usage top [--limit 20] [--json]
 
 OPTIONS
@@ -1037,6 +1142,18 @@ async function cmdProviders(flags, json) {
     db.close();
   }
 }
+async function cmdProbe(flags, json) {
+  const results = await runProbes();
+  if (json) {
+    console.log(JSON.stringify({ probes: results }, null, 2));
+    return;
+  }
+  const auth = await readAuth();
+  const withoutProbe = Object.entries(auth).filter(([, e]) => e.key || e.access).map(([p]) => p).filter((p) => !["cloudflare-workers-ai", "nvidia", "orcarouter", "zai"].includes(p)).sort();
+  console.log(probeTable(results));
+  if (withoutProbe.length > 0)
+    console.log(`no free-model probe: ${withoutProbe.join(", ")}`);
+}
 var TOP_HEADERS = ["MODEL", "NAME", "IQ", "CODING", "$/M", "IQ/$"];
 var TOP_WIDTHS = [28, 26, 5, 7, 8, 6];
 function topLine(cells) {
@@ -1074,6 +1191,8 @@ async function run(argv) {
     await cmdUsage(flags, json);
   else if (cmd === "providers")
     await cmdProviders(flags, json);
+  else if (cmd === "probe")
+    await cmdProbe(flags, json);
   else if (cmd === "top")
     await cmdTop(flags, json);
   else

package/package.json

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