pi-better-openai 0.1.6 → 0.1.8

package/README.md

@@ -7,7 +7,7 @@ A pi extension for OpenAI subscription workflows: fast mode, usage visibility, f
 Install from GitHub:
 
 ```bash
-pi install github:mattleong/pi-better-openai
+pi install git:github.com/mattleong/pi-better-openai
 ```
 
 Or install from npm:

package/{extensions/pi-better-openai.ts → index.ts}

@@ -5,8 +5,8 @@
  * enabled and the selected model is in the configured allow-list.
  */
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
-import { CONFIG_BASENAME, STATUS_KEY } from "./identity.ts";
-import { formatTokens, sanitizeStatusText, truncateToWidth, visibleWidth } from "./format.ts";
+import { CONFIG_BASENAME, STATUS_KEY } from "./src/identity.ts";
+import { formatTokens, sanitizeStatusText, truncateToWidth, visibleWidth } from "./src/format.ts";
 import {
   DEFAULT_CONFIG,
   DEFAULT_IMAGE_CONFIG,
@@ -15,7 +15,6 @@ import {
   IMAGE_OUTPUT_FORMATS,
   IMAGE_SAVE_MODES,
   configPaths,
-  type FooterMode,
   type ResolvedConfig,
   type SupportedModel,
   isRecord,
@@ -24,8 +23,8 @@ import {
   parseModels,
   readRawConfig,
   resolveConfig,
-  writeConfig
-} from "./config.ts";
+  writeConfig,
+} from "./src/config.ts";
 import {
   AUTH_FILE,
   type UsageSnapshot,
@@ -34,9 +33,9 @@ import {
   formatUsageSnapshot,
   parseUsageSnapshot,
   readCodexAuth,
-  requestCodexUsage
-} from "./usage.ts";
-import { registerOpenAIImage, _imageTest } from "./image.ts";
+  requestCodexUsage,
+} from "./src/usage.ts";
+import { registerOpenAIImage, _imageTest } from "./src/image.ts";
 
 const COMMAND = "fast";
 const OPENAI_STATUS_COMMAND = "openai-usage";
@@ -49,7 +48,10 @@ type SettingsPickerItem = {
   description?: string;
   currentValue: string;
   values?: string[];
-  submenu?: (currentValue: string, done: (selectedValue?: string) => void) => { render(width: number): string[]; invalidate(): void; handleInput?(data: string): void };
+  submenu?: (
+    currentValue: string,
+    done: (selectedValue?: string) => void,
+  ) => { render(width: number): string[]; invalidate(): void; handleInput?(data: string): void };
 };
 
 function currentModelKey(ctx: ExtensionContext): string {
@@ -57,8 +59,11 @@ function currentModelKey(ctx: ExtensionContext): string {
 }
 
 function supportsFast(ctx: ExtensionContext, supportedModels: SupportedModel[]): boolean {
-  if (!ctx.model) return false;
-  return supportedModels.some((model) => model.provider === ctx.model.provider && model.id === ctx.model.id);
+  const current = ctx.model;
+  if (!current) return false;
+  return supportedModels.some(
+    (model) => model.provider === current.provider && model.id === current.id,
+  );
 }
 
 function modelList(supportedModels: SupportedModel[]): string {
@@ -67,7 +72,12 @@ function modelList(supportedModels: SupportedModel[]): string {
     : "none configured";
 }
 
-function stateText(ctx: ExtensionContext, desiredActive: boolean, active: boolean, supportedModels: SupportedModel[]): string {
+function stateText(
+  ctx: ExtensionContext,
+  desiredActive: boolean,
+  active: boolean,
+  supportedModels: SupportedModel[],
+): string {
   const model = currentModelKey(ctx);
   if (active) return `Fast mode is on for ${model}.`;
   if (desiredActive) {
@@ -77,7 +87,8 @@ function stateText(ctx: ExtensionContext, desiredActive: boolean, active: boolea
 }
 
 function isOpenAISubscriptionModel(ctx: ExtensionContext, cfg: ResolvedConfig): boolean {
-  if (!ctx.model || (ctx.model.provider !== "openai" && ctx.model.provider !== "openai-codex")) return false;
+  if (!ctx.model || (ctx.model.provider !== "openai" && ctx.model.provider !== "openai-codex"))
+    return false;
   return !cfg.usage.showOnlyOnSubscriptionModels || ctx.modelRegistry.isUsingOAuth(ctx.model);
 }
 
@@ -113,7 +124,11 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   function persist(nextConfig: ResolvedConfig): void {
     cachedConfig = { ...nextConfig, active, desiredActive };
     if (!nextConfig.persistState) return;
-    writeConfig(nextConfig.configPath, { ...readRawConfig(nextConfig.configPath), active, desiredActive });
+    writeConfig(nextConfig.configPath, {
+      ...readRawConfig(nextConfig.configPath),
+      active,
+      desiredActive,
+    });
   }
 
   function applyDesiredFastState(ctx: ExtensionContext, cfg = config(ctx)): void {
@@ -127,13 +142,20 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     persist(nextConfig);
     updateFooter(ctx);
     if (next && !active) {
-      ctx.ui.notify(`Fast mode requested, but ${currentModelKey(ctx)} is unsupported. It will activate automatically when you switch to a supported model: ${modelList(nextConfig.supportedModels)}.`, "warning");
+      ctx.ui.notify(
+        `Fast mode requested, but ${currentModelKey(ctx)} is unsupported. It will activate automatically when you switch to a supported model: ${modelList(nextConfig.supportedModels)}.`,
+        "warning",
+      );
       return;
     }
     ctx.ui.notify(stateText(ctx, desiredActive, active, nextConfig.supportedModels), "info");
   }
 
-  async function refreshUsage(ctx: ExtensionContext, modelId = ctx.model?.id, options?: { notify?: boolean }): Promise<void> {
+  async function refreshUsage(
+    ctx: ExtensionContext,
+    modelId = ctx.model?.id,
+    options?: { notify?: boolean },
+  ): Promise<void> {
     if (shuttingDown || !ctx.hasUI) return;
     if (usageRefreshInFlight) {
       queuedUsageRefresh = { ctx, modelId, notify: queuedUsageRefresh?.notify || options?.notify };
@@ -163,7 +185,8 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
       usageUpdatedAt = usageSnapshot ? Date.now() : undefined;
       usageError = data ? undefined : `Missing openai-codex OAuth credentials in ${AUTH_FILE}.`;
       if (!shuttingDown) updateFooter(ctx);
-      if (!shuttingDown && options?.notify) ctx.ui.notify(formatUsageStatus(ctx), usageSnapshot ? "info" : "warning");
+      if (!shuttingDown && options?.notify)
+        ctx.ui.notify(formatUsageStatus(ctx), usageSnapshot ? "info" : "warning");
     } catch (error) {
       if (shuttingDown) return;
       usageError = error instanceof Error ? error.message : String(error);
@@ -215,25 +238,27 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
       `Last successful update: ${usageUpdatedAt ? new Date(usageUpdatedAt).toLocaleTimeString() : "never"}`,
       `Last error: ${usageError ?? "none"}`,
       `Refresh interval: ${cfg.usage.refreshIntervalMs}ms`,
-      `Endpoint: https://chatgpt.com/backend-api/wham/usage`
+      `Endpoint: https://chatgpt.com/backend-api/wham/usage`,
     ].join("\n");
   }
 
   function formatUsageStatus(ctx: ExtensionContext): string {
     const cfg = config(ctx);
     if (!cfg.usage.enabled) return "Usage display is disabled.";
-    if (!isOpenAISubscriptionModel(ctx, cfg)) return "Usage hidden: current model is not an OpenAI subscription model.";
+    if (!isOpenAISubscriptionModel(ctx, cfg))
+      return "Usage hidden: current model is not an OpenAI subscription model.";
     if (!usageSnapshot) return `Usage unavailable${usageError ? `: ${usageError}` : "."}`;
-    const stale = usageUpdatedAt && Date.now() - usageUpdatedAt > cfg.usage.refreshIntervalMs * 2
-      ? ` | stale ${formatResetCountdown((Date.now() - usageUpdatedAt) / 1000)}`
-      : "";
+    const stale =
+      usageUpdatedAt && Date.now() - usageUpdatedAt > cfg.usage.refreshIntervalMs * 2
+        ? ` | stale ${formatResetCountdown((Date.now() - usageUpdatedAt) / 1000)}`
+        : "";
     return `${formatUsageSnapshot(usageSnapshot, cfg.usage)}${stale}`;
   }
 
   pi.registerFlag(FLAG, {
     description: "Start with OpenAI fast mode enabled (service_tier=priority)",
     type: "boolean",
-    default: false
+    default: false,
   });
 
   function formatDebugStatus(ctx: ExtensionContext): string {
@@ -246,10 +271,12 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
       `Configured service_tier: ${SERVICE_TIER}`,
       `Last injected: ${lastInjectedAt ? `${new Date(lastInjectedAt).toLocaleTimeString()} (${lastInjectedModel}, ${lastInjectedTier})` : "never"}`,
       `Footer mode: ${cfg.footer.mode}`,
-      `Usage enabled: ${cfg.usage.enabled}`,
+      "",
+      formatUsageDebug(ctx),
+      "",
       `Image enabled: ${cfg.image.enabled}`,
       `Image default save: ${cfg.image.defaultSave}`,
-      `Config: ${cfg.configPath}`
+      `Config: ${cfg.configPath}`,
     ].join("\n");
   }
 
@@ -264,14 +291,14 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
       const arg = args.trim().toLowerCase();
       if (!arg) return setActive(ctx, !desiredActive);
       ctx.ui.notify("Usage: /fast", "error");
-    }
+    },
   });
 
   pi.registerCommand(OPENAI_STATUS_COMMAND, {
     description: "Show OpenAI subscription usage status",
     handler: async (_args, ctx) => {
       ctx.ui.notify(formatOpenAIStatus(ctx), "info");
-    }
+    },
   });
 
   function textPanel(title: string, lines: string[], done: () => void) {
@@ -283,27 +310,124 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
       invalidate() {},
       handleInput(data: string) {
         if (data.includes("\x1b") || data === "escape" || data === "q" || data === "\x03") done();
-      }
+      },
     };
   }
 
   function buildSettingsItems(ctx: ExtensionContext, cfg: ResolvedConfig): SettingsPickerItem[] {
     return [
-      { id: "fast.enabled", label: "Fast mode", currentValue: String(desiredActive), values: ["true", "false"], description: `Request OpenAI fast mode. Activates for supported models: ${modelList(cfg.supportedModels)}.` },
-      { id: "persistState", label: "Persist fast state", currentValue: String(cfg.persistState), values: ["true", "false"], description: "Remember fast-mode state across sessions." },
-      { id: "footer.mode", label: "Footer mode", currentValue: cfg.footer.mode, values: [...FOOTER_MODES], description: "replace = custom footer, status = pi footer plus status line, off = no Better OpenAI footer/status." },
-      { id: "usage.enabled", label: "Usage display", currentValue: String(cfg.usage.enabled), values: ["true", "false"], description: "Fetch and display OpenAI subscription usage windows." },
-      { id: "usage.refreshIntervalMs", label: "Usage refresh", currentValue: String(cfg.usage.refreshIntervalMs), values: ["15000", "30000", "60000", "120000", "300000", "600000"], description: "Usage refresh interval in milliseconds." },
-      { id: "usage.showOnlyOnSubscriptionModels", label: "Usage only on OAuth", currentValue: String(cfg.usage.showOnlyOnSubscriptionModels), values: ["true", "false"], description: "Only show usage when the current OpenAI model uses subscription/OAuth auth." },
-      { id: "usage.showResetTimes", label: "Usage reset times", currentValue: String(cfg.usage.showResetTimes), values: ["true", "false"], description: "Include compact reset countdowns and local reset times." },
-      { id: "image.enabled", label: "Image tool", currentValue: String(cfg.image.enabled), values: ["true", "false"], description: "Allow the openai_image tool to make image requests." },
-      { id: "image.defaultModel", label: "Image model", currentValue: cfg.image.defaultModel, values: ["gpt-5.5", "gpt-5.4", "gpt-5.2", "gpt-5"], description: "Mainline model used for image generation when current model is not openai-codex." },
-      { id: "image.defaultSave", label: "Image save", currentValue: cfg.image.defaultSave, values: [...IMAGE_SAVE_MODES], description: "Where generated images are saved by default." },
-      { id: "image.outputFormat", label: "Image format", currentValue: cfg.image.outputFormat, values: [...IMAGE_OUTPUT_FORMATS], description: "Generated image file format." },
-      { id: "image.timeoutMs", label: "Image timeout", currentValue: String(cfg.image.timeoutMs), values: ["30000", "60000", "120000", "180000", "300000"], description: "Image request timeout in milliseconds." },
-      { id: "debug", label: "Debug info", currentValue: "open", description: "Show Better OpenAI diagnostics.", submenu: (_value, done) => textPanel("Debug info", formatDebugStatus(ctx).split("\n"), () => done()) },
-      { id: "config.path", label: "Config path", currentValue: cfg.configPath, description: `Project: ${cfg.projectConfigPath}\nGlobal: ${cfg.globalConfigPath}` },
-      { id: "config.print", label: "Print config", currentValue: "open", description: "Show the selected raw config JSON.", submenu: (_value, done) => textPanel("Config", JSON.stringify(readRawConfig(cfg.configPath), null, 2).split("\n"), () => done()) }
+      {
+        id: "fast.enabled",
+        label: "Fast mode",
+        currentValue: String(desiredActive),
+        values: ["true", "false"],
+        description: `Request OpenAI fast mode. Activates for supported models: ${modelList(cfg.supportedModels)}.`,
+      },
+      {
+        id: "persistState",
+        label: "Persist fast state",
+        currentValue: String(cfg.persistState),
+        values: ["true", "false"],
+        description: "Remember fast-mode state across sessions.",
+      },
+      {
+        id: "footer.mode",
+        label: "Footer mode",
+        currentValue: cfg.footer.mode,
+        values: [...FOOTER_MODES],
+        description:
+          "replace = custom footer, status = pi footer plus status line, off = no Better OpenAI footer/status.",
+      },
+      {
+        id: "usage.enabled",
+        label: "Usage display",
+        currentValue: String(cfg.usage.enabled),
+        values: ["true", "false"],
+        description: "Fetch and display OpenAI subscription usage windows.",
+      },
+      {
+        id: "usage.refreshIntervalMs",
+        label: "Usage refresh",
+        currentValue: String(cfg.usage.refreshIntervalMs),
+        values: ["15000", "30000", "60000", "120000", "300000", "600000"],
+        description: "Usage refresh interval in milliseconds.",
+      },
+      {
+        id: "usage.showOnlyOnSubscriptionModels",
+        label: "Usage only on OAuth",
+        currentValue: String(cfg.usage.showOnlyOnSubscriptionModels),
+        values: ["true", "false"],
+        description: "Only show usage when the current OpenAI model uses subscription/OAuth auth.",
+      },
+      {
+        id: "usage.showResetTimes",
+        label: "Usage reset times",
+        currentValue: String(cfg.usage.showResetTimes),
+        values: ["true", "false"],
+        description: "Include compact reset countdowns and local reset times.",
+      },
+      {
+        id: "image.enabled",
+        label: "Image tool",
+        currentValue: String(cfg.image.enabled),
+        values: ["true", "false"],
+        description: "Allow the openai_image tool to make image requests.",
+      },
+      {
+        id: "image.defaultModel",
+        label: "Image model",
+        currentValue: cfg.image.defaultModel,
+        values: ["gpt-5.5", "gpt-5.4", "gpt-5.2", "gpt-5"],
+        description:
+          "Mainline model used for image generation when current model is not openai-codex.",
+      },
+      {
+        id: "image.defaultSave",
+        label: "Image save",
+        currentValue: cfg.image.defaultSave,
+        values: [...IMAGE_SAVE_MODES],
+        description: "Where generated images are saved by default.",
+      },
+      {
+        id: "image.outputFormat",
+        label: "Image format",
+        currentValue: cfg.image.outputFormat,
+        values: [...IMAGE_OUTPUT_FORMATS],
+        description: "Generated image file format.",
+      },
+      {
+        id: "image.timeoutMs",
+        label: "Image timeout",
+        currentValue: String(cfg.image.timeoutMs),
+        values: ["30000", "60000", "120000", "180000", "300000"],
+        description: "Image request timeout in milliseconds.",
+      },
+      {
+        id: "debug",
+        label: "Debug info",
+        currentValue: "open",
+        description: "Show Better OpenAI diagnostics.",
+        submenu: (_value, done) =>
+          textPanel("Debug info", formatDebugStatus(ctx).split("\n"), () => done()),
+      },
+      {
+        id: "config.path",
+        label: "Config path",
+        currentValue: cfg.configPath,
+        description: `Project: ${cfg.projectConfigPath}\nGlobal: ${cfg.globalConfigPath}`,
+      },
+      {
+        id: "config.print",
+        label: "Print config",
+        currentValue: "open",
+        description: "Show the selected raw config JSON.",
+        submenu: (_value, done) =>
+          textPanel(
+            "Config",
+            JSON.stringify(readRawConfig(cfg.configPath), null, 2).split("\n"),
+            () => done(),
+          ),
+      },
     ];
   }
 
