@oai404iao/pi-codex-imagegen 2.0.0 → 2.1.1

package/src/background-image-generation.ts

@@ -1,218 +0,0 @@
-import type { Api, Model } from "@earendil-works/pi-ai";
-import type { ExtensionAPI, ExtensionCommandContext, Theme } from "@earendil-works/pi-coding-agent";
-import { truncateToWidth, type Component } from "@earendil-works/pi-tui";
-import { relative } from "node:path";
-import { supportsImageInput, type ModelLike } from "@oai404iao/pi-codex-runtime/internal/capabilities";
-import { createBackgroundImageJobs, panelBranch } from "./background-image-jobs.js";
-import { listResolvedModelProfiles } from "@oai404iao/pi-codex-runtime/internal/model-catalog/catalog";
-import { loadModelSettings } from "@oai404iao/pi-codex-runtime/internal/model-catalog/runtime";
-import { loadSettings } from "@oai404iao/pi-codex-runtime/internal/settings";
-import { standaloneImageGeneration } from "./tools/image-generation.js";
-import { buildGeneratedImageDisplayText } from "./tools/image-generation/storage.js";
-import { IMAGE_SAVE_DISPLAY_MESSAGE_TYPE, type SavedGeneratedImage } from "./tools/image-generation/types.js";
-import { projectRoot } from "./utils/images.js";
-
-const IMAGE_GEN_ERROR_MESSAGE_TYPE = "codex-image-generation-error";
-
-export interface ParsedImageGenCommand {
-	prompt: string;
-	imagePaths: string[];
-}
-
-interface ImageGenerationErrorDetails {
-	message: string;
-	prompt?: string;
-	imageModel?: string;
-	referenceCount?: number;
-}
-
-interface ModelRegistryLike {
-	getAll?: () => unknown;
-	getAvailable?: () => unknown;
-	find?: (provider: string, id: string) => unknown;
-}
-
-function isModelLike(value: unknown): value is ModelLike {
-	if (!value || typeof value !== "object") return false;
-	const record = value as Record<string, unknown>;
-	return typeof record.provider === "string" && (typeof record.id === "string" || typeof record.name === "string");
-}
-
-function registryModels(registry: ModelRegistryLike | undefined): ModelLike[] {
-	if (!registry) return [];
-	for (const method of [registry.getAll, registry.getAvailable]) {
-		if (typeof method !== "function") continue;
-		try {
-			const value = method.call(registry);
-			if (Array.isArray(value)) return value.filter(isModelLike);
-		} catch {
-			// Try the next registry shape.
-		}
-	}
-	return [];
-}
-
-function isConfiguredImageModel(model: ModelLike | undefined): boolean {
-	if (!model || !supportsImageInput(model)) return false;
-	const settings = loadModelSettings({ ...model, baseUrl: undefined });
-	return Boolean(settings.modelProfile?.effective.enabled && settings.imageGenerationImplementation === "standalone");
-}
-
-export function selectCodexImageModel(currentModel: ModelLike | undefined, registry: ModelRegistryLike | undefined): ModelLike | undefined {
-	const profile = currentModel && loadModelSettings({ ...currentModel, baseUrl: undefined }).modelProfile;
-	if (profile?.removedHostedImageGeneration) {
-		throw new Error(`Hosted image generation was removed for ${profile.id}. Change tools.imageGeneration to standalone or false in models.json; no alternate model was selected.`);
-	}
-	if (isConfiguredImageModel(currentModel)) return currentModel;
-	const discovered = registryModels(registry).find(isConfiguredImageModel);
-	if (discovered) return discovered;
-	for (const profile of listResolvedModelProfiles()) {
-		if (!profile.effective.enabled || profile.effective.tools.imageGeneration !== "standalone") continue;
-		const slash = profile.id.indexOf("/");
-		const candidate = registry?.find?.(profile.id.slice(0, slash), profile.id.slice(slash + 1));
-		if (isModelLike(candidate) && isConfiguredImageModel(candidate)) return candidate;
-	}
-	return undefined;
-}
-
-function tokenizeArgs(input: string): string[] {
-	const tokens: string[] = [];
-	let current = "";
-	let quote: '"' | "'" | undefined;
-	let escaped = false;
-	for (const ch of input) {
-		if (escaped) {
-			current += ch;
-			escaped = false;
-			continue;
-		}
-		if (ch === "\\") {
-			escaped = true;
-			continue;
-		}
-		if (quote) {
-			if (ch === quote) quote = undefined;
-			else current += ch;
-			continue;
-		}
-		if (ch === '"' || ch === "'") {
-			quote = ch;
-			continue;
-		}
-		if (/\s/.test(ch)) {
-			if (current) tokens.push(current);
-			current = "";
-			continue;
-		}
-		current += ch;
-	}
-	if (current) tokens.push(current);
-	return tokens;
-}
-
-function isSupportedImagePathToken(token: string): boolean {
-	const normalized = token.replace(/^file:\/\//, "").replace(/[),.;:]+$/, "");
-	if (/^https?:\/\//i.test(normalized) || normalized.startsWith("data:")) return false;
-	return /\.(?:png|jpe?g|webp|gif|bmp)$/i.test(normalized);
-}
-
-export function parseImageGenCommandArgs(input: string): ParsedImageGenCommand {
-	const imagePaths: string[] = [];
-	const promptParts: string[] = [];
-	for (const token of tokenizeArgs(input.trim())) {
-		if (token.startsWith("@") && token.length > 1) imagePaths.push(token.slice(1));
-		else if (isSupportedImagePathToken(token)) imagePaths.push(token.replace(/^file:\/\//, "").replace(/[),.;:]+$/, ""));
-		else promptParts.push(token);
-	}
-	return { prompt: promptParts.join(" ").trim(), imagePaths };
-}
-
-function renderImageGenError(details: ImageGenerationErrorDetails, theme: Theme): Component {
-	return {
-		invalidate() {},
-		render(width: number): string[] {
-			const safeWidth = Math.max(1, width);
-			const message = details.message || "Image generation failed.";
-			const promptWidth = Math.max(16, safeWidth - 28);
-			const lines = [
-				`${theme.fg("error", "● ")}${theme.fg("text", theme.bold("Image Generation "))}${theme.fg("error", "failed")}`,
-			];
-			if (details.imageModel) lines.push(`${panelBranch(theme, "├")}${theme.fg("text", "Model ")}${theme.fg("muted", details.imageModel)}`);
-			if (details.referenceCount && details.referenceCount > 0) lines.push(`${panelBranch(theme, "├")}${theme.fg("text", "Refs ")}${theme.fg("muted", `${details.referenceCount}`)}`);
-			if (details.prompt) lines.push(`${panelBranch(theme, "├")}${theme.fg("text", "Prompt ")}${theme.fg("muted", truncateToWidth(details.prompt, promptWidth, "…"))}`);
-			lines.push(`${panelBranch(theme, "└")}${theme.fg("error", truncateToWidth(message, Math.max(16, safeWidth - 5), "…"))}`);
-			return lines.map((line) => truncateToWidth(line, safeWidth, ""));
-		},
-	};
-}
-
-async function runBackgroundImageGeneration(pi: ExtensionAPI, ctx: ExtensionCommandContext, parsed: ParsedImageGenCommand, signal: AbortSignal): Promise<void> {
-	signal.throwIfAborted();
-	const model = selectCodexImageModel(ctx.model as ModelLike | undefined, ctx.modelRegistry as ModelRegistryLike | undefined) as Model<Api> | undefined;
-	if (!model) throw new Error("No image-capable model with an enabled standalone model catalog profile is available.");
-	const settings = loadModelSettings({ ...model, baseUrl: undefined }, ctx.cwd);
-	const result = await standaloneImageGeneration({
-		prompt: parsed.prompt, referenced_image_paths: parsed.imagePaths,
-	}, {
-		cwd: ctx.cwd, model, modelRegistry: ctx.modelRegistry, sessionManager: ctx.sessionManager,
-	}, settings, signal, { callId: "standalone" });
-	signal.throwIfAborted();
-	const saved = result.details.saved;
-	const workspaceRoot = projectRoot(ctx.cwd);
-	const relativePath = relative(workspaceRoot, saved.path);
-	const latestAbsolutePath = saved.latestPath ?? saved.path;
-	const latestRelativePath = relative(workspaceRoot, latestAbsolutePath);
-	const savedImage: SavedGeneratedImage = {
-		absolutePath: saved.path,
-		relativePath: relativePath && !relativePath.startsWith("..") ? relativePath : saved.path,
-		latestAbsolutePath,
-		latestRelativePath: latestRelativePath && !latestRelativePath.startsWith("..") ? latestRelativePath : latestAbsolutePath,
-		responseId: result.details.requestId,
-		callId: "standalone", outputFormat: saved.format, imageModel: settings.imageModel,
-		revisedPrompt: parsed.prompt,
-	};
-	pi.sendMessage({
-		customType: IMAGE_SAVE_DISPLAY_MESSAGE_TYPE,
-		content: [{ type: "text", text: buildGeneratedImageDisplayText(savedImage, { expanded: false }) }],
-		display: true, details: { savedImages: [savedImage] },
-	}, { triggerTurn: false });
-}
-
-export function registerBackgroundImageGenerationCommand(pi: ExtensionAPI): void {
-	const jobs = createBackgroundImageJobs();
-	pi.on("session_start", (_event, ctx) => jobs.reset(ctx));
-	pi.on("session_shutdown", (_event, ctx) => jobs.reset(ctx));
-	pi.registerMessageRenderer<ImageGenerationErrorDetails>(IMAGE_GEN_ERROR_MESSAGE_TYPE, (message, _options, theme) => {
-		const rawContent = typeof message.content === "string" ? message.content
-			: message.content.filter((item) => item.type === "text").map((item) => item.text).join("\n");
-		return renderImageGenError({
-			message: message.details?.message ?? rawContent.replace(/^Image generation failed:\s*/i, "") ?? "Image generation failed.",
-			prompt: message.details?.prompt, imageModel: message.details?.imageModel, referenceCount: message.details?.referenceCount,
-		}, theme);
-	});
-	pi.registerCommand("image-gen", {
-		description: "Generate or edit an image in the background with the standalone Images API. Usage: /image-gen prompt text [@reference.png]",
-		handler: async (args, ctx) => {
-			const parsed = parseImageGenCommandArgs(args);
-			if (!parsed.prompt) {
-				ctx.ui.notify("Usage: /image-gen prompt text [@reference.png]", "warning");
-				return;
-			}
-			const settings = loadSettings(ctx.cwd);
-			const job = jobs.start(ctx, parsed, settings.imageModel);
-			ctx.ui.notify(`Queued image generation with ${settings.imageModel}${parsed.imagePaths.length ? ` (${parsed.imagePaths.length} reference image${parsed.imagePaths.length === 1 ? "" : "s"})` : ""}.`, "info");
-			void runBackgroundImageGeneration(pi, ctx, parsed, job.signal)
-				.catch((error) => {
-					if (!job.isCurrent()) return;
-					const message = error instanceof Error ? error.message : String(error);
-					pi.sendMessage({
-						customType: IMAGE_GEN_ERROR_MESSAGE_TYPE,
-						content: `Image generation failed: ${message}`, display: true,
-						details: { message, prompt: parsed.prompt, imageModel: settings.imageModel, referenceCount: parsed.imagePaths.length } satisfies ImageGenerationErrorDetails,
-					}, { triggerTurn: false });
-				})
-				.finally(() => job.finish())
-				.catch(() => {}); // Closed UI/API failures must not escape a detached job.
-		},
-	});
-}

package/src/background-image-jobs.ts

@@ -1,116 +0,0 @@
-import type { ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
-import { truncateToWidth, visibleWidth, type Component } from "@earendil-works/pi-tui";
-import { frameGlyphs, glyphs, treeGlyph } from "@oai404iao/pi-codex-runtime/internal/glyphs";
-
-const STATUS_KEY = "codex-image-gen";
-const PADDING = 1;
-
-interface Job {
-	id: string;
-	startedAt: number;
-	prompt: string;
-	referenceCount: number;
-	imageModel: string;
-	controller: AbortController;
-}
-
-function padAnsi(text: string, width: number): string {
-	const truncated = truncateToWidth(text, width, "");
-	return `${truncated}${" ".repeat(Math.max(0, width - visibleWidth(truncated)))}`;
-}
-
-function panelFrame(lines: string[], width: number, theme: Theme): string[] {
-	const safeWidth = Math.max(1, width);
-	if (safeWidth < 8) return lines.map(line => truncateToWidth(line, safeWidth, ""));
-	const inner = Math.max(1, width - 2);
-	const contentWidth = Math.max(1, width - 2 - PADDING * 2);
-	const border = (text: string) => theme.fg("borderAccent", text);
-	const frame = frameGlyphs();
-	return [
-		`${border(frame.tl)}${border(frame.h.repeat(inner))}${border(frame.tr)}`,
-		...lines.map(line => `${border(frame.v)}${" ".repeat(PADDING)}${padAnsi(line, contentWidth)}${" ".repeat(PADDING)}${border(frame.v)}`),
-		`${border(frame.bl)}${border(frame.h.repeat(inner))}${border(frame.br)}`,
-	].map(line => truncateToWidth(line, safeWidth, ""));
-}
-
-export function panelBranch(theme: Theme, branch: "├" | "└" | "│"): string {
-	return theme.fg("muted", treeGlyph(branch));
-}
-
-function renderJobs(active: Map<string, Job>, theme: Theme, width: number): string[] {
-	const jobs = [...active.values()].sort((a, b) => a.startedAt - b.startedAt);
-	if (!jobs.length) return [];
-	const elapsed = Math.max(0, Math.round((Date.now() - jobs[0]!.startedAt) / 1000));
-	const imageModels = [...new Set(jobs.map(job => job.imageModel))];
-	const modelText = imageModels.length === 1 ? imageModels[0] : `${imageModels.length} models`;
-	const refs = jobs.reduce((total, job) => total + job.referenceCount, 0);
-	const refText = refs > 0 ? ` · ${refs} ref${refs === 1 ? "" : "s"}` : "";
-	const lines = [`${theme.fg("customMessageLabel", theme.bold("Image Generation"))} ${theme.fg("muted", `${jobs.length} running · ${modelText}${refText} · ${elapsed}s`)}`];
-	const dot = theme.fg("dim", glyphs().dot);
-	const shown = jobs.slice(0, 4);
-	for (const [index, job] of shown.entries()) {
-		const age = Math.max(0, Math.round((Date.now() - job.startedAt) / 1000));
-		const isLast = index === shown.length - 1 && jobs.length <= shown.length;
-		const references = job.referenceCount > 0 ? `${dot}${theme.fg("dim", `${job.referenceCount} ref${job.referenceCount === 1 ? "" : "s"}`)}` : "";
-		const promptWidth = Math.max(16, width - 36);
-		lines.push(`${panelBranch(theme, isLast ? "└" : "├")}${theme.fg("accent", glyphs().bullet.trim())} ${theme.fg("accent", truncateToWidth(job.prompt, promptWidth, glyphs().ellipsis))}${dot}${theme.fg("muted", job.imageModel)}${references}${dot}${theme.fg("dim", `${age}s`)}`);
-	}
-	if (jobs.length > shown.length) lines.push(`${panelBranch(theme, "└")}${theme.fg("muted", `${glyphs().ellipsis} ${jobs.length - shown.length} more`)}`);
-	return panelFrame(lines, width, theme);
-}
-
-export function createBackgroundImageJobs() {
-	const jobs = new Map<string, Job>();
-	let generation = 0;
-	let timer: ReturnType<typeof setInterval> | undefined;
-	let statusContext: ExtensionContext | undefined;
-	const stopTimer = () => {
-		if (timer !== undefined) clearInterval(timer);
-		timer = undefined;
-	};
-	const update = (ctx: ExtensionContext) => {
-		statusContext = ctx;
-		ctx.ui?.setStatus(STATUS_KEY, jobs.size ? `image-gen ${jobs.size}` : undefined);
-		ctx.ui?.setWidget(STATUS_KEY, jobs.size ? (_tui: unknown, theme: Theme): Component => ({
-			invalidate() {},
-			render: width => renderJobs(jobs, theme, width),
-		}) : undefined, { placement: "aboveEditor" });
-		if (!jobs.size) stopTimer();
-		else if (timer === undefined) {
-			timer = setInterval(() => { if (statusContext) update(statusContext); }, 1000);
-			timer.unref?.();
-		}
-	};
-	return {
-		start(ctx: ExtensionContext, parsed: { prompt: string; imagePaths: string[] }, imageModel: string) {
-			const epoch = generation;
-			const job: Job = {
-				id: `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`,
-				startedAt: Date.now(), prompt: parsed.prompt, referenceCount: parsed.imagePaths.length,
-				imageModel, controller: new AbortController(),
-			};
-			jobs.set(job.id, job);
-			update(ctx);
-			const isCurrent = () => epoch === generation && !job.controller.signal.aborted;
-			return {
-				signal: job.controller.signal,
-				isCurrent,
-				finish() {
-					if (!isCurrent()) return;
-					jobs.delete(job.id);
-					update(ctx);
-				},
-			};
-		},
-		reset(ctx?: ExtensionContext) {
-			generation++;
-			for (const job of jobs.values()) job.controller.abort(new Error("Image generation session ended"));
-			jobs.clear();
-			stopTimer();
-			const current = ctx ?? statusContext;
-			statusContext = undefined;
-			current?.ui?.setStatus(STATUS_KEY, undefined);
-			current?.ui?.setWidget(STATUS_KEY, undefined);
-		},
-	};
-}

package/src/tools/image-generation/capture.ts

@@ -1,42 +0,0 @@
-import type { ProviderEventContext } from "@oai404iao/pi-codex-runtime/internal/providers/openai-codex/stream-effects";
-import type { StreamEventShape } from "@oai404iao/pi-codex-runtime/internal/providers/openai-codex/types";
-import { loadSettings } from "@oai404iao/pi-codex-runtime/internal/settings";
-import { normalizeImageOutputFormat, saveOpenAICodexGeneratedImage } from "./storage.js";
-import type { SavedGeneratedImage } from "./types.js";
-
-export function createImageCapture(
-	context: ProviderEventContext,
-	onImageSaved: (image: SavedGeneratedImage, data: { data: string; mimeType: string }) => void,
-	saveImage = saveOpenAICodexGeneratedImage,
-) {
-	let responseId: string | undefined;
-	return async (event: StreamEventShape): Promise<void> => {
-		if (event.type === "response.created" && event.response?.id) responseId = event.response.id;
-		if (context.signal?.aborted) return;
-		if (event.type !== "response.output_item.done" || event.item?.type !== "image_generation_call") return;
-		const callId = typeof event.item.id === "string" ? event.item.id : undefined;
-		const result = typeof event.item.result === "string" ? event.item.result : undefined;
-		if (!callId || !result) return;
-		try {
-			const outputFormat = typeof event.item.output_format === "string" ? event.item.output_format : undefined;
-			const normalizedOutputFormat = normalizeImageOutputFormat(outputFormat);
-			const settings = loadSettings(context.cwd);
-			const imageModel = typeof event.item.model === "string" ? event.item.model : settings.imageModel;
-			const saved = await saveImage(context.cwd, {
-				responseId,
-				callId,
-				result,
-				outputFormat: normalizedOutputFormat,
-				imageModel,
-				revisedPrompt:
-					typeof event.item.revised_prompt === "string" ? event.item.revised_prompt : context.requestPrompt,
-			});
-			if (!context.signal?.aborted) {
-				onImageSaved(saved, { data: result, mimeType: `image/${normalizedOutputFormat}` });
-			}
-		} catch {
-			// Persistence is best-effort. Raw diagnostics here would corrupt the
-			// active TUI; failures must not discard the provider response/replay.
-		}
-	};
-}

package/src/tools/image-generation/display.ts

@@ -1,73 +0,0 @@
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { makeCachedImagePreview, renderImageGenerationMessage } from "./preview.js";
-import { buildGeneratedImageDisplayText } from "./storage.js";
-import {
-	IMAGE_SAVE_DISPLAY_MESSAGE_TYPE,
-	type CachedImagePreview,
-	type ImageDisplayMessageDetails,
-	type PendingActivity,
-	type SavedGeneratedImage,
-} from "./types.js";
-
-export function createImageDisplay(pi: ExtensionAPI) {
-	const pendingActivities: PendingActivity[] = [];
-	const imagePreviewCache = new Map<string, CachedImagePreview>();
-	let pendingFlushTimer: ReturnType<typeof setTimeout> | undefined;
-	let generation = 0;
-
-	const cancelFlush = () => {
-		if (pendingFlushTimer !== undefined) clearTimeout(pendingFlushTimer);
-		pendingFlushTimer = undefined;
-	};
-
-	const flush = () => {
-		cancelFlush();
-		const activities = pendingActivities.splice(0, pendingActivities.length);
-		for (const activity of activities) {
-			imagePreviewCache.set(activity.savedImage.absolutePath, makeCachedImagePreview(activity.imageData.data, activity.imageData.mimeType));
-			pi.sendMessage(
-				{
-					customType: IMAGE_SAVE_DISPLAY_MESSAGE_TYPE,
-					content: [{ type: "text", text: buildGeneratedImageDisplayText(activity.savedImage, { expanded: false }) }],
-					display: true,
-					details: { savedImages: [activity.savedImage] } satisfies ImageDisplayMessageDetails,
-				},
-				{ triggerTurn: false },
-			);
-		}
-	};
-
-	return {
-		flush,
-		scheduleFlush() {
-			if (pendingFlushTimer !== undefined || pendingActivities.length === 0) return;
-			pendingFlushTimer = setTimeout(flush, 0);
-		},
-		clear() {
-			generation++;
-			cancelFlush();
-			pendingActivities.length = 0;
-			imagePreviewCache.clear();
-		},
-		/** Ignore completion callbacks belonging to a replaced/closed session. */
-		captureSink() {
-			const requestGeneration = generation;
-			return (savedImage: SavedGeneratedImage, imageData: { data: string; mimeType: string }) => {
-				if (requestGeneration !== generation) return;
-				pendingActivities.push({ kind: "image", savedImage, imageData });
-			};
-		},
-		registerRenderer() {
-			pi.registerMessageRenderer<ImageDisplayMessageDetails>(IMAGE_SAVE_DISPLAY_MESSAGE_TYPE, (message, options, theme) => {
-				const savedImage = message.details?.savedImages?.[0];
-				const textContent = typeof message.content === "string"
-					? message.content
-					: message.content
-							.filter((item) => item.type === "text")
-							.map((item) => item.text)
-							.join("\n");
-				return renderImageGenerationMessage(savedImage, textContent, options, theme, imagePreviewCache);
-			});
-		},
-	};
-}

package/src/tools/image-generation/output.ts

@@ -1,29 +0,0 @@
-import type { SavedImageInfo } from "../../utils/images.js";
-
-export const imageGenerationOutputSchema = {
-	type: "object",
-	additionalProperties: false,
-	required: ["path", "image"],
-	properties: {
-		path: { type: "string" },
-		latestPath: { type: "string" },
-		image: {
-			type: "object",
-			additionalProperties: false,
-			required: ["type", "data", "mimeType"],
-			properties: {
-				type: { type: "string", const: "image" },
-				data: { type: "string", description: "Base64 image data. Use image(result.image) to show it to the model." },
-				mimeType: { type: "string" },
-			},
-		},
-	},
-};
-
-export function imageGenerationOutput(saved: SavedImageInfo, base64: string) {
-	return {
-		path: saved.path,
-		...(saved.latestPath ? { latestPath: saved.latestPath } : {}),
-		image: { type: "image" as const, data: base64, mimeType: saved.mimeType },
-	};
-}

package/src/tools/image-generation/preview.ts

@@ -1,79 +0,0 @@
-import { Container, Image, Spacer, Text, getCapabilities, getImageDimensions } from "@earendil-works/pi-tui";
-import { glyphs, treeGlyph } from "@oai404iao/pi-codex-runtime/internal/glyphs";
-import { themeBold, themeFg } from "@oai404iao/pi-codex-runtime/internal/utils/theme";
-import { type CachedImagePreview, type SavedGeneratedImage } from "./types.js";
-
-function getNodeFsSync(): { readFileSync(path: string): Buffer } | null {
-	if (typeof process === "undefined" || !(process.versions?.node || process.versions?.bun)) {
-		return null;
-	}
-	const builtinProcess = process as typeof process & { getBuiltinModule?: (specifier: string) => unknown };
-	if (typeof builtinProcess.getBuiltinModule !== "function") {
-		return null;
-	}
-	try {
-		const module = builtinProcess.getBuiltinModule("node:fs") as { readFileSync?: (path: string) => Buffer } | undefined;
-		return typeof module?.readFileSync === "function" ? { readFileSync: module.readFileSync } : null;
-	} catch {
-		return null;
-	}
-}
-
-export function makeCachedImagePreview(data: string, mimeType: string, bytes?: number): CachedImagePreview {
-	const dimensions = getImageDimensions(data, mimeType) ?? undefined;
-	return { data, mimeType, bytes: bytes ?? Buffer.from(data, "base64").byteLength, widthPx: dimensions?.widthPx, heightPx: dimensions?.heightPx };
-}
-
-function loadCachedImagePreview(savedImage: SavedGeneratedImage, imagePreviewCache: Map<string, CachedImagePreview>): CachedImagePreview | undefined {
-	const cached = imagePreviewCache.get(savedImage.absolutePath);
-	if (cached) return cached;
-	const fs = getNodeFsSync();
-	if (!fs) return undefined;
-	try {
-		const buffer = fs.readFileSync(savedImage.absolutePath);
-		const data = buffer.toString("base64");
-		const mimeType = `image/${savedImage.outputFormat}`;
-		const preview = makeCachedImagePreview(data, mimeType, buffer.byteLength);
-		imagePreviewCache.set(savedImage.absolutePath, preview);
-		return preview;
-	} catch {
-		return undefined;
-	}
-}
-
-function formatImageBytes(bytes: number | undefined): string | undefined {
-	if (!Number.isFinite(bytes) || !bytes) return undefined;
-	if (bytes < 1024) return `${bytes} B`;
-	if (bytes < 1024 * 1024) return `${Math.round(bytes / 102.4) / 10}K`;
-	return `${Math.round(bytes / (1024 * 102.4)) / 10}M`;
-}
-
-function shouldRenderInlineImage(): { ok: boolean; reason?: string } {
-	if (process.env.TMUX) return { ok: false, reason: "inline preview disabled in tmux to avoid overlay/stale image artifacts" };
-	const protocol = getCapabilities().images;
-	if (!protocol) return { ok: false, reason: "terminal image protocol unavailable" };
-	return { ok: true };
-}
-
-export function renderImageGenerationMessage(savedImage: SavedGeneratedImage | undefined, messageContent: unknown, options: any, theme: any, imagePreviewCache: Map<string, CachedImagePreview>): Container {
-	const container = new Container();
-	const preview = savedImage ? loadCachedImagePreview(savedImage, imagePreviewCache) : undefined;
-	const type = savedImage?.outputFormat?.toUpperCase() ?? preview?.mimeType?.replace(/^image\//, "").toUpperCase() ?? "IMAGE";
-	const dimensions = preview?.widthPx && preview?.heightPx ? `${preview.widthPx}x${preview.heightPx}` : undefined;
-	const size = formatImageBytes(preview?.bytes);
-	const imageModel = savedImage?.imageModel ? `model ${savedImage.imageModel}` : undefined;
-	const meta = [imageModel, type, dimensions, size].filter(Boolean).join(glyphs().dot);
-	const label = `${themeFg(theme, "accent", glyphs().bullet)}${themeFg(theme, "text", themeBold(theme, "Image Generation "))}`;
-	const pathText = savedImage?.relativePath ?? (typeof messageContent === "string" ? messageContent : "generated image");
-	const lines = [`${label}${themeFg(theme, "accent", pathText)}${meta ? themeFg(theme, "dim", `${glyphs().dot}${meta}`) : ""}`];
-	if (savedImage?.latestRelativePath) lines.push(`${themeFg(theme, "muted", `  ${treeGlyph("├")}`)}${themeFg(theme, "text", "Latest ")}${themeFg(theme, "accent", savedImage.latestRelativePath)}`);
-	if (options?.expanded && savedImage?.revisedPrompt) lines.push(`${themeFg(theme, "muted", `  ${treeGlyph("├")}`)}${themeFg(theme, "text", "Prompt ")}${themeFg(theme, "dim", savedImage.revisedPrompt)}`);
-	const inline = shouldRenderInlineImage();
-	if (!inline.ok) lines.push(`${themeFg(theme, "muted", `  ${treeGlyph("└")}`)}${themeFg(theme, "warning", inline.reason ?? "inline preview unavailable")}`);
-	container.addChild(new Text(lines.join("\n"), 0, 0));
-	if (savedImage && preview && inline.ok) {
-		container.addChild(new Spacer(1));
-		container.addChild(new Image(preview.data, preview.mimeType, { fallbackColor: (text) => themeFg(theme, "dim", text) }, { maxWidthCells: 72, maxHeightCells: options?.expanded ? 24 : 14, filename: savedImage.relativePath }));
-	}
-	return container;
-}