@bacnh85/pi-sub 0.1.31 → 0.1.33

package/CHANGELOG.md

@@ -1,3 +1,17 @@
+## 0.1.32 (2026-08-22)
+
+- **DeepSeek via OmniRoute now shows the real USD balance** (e.g. `M:$18.25`)
+  instead of the misleading aggregate windows (R:100%/5H W:0%/2D). Credit-based
+  upstreams report "Unavailable" session/weekly in the usage text; pi-sub now
+  recognizes that shape and fetches `credits_usd` from the management usage
+  API — grant the router key the `manage` scope in the OmniRoute dashboard
+  (done), or set `ROUTER_MGMT_TOKEN`. Without it, the footer degrades to the
+  clean endpoint display (no aggregate leak).
+- The plain-retry fallback (drop `?provider=`) now fires only on "No cached
+  usage data" (wrong/unknown slug), not on "Unavailable" windows.
+- pi-sub now reads `.env.local`/`.env` (cwd then `~/.pi/agent`) for its own
+  env vars — same discovery chain as pi-munin, stdlib parser, no deps.
+
 ## 0.1.31 (2026-08-22)
 
 - **Fixed glm-cn quota via OmniRoute**: OmniRoute's connection slug for the
@@ -52,6 +66,12 @@
 
 # Changelog
 
+## 0.1.33 (2026-08-29)
+
+### Added
+
+- `/sub` argument completion offers `refresh`.
+
 ## 0.1.29 (2026-08-15)
 
 ### Features

package/extensions/index.ts

@@ -4,6 +4,28 @@ import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
 
+/** Pi config dirs + .env.local/.env discovery (pi-munin convention, stdlib parse). */
+function loadEnvFiles(): void {
+  const dirs = process.env.PI_CODING_AGENT_DIR
+    ? [process.env.PI_CODING_AGENT_DIR]
+    : [path.join(os.homedir(), ".pi", "agent"), path.join(os.homedir(), ".pi", "agents")];
+  const candidates = [path.resolve(process.cwd(), ".env.local"), path.resolve(process.cwd(), ".env")]
+    .concat(dirs.flatMap((d) => [path.join(d, ".env.local"), path.join(d, ".env")]));
+  for (const file of candidates) {
+    try {
+      const text = fs.readFileSync(file, "utf8");
+      for (const line of text.split(/\r?\n/)) {
+        const m = line.match(/^\s*(?:export\s+)?([A-Za-z_][A-Za-z0-9_]*)\s*=\s*(.*)\s*$/);
+        if (!m) continue;
+        let v = m[2].trim();
+        if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'"))) v = v.slice(1, -1);
+        if (process.env[m[1]] === undefined) process.env[m[1]] = v;
+      }
+    } catch { /* optional file */ }
+  }
+}
+loadEnvFiles();
+
 const STATUS_KEY = "pi-sub";
 const MESSAGE_TYPE = "pi-sub-status";
 const USAGE_ENDPOINT = "https://chatgpt.com/backend-api/wham/usage";
@@ -340,6 +362,15 @@ function routerOrigin(baseUrl: string): string {
   return baseUrl.replace(/\/v1\/?$/, "");
 }
 