@@ -332,7 +456,14 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     } else if (id.startsWith("image.")) {
       const image = isRecord(current.image) ? current.image : {};
       const key = id.slice("image.".length);
-      image[key] = key === "timeoutMs" ? num : rawValue === "true" ? true : rawValue === "false" ? false : rawValue;
+      image[key] =
+        key === "timeoutMs"
+          ? num
+          : rawValue === "true"
+            ? true
+            : rawValue === "false"
+              ? false
+              : rawValue;
       current.image = image;
     }
     writeConfig(cfg.configPath, current);
@@ -352,37 +483,51 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
   async function showSettingsPicker(ctx: ExtensionContext): Promise<void> {
     const [{ getSettingsListTheme }, { Container, SettingsList }] = await Promise.all([
       import("@mariozechner/pi-coding-agent"),
-      import("@mariozechner/pi-tui")
+      import("@mariozechner/pi-tui"),
     ]);
     await ctx.ui.custom((tui, theme, _kb, done) => {
       const container = new Container();
-      container.addChild(new (class {
-        render(_width: number) {
-          const cfg = config(ctx);
-          return [theme.fg("accent", theme.bold("Better OpenAI Settings")), theme.fg("dim", cfg.configPath), ""];
-        }
-        invalidate() {}
-      })());
+      container.addChild(
+        new (class {
+          render(_width: number) {
+            const cfg = config(ctx);
+            return [
+              theme.fg("accent", theme.bold("Better OpenAI Settings")),
+              theme.fg("dim", cfg.configPath),
+              "",
+            ];
+          }
+          invalidate() {}
+        })(),
+      );
       const settingsList = new SettingsList(
         buildSettingsItems(ctx, refresh(ctx)),
         13,
         getSettingsListTheme(),
         (id, newValue) => {
           writeSetting(ctx, id, newValue);
-          settingsList.updateValue(id, buildSettingsItems(ctx, config(ctx)).find((item) => item.id === id)?.currentValue ?? newValue);
+          settingsList.updateValue(
+            id,
+            buildSettingsItems(ctx, config(ctx)).find((item) => item.id === id)?.currentValue ??
+              newValue,
+          );
           tui.requestRender();
         },
         () => done(undefined),
-        { enableSearch: true }
+        { enableSearch: true },
       );
       container.addChild(settingsList);
       return {
-        render(width: number) { return container.render(width); },
-        invalidate() { container.invalidate(); },
+        render(width: number) {
+          return container.render(width);
+        },
+        invalidate() {
+          container.invalidate();
+        },
         handleInput(data: string) {
           settingsList.handleInput(data);
           tui.requestRender();
-        }
+        },
       };
     });
   }
