pi-better-openai 0.1.19 → 0.1.22

package/README.md

@@ -4,6 +4,8 @@ A pi extension for OpenAI subscription workflows: fast mode, usage visibility, f
 
 ## Install
 
+Requires Node.js 22.19.0 or newer.
+
 Install from GitHub:
 
 ```bash
@@ -23,7 +25,7 @@ Usage display and image generation require pi's `openai-codex` OAuth credentials
 1. In pi, run `/login openai-codex`.
 2. Verify subscription usage with `/openai-usage`, or open `/openai-settings` and check **Diagnostics**.
 3. The extension reads auth from pi's agent auth store, normally `~/.pi/agent/auth.json`. Do not copy, paste, or commit values from this file.
-4. If `PI_CODING_AGENT_DIR` is set, the auth store and global generated-image directory use that agent directory instead of `~/.pi/agent`.
+4. If `PI_CODING_AGENT_DIR` is set, the auth store, global extension config, and global generated-image directory use that agent directory instead of `~/.pi/agent`. A leading `~/` is expanded to your home directory.
 
 ## Features
 
@@ -45,14 +47,22 @@ Usage display and image generation require pi's `openai-codex` OAuth credentials
 The extension reads JSON config from two locations:
 
 - Project config: `.pi/extensions/pi-better-openai.json`
-- Global config: `~/.pi/agent/extensions/pi-better-openai.json`
+- Global config: `$PI_CODING_AGENT_DIR/extensions/pi-better-openai.json`, defaulting to `~/.pi/agent/extensions/pi-better-openai.json`
 
 Project overrides global. Global values fill fields omitted by the project file. Invalid enum values are ignored, and numeric settings are clamped to safe ranges.
 
 Default supported models:
 
 ```json
-["openai/gpt-5.4", "openai/gpt-5.5", "openai-codex/gpt-5.4", "openai-codex/gpt-5.5"]
+[
+  "openai/gpt-5.4",
+  "openai/gpt-5.5",
+  "openai-codex/gpt-5.6-sol",
+  "openai-codex/gpt-5.6-terra",
+  "openai-codex/gpt-5.6-luna",
+  "openai-codex/gpt-5.4",
+  "openai-codex/gpt-5.5"
+]
 ```
 
 Example config:
@@ -105,7 +115,7 @@ Agents can call the `openai_image` tool directly. Supported parameters:
 
 - `prompt` (required): pass the user's image wording verbatim.
 - `action`: `auto`, `generate`, or `edit`.
-- `images`: project-local reference/edit image paths. Paths must stay inside the current workspace and point to readable PNG, JPEG, WebP, or GIF files.
+- `images`: up to five distinct project-local reference/edit image paths. Paths must stay inside the current workspace and point to readable PNG, JPEG, WebP, or GIF files; each file is limited to 20 MB and the combined input to 50 MB.
 - `model`: Codex image model override, for example `openai-codex/gpt-5.5`.
 - `outputFormat`: `png`, `jpeg`, or `webp`.
 - `save`: `project`, `global`, `custom`, or `none`.
@@ -115,7 +125,7 @@ Save modes:
 
 - `project` writes to `.pi/generated-images/` in the current project.
 - `global` writes to the agent `generated-images` directory, normally `~/.pi/agent/generated-images/` or `$PI_CODING_AGENT_DIR/generated-images/`.
-- `custom` writes to `saveDir` or `PI_IMAGE_SAVE_DIR`.
+- `custom` writes to `saveDir` or `PI_IMAGE_SAVE_DIR`; relative paths are resolved from the current project.
 - `none` returns the image without saving it.
 
 The repository ignores `.pi/`, so generated images and local config should not be committed.
@@ -146,7 +156,7 @@ Then reload Codex skills (`Cmd/Ctrl+K → Force Reload Skills`) and ask:
 $hatch-pet create a new pet inspired by pi-better-openai
 ```
 
