@latentminds/pi-quotas 0.2.6 → 0.3.1

package/README.md

@@ -1,6 +1,6 @@
 # @latentminds/pi-quotas
 
-Quota monitoring for Pi. Shows remaining usage and rate limits for Anthropic, OpenAI Codex, GitHub Copilot, OpenRouter, and Synthetic — directly in your Pi session.
+Quota monitoring for Pi. Shows remaining usage and rate limits for Anthropic, OpenAI Codex, GitHub Copilot, OpenRouter, Synthetic, Z.ai, and OpenCode Go — directly in your Pi session.
 
 ## Screenshots
 
@@ -42,6 +42,9 @@ pi -e npm:@latentminds/pi-quotas
 | `/github:quotas`     | GitHub Copilot quotas only                 |
 | `/openrouter:quotas` | OpenRouter quotas only                     |
 | `/synthetic:quotas`  | Synthetic quotas only                      |
+| `/zai:quotas`        | Z.ai quotas only                           |
+| `/opencode-go:quotas`| OpenCode Go quotas only                    |
+| `/tokens`            | Cross-session token/cost usage            |
 | `/quotas:settings`   | Toggle individual features on or off       |
 
 
@@ -64,7 +67,7 @@ Automatic notifications when projected usage is on track to exceed limits before
 Use `/quotas:settings` to enable or disable:
 
 - Combined `/quotas` command