@@ -391,10 +536,9 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     description: "Open Better OpenAI settings picker",
     handler: async (_args, ctx) => {
       await showSettingsPicker(ctx);
-    }
+    },
   });
 
-
   registerOpenAIImage(pi, config);
 
   function installFooter(ctx: ExtensionContext): void {
@@ -437,18 +581,24 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
           if (totalCacheWrite) parts.push(`W${formatTokens(totalCacheWrite)}`);
 
           const usingSubscription = ctx.model ? ctx.modelRegistry.isUsingOAuth(ctx.model) : false;
-          if (totalCost || usingSubscription) parts.push(`$${totalCost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`);
+          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 contextPercent = contextUsage?.percent !== null ? contextPercentValue.toFixed(1) : "?";
-          const contextDisplay = contextPercent === "?" ? `?/${formatTokens(contextWindow)} (auto)` : `${contextPercent}%/${formatTokens(contextWindow)} (auto)`;
-          const contextText = contextPercentValue > 90
-            ? theme.fg("error", contextDisplay)
-            : contextPercentValue > 70
-              ? theme.fg("warning", contextDisplay)
-              : contextDisplay;
+          const contextPercent =
+            contextUsage?.percent !== null ? contextPercentValue.toFixed(1) : "?";
+          const contextDisplay =
+            contextPercent === "?"
+              ? `?/${formatTokens(contextWindow)} (auto)`
+              : `${contextPercent}%/${formatTokens(contextWindow)} (auto)`;
+          const contextText =
+            contextPercentValue > 90
+              ? theme.fg("error", contextDisplay)
+              : contextPercentValue > 70
+                ? theme.fg("warning", contextDisplay)
+                : contextDisplay;
           parts.push(contextText);
 
           let usageLine: string | undefined;
@@ -466,12 +616,14 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
 
           const modelName = ctx.model?.id || "no-model";
           const thinkingLevel = pi.getThinkingLevel();
-          const fastSuffix = active && supportsFast(ctx, config(ctx).supportedModels) ? " fast" : "";
+          const fastSuffix =
+            active && supportsFast(ctx, config(ctx).supportedModels) ? " fast" : "";
           let rightWithoutProvider = modelName;
           if (ctx.model?.reasoning) {
-            rightWithoutProvider = thinkingLevel === "off"
-              ? `${modelName}${fastSuffix} • thinking off`
-              : `${modelName}${fastSuffix} • ${thinkingLevel}`;
+            rightWithoutProvider =
+              thinkingLevel === "off"
+                ? `${modelName}${fastSuffix} • thinking off`
+                : `${modelName}${fastSuffix} • ${thinkingLevel}`;
           } else if (fastSuffix) {
             rightWithoutProvider = `${modelName}${fastSuffix}`;
           }
@@ -491,7 +643,10 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
             const availableForRight = width - statsLeftWidth - 2;
             if (availableForRight > 0) {
               const truncatedRight = truncateToWidth(rightSide, availableForRight, "");
-              statsLine = statsLeft + " ".repeat(Math.max(0, width - statsLeftWidth - visibleWidth(truncatedRight))) + truncatedRight;
+              statsLine =
+                statsLeft +
+                " ".repeat(Math.max(0, width - statsLeftWidth - visibleWidth(truncatedRight))) +
+                truncatedRight;
             } else {
               statsLine = statsLeft;
             }
@@ -499,7 +654,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
 
           const lines = [
             truncateToWidth(theme.fg("dim", pwd), width, theme.fg("dim", "...")),
-            theme.fg("dim", statsLeft) + theme.fg("dim", statsLine.slice(statsLeft.length))
+            theme.fg("dim", statsLeft) + theme.fg("dim", statsLine.slice(statsLeft.length)),
           ];
 
           if (usageLine) {
@@ -516,7 +671,7 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
           }
 
           return lines;
-        }
+        },
       };
     });
   }
