pi-better-openai 0.1.6 → 0.1.7

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,
@@ -25,7 +25,7 @@ import {
   readRawConfig,
   resolveConfig,
   writeConfig
-} from "./config.ts";
+} from "./src/config.ts";
 import {
   AUTH_FILE,
   type UsageSnapshot,
@@ -35,8 +35,8 @@ import {
   parseUsageSnapshot,
   readCodexAuth,
   requestCodexUsage
-} from "./usage.ts";
-import { registerOpenAIImage, _imageTest } from "./image.ts";
+} from "./src/usage.ts";
+import { registerOpenAIImage, _imageTest } from "./src/image.ts";
 
 const COMMAND = "fast";
 const OPENAI_STATUS_COMMAND = "openai-usage";
@@ -57,8 +57,9 @@ 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 {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-better-openai",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "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,8 @@
     "access": "public"
   },
   "scripts": {
-    "test": "node tests/basic.mjs"
+    "test": "node tests/basic.mjs",
+    "typecheck": "tsc --noEmit"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": ">=0.57.0",
@@ -36,7 +38,11 @@
   },
   "pi": {
     "extensions": [
-      "./extensions/pi-better-openai.ts"
+      "./index.ts"
     ]
+  },
+  "devDependencies": {
+    "@types/node": "^25.6.0",
+    "typescript": "^6.0.3"
   }
 }

package/{extensions → src}/image.ts

@@ -74,6 +74,8 @@ export type CodexImageResult = {
   outputFormat: ImageOutputFormat;
 };
 
+type ExtractedImageResult = Omit<CodexImageResult, "prompt" | "savedPath" | "model" | "action" | "outputFormat">;
+
 export type ImageGenerationDebug = {
   authFound: boolean;
   authSource?: string;
@@ -224,7 +226,11 @@ function asImageResultItem(value: unknown): { id?: string; status?: string; revi
   return value as { id?: string; status?: string; revised_prompt?: string; result?: string; b64_json?: string };
 }
 
-function extractImageFromEvent(event: unknown, fallbackMimeType: string): Omit<CodexImageResult, "savedPath" | "model" | "action" | "outputFormat"> | undefined {
+function isImageContent(value: unknown): value is { type: "image"; data: string; mimeType: string } {
+  return isRecord(value) && value.type === "image" && typeof value.data === "string" && typeof value.mimeType === "string";
+}
+
+function extractImageFromEvent(event: unknown, fallbackMimeType: string): ExtractedImageResult | undefined {
   if (!isRecord(event)) return undefined;
   const item = asImageResultItem(event.item) ?? asImageResultItem(event);
   if (item) {
@@ -247,12 +253,12 @@ function extractImageFromEvent(event: unknown, fallbackMimeType: string): Omit<C
   return undefined;
 }
 
-async function parseSseForImage(response: Response, fallbackMimeType: string, signal?: AbortSignal): Promise<Omit<CodexImageResult, "savedPath" | "model" | "action" | "outputFormat">> {
+async function parseSseForImage(response: Response, fallbackMimeType: string, signal?: AbortSignal): Promise<ExtractedImageResult> {
   if (!response.body) throw new Error("No response body from Codex image request.");
   const reader = response.body.getReader();
   const decoder = new TextDecoder();
   let buffer = "";
-  let lastImage: Omit<CodexImageResult, "savedPath" | "model" | "action" | "outputFormat"> | undefined;
+  let lastImage: ExtractedImageResult | undefined;
   try {
     while (true) {
       if (signal?.aborted) throw new Error("Image request was aborted.");
@@ -429,11 +435,13 @@ export function registerOpenAIImage(pi: ExtensionAPI, getConfig: (ctx: Extension
         : typeof message.content === "string"
           ? message.content
           : message.content.filter((part) => part.type === "text").map((part) => part.text).join("\n");
-      const image = result && isRecord(result) && typeof result.data === "string" && typeof result.mimeType === "string"
-        ? { data: result.data, mimeType: result.mimeType, savedPath: typeof result.savedPath === "string" ? result.savedPath : undefined }
-        : Array.isArray(message.content)
-          ? message.content.find((part) => part.type === "image" && typeof part.data === "string" && typeof part.mimeType === "string")
-          : undefined;
+      let image: { data: string; mimeType: string; savedPath?: string } | undefined;
+      if (result && isRecord(result) && typeof result.data === "string" && typeof result.mimeType === "string") {
+        image = { data: result.data, mimeType: result.mimeType, savedPath: typeof result.savedPath === "string" ? result.savedPath : undefined };
+      } else if (Array.isArray(message.content)) {
+        const imagePart = message.content.find(isImageContent);
+        if (imagePart) image = { data: imagePart.data, mimeType: imagePart.mimeType };
+      }
 
       const container = new Container();
       const box = new Box(1, 1, (line) => theme.bg("customMessageBg", line));
@@ -487,7 +495,7 @@ export function registerOpenAIImage(pi: ExtensionAPI, getConfig: (ctx: Extension
       const cfg = getConfig(ctx);
       const model = resolveModel(params, ctx, cfg);
       const requestParams = { ...params, prompt: resolveToolPrompt(params, ctx) };
-      onUpdate?.({ content: [{ type: "text", text: `Requesting OpenAI image via openai-codex/${model}...` }] });
+      onUpdate?.({ content: [{ type: "text", text: `Requesting OpenAI image via openai-codex/${model}...` }], details: undefined });
       const result = await generate(requestParams, ctx, signal);
       return {
         content: [