-- Per-provider commands (`/anthropic:quotas`, `/codex:quotas`, `/github:quotas`, `/openrouter:quotas`, `/synthetic:quotas`)
+- Per-provider commands (`/anthropic:quotas`, `/codex:quotas`, `/github:quotas`, `/openrouter:quotas`, `/synthetic:quotas`, `/zai:quotas`, `/opencode-go:quotas`)
 - Footer status widget
 - Quota warning notifications
 - **Defer to Synthetic** — when both pi-quotas and [pi-synthetic](https://www.npmjs.com/package/@aliou/pi-synthetic) are loaded, pi-quotas hides its own Synthetic footer to avoid showing duplicate quota information. Enabled by default; disable if you prefer to see both footers.
@@ -81,6 +84,8 @@ Settings can be saved globally (`~/.pi/agent/extensions/quotas.json`) or per-pro
 | GitHub Copilot | Premium/chat/completions per month                             | Remaining/entitlement counts with overage indicators                                                |
 | OpenRouter     | Monthly budget, daily/weekly/monthly usage                     | USD spending tracking with cents precision; optional per-key budget limits; UTC-based period resets |
 | Synthetic      | Subscription, search/hour, free tools, weekly tokens, 5h limit | Request counts and token budgets; rolling five-hour rate limit; weekly token regen                  |
+| Z.ai           | 5h, 7d, monthly web searches                                  | Token utilisation percentages (rolling 5h/7d windows); monthly web-search count limit               |
+| OpenCode Go    | Rolling 5h, weekly, monthly USD                              | USD spend tracking against tier limits; cross-session token/cost aggregation via the `/tokens` command |
 
 
 ## Credentials
@@ -92,6 +97,8 @@ pi-quotas reads existing Pi auth entries from `~/.pi/agent/auth.json`:
 - `github-copilot` — GitHub Copilot OAuth token (falls back to `gh auth token` if needed)
 - `openrouter` — OpenRouter API key (Bearer token)
 - `synthetic` — Synthetic API key (set the `SYNTHETIC_API_KEY` environment variable)
+- `zai` — Z.ai (Zhipu AI / GLM Coding Plan) API key
+- `opencode-go` — OpenCode Go workspace ID and auth cookie (set the `OPENCODE_GO_WORKSPACE_ID` and `OPENCODE_GO_AUTH_COOKIE` environment variables, or configure them in the OpenCode Go config file)
 
 No additional setup is required - if Pi can use the provider, pi-quotas can check its quotas. For Synthetic, export `SYNTHETIC_API_KEY` in your shell or Pi environment.
 

package/package.json

@@ -1,11 +1,15 @@
 {
   "name": "@latentminds/pi-quotas",
-  "version": "0.2.6",
+  "version": "0.3.1",
   "license": "MIT",
   "type": "module",
   "private": false,
   "author": "Markus Strazds <markus@latentminds.ai>",
-  "keywords": ["pi-package", "pi-extension", "pi"],
+  "keywords": [
+    "pi-package",
+    "pi-extension",
+    "pi"
+  ],
   "repository": {
     "type": "git",
     "url": "https://github.com/latentminds-ai/pi-quotas"
@@ -14,30 +18,46 @@
     "extensions": [
       "./src/extensions/core/index.ts",
       "./src/extensions/command-quotas/index.ts",
+      "./src/extensions/command-tokens/index.ts",
       "./src/extensions/usage-status/index.ts",
+      "./src/extensions/token-status/index.ts",
       "./src/extensions/quota-warnings/index.ts"
     ],
     "image": "https://raw.githubusercontent.com/latentminds-ai/pi-quotas/main/docs/quotas-dashboard.png"
   },
-  "files": ["src", "README.md", "LICENSE"],
+  "files": [
+    "src",
+    "README.md",
+    "LICENSE"
+  ],
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": "*",
     "@mariozechner/pi-tui": "*"
   },
   "peerDependenciesMeta": {
-    "@mariozechner/pi-coding-agent": { "optional": true },
-    "@mariozechner/pi-tui": { "optional": true }
+    "@mariozechner/pi-coding-agent": {
+      "optional": true
+    },
+    "@mariozechner/pi-tui": {
+      "optional": true
+    }
   },
   "devDependencies": {
+    "@eslint/js": "^10.0.1",
     "@mariozechner/pi-coding-agent": "0.61.0",
     "@mariozechner/pi-tui": "0.61.0",
     "@sinclair/typebox": "^0.34.48",
     "@types/node": "^25.0.10",
+    "eslint": "^10.6.0",
+    "globals": "^17.7.0",
     "typescript": "^5.9.3",
+    "typescript-eslint": "^8.63.0",
     "vitest": "^4.0.18"
   },
   "scripts": {
     "typecheck": "tsc --noEmit",
+    "lint": "eslint src",
+    "lint:fix": "eslint src --fix",
     "test": "vitest run",
     "test:watch": "vitest"
   }

package/src/config.ts

@@ -9,6 +9,7 @@ export type QuotasFeatureId =
   | "quotasCommand"
   | "providerCommands"
   | "usageStatus"
+  | "tokenStatus"
   | "quotaWarnings"
   | "deferToSynthetic";
 
@@ -16,8 +17,7 @@ export const QUOTAS_EXTENSIONS_REQUEST_EVENT =
   "quotas:extensions:request" as const;
 export const QUOTAS_EXTENSIONS_REGISTER_EVENT =
   "quotas:extensions:register" as const;
-export const QUOTAS_CONFIG_UPDATED_EVENT =
-  "quotas:config:updated" as const;
+export const QUOTAS_CONFIG_UPDATED_EVENT = "quotas:config:updated" as const;
 
 export interface QuotasExtensionsRegisterPayload {
   feature: QuotasFeatureId;
@@ -28,6 +28,7 @@ export interface QuotasConfig {
   quotasCommand?: boolean;
   providerCommands?: boolean;
   usageStatus?: boolean;
+  tokenStatus?: boolean;
   quotaWarnings?: boolean;
   /** When true and pi-synthetic's usage footer is active, hide pi-quotas' Synthetic footer. */
   deferToSynthetic?: boolean;
@@ -38,6 +39,7 @@ export interface ResolvedQuotasConfig {
   quotasCommand: boolean;
   providerCommands: boolean;
   usageStatus: boolean;
+  tokenStatus: boolean;
   quotaWarnings: boolean;
   deferToSynthetic: boolean;
 }
@@ -47,6 +49,7 @@ const DEFAULT_CONFIG: ResolvedQuotasConfig = {
   quotasCommand: true,
   providerCommands: true,
   usageStatus: true,
+  tokenStatus: true,
   quotaWarnings: true,
   deferToSynthetic: true,
 };
@@ -81,10 +84,13 @@ class QuotasConfigStore {
     return {
       configVersion: input?.configVersion ?? DEFAULT_CONFIG.configVersion,
       quotasCommand: input?.quotasCommand ?? DEFAULT_CONFIG.quotasCommand,
-      providerCommands: input?.providerCommands ?? DEFAULT_CONFIG.providerCommands,
+      providerCommands:
+        input?.providerCommands ?? DEFAULT_CONFIG.providerCommands,
       usageStatus: input?.usageStatus ?? DEFAULT_CONFIG.usageStatus,
+      tokenStatus: input?.tokenStatus ?? DEFAULT_CONFIG.tokenStatus,
       quotaWarnings: input?.quotaWarnings ?? DEFAULT_CONFIG.quotaWarnings,
-      deferToSynthetic: input?.deferToSynthetic ?? DEFAULT_CONFIG.deferToSynthetic,
+      deferToSynthetic:
+        input?.deferToSynthetic ?? DEFAULT_CONFIG.deferToSynthetic,
     };
   }
 
@@ -118,7 +124,11 @@ class QuotasConfigStore {
   async save(scope: "global" | "local", config: QuotasConfig): Promise<void> {
     const path = scope === "global" ? this.globalPath() : this.localPath();
     await mkdir(dirname(path), { recursive: true });
-    await writeFile(path, JSON.stringify(this.resolve(config), null, 2) + "\n", "utf8");
+    await writeFile(
+      path,
+      JSON.stringify(this.resolve(config), null, 2) + "\n",
+      "utf8",
+    );
     await this.load(this.cwd);
   }
 }
@@ -126,7 +136,8 @@ class QuotasConfigStore {
 export const configLoader = new QuotasConfigStore();
 
 export async function seedQuotasConfigIfMissing(): Promise<void> {
-  if (configLoader.hasConfig("global") || configLoader.hasConfig("local")) return;
+  if (configLoader.hasConfig("global") || configLoader.hasConfig("local"))
+    return;
   markMigrationNoticePending();
   try {
     await configLoader.save("global", DEFAULT_CONFIG);
@@ -158,13 +169,19 @@ const FEATURE_META: Array<{
   {
     id: "providerCommands",
     label: "Provider quota commands",
-    description: "Toggle `/anthropic:quotas`, `/codex:quotas`, `/github:quotas`, `/openrouter:quotas`, and `/synthetic:quotas`",
+    description:
+      "Toggle `/anthropic:quotas`, `/codex:quotas`, `/github:quotas`, `/openrouter:quotas`, and `/synthetic:quotas`",
   },
   {
     id: "usageStatus",
     label: "Usage status",
     description: "Toggle footer quota status for the active provider",
   },
+  {
+    id: "tokenStatus",
+    label: "Token usage status",
+    description: "Toggle the footer token-usage status and /tokens command",
+  },
   {
     id: "quotaWarnings",
     label: "Quota warnings",
@@ -173,7 +190,8 @@ const FEATURE_META: Array<{
   {
     id: "deferToSynthetic",
     label: "Defer to Synthetic",
-    description: "When pi-synthetic is loaded, hide pi-quotas' Synthetic footer to avoid duplicates",
+    description:
+      "When pi-synthetic is loaded, hide pi-quotas' Synthetic footer to avoid duplicates",
   },
 ];
 
@@ -196,7 +214,9 @@ export function registerQuotasSettings(
 
       while (true) {
         const choices = FEATURE_META.map((feature) => {
-          const loaded = getLoadedFeatures().has(feature.id) ? "" : " (not loaded)";
+          const loaded = getLoadedFeatures().has(feature.id)
+            ? ""
+            : " (not loaded)";
           const enabled = draft[feature.id] ? "enabled" : "disabled";
           return `${feature.label}: ${enabled}${loaded}`;
         });
@@ -214,13 +234,23 @@ export function registerQuotasSettings(
           return;
         }
 
-        const feature = FEATURE_META.find((item) => selected.startsWith(item.label));
+        const feature = FEATURE_META.find((item) =>
+          selected.startsWith(item.label),
+        );
         if (!feature) continue;
-        if (feature.id !== "deferToSynthetic" && !getLoadedFeatures().has(feature.id)) {
-          ctx.ui.notify(`${feature.label} is not loaded by Pi in this session.`, "warning");
+        if (
+          feature.id !== "deferToSynthetic" &&
+          !getLoadedFeatures().has(feature.id)
+        ) {
+          ctx.ui.notify(
+            `${feature.label} is not loaded by Pi in this session.`,
+            "warning",
+          );
           continue;
         }
-        (draft as unknown as Record<string, boolean>)[feature.id] = !(draft as unknown as Record<string, boolean>)[feature.id];
+        (draft as unknown as Record<string, boolean>)[feature.id] = !(
+          draft as unknown as Record<string, boolean>
+        )[feature.id];
       }
     },
   });

package/src/extensions/command-quotas/command.ts

@@ -59,10 +59,32 @@ async function openQuotaView(
 
   if (result === undefined) {
     const snapshots = await loadSnapshots(true);
-    ctx.ui.notify(JSON.stringify(snapshots, null, 2), "info");
+    ctx.ui.notify(formatSnapshotsForNotify(snapshots), "info");
   }
 }
 
+/**
+ * Render quota snapshots as a readable multi-line summary for the
+ * non-interactive fallback (when `ctx.ui.custom` returns undefined). Avoids
+ * dumping raw JSON — which previously leaked raw HTTP error bodies — and
+ * skips "not_applicable" providers since they have nothing to report.
+ */
+function formatSnapshotsForNotify(snapshots: Snapshot[]): string {
+  const lines: string[] = [];
+  for (const { provider, result } of snapshots) {
+    if (!result.success) {
+      if (result.error.kind === "not_applicable") continue;
+      lines.push(`${provider}: ${result.error.message}`);
+      continue;
+    }
+    const summary = result.data.windows
+      .map((w) => `${w.label} ${w.usedPercent}%`)
+      .join(", ");
+    lines.push(`${provider}: ${summary || "no windows"}`);
+  }
+  return lines.join("\n") || "No quota data available";
+}
+
 export function registerQuotasCommands(pi: ExtensionAPI): void {
   pi.registerCommand("quotas", {
     description: "Display remaining quotas for Anthropic, Codex, GitHub Copilot, and OpenRouter",

package/src/extensions/command-quotas/components/quotas-display.test.ts

@@ -122,4 +122,33 @@ describe("QuotasComponent", () => {
 
     vi.useRealTimers();
   });
+
+  it("renders a not_applicable provider silently with a dim note, not a warning", () => {
+    const component = makeComponent();
+    component.setState({
+      type: "loaded",
+      snapshots: [
+        {
+          provider: "anthropic",
+          result: {
+            success: false,
+            error: {
+              kind: "not_applicable",
+              message: "Direct API key — no subscription usage to report",
+            },
+          },
+        },
+      ],
+    });
+
+    const output = component.render(70).join("\n");
+
+    expect(output).toContain("Anthropic");
+    expect(output).toContain("Direct API key");
+    // Rendered as a dim informational note, not a warning.
+    expect(output).toContain("\x1b[2m");
+    expect(output).not.toContain("\x1b[33m");
+    expect(output).not.toContain("usage unavailable");
+    expect(output).not.toContain("{");
+  });
 });

package/src/extensions/command-quotas/components/quotas-display.ts

@@ -38,7 +38,6 @@ function renderProgressBar(
   const paceIndex = showPace
     ? Math.min(width - 1, Math.round((Math.max(0, Math.min(100, pacePercent ?? 0)) / 100) * width))
     : null;
-  const reset = "\x1b[0m";
   const parts: string[] = [];
   for (let idx = 0; idx < width; idx++) {
     if (paceIndex !== null && idx === paceIndex) {
@@ -141,7 +140,14 @@ export class QuotasComponent implements Component {
     lines.push(truncateToWidth(`  ${this.theme.fg("accent", title)}`, maxWidth));
 
     if (!snapshot.result.success) {
-      lines.push(truncateToWidth(`  ${this.theme.fg("warning", snapshot.result.error.message)}`, maxWidth));
+      // "not applicable" is an expected, non-error state (e.g. a direct
+      // Anthropic API key with no subscription usage) — show a dim note,
+      // not a warning-coloured error.
+      const { error } = snapshot.result;
+      const tone = error.kind === "not_applicable" ? "dim" : "warning";
+      lines.push(
+        truncateToWidth(`  ${this.theme.fg(tone, error.message)}`, maxWidth),
+      );
       return lines;
     }
 

package/src/extensions/command-quotas/provider-commands.test.ts

@@ -40,4 +40,13 @@ describe("getProviderCommandInfo", () => {
       title: "OpenRouter Quotas",
     });
   });
+
+  it("maps zai to zai:quotas", () => {
+    const info = getProviderCommandInfo("zai");
+    expect(info).toMatchObject<Partial<ProviderCommandInfo>>({
+      provider: "zai",
+      commandName: "zai:quotas",
+      title: "Z.ai Quotas",
+    });
+  });
 });

package/src/extensions/command-quotas/provider-commands.ts

@@ -40,5 +40,17 @@ export function getProviderCommandInfo(
         commandName: "synthetic:quotas",
         title: "Synthetic Quotas",
       };
+    case "zai":
+      return {
+        provider,
+        commandName: "zai:quotas",
+        title: "Z.ai Quotas",
+      };
+    case "opencode-go":
+      return {
+        provider,
+        commandName: "opencode-go:quotas",
+        title: "OpenCode Go Quotas",
+      };
   }
 }

package/src/extensions/command-tokens/command.ts

@@ -0,0 +1,102 @@
+import type {
+  ExtensionAPI,
+  ExtensionCommandContext,
+} from "@mariozechner/pi-coding-agent";
+import { aggregateAllSessions } from "../../lib/session-tokens.js";
+import { TokensComponent } from "./tokens-display.js";
+
+async function openTokensView(
+  ctx: ExtensionCommandContext,
+  cwd?: string,
+): Promise<void> {
+  const result = await ctx.ui.custom<null>((tui, theme, _kb, done) => {
+    const controller = new AbortController();
+    const component = new TokensComponent(
+      theme,
+      tui,
+      () => {
+        controller.abort();
+        done(null);
+      },
+      () => {
+        component.setState({ type: "loading" });
+        tui.requestRender();
+        void load();
+      },
+      cwd,
+    );
+
+    async function load(): Promise<void> {
+      try {
+        const aggregateResult = await aggregateAllSessions({ cwd });
+        if (controller.signal.aborted) return;
+        component.setState({ type: "loaded", result: aggregateResult });
+        tui.requestRender();
+      } catch {
+        if (controller.signal.aborted) return;
+        component.setState({
+          type: "loaded",
+          result: {
+            totals: {
+              input: 0,
+              output: 0,
+              cacheRead: 0,
+              cacheWrite: 0,
+              totalTokens: 0,
+              costTotal: 0,
+            },
+            byModel: [],
+            byProvider: [],
+            bySession: [],
+            sessionCount: 0,
+            messageCount: 0,
+          },
+        });
+        tui.requestRender();
+      }
+    }
+
+    void load();
+
+    return {
+      render: (width: number) => component.render(width),
+      invalidate: () => component.invalidate(),
+      handleInput: (data: string) => component.handleInput(data),
+      dispose: () => {
+        controller.abort();
+        component.destroy();
+      },
+    };
+  });
+
+  // Fallback for non-interactive mode
+  if (result === undefined) {
+    const aggregateResult = await aggregateAllSessions({ cwd });
+    const { totals } = aggregateResult;
+    const lines = [
+      `Token Usage: ${aggregateResult.sessionCount} sessions, ${aggregateResult.messageCount} messages`,
+      `  Input: ${totals.input.toLocaleString()} · Output: ${totals.output.toLocaleString()}`,
+      `  Cache Read: ${totals.cacheRead.toLocaleString()} · Cache Write: ${totals.cacheWrite.toLocaleString()}`,
+      `  Total: ${totals.totalTokens.toLocaleString()} tokens · $${totals.costTotal.toFixed(2)}`,
+    ];
+    for (const model of aggregateResult.byModel) {
+      lines.push(
+        `  ${model.provider}/${model.model}: $${model.tokens.costTotal.toFixed(2)} (${model.messageCount} msgs)`,
+      );
+    }
+    ctx.ui.notify(lines.join("\n"), "info");
+  }
+}
+
+export function registerTokensCommand(pi: ExtensionAPI): void {
+  pi.registerCommand("tokens", {
+    description: "Display token usage and cost across all sessions",
+    handler: async (_args, ctx) => {
+      await openTokensView(ctx, ctx.cwd);
+    },
+  });
+}
+
+export default async function (pi: ExtensionAPI) {
+  registerTokensCommand(pi);
+}

package/src/extensions/command-tokens/index.ts

@@ -0,0 +1 @@
+export { default } from "./command.js";