@@ -535,8 +690,14 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
       ctx.ui.setStatus(STATUS_KEY, undefined);
       return;
     }
-    const fast = active && supportsFast(ctx, cfg.supportedModels) ? `${ctx.model?.id ?? "model"} fast` : undefined;
-    const usage = usageSnapshot && cfg.usage.enabled && isOpenAISubscriptionModel(ctx, cfg) ? formatUsageSnapshot(usageSnapshot, cfg.usage) : undefined;
+    const fast =
+      active && supportsFast(ctx, cfg.supportedModels)
+        ? `${ctx.model?.id ?? "model"} fast`
+        : undefined;
+    const usage =
+      usageSnapshot && cfg.usage.enabled && isOpenAISubscriptionModel(ctx, cfg)
+        ? formatUsageSnapshot(usageSnapshot, cfg.usage)
+        : undefined;
     ctx.ui.setStatus(STATUS_KEY, [fast, usage].filter(Boolean).join(" | ") || undefined);
   }
 
@@ -545,14 +706,19 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     desiredActive = nextConfig.persistState ? nextConfig.desiredActive : false;
     if (pi.getFlag(FLAG) === true) desiredActive = true;
     applyDesiredFastState(ctx, nextConfig);
-    if (desiredActive !== nextConfig.desiredActive || active !== nextConfig.active) persist(nextConfig);
+    if (desiredActive !== nextConfig.desiredActive || active !== nextConfig.active)
+      persist(nextConfig);
     if (desiredActive && !active) {
-      ctx.ui.notify(`Fast mode requested, but ${currentModelKey(ctx)} is unsupported. It will activate automatically when you switch to a supported model: ${modelList(nextConfig.supportedModels)}.`, "warning");
+      ctx.ui.notify(
+        `Fast mode requested, but ${currentModelKey(ctx)} is unsupported. It will activate automatically when you switch to a supported model: ${modelList(nextConfig.supportedModels)}.`,
+        "warning",
+      );
     }
     refreshFooterTotals(ctx);
     updateFooter(ctx);
     startUsageRefresh(ctx);
