@latentminds/pi-quotas 0.2.4 → 0.2.6

package/README.md

@@ -1,12 +1,14 @@
 # @latentminds/pi-quotas
 
-Quota monitoring for the [Pi coding agent](https://github.com/mariozechner/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, and Synthetic — directly in your Pi session.
 
 ## Screenshots
 
+
 | `/quotas` dashboard | Footer status |
-|---|---|
-| ![Quotas dashboard](docs/quotas-dashboard.png) | ![Footer status](docs/footer-status.png) |
+| ------------------- | ------------- |
+| Quotas dashboard    | Footer status |
+
 
 ## Install
 
@@ -31,15 +33,17 @@ pi -e npm:@latentminds/pi-quotas
 
 ## Commands
 
-| Command | Description |
-|---------|-------------|
-| `/quotas` | Combined quota dashboard for all providers |
-| `/anthropic:quotas` | Anthropic quotas only |
-| `/codex:quotas` | OpenAI Codex quotas only |
-| `/github:quotas` | GitHub Copilot quotas only |
-| `/openrouter:quotas` | OpenRouter quotas only |
-| `/synthetic:quotas` | Synthetic quotas only |
-| `/quotas:settings` | Toggle individual features on or off |
+
+| Command              | Description                                |
+| -------------------- | ------------------------------------------ |
+| `/quotas`            | Combined quota dashboard for all providers |
+| `/anthropic:quotas`  | Anthropic quotas only                      |
+| `/codex:quotas`      | OpenAI Codex quotas only                   |
+| `/github:quotas`     | GitHub Copilot quotas only                 |
+| `/openrouter:quotas` | OpenRouter quotas only                     |
+| `/synthetic:quotas`  | Synthetic quotas only                      |
+| `/quotas:settings`   | Toggle individual features on or off       |
+
 
 ## Features
 
@@ -58,6 +62,7 @@ Automatic notifications when projected usage is on track to exceed limits before
 ### Per-feature toggles
 
 Use `/quotas:settings` to enable or disable:
+
 - Combined `/quotas` command
 - Per-provider commands (`/anthropic:quotas`, `/codex:quotas`, `/github:quotas`, `/openrouter:quotas`, `/synthetic:quotas`)
 - Footer status widget
@@ -68,13 +73,15 @@ Settings can be saved globally (`~/.pi/agent/extensions/quotas.json`) or per-pro
 
 ## Supported providers
 
-| Provider | Windows | Details |
-|----------|---------|---------|
-| Anthropic | 5h, 7d, per-model 7d, extra usage | Utilization percentages; optional overage budget in local currency |
-| OpenAI Codex | 5h, 7d, credits, spend cap | Rate-limit percentages; credit balance; spend-cap reached/OK |
-| 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 |
+
+| Provider       | Windows                                                        | Details                                                                                             |
+| -------------- | -------------------------------------------------------------- | --------------------------------------------------------------------------------------------------- |
+| Anthropic      | 5h, 7d, per-model 7d, extra usage                              | Utilization percentages; optional overage budget in local currency                                  |
+| OpenAI Codex   | 5h, 7d, credits, spend cap                                     | Rate-limit percentages; credit balance; spend-cap reached/OK                                        |
+| 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                  |
+
 
 ## Credentials
 
@@ -104,6 +111,3 @@ See [CHANGELOG.md](CHANGELOG.md) for release notes and recent changes.
 
 This project was inspired by [@aliou/pi-synthetic](https://www.npmjs.com/package/@aliou/pi-synthetic).
 
-<p align="center">
-  <img src="docs/latent-minds@2x.png" alt="Latent Minds" width="320" />
-</p>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@latentminds/pi-quotas",
-  "version": "0.2.4",
+  "version": "0.2.6",
   "license": "MIT",
   "type": "module",
   "private": false,

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

@@ -0,0 +1,125 @@
+import { describe, expect, it, vi } from "vitest";
+import pkg from "../../../../package.json" with { type: "json" };
+import { QuotasComponent } from "./quotas-display.js";
+
+const ansi = {
+  accent: "\x1b[32m",
+  border: "\x1b[36m",
+  dim: "\x1b[2m",
+  error: "\x1b[31m",
+  muted: "\x1b[90m",
+  success: "\x1b[32m",
+  warning: "\x1b[33m",
+} as const;
+
+function fakeTheme() {
+  return {
+    bold: (text: string) => text,
+    fg: (color: keyof typeof ansi, text: string) => `${ansi[color] ?? ""}${text}\x1b[0m`,
+    getFgAnsi: (color: keyof typeof ansi) => ansi[color] ?? "",
+  };
+}
+
+function makeComponent(): QuotasComponent {
+  return new QuotasComponent(
+    fakeTheme() as any,
+    { requestRender: () => {} } as any,
+    "Quotas",
+    () => {},
+    () => {},
+  );
+}
+
+describe("QuotasComponent", () => {
+  it("renders the pi-quotas package version in the dashboard footer", () => {
+    const component = makeComponent();
+    component.setState({ type: "loaded", snapshots: [] });
+
+    expect(component.render(70).join("\n")).toContain(`pi-quotas v${pkg.version}`);
+  });
+
+  it("does not render an accent-colored pace marker inside filled warning bars", () => {
+    vi.useFakeTimers();
+    vi.setSystemTime(new Date("2026-05-14T13:42:11Z"));
+
+    const component = makeComponent();
+    component.setState({
+      type: "loaded",
+      snapshots: [
+        {
+          provider: "github-copilot",
+          result: {
+            success: true,
+            data: {
+              provider: "github-copilot",
+              windows: [
+                {
+                  provider: "github-copilot",
+                  label: "Premium / month",
+                  usedPercent: 83,
+                  resetsAt: new Date("2026-06-01T10:00:00Z"),
+                  windowSeconds: 31 * 24 * 3600,
+                  usedValue: 249,
+                  limitValue: 300,
+                  showPace: true,
+                  nextLabel: "Resets",
+                  nextAmount: "overage allowed",
+                },
+              ],
+            },
+          },
+        },
+      ],
+    });
+
+    const output = component.render(70).join("\n");
+
+    expect(output).toContain("51/300 left");
+    expect(output).not.toContain(`${ansi.accent}|`);
+
+    vi.useRealTimers();
+  });
+
+  it("does not render a pace marker for zero-usage windows", () => {
+    vi.useFakeTimers();
+    vi.setSystemTime(new Date("2026-05-14T13:49:29Z"));
+
+    const component = makeComponent();
+    component.setState({
+      type: "loaded",
+      snapshots: [
+        {
+          provider: "synthetic",
+          result: {
+            success: true,
+            data: {
+              provider: "synthetic",
+              windows: [
+                {
+                  provider: "synthetic",
+                  label: "Credits / week",
+                  usedPercent: 0,
+                  resetsAt: new Date("2026-05-14T16:40:29Z"),
+                  windowSeconds: 7 * 24 * 3600,
+                  usedValue: 0,
+                  limitValue: 24,
+                  isCurrency: true,
+                  showPace: true,
+                  nextLabel: "Next regen",
+                  nextAmount: "+$0.48",
+                },
+              ],
+            },
+          },
+        },
+      ],
+    });
+
+    const output = component.render(70).join("\n");
+
+    expect(output).toContain("$0.00 / $24.00");
+    expect(output).not.toContain("|");
+
+    vi.useRealTimers();
+  });
+});

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

@@ -2,6 +2,7 @@ import type { Theme } from "@mariozechner/pi-coding-agent";
 import { DynamicBorder } from "@mariozechner/pi-coding-agent";
 import type { Component } from "@mariozechner/pi-tui";
 import { Loader, matchesKey, truncateToWidth } from "@mariozechner/pi-tui";
+import pkg from "../../../../package.json" with { type: "json" };
 import { PROVIDER_LABELS } from "../../../lib/quotas.js";
 import type { QuotasResult, SupportedQuotaProvider } from "../../../types/quotas.js";
 import {
@@ -19,10 +20,6 @@ type QuotasState =
   | { type: "loading" }
   | { type: "loaded"; snapshots: Snapshot[] };
 
-function fgAnsiToBg(fgAnsi: string): string {
-  return fgAnsi.split("[38;").join("[48;").replace(/\[3([0-9])m/g, "[4$1m");
-}
-
 function renderProgressBar(
   percent: number,
   width: number,
@@ -33,6 +30,7 @@ function renderProgressBar(
   const clamped = Math.max(0, Math.min(100, Math.round(percent)));
   const filled = Math.round((clamped / 100) * width);
   const showPace =
+    percent > 0 &&
     pacePercent !== null &&
     pacePercent !== undefined &&
     pacePercent >= 5 &&
@@ -44,11 +42,10 @@ function renderProgressBar(
   const parts: string[] = [];
   for (let idx = 0; idx < width; idx++) {
     if (paceIndex !== null && idx === paceIndex) {
-      const markerColor = idx < filled ? "accent" : fillColor;
       if (idx < filled) {
-        parts.push(`${fgAnsiToBg(theme.getFgAnsi(fillColor))}${theme.getFgAnsi(markerColor)}|${reset}`);
+        parts.push(theme.fg(fillColor, "█"));
       } else {
-        parts.push(theme.fg(markerColor, "|"));
+        parts.push(theme.fg(fillColor, "|"));
       }
     } else if (idx < filled) {
       parts.push(theme.fg(fillColor, "█"));
@@ -123,7 +120,7 @@ export class QuotasComponent implements Component {
     }
 
     lines.push("");
-    lines.push(this.theme.fg("dim", "  r to refresh  q/Esc to close"));
+    lines.push(this.theme.fg("dim", `  pi-quotas v${pkg.version}  ·  r to refresh  q/Esc to close`));
     lines.push(...border.render(width));
     return lines;
   }

package/src/extensions/quota-warnings/index.ts

@@ -85,6 +85,12 @@ export default async function (pi: ExtensionAPI) {
     ctx.ui.notify(`${providerName} quota warning:\n${lines.join("\n")}`, level);
   }
 
+  function scheduleCheck(ctx: ExtensionContext, onlyNew: boolean): void {
+    void check(ctx, onlyNew).catch(() => {
+      // Quota warnings are opportunistic; never let a failed quota check block Pi events.
+    });
+  }
+
   pi.events.on(QUOTAS_CONFIG_UPDATED_EVENT, (data: unknown) => {
     enabled = (data as QuotasConfigUpdatedPayload).config.quotaWarnings;
     if (!enabled) {
@@ -93,21 +99,21 @@ export default async function (pi: ExtensionAPI) {
     }
     if (currentContext) {
       clearAlertState();
-      void check(currentContext, false);
+      scheduleCheck(currentContext, false);
     }
   });
 
-  pi.on("session_start", async (_event, ctx) => {
+  pi.on("session_start", (_event, ctx) => {
     currentContext = ctx;
     clearAlertState();
     if (!enabled) return;
-    await check(ctx, false);
+    scheduleCheck(ctx, false);
   });
 
-  pi.on("turn_end", async (_event, ctx) => {
+  pi.on("turn_end", (_event, ctx) => {
     currentContext = ctx;
     if (!enabled) return;
-    await check(ctx, true);
+    scheduleCheck(ctx, true);
   });
 
   pi.on("model_select", async (_event, ctx) => {

package/src/extensions/usage-status/index.ts

@@ -176,6 +176,12 @@ export default async function (pi: ExtensionAPI) {
     }
   });
 
+  function scheduleRefresh(ctx: ExtensionContext): void {
+    void refresher.refreshFor(ctx).catch(() => {
+      if (ctx.hasUI) ctx.ui.setStatus(EXTENSION_ID, ctx.ui.theme.fg("warning", "usage unavailable"));
+    });
+  }
+
   pi.events.on(QUOTAS_CONFIG_UPDATED_EVENT, (data: unknown) => {
     const config = (data as QuotasConfigUpdatedPayload).config;
     enabled = config.usageStatus;
@@ -186,7 +192,7 @@ export default async function (pi: ExtensionAPI) {
     }
     if (currentContext) {
       refresher.start();
-      void refresher.refreshFor(currentContext);
+      scheduleRefresh(currentContext);
     }
   });
 
@@ -198,7 +204,7 @@ export default async function (pi: ExtensionAPI) {
     return deferToSynthetic && syntheticUsageActive && provider === "synthetic";
   }
 
-  pi.on("session_start", async (_event, ctx) => {
+  pi.on("session_start", (_event, ctx) => {
     currentContext = ctx;
     if (!enabled) return;
     if (shouldDeferToSynthetic(ctx.model?.provider)) {
@@ -206,17 +212,17 @@ export default async function (pi: ExtensionAPI) {
       return;
     }
     refresher.start();
-    await refresher.refreshFor(ctx);
+    scheduleRefresh(ctx);
   });
 
-  pi.on("turn_end", async (_event, ctx) => {
+  pi.on("turn_end", (_event, ctx) => {
     currentContext = ctx;
     if (!enabled) return;
     if (shouldDeferToSynthetic(ctx.model?.provider)) return;
-    await refresher.refreshFor(ctx);
+    scheduleRefresh(ctx);
   });
 
-  pi.on("model_select", async (_event, ctx) => {
+  pi.on("model_select", (_event, ctx) => {
     currentContext = ctx;
     if (!enabled) {
       refresher.stop(ctx);
@@ -226,7 +232,7 @@ export default async function (pi: ExtensionAPI) {
       ctx.ui.setStatus(EXTENSION_ID, undefined);
       return;
     }
-    await refresher.refreshFor(ctx);
+    scheduleRefresh(ctx);
   });
 
   pi.on("session_shutdown", async (_event, ctx) => {

package/src/providers/fetch.test.ts

@@ -1,7 +1,9 @@
+import { AuthStorage } from "@mariozechner/pi-coding-agent";
 import { afterEach, describe, expect, it, vi } from "vitest";
 import {
   fetchAnthropicQuotasWithToken,
   fetchCodexQuotasWithToken,
+  fetchGitHubCopilotQuotas,
   fetchGitHubCopilotQuotasWithToken,
   fetchOpenRouterQuotasWithToken,
 } from "./fetch.js";
@@ -135,6 +137,44 @@ describe("fetchGitHubCopilotQuotasWithToken", () => {
     expect(result.success).toBe(true);
     expect(globalThis.fetch).toHaveBeenCalledTimes(2);
   });
+
+  it("uses the stored GitHub OAuth refresh token for Pi 0.74 Copilot quota checks", async () => {
+    const auth = AuthStorage.inMemory({
+      "github-copilot": {
+        type: "oauth",
+        refresh: "ghu-refresh-token",
+        access: "tid=abc;proxy-ep=proxy.individual.githubcopilot.com;exp=1778611280",
+        expires: Date.now() + 60_000,
+      },
+    });
+
+    globalThis.fetch = vi.fn(async (_url, init) => {
+      const authorization = new Headers(init?.headers).get("authorization");
+      if (authorization === "Bearer ghu-refresh-token") {
+        return new Response(
+          JSON.stringify({
+            quota_reset_date: "2026-05-01T00:00:00Z",
+            quota_snapshots: {
+              premium_interactions: { entitlement: 300, remaining: 210 },
+            },
+          }),
+          { status: 200 },
+        );
+      }
+      return new Response(JSON.stringify({ message: "Bad credentials" }), { status: 401 });
+    }) as any;
+
+    const result = await fetchGitHubCopilotQuotas(auth);
+
+    expect(result.success).toBe(true);
+    expect(globalThis.fetch).toHaveBeenCalledTimes(1);
+    expect(globalThis.fetch).toHaveBeenCalledWith(
+      "https://api.github.com/copilot_internal/user",
+      expect.objectContaining({
+        headers: expect.objectContaining({ Authorization: "Bearer ghu-refresh-token" }),
+      }),
+    );
+  });
 });
 
 describe("fetchOpenRouterQuotasWithToken", () => {

package/src/providers/fetch.ts

@@ -179,6 +179,32 @@ async function tryGitHubUserEndpoint(
   );
 }
 
+function githubOAuthToken(authStorage: AuthStorage): string | undefined {
+  // Pi's GitHub Copilot OAuth credential stores the GitHub OAuth token in
+  // `refresh`; `access` is a Copilot proxy token (tid=...;proxy-ep=...) that
+  // is valid for model calls but rejected by api.github.com quota endpoints.
+  const credential = authStorage.get("github-copilot") as any;
+  if (credential?.type !== "oauth") return undefined;
+  return typeof credential.refresh === "string" && credential.refresh.length > 0
+    ? credential.refresh
+    : undefined;
+}
+
+async function fetchGitHubCopilotQuotasWithGitHubToken(
+  githubToken: string | undefined,
+  signal?: AbortSignal,
+): Promise<QuotasResult> {
+  if (!githubToken) return failure("No GitHub Copilot OAuth token found", "config");
+
+  const bearerUsage = await tryGitHubUserEndpoint(`Bearer ${githubToken}`, signal);
+  if (bearerUsage.ok) return success("github-copilot", parseGitHubCopilotUsage(bearerUsage.data));
+
+  const tokenUsage = await tryGitHubUserEndpoint(`token ${githubToken}`, signal);
+  if (tokenUsage.ok) return success("github-copilot", parseGitHubCopilotUsage(tokenUsage.data));
+
+  return failure(bearerUsage.message, bearerUsage.kind);
+}
+
 export async function fetchGitHubCopilotQuotasWithToken(
   accessToken: string | undefined,
   signal?: AbortSignal,
@@ -234,6 +260,18 @@ export async function fetchGitHubCopilotQuotas(
   authStorage: AuthStorage,
   signal?: AbortSignal,
 ): Promise<QuotasResult> {
+  const oauthResult = await fetchGitHubCopilotQuotasWithGitHubToken(
+    githubOAuthToken(authStorage),
+    signal,
+  );
+  if (
+    oauthResult.success ||
+    oauthResult.error.kind === "cancelled" ||
+    oauthResult.error.kind === "timeout"
+  ) {
+    return oauthResult;
+  }
+
   return fetchGitHubCopilotQuotasWithToken(
     await providerAccessToken(authStorage, "github-copilot"),
     signal,

package/src/utils/quotas-severity.test.ts

@@ -74,13 +74,33 @@ describe("assessWindow", () => {
     expect(assessWindow(w).severity).toBe("critical");
   });
 
+  it("warns for pace-aware windows once absolute usage reaches 80%", () => {
+    const now = Date.now();
+    const monthSeconds = 31 * 24 * 3600;
+    const w = makeWindow({
+      provider: "github-copilot",
+      label: "Premium / month",
+      usedPercent: 83,
+      usedValue: 249,
+      limitValue: 300,
+      showPace: true,
+      windowSeconds: monthSeconds,
+      // Screenshot repro: 51/300 left with reset in about 428h35m in a 31-day
+      // monthly window. Dynamic pace alone projects just under warning, but
+      // only 17% remains, so low-quota color should still be amber.
+      resetsAt: new Date(now + ((428 * 60 + 35) * 60 * 1000)),
+    });
+
+    expect(assessWindow(w).severity).toBe("warning");
+  });
+
   it("returns warning when projected exceeds dynamic warn threshold", () => {
     const w = makeWindow({
-      usedPercent: 95,
+      usedPercent: 79,
       showPace: true,
       paceScale: 1,
       windowSeconds: 3600,
-      resetsAt: new Date(Date.now() + 1800 * 1000),
+      resetsAt: new Date(Date.now() + 2520 * 1000),
     });
     expect(assessWindow(w).severity).toBe("warning");
   });

package/src/utils/quotas-severity.ts

@@ -53,11 +53,24 @@ export function getProjectedPercent(
   return Math.max(0, (usedPercent / effectivePace) * 100);
 }
 
+function absoluteUsageSeverity(window: QuotaWindow, percent: number): RiskSeverity {
+  if (window.limited || percent >= 100) return "critical";
+  if (percent >= 90) return "high";
+  if (percent >= 80) return "warning";
+  return "none";
+}
+
+function maxSeverity(a: RiskSeverity, b: RiskSeverity): RiskSeverity {
+  const order: RiskSeverity[] = ["none", "warning", "high", "critical"];
+  return order[Math.max(order.indexOf(a), order.indexOf(b))] ?? "none";
+}
+
 export function assessWindow(window: QuotaWindow): RiskAssessment {
   const rawPace = window.showPace ? getPacePercent(window) : null;
   const pacePercent =
     rawPace !== null ? rawPace * (window.paceScale ?? 1) : null;
   const projectedPercent = getProjectedPercent(window.usedPercent, pacePercent);
+  const absoluteSeverity = absoluteUsageSeverity(window, window.usedPercent);
 
   let progress: number | null = null;
   if (pacePercent !== null) progress = pacePercent / 100;
@@ -70,10 +83,7 @@ export function assessWindow(window: QuotaWindow): RiskAssessment {
   };
 
   if (progress === null) {
-    let severity: RiskSeverity = "none";
-    if (window.limited || projectedPercent >= 100) severity = "critical";
-    else if (projectedPercent >= 90) severity = "high";
-    else if (projectedPercent >= 80) severity = "warning";
+    const severity = absoluteUsageSeverity(window, projectedPercent);
 
     return {
       ...base,
@@ -121,7 +131,7 @@ export function assessWindow(window: QuotaWindow): RiskAssessment {
     warnProjectedPercent,
     highProjectedPercent,
     criticalProjectedPercent,
-    severity,
+    severity: maxSeverity(severity, absoluteSeverity),
   };
 }