pi-better-openai 0.1.4 → 0.1.7

package/README.md

@@ -2,40 +2,18 @@
 
 A pi extension for OpenAI subscription workflows: fast mode, usage visibility, footer polish, and image generation through `openai-codex` auth.
 
-## Screenshots
-
-<!-- Add screenshots here. -->
-
 ## Install
 
-Install from a local checkout:
+Install from GitHub:
 
 ```bash
-pi install /path/to/pi-better-openai
+pi install github:mattleong/pi-better-openai
 ```
 
-For a project-local install instead of a global install, add `-l`:
+Or install from npm:
 
 ```bash
-pi install -l /path/to/pi-better-openai
-```
-
-Reload pi after installing:
-
-```text
-/reload
-```
-
-Sign in to OpenAI Codex subscription auth if you want usage stats or image generation:
-
-```text
-/login openai-codex
-```
-
-Start pi with fast mode enabled:
-
-```bash
-pi --fast
+pi install npm:pi-better-openai
 ```
 
 ## Features
@@ -50,3 +28,10 @@ pi --fast
   - `/openai-image <prompt>` generates an image directly.
   - `/openai-usage` shows current OpenAI subscription usage.
   - `/openai-settings` opens settings, diagnostics, and config details.
+
+## Screenshots
+
+<!-- Add screenshots here. -->
+
+<img width="983" height="851" alt="Screenshot 2026-04-29 at 11 53 23 PM" src="https://github.com/user-attachments/assets/07a2fb87-ef48-4396-8b12-124825c8d360" />
+<img width="1327" height="102" alt="Screenshot 2026-04-29 at 11 34 49 PM" src="https://github.com/user-attachments/assets/22042782-c94e-491d-b5af-095f7f0810f9" />

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.4",
+  "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

@@ -1,7 +1,7 @@
 import { randomUUID } from "node:crypto";
 import { mkdir, readFile, writeFile } from "node:fs/promises";
 import { homedir } from "node:os";
-import { extname, isAbsolute, join, resolve } from "node:path";
+import { extname, isAbsolute, join, resolve, sep } from "node:path";
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 import type { ResolvedConfig } from "./config.ts";
 import { isRecord } from "./config.ts";
@@ -23,7 +23,7 @@ export type ImageOutputFormat = typeof IMAGE_OUTPUT_FORMATS[number];
 const TOOL_PARAMS = {
   type: "object",
   properties: {
-    prompt: { type: "string", description: "Image generation/editing prompt." },
+    prompt: { type: "string", description: "Image generation/editing prompt. Pass the user's wording verbatim unless they explicitly ask you to refine or expand it." },
     action: { type: "string", enum: IMAGE_ACTIONS, description: "Whether to generate a new image, edit/reference provided images, or let the model decide." },
     images: {
       type: "array",
@@ -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.");
@@ -338,6 +344,39 @@ async function requestCodexImage(params: ToolParams, ctx: ExtensionContext, cfg:
   return { ...parsed, prompt: params.prompt, savedPath, model, action, outputFormat };
 }
 
+function displayPath(path: string): string {
+  const home = homedir();
+  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 textFromMessageContent(content: unknown): string | undefined {
+  if (typeof content === "string") return content.trim() || undefined;
+  if (!Array.isArray(content)) return undefined;
+  const text = content
+    .filter((part) => isRecord(part) && part.type === "text" && typeof part.text === "string")
+    .map((part) => (part as { text: string }).text)
+    .join("\n")
+    .trim();
+  return text || undefined;
+}
+
+function latestUserPromptFromEntries(entries: unknown[]): string | undefined {
+  for (let i = entries.length - 1; i >= 0; i--) {
+    const entry = entries[i];
+    if (!isRecord(entry) || entry.type !== "message" || !isRecord(entry.message) || entry.message.role !== "user") continue;
+    const text = textFromMessageContent(entry.message.content);
+    if (text) return text;
+  }
+  return undefined;
+}
+
+function resolveToolPrompt(params: ToolParams, ctx: ExtensionContext): string {
+  return latestUserPromptFromEntries(ctx.sessionManager.getEntries()) ?? params.prompt;
+}
+
 function resultText(result: CodexImageResult): string {
   const parts = [
     `Generated image via openai-codex/${result.model}.`,
@@ -345,7 +384,7 @@ function resultText(result: CodexImageResult): string {
     `Prompt: ${result.prompt}`
   ];
   if (result.revisedPrompt) parts.push(`Revised prompt: ${result.revisedPrompt}`);
-  if (result.savedPath) parts.push(`Saved: ${result.savedPath}`);
+  if (result.savedPath) parts.push(`Saved: ${displayPath(result.savedPath)}`);
   return parts.join("\n");
 }
 
@@ -388,6 +427,37 @@ export function registerOpenAIImage(pi: ExtensionAPI, getConfig: (ctx: Extension
     };
   }
 
+  void import("@mariozechner/pi-tui").then(({ Box, Container, Image, Text }) => {
+    pi.registerMessageRenderer<CodexImageResult>("openai-image", (message, _options, theme) => {
+      const result = message.details;
+      const text = result && isRecord(result)
+        ? resultText(result as CodexImageResult)
+        : typeof message.content === "string"
+          ? message.content
+          : message.content.filter((part) => part.type === "text").map((part) => part.text).join("\n");
+      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));
+      box.addChild(new Text(`${theme.fg("accent", theme.bold("[openai-image]"))}\n\n${text}`, 0, 0));
+      if (image) {
+        box.addChild(new Image(image.data, image.mimeType, { fallbackColor: (line) => theme.fg("dim", line) }, {
+          maxWidthCells: 80,
+          maxHeightCells: 24,
+          filename: "savedPath" in image && typeof image.savedPath === "string" ? image.savedPath : undefined
+        }));
+      }
+      container.addChild(box);
+      return container;
+    });
+  }).catch(() => undefined);
+
   pi.registerCommand(OPENAI_IMAGE_COMMAND, {
     description: "Generate an image with OpenAI Codex image generation",
     handler: async (args, ctx) => {
@@ -417,14 +487,16 @@ export function registerOpenAIImage(pi: ExtensionAPI, getConfig: (ctx: Extension
     promptSnippet: "Generate or edit raster images via OpenAI Codex subscription auth.",
     promptGuidelines: [
       "Use openai_image when the user asks to generate or edit a raster image, photo, illustration, mockup, texture, sprite, or bitmap asset.",
+      "Pass the user's image prompt verbatim. Do not embellish, rewrite, add camera/style details, or add negative prompt terms unless the user explicitly asks you to refine the prompt.",
       "Use openai_image with images for local reference images or edit targets; save project assets into the workspace when requested."
     ],
     parameters: TOOL_PARAMS,
     async execute(_toolCallId, params: ToolParams, signal, onUpdate, ctx) {
       const cfg = getConfig(ctx);
       const model = resolveModel(params, ctx, cfg);
-      onUpdate?.({ content: [{ type: "text", text: `Requesting OpenAI image via openai-codex/${model}...` }] });
-      const result = await generate(params, ctx, signal);
+      const requestParams = { ...params, prompt: resolveToolPrompt(params, ctx) };
+      onUpdate?.({ content: [{ type: "text", text: `Requesting OpenAI image via openai-codex/${model}...` }], details: undefined });
+      const result = await generate(requestParams, ctx, signal);
       return {
         content: [
           { type: "text", text: resultText(result) },
@@ -447,5 +519,7 @@ export const _imageTest = {
   imageMimeType,
   dataUrlParts,
   extractImageFromEvent,
+  displayPath,
+  latestUserPromptFromEntries,
   buildRequest
 };