-    if (active) ctx.ui.notify(stateText(ctx, desiredActive, active, nextConfig.supportedModels), "info");
+    if (active)
+      ctx.ui.notify(stateText(ctx, desiredActive, active, nextConfig.supportedModels), "info");
   });
 
   pi.on("turn_end", (_event, ctx) => {
@@ -577,7 +743,12 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
     applyDesiredFastState(ctx, cfg);
     if (active !== wasActive) {
       persist(cfg);
-      ctx.ui.notify(active ? stateText(ctx, desiredActive, active, cfg.supportedModels) : `Fast mode inactive for unsupported model ${currentModelKey(ctx)}.`, active ? "info" : "warning");
+      ctx.ui.notify(
+        active
+          ? stateText(ctx, desiredActive, active, cfg.supportedModels)
+          : `Fast mode inactive for unsupported model ${currentModelKey(ctx)}.`,
+        active ? "info" : "warning",
+      );
     }
     updateFooter(ctx);
     void refreshUsage(ctx, event.model.id);
@@ -594,7 +765,8 @@ export default function betterOpenAI(pi: ExtensionAPI): void {
 
   pi.on("before_provider_request", (event, ctx) => {
     const nextConfig = config(ctx);
-    if (!active || !supportsFast(ctx, nextConfig.supportedModels) || !isRecord(event.payload)) return;
+    if (!active || !supportsFast(ctx, nextConfig.supportedModels) || !isRecord(event.payload))
+      return;
     lastInjectedAt = Date.now();
     lastInjectedModel = currentModelKey(ctx);
     lastInjectedTier = SERVICE_TIER;
@@ -619,5 +791,5 @@ export const _test = {
   formatPercent,
   formatUsageSnapshot,
   readCodexAuth,
-  imageTest: _imageTest
+  imageTest: _imageTest,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-better-openai",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "Personal pi extension that improves OpenAI with fast mode, usage stats, and footer polish.",
   "keywords": [
     "fast",
@@ -19,7 +19,8 @@
     "url": "git+https://github.com/mattleong/pi-better-openai.git"
   },
   "files": [
-    "extensions/",
+    "index.ts",
+    "src/",
     "README.md",
     "LICENSE"
   ],
@@ -28,7 +29,19 @@
     "access": "public"
   },
   "scripts": {
-    "test": "node tests/basic.mjs"
+    "check": "npm run typecheck && npm run lint && npm run format:check && npm test",
+    "format": "oxfmt .",
+    "format:check": "oxfmt --check .",
+    "lint": "oxlint --deny-warnings .",
+    "lint:fix": "oxlint --fix .",
+    "test": "node tests/basic.mjs",
+    "typecheck": "tsc --noEmit"
+  },
+  "devDependencies": {
+    "@types/node": "^25.6.0",
+    "oxfmt": "^0.47.0",
+    "oxlint": "^1.62.0",
+    "typescript": "^6.0.3"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.57.0",
@@ -36,7 +49,7 @@
   },
   "pi": {
     "extensions": [
-      "./extensions/pi-better-openai.ts"
+      "./index.ts"
     ]
   }
 }