+/** OmniRoute management credential (manage-scope key or oma_ CLI token) from
+ *  env — optional override. The router key itself works when it holds the
+ *  `manage` scope (API Keys dashboard), which unlocks
+ *  /api/usage/<connectionId> carrying the raw USD balance for credit-based
+ *  upstreams (deepseek) that the key-authable endpoints normalize away. */
+function readRouterMgmtToken(apiKey: string | undefined): string | undefined {
+  return process.env.ROUTER_MGMT_TOKEN || process.env.OMNIROUTE_MGMT_TOKEN || apiKey;
+}
+
 /** Parse OmniRoute's `/api/usage/om-usage` plain-text report into windows.
  *  Sections: "Personal quota" (per-key USD budgets: Daily/Weekly) and
  *  "Provider quota" (connection session/weekly). Lines: `<Label>`,
@@ -420,7 +451,7 @@ if (process.env.PI_SUB_SELF_CHECK === "1") {
   assert(p.personalWeekly?.remaining === 90, "personal weekly 90");
   assert(p.session?.remaining === 47, "session 47");
   assert(p.providerWeekly?.remaining === 28, "provider weekly 28");
-  assert(p.personalDaily?.resetLabel?.includes("15h"), "daily reset label");
+  assert(p.personalDaily?.resetLabel?.includes("15h") === true, "daily reset label");
   const disabled = parseOmniUsageText("Usage command is disabled for this API key.");
   assert(Object.keys(disabled).length === 0, "disabled text parses empty");
   // Live-verified: provider without cached data → no windows (endpoint fallback).
@@ -432,7 +463,7 @@ if (process.env.PI_SUB_SELF_CHECK === "1") {
   );
   assert(live.session?.remaining === 90, "session 90");
   assert(live.providerWeekly?.remaining === 0, "weekly 0");
-  assert(live.session?.resetLabel?.includes("1h 59m"), "session reset");
+  assert(live.session?.resetLabel?.includes("1h 59m") === true, "session reset");
   // Live-verified 2026-08-22: glm-cn scoped quota — Session 99%, Weekly
   // "Unavailable" (skipped, so W stays absent like the direct Z.ai footer).
   const glmCn = parseOmniUsageText(
@@ -553,9 +584,11 @@ async function fetchRouterUsage(signal?: AbortSignal, provider?: string): Promis
         let text = await response.text();
         if (text && !text.includes("disabled")) {
           let w = parseOmniUsageText(text);
-          // Provider-scoped call but no cached quota for that upstream — retry
-          // without ?provider= so the report shows the best/all snapshot.
-          if (provider && !w.session && !w.providerWeekly) {
+          // "No cached usage data" = unknown/wrong slug — retry without
+          // ?provider= for the best/all snapshot. "Unavailable" windows mean a
+          // known credit-based upstream (deepseek) — keep them empty so the
+          // USD-balance path below takes over instead of showing the aggregate.
+          if (provider && !w.session && !w.providerWeekly && text.includes("No cached usage data")) {
             const plain = await fetch(url.replace(/\?provider=.*$/, ""), {
               headers: { Accept: "text/plain", Authorization: `Bearer ${apiKey}` },
               signal: combined,
@@ -568,6 +601,23 @@ async function fetchRouterUsage(signal?: AbortSignal, provider?: string): Promis
               }
             }
           }
+          // Credit-based upstreams (deepseek): the usage text prints "Unavailable"
+          // windows — pull the real USD balance from the management API instead.
+          const credits = await fetchRouterCredits(provider, w);
+          if (credits) {
+            const account: SubscriptionAccountSnapshot = {
+              ...baseAccount,
+              plan: `Router · ${provider}`,
+              monthlyCredits: credits.balanceUsd,
+              usageBreakdown: credits.breakdown,
+            };
+            return {
+              providerDisplayName: "Router",
+              accounts: [account],
+              activeAccount: account,
+              fetchedAt: Date.now(),
+            };
+          }
           // personalDaily = per-key budget (nearest reset → R slot),
           // provider weekly/session = upstream quota (W slot). Fall back sensibly.
           const account: SubscriptionAccountSnapshot = {
@@ -612,6 +662,56 @@ async function fetchRouterUsage(signal?: AbortSignal, provider?: string): Promis
   };
 }
 
+interface RouterCredits {
+  balanceUsd: number;
+  breakdown: string;
+}
+
+/** Fetch the raw USD balance for credit-based upstreams (deepseek: `credits_usd`)
+ *  via OmniRoute's management usage API. Only called when the key-authable
+ *  om-usage text reports no usable windows — that surface normalizes credits
+ *  to meaningless percentages. Needs the router key to hold the `manage`
+ *  scope (or ROUTER_MGMT_TOKEN as override). Connection discovery comes from
+ *  /api/v1/me/status (key-authable); the balance from /api/usage/<id>. */
+async function fetchRouterCredits(provider: string | undefined, w: ReturnType<typeof parseOmniUsageText>): Promise<RouterCredits | undefined> {
+  if (!provider || w.session || w.providerWeekly) return undefined;
+  const cfg = readRouterConfig();
+  const apiKey = readRouterApiKey();
+  const mgmtToken = readRouterMgmtToken(apiKey);
+  if (!cfg || !apiKey || !mgmtToken) return undefined;
+  const origin = routerOrigin(cfg.baseUrl);
+  try {
+    const combined = AbortSignal.timeout(7_000);
+    // 1. Connection id for this upstream via the key-authable status endpoint.
+    const statusRes = await fetch(`${origin}/api/v1/me/status`, {
+      headers: { Authorization: `Bearer ${apiKey}` },
+      signal: combined,
+    });
+    if (!statusRes.ok) return undefined;
+    const status = (await statusRes.json()) as { accountQuotas?: Array<{ provider?: string; connectionId?: string }> };
+    const connectionId = status.accountQuotas?.find((q) => q.provider === provider)?.connectionId;
+    if (!connectionId) return undefined;
+    // 2. Raw usage (management token) — quotas.credits_usd.remaining is the USD balance.
+    const usageRes = await fetch(`${origin}/api/usage/${connectionId}`, {
+      headers: { Authorization: `Bearer ${mgmtToken}` },
+      signal: combined,
+    });
+    if (!usageRes.ok) return undefined;
+    const usage = (await usageRes.json()) as { quotas?: Record<string, { remaining?: number }> };
+    const credits = usage.quotas?.credits_usd ?? usage.quotas?.credits;
+    const remaining = credits?.remaining;
+    if (typeof remaining !== "number" || !Number.isFinite(remaining)) return undefined;
+    const cny = usage.quotas?.credits_cny?.remaining;
+    return {
+      balanceUsd: remaining,
+      breakdown: `🪙 Balance (USD) $${remaining.toFixed(2)}` +
+        (typeof cny === "number" ? ` · ¥${cny.toFixed(2)} CNY` : ""),
+    };
+  } catch {
+    return undefined; // no mgmt token / upstream down — fall back to endpoint display
+  }
+}
+
 // Command Code's /alpha/billing/credits endpoint (auth: same Provider API key
 // as /provider/v1 models) returns live 5-hour and weekly rolling windows plus
 // the monthly credit balance — the same data as the cmd /usage CLI.
@@ -1195,6 +1295,12 @@ export default function (pi: ExtensionAPI) {
 
   pi.registerCommand("sub", {
     description: "Show subscription usage for the current supported model provider (use /sub refresh to force refresh).",
+    getArgumentCompletions: (prefix) => {
+      const items = ["refresh"]
+        .filter((k) => k.startsWith(String(prefix || "").trim().toLowerCase()))
+        .map((k) => ({ value: k, label: k, description: "force refresh" }));
+      return items.length > 0 ? items : null;
+    },
     handler: async (args, ctx) => {
       updateActiveAdapter(ctx, state, ctx.model);
       const command = args.trim().toLowerCase();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bacnh85/pi-sub",
-  "version": "0.1.31",
+  "version": "0.1.33",
   "description": "Pi extension showing subscription usage for supported model providers.",
   "type": "module",
   "license": "MIT",