pi-usage-limits 0.1.2 → 0.1.4

package/README.md

@@ -25,7 +25,9 @@ A [pi](https://github.com/earendil-works/pi) extension that shows your **Claude*
 | Active model | Source | Shows |
 |---|---|---|
 | Model id contains `claude` | Claude Agent SDK (Claude Code login) | 5-hour and 7-day utilization |
-| `openai-codex` provider, or model id contains `codex` | ChatGPT usage endpoint (pi's `openai-codex` login) | 5-hour and weekly usage |
+| `openai` or `openai-codex` provider, or model id contains `codex` | ChatGPT usage endpoint (pi's `openai-codex` login) | 5-hour and weekly usage |
+
+OpenAI models such as `(openai) gpt-6.1-sol` show the same ChatGPT subscription limits using your `openai-codex` login, not OpenAI API limits.
 
 Other models show `<model> not supported yet`.
 

package/index.ts

@@ -5,17 +5,25 @@ import type {
 } from "@earendil-works/pi-coding-agent";
 import { getUsage } from "./usage";
 
-// pi awaits handler promises, so awaiting here blocks the redraw.
-function refresh(_event: ExtensionEvent, ctx: ExtensionContext) {
-  if (!ctx.model) return;
-  getUsage(ctx.model, ctx.modelRegistry)
-    .then((usage) =>
-      ctx.ui.setStatus("model-usage", ctx.ui.theme.fg("dim", usage)),
-    )
-    .catch((e) => ctx.ui.setStatus("model-usage", ctx.ui.theme.fg("dim", e)));
-}
-
 export default function (pi: ExtensionAPI) {
+  let generation = 0;
+
+  // pi awaits handler promises, so awaiting here blocks the redraw.
+  function refresh(_event: ExtensionEvent, ctx: ExtensionContext) {
+    const current = ++generation;
+    if (ctx.mode !== "tui" || !ctx.model) return;
+
+    const setStatus = (text: string) => {
+      // Shutdown or a newer refresh invalidates this captured context.
+      if (current !== generation) return;
+      ctx.ui.setStatus("model-usage", ctx.ui.theme.fg("dim", text));
+    };
+    getUsage(ctx.model, ctx.modelRegistry).then(setStatus, setStatus);
+  }
+
+  pi.on("session_shutdown", () => {
+    generation++;
+  });
   pi.on("session_start", refresh);
   pi.on("model_select", refresh);
   pi.on("agent_settled", refresh);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-usage-limits",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "",
   "keywords": [
     "pi-package",

package/usage.ts

@@ -14,7 +14,11 @@ export async function getUsage(
   model: { id: string; provider?: string },
   auth: TokenSource,
 ): Promise<string> {
-  if (model.provider === "openai-codex" || model.id.includes("codex")) {
+  if (
+    model.provider === "openai" ||
+    model.provider === "openai-codex" ||
+    model.id.includes("codex")
+  ) {
     return getCodexUsage(auth);
   }