-Custom pets should end up in `${CODEX_HOME:-~/.codex}/pets/<pet-name>/` with `pet.json` and `spritesheet.webp`. Refresh custom pets in Codex settings and toggle the overlay with `/pet`.
+Custom pets should end up in `${CODEX_HOME:-~/.codex}/pets/<pet-name>/` with `pet.json` and `spritesheet.webp`. The spritesheet must be a 1536×1872 atlas arranged as 8 columns by 9 animation rows. Animated footer rendering also requires a terminal image protocol supported by pi. Refresh custom pets in Codex settings and toggle the overlay with `/pet`.
 
 ## Screenshots
 

package/index.ts

@@ -4,15 +4,20 @@
  * Adds `service_tier: "priority"` to OpenAI provider payloads while fast mode is
  * enabled and the selected model is in the configured allow-list.
  */
-import { sep } from "node:path";
 import {
   getSettingsListTheme,
   type ExtensionAPI,
   type ExtensionContext,
-} from "@mariozechner/pi-coding-agent";
-import { Container, Key, matchesKey, SettingsList } from "@mariozechner/pi-tui";
+} from "@earendil-works/pi-coding-agent";
+import { Container, Key, matchesKey, SettingsList } from "@earendil-works/pi-tui";
 import { CONFIG_BASENAME, STATUS_KEY } from "./src/identity.ts";
-import { formatTokens, sanitizeStatusText, truncateToWidth, visibleWidth } from "./src/format.ts";
+import {
+  formatTokens,
+  redactDiagnosticValue,
+  sanitizeStatusText,
+  truncateToWidth,
+  visibleWidth,
+} from "./src/format.ts";
 import {
   DEFAULT_CONFIG,
   DEFAULT_IMAGE_CONFIG,
@@ -27,7 +32,6 @@ import {
   type SettingsOptionDescriptor,
   configPaths,
   type ResolvedConfig,
-  type PetPlacement,
   type PetState,
   isRecord,
   parseModelKey,
@@ -49,6 +53,7 @@ import {
   type CodexPetPackage,
   codexHome,
   describeCodexPetSelectionIssue,
+  findCodexPet,
   findReadyCodexPet,
   formatNoReadyCodexPetsMessage,
   listCodexPets,
@@ -58,6 +63,13 @@ import {
 import { FastController, modelList, supportsFast } from "./src/fast-controller.ts";
 import { UsageController } from "./src/usage-controller.ts";
 import { PetFooterController } from "./src/pet-footer-controller.ts";
+import {
+  abbreviateHomePath,
+  combineInlinePetFooter,
+  isInlinePetPlacement,
+  isTerminalImageLine,
+  petSizeCellsForPlacement,
+} from "./src/footer-layout.ts";
 
 const COMMAND = "fast";
 const OPENAI_STATUS_COMMAND = "openai-usage";
@@ -77,22 +89,8 @@ type SettingsPickerItem = {
   ) => { render(width: number): string[]; invalidate(): void; handleInput?(data: string): void };
 };
 
-function abbreviateHomePath(
-  path: string,
-  home = process.env.HOME || process.env.USERPROFILE,
-): string {
-  if (!home) return path;
-  if (path === home) return "~";
-  const homePrefix = home.endsWith(sep) ? home : `${home}${sep}`;
-  return path.startsWith(homePrefix) ? `~/${path.slice(homePrefix.length)}` : path;
-}
-
-function isInlinePetPlacement(placement: PetPlacement): boolean {
-  return placement === "inline-left" || placement === "inline-right" || placement === "badge";
-}
-
-function petSizeCellsForPlacement(placement: PetPlacement, sizeCells: number): number {
-  return placement === "badge" ? Math.min(6, sizeCells) : sizeCells;
+function hasTerminalUI(ctx: ExtensionContext): boolean {
+  return ctx.mode === "tui" || (ctx.mode === undefined && ctx.hasUI);
 }
 
 function readyPetPickerValues(pets: CodexPetPackage[]): string[] | undefined {
@@ -108,19 +106,10 @@ function petSlugFromPickerValue(value: string): string {
   return value === PET_EMPTY_VALUE ? "" : value;
 }
 
-function petPickerLookupKey(value: string): string {
-  return value.toLowerCase().replace(/[^a-z0-9]+/g, "");
-}
-
 function findPickerPet(value: string, pets: CodexPetPackage[]): CodexPetPackage | undefined {
-  const requested = petPickerLookupKey(value.trim());
-  if (!requested) return undefined;
-  return pets.find(
-    (pet) =>
-      pet.hasSpritesheet &&
-      (petPickerLookupKey(pet.slug) === requested ||
-        petPickerLookupKey(pet.name) === requested ||
-        (pet.id !== undefined && petPickerLookupKey(pet.id) === requested)),
+  return findCodexPet(
+    pets.filter((pet) => pet.hasSpritesheet),
+    value,
   );
 }
 
@@ -129,8 +118,8 @@ function petPickerDescription(cfg: ResolvedConfig, pets: CodexPetPackage[]): str
   if (readyPets.length === 0)
     return "No ready custom pets found. Use /pets help to create one, then return here.";
   if (!cfg.pets.slug) {
-    const firstPet = readyPets[0];
-    const lastPet = readyPets[readyPets.length - 1];
+    const firstPet = readyPets[0]!;
+    const lastPet = readyPets[readyPets.length - 1]!;
     return `No pet selected. Enter/Space/→ selects ${firstPet.name}; ← selects ${lastPet.name}.`;
   }
   const selectedPet = findPickerPet(cfg.pets.slug, readyPets);
@@ -167,82 +156,17 @@ function formatPetSelectPrompt(
   return { message: lines.join("\n"), level: "info" };
 }
 
-function spaces(width: number): string {
-  return " ".repeat(Math.max(0, width));
-}
-
-function padTextToWidth(value: string, width: number): string {
-  return value + spaces(width - visibleWidth(value));
-}
-
-function isTerminalImageLine(line: string): boolean {
-  return line.includes("\x1b_G") || line.includes("\x1b]1337;File=");
-}
-
-function petLineCell(line: string, width: number): string {
-  if (!line) return spaces(width);
-  if (isTerminalImageLine(line)) return `\x1b[0m${line}`;
-  const clipped = truncateToWidth(line, width, "");
-  return clipped + spaces(width - visibleWidth(clipped));
-}
-
-function stripLeadingCursorUp(line: string): string {
-  if (!line.startsWith("\x1b[")) return line;
-  const end = line.indexOf("A", 2);
-  if (end === -1) return line;
-  return /^\d+$/.test(line.slice(2, end)) ? line.slice(end + 1) : line;
-}
-
-function stripTrailingCursorDown(line: string): string {
-  if (!line.endsWith("B")) return line;
-  const start = line.lastIndexOf("\x1b[");
-  if (start === -1) return line;
-  return /^\d+$/.test(line.slice(start + 2, -1)) ? line.slice(0, start) : line;
-}
-
-function terminalImageInlineLeftSequence(line: string, totalRows: number): string {
-  const moveUp = totalRows > 1 ? `\x1b[${totalRows - 1}A` : "";
-  const moveDown = totalRows > 1 ? `\x1b[${totalRows - 1}B` : "";
-  const balancedLine = stripTrailingCursorDown(stripLeadingCursorUp(line));
-  return `\x1b[0m\r${moveUp}${balancedLine}${moveDown}`;
-}
-
-function combineInlinePetFooter(
-  petLines: string[],
-  textLines: string[],
-  width: number,
-  placement: PetPlacement,
-  petWidth: number,
-): string[] {
-  const gap = 2;
-  const textWidth = Math.max(1, width - petWidth - gap);
-  const totalRows = Math.max(petLines.length, textLines.length);
-  const petStart = 0;
-  const textStart = 0;
-  const hasTerminalImageLine = petLines.some(isTerminalImageLine);
-  const leftImageLine =
-    placement === "inline-left" ? petLines.find(isTerminalImageLine) : undefined;
-  const renderPetOnRight =
-    placement === "inline-right" || (placement !== "inline-left" && hasTerminalImageLine);
-  const lines: string[] = [];
-
-  for (let row = 0; row < totalRows; row++) {
-    const petLine = row >= petStart ? (petLines[row - petStart] ?? "") : "";
-    const textLine = row >= textStart ? (textLines[row - textStart] ?? "") : "";
-    const textPart = truncateToWidth(textLine, textWidth, "...");
-    const petPart = leftImageLine ? spaces(petWidth) : petLineCell(petLine, petWidth);
-    if (renderPetOnRight) {
-      lines.push(`${padTextToWidth(textPart, textWidth)}${spaces(gap)}${petPart}`);
-    } else {
-      lines.push(`${petPart}${spaces(gap)}${textPart}`);
-    }
-  }
-
-  if (leftImageLine && lines.length > 0) {
-    lines[lines.length - 1] += terminalImageInlineLeftSequence(leftImageLine, totalRows);
-  }
-
-  return lines;
+function textPanel(title: string, lines: string[], done: () => void) {
+  return {
+    render(width: number) {
+      const clipped = lines.map((line) => truncateToWidth(line, width, "..."));
+      return [title, "", ...clipped, "", "Esc/q to go back"];
+    },
+    invalidate() {},
+    handleInput(data: string) {
+      if (matchesKey(data, Key.escape) || matchesKey(data, Key.ctrl("c")) || data === "q") done();
+    },
+  };
 }
 
 export default function betterOpenAI(pi: ExtensionAPI): void {
@@ -251,6 +175,13 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   let footerTotals = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 };
   let footerInstalled = false;
   let statusInstalled = false;
+  let contextUsageCached = false;
+  let cachedContextUsage: ReturnType<ExtensionContext["getContextUsage"]>;
+  let cachedContextLeafId: string | null | undefined;
+  let cachedContextModel: ExtensionContext["model"];
+  let sessionNameCached = false;
+  let cachedSessionNameLeafId: string | null | undefined;
+  let cachedSessionName: string | undefined;
   const usageController = new UsageController(config, updateFooter);
   const petController = new PetFooterController(config, updateFooter, () => footerInstalled);
 
@@ -310,6 +241,41 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     }
   }
 
+  function invalidateContextUsage(): void {
+    contextUsageCached = false;
+    cachedContextUsage = undefined;
+    cachedContextLeafId = undefined;
+    cachedContextModel = undefined;
+  }
+
+  function contextUsage(ctx: ExtensionContext): ReturnType<ExtensionContext["getContextUsage"]> {
+    const leafId = ctx.sessionManager.getLeafId();
+    const model = ctx.model;
+    if (!contextUsageCached || leafId !== cachedContextLeafId || model !== cachedContextModel) {
+      cachedContextUsage = ctx.getContextUsage();
+      contextUsageCached = true;
+      cachedContextLeafId = leafId;
+      cachedContextModel = model;
+    }
+    return cachedContextUsage;
+  }
+
+  function sessionName(ctx: ExtensionContext): string | undefined {
+    const leafId = ctx.sessionManager.getLeafId();
+    if (!sessionNameCached || leafId !== cachedSessionNameLeafId) {
+      cachedSessionName = ctx.sessionManager.getSessionName();
+      cachedSessionNameLeafId = leafId;
+      sessionNameCached = true;
+    }
+    return cachedSessionName;
+  }
+
+  function invalidateSessionName(): void {
+    sessionNameCached = false;
+    cachedSessionNameLeafId = undefined;
+    cachedSessionName = undefined;
+  }
+
   pi.registerFlag(FLAG, {
     description: "Start with OpenAI fast mode enabled (service_tier=priority)",
     type: "boolean",
@@ -353,19 +319,6 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     },
   });
 
-  function textPanel(title: string, lines: string[], done: () => void) {
-    return {
-      render(width: number) {
-        const clipped = lines.map((line) => truncateToWidth(line, width, "..."));
-        return [title, "", ...clipped, "", "Esc/q to go back"];
-      },
-      invalidate() {},
-      handleInput(data: string) {
-        if (data.includes("\x1b") || data === "escape" || data === "q" || data === "\x03") done();
-      },
-    };
-  }
-
   function buildPetSettingsItems(cfg: ResolvedConfig): SettingsPickerItem[] {
     return settingsItemsFromDescriptors(PET_SETTING_DESCRIPTORS, cfg, {
       "pets.slug": {
@@ -608,10 +561,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     });
   }
 
-  function buildFastSettingsItems(
-    ctx: ExtensionContext,
-    cfg: ResolvedConfig,
-  ): SettingsPickerItem[] {
+  function buildFastSettingsItems(cfg: ResolvedConfig): SettingsPickerItem[] {
     return [
       {
         id: "fast.enabled",
@@ -659,11 +609,13 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
         id: "config.print",
         label: "Print config",
         currentValue: "open",
-        description: "Show the selected raw config JSON.",
+        description: "Show the selected config JSON with sensitive fields redacted.",
         submenu: (_value, done) =>
           textPanel(
             "Config",
-            JSON.stringify(readRawConfig(cfg.configPath), null, 2).split("\n"),
+            JSON.stringify(redactDiagnosticValue(readRawConfig(cfg.configPath)), null, 2).split(
+              "\n",
+            ),
             () => done(),
           ),
       },
@@ -680,7 +632,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
         submenu: (_value, done) =>
           settingsSubmenu(
             "Fast mode settings",
-            () => buildFastSettingsItems(ctx, config(ctx)),
+            () => buildFastSettingsItems(config(ctx)),
             ctx,
             () => done(fastSettingsSummary(ctx, config(ctx))),
           ),
@@ -803,6 +755,10 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   }
 
   async function showSettingsPicker(ctx: ExtensionContext): Promise<void> {
+    if (!hasTerminalUI(ctx)) {
+      ctx.ui.notify("Better OpenAI settings require interactive TUI mode.", "warning");
+      return;
+    }
     try {
       petController.settingsPets = await listCodexPets();
     } catch {
@@ -887,7 +843,12 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
         slug: selectedPet.slug,
       });
       updateFooter(ctx);
-      await petController.refresh(ctx, next, true);
+      if (hasTerminalUI(ctx)) await petController.refresh(ctx, next, true);
+      else
+        ctx.ui.notify(
+          `Enabled ${selectedPet.name} (${selectedPet.slug}); footer pets render in interactive TUI mode.`,
+          "info",
+        );
     },
     tuck: (ctx) => {
       writePetConfig(ctx, { enabled: false });
@@ -912,8 +873,14 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
 
       const next = writePetConfig(ctx, { slug: selectedPet.slug });
       updateFooter(ctx);
-      if (petController.shouldLoadForConfig(next)) await petController.refresh(ctx, next, true);
-      else {
+      if (petController.shouldLoadForConfig(next)) {
+        if (hasTerminalUI(ctx)) await petController.refresh(ctx, next, true);
+        else
+          ctx.ui.notify(
+            `Selected ${selectedPet.name} (${selectedPet.slug}); footer pets render in interactive TUI mode.`,
+            "info",
+          );
+      } else {
         ctx.ui.notify(
           `Selected ${selectedPet.name} (${selectedPet.slug}) for the footer pet. Use /pets wake to show it.`,
           "info",
@@ -966,8 +933,8 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
           const branch = footerData.getGitBranch?.();
           if (branch) pwd = `${pwd} (${branch})`;
 
-          const sessionName = ctx.sessionManager.getSessionName();
-          if (sessionName) pwd = `${pwd} • ${sessionName}`;
+          const currentSessionName = sessionName(ctx);
+          if (currentSessionName) pwd = `${pwd} • ${currentSessionName}`;
 
           const parts: string[] = [];
           if (totalInput) parts.push(`↑${formatTokens(totalInput)}`);
@@ -979,11 +946,11 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
           if (totalCost || usingSubscription)
             parts.push(`$${totalCost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`);
 
-          const contextUsage = ctx.getContextUsage();
-          const contextWindow = contextUsage?.contextWindow ?? ctx.model?.contextWindow ?? 0;
-          const contextPercentValue = contextUsage?.percent ?? 0;
+          const currentContextUsage = contextUsage(ctx);
+          const contextWindow = currentContextUsage?.contextWindow ?? ctx.model?.contextWindow ?? 0;
+          const contextPercentValue = currentContextUsage?.percent ?? 0;
           const contextPercent =
-            contextUsage?.percent !== null ? contextPercentValue.toFixed(1) : "?";
+            currentContextUsage?.percent !== null ? contextPercentValue.toFixed(1) : "?";
           const contextDisplay =
             contextPercent === "?"
               ? `?/${formatTokens(contextWindow)} (auto)`
@@ -997,12 +964,11 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
           parts.push(contextText);
 
           const cfg = config(ctx);
-          const cfgForPets = cfg;
-          const shouldRenderPet = petController.shouldRenderInFooter(cfgForPets);
-          const requestedPetPlacement = cfgForPets.pets.placement;
+          const shouldRenderPet = petController.shouldRenderInFooter(cfg);
+          const requestedPetPlacement = cfg.pets.placement;
           const requestedPetSizeCells = petSizeCellsForPlacement(
             requestedPetPlacement,
-            cfgForPets.pets.sizeCells,
+            cfg.pets.sizeCells,
           );
           const inlinePet = Boolean(
             shouldRenderPet &&
@@ -1010,11 +976,11 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
             isInlinePetPlacement(requestedPetPlacement) &&
             width >= requestedPetSizeCells + 32,
           );
-          const petRenderSizeCells = inlinePet ? requestedPetSizeCells : cfgForPets.pets.sizeCells;
+          const petRenderSizeCells = inlinePet ? requestedPetSizeCells : cfg.pets.sizeCells;
           const petColumnWidth = Math.min(petRenderSizeCells, Math.max(1, width - 1));
           const footerTextWidth = inlinePet ? Math.max(1, width - petColumnWidth - 2) : width;
 
-          const usageStatusLine = usageController.statusLine(ctx, cfg);
+          const usageStatusLine = usageController.statusLine(ctx, cfg, usingSubscription);
           const usageLine = usageStatusLine ? theme.fg("dim", usageStatusLine) : undefined;
 
           let statsLeft = parts.join(" ");
@@ -1027,7 +993,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
           const modelName = ctx.model?.id || "no-model";
           const thinkingLevel = pi.getThinkingLevel();
           const fastSuffix =
-            fastController.active && supportsFast(ctx, config(ctx).supportedModels) ? " fast" : "";
+            fastController.active && supportsFast(ctx, cfg.supportedModels) ? " fast" : "";
           let rightWithoutProvider = modelName;
           if (ctx.model?.reasoning) {
             rightWithoutProvider =
@@ -1084,7 +1050,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
             textLines.push(truncateToWidth(statusLine, footerTextWidth, theme.fg("dim", "...")));
           }
 
-          const petLines = petController.renderPetLines(ctx, cfgForPets, {
+          const petLines = petController.renderPetLines(ctx, cfg, {
             shouldRenderPet,
             freezePetFrame,
             requestedPetPlacement,
@@ -1140,6 +1106,18 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
 
   function updateFooter(ctx: ExtensionContext): void {
     const cfg = config(ctx);
+
+    if (!hasTerminalUI(ctx)) {
+      if (cfg.footer.mode === "off") {
+        setStatus(ctx, undefined);
+        return;
+      }
+      const fast = fastController.statusSegment(ctx, cfg);
+      const usage = usageController.statusLine(ctx, cfg);
+      setStatus(ctx, [fast, usage].filter(Boolean).join(" | ") || undefined);
+      return;
+    }
+
     petController.updateActivity(ctx, cfg);
     const shouldRenderPet = petController.shouldRenderInFooter(cfg);
 
@@ -1162,6 +1140,8 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   }
 
   pi.on("session_start", (_event, ctx) => {
+    invalidateContextUsage();
+    invalidateSessionName();
     const nextConfig = refresh(ctx);
     fastController.initializeForSession(ctx, nextConfig, pi.getFlag(FLAG) === true);
     if (
@@ -1172,15 +1152,16 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     if (fastController.desiredActive && !fastController.active) {
       ctx.ui.notify(fastController.unsupportedRequestMessage(ctx, nextConfig), "warning");
     }
-    petController.installResizeGuard(ctx);
+    if (hasTerminalUI(ctx)) petController.installResizeGuard(ctx);
     refreshFooterTotals(ctx);
     updateFooter(ctx);
-    if (nextConfig.pets.enabled) void petController.refresh(ctx, nextConfig);
+    if (hasTerminalUI(ctx) && nextConfig.pets.enabled) void petController.refresh(ctx, nextConfig);
     usageController.start(ctx);
     if (fastController.active) ctx.ui.notify(fastController.stateText(ctx, nextConfig), "info");
   });
 
   pi.on("agent_start", (_event, ctx) => {
+    invalidateContextUsage();
     petController.agentStart(ctx);
     updateFooter(ctx);
   });
@@ -1200,13 +1181,21 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     updateFooter(ctx);
   });
 
-  pi.on("turn_end", (_event, ctx) => {
-    refreshFooterTotals(ctx);
+  pi.on("turn_end", (event, ctx) => {
+    invalidateContextUsage();
+    if (event.message?.role === "assistant") {
+      footerTotals.input += event.message.usage.input;
+      footerTotals.output += event.message.usage.output;
+      footerTotals.cacheRead += event.message.usage.cacheRead;
+      footerTotals.cacheWrite += event.message.usage.cacheWrite;
+      footerTotals.cost += event.message.usage.cost.total;
+    } else refreshFooterTotals(ctx);
     updateFooter(ctx);
     void usageController.refresh(ctx);
   });
 
   pi.on("session_compact", (_event, ctx) => {
+    invalidateContextUsage();
     refreshFooterTotals(ctx);
     petController.queueKittyCleanup();
     petController.resetRenderCache();
@@ -1214,6 +1203,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   });
 
   pi.on("session_tree", (_event, ctx) => {
+    invalidateContextUsage();
     refreshFooterTotals(ctx);
     petController.queueKittyCleanup();
     petController.resetRenderCache();
@@ -1221,6 +1211,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   });
 
   pi.on("model_select", (event, ctx) => {
+    invalidateContextUsage();
     const cfg = config(ctx);
     const wasActive = fastController.active;
     fastController.applyDesiredState(ctx, cfg);
@@ -1238,6 +1229,8 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   });
 
   pi.on("session_shutdown", () => {
+    invalidateContextUsage();
+    invalidateSessionName();
     usageController.shutdown();
     petController.shutdown();
   });
@@ -1245,6 +1238,10 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   pi.on("before_provider_request", (event, ctx) => {
     return fastController.injectProviderPayload(event, ctx, config(ctx));
   });
+
+  pi.on("message_start", invalidateContextUsage);
+  pi.on("message_update", invalidateContextUsage);
+  pi.on("message_end", invalidateContextUsage);
 }
 
 export const _test = {
@@ -1273,6 +1270,7 @@ export const _test = {
   formatPercent,
   formatUsageSnapshot,
   readCodexAuth,
+  textPanel,
   imageTest: _imageTest,
   petsTest: _petsTest,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-better-openai",
-  "version": "0.1.19",
+  "version": "0.1.22",
   "description": "Personal pi extension that improves OpenAI with fast mode, usage stats, and footer polish.",
   "keywords": [
     "fast",
@@ -41,18 +41,20 @@
     "sharp": "^0.34.5"
   },
   "devDependencies": {
-    "@types/node": "^24.13.2",
+    "@earendil-works/pi-coding-agent": "0.80.6",
+    "@earendil-works/pi-tui": "0.80.6",
+    "@types/node": "~22.19.0",
     "oxfmt": "^0.47.0",
     "oxlint": "^1.62.0",
     "typescript": "^6.0.3",
     "vitest": "^4.1.5"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": ">=0.57.0",
-    "@mariozechner/pi-tui": "*"
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
   },
   "engines": {
-    "node": ">=20.6.0"
+    "node": ">=22.19.0"
   },
   "pi": {
     "extensions": [