@howaboua/pi-codex-conversion 1.0.19 → 1.0.21

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

@@ -0,0 +1,112 @@
+import type { ExtensionAPI, ExtensionContext, ToolDefinition } from "@mariozechner/pi-coding-agent";
+import { Type } from "typebox";
+import { Container, Text } from "@mariozechner/pi-tui";
+import { isOpenAICodexModel } from "../adapter/codex-model.ts";
+
+export const IMAGE_GENERATION_UNSUPPORTED_MESSAGE =
+	"image_generation is only available with image-capable openai-codex models";
+const IMAGE_GENERATION_LOCAL_EXECUTION_MESSAGE =
+	"image_generation is a native openai-codex provider tool and should not execute locally";
+
+const IMAGE_GENERATION_PARAMETERS = Type.Unsafe<Record<string, never>>({
+	type: "object",
+	additionalProperties: false,
+});
+
+interface FunctionToolPayload {
+	type?: unknown;
+	name?: unknown;
+}
+
+interface ResponsesPayload {
+	tools?: unknown[];
+	[key: string]: unknown;
+}
+
+interface ResponsesImageGenerationTool {
+	type: "image_generation";
+	output_format: "png";
+}
+
+function supportsImageInputs(model: ExtensionContext["model"]): boolean {
+	return Array.isArray(model?.input) && model.input.includes("image");
+}
+
+export function supportsNativeImageGeneration(model: ExtensionContext["model"]): boolean {
+	return isOpenAICodexModel(model) && supportsImageInputs(model);
+}
+
+function isImageGenerationFunctionTool(tool: unknown): tool is FunctionToolPayload {
+	return !!tool && typeof tool === "object" && (tool as FunctionToolPayload).type === "function" && (tool as FunctionToolPayload).name === "image_generation";
+}
+
+function createEmptyResultComponent(): Container {
+	return new Container();
+}
+
+export function rewriteNativeImageGenerationTool(payload: unknown, model: ExtensionContext["model"]): unknown {
+	if (!supportsNativeImageGeneration(model) || !payload || typeof payload !== "object") {
+		return payload;
+	}
+
+	const tools = (payload as ResponsesPayload).tools;
+	if (!Array.isArray(tools)) {
+		return payload;
+	}
+
+	let rewritten = false;
+	const nextTools = tools.map((tool) => {
+		if (!isImageGenerationFunctionTool(tool)) {
+			return tool;
+		}
+		rewritten = true;
+		const nativeTool: ResponsesImageGenerationTool = {
+			type: "image_generation",
+			output_format: "png",
+		};
+		return nativeTool;
+	});
+
+	if (!rewritten) {
+		return payload;
+	}
+
+	return {
+		...(payload as ResponsesPayload),
+		tools: nextTools,
+	};
+}
+
+export function createImageGenerationTool(): ToolDefinition<typeof IMAGE_GENERATION_PARAMETERS> {
+	const description =
+		"Generate an image. Native openai-codex image_generation outputs are saved under `.pi/openai-codex-images/` and mirrored to `.pi/openai-codex-images/latest.png`. Use `view_image` only when pixel-level inspection is necessary.";
+	return {
+		name: "image_generation",
+		label: "image_generation",
+		description,
+		promptSnippet: description,
+		parameters: IMAGE_GENERATION_PARAMETERS,
+		prepareArguments: () => ({}),
+		async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
+			if (!supportsNativeImageGeneration(ctx.model)) {
+				throw new Error(IMAGE_GENERATION_UNSUPPORTED_MESSAGE);
+			}
+			throw new Error(IMAGE_GENERATION_LOCAL_EXECUTION_MESSAGE);
+		},
+		renderCall(_args, theme) {
+			return new Text(`${theme.fg("toolTitle", theme.bold("image_generation"))}`, 0, 0);
+		},
+		renderResult(result, { expanded }, theme) {
+			if (!expanded) {
+				return createEmptyResultComponent();
+			}
+			const textBlock = result.content.find((item) => item.type === "text");
+			const text = textBlock?.type === "text" ? textBlock.text : "(no output)";
+			return new Text(theme.fg("dim", text), 0, 0);
+		},
+	};
+}
+
+export function registerImageGenerationTool(pi: ExtensionAPI): void {
+	pi.registerTool(createImageGenerationTool());
+}

package/src/tools/view-image-tool.ts

@@ -7,7 +7,7 @@ import {
 	type ExtensionContext,
 	type ToolDefinition,
 } from "@mariozechner/pi-coding-agent";
-import { Type, type TSchema } from "@sinclair/typebox";
+import { Type, type TSchema } from "typebox";
 import { Text } from "@mariozechner/pi-tui";
 
 const VIEW_IMAGE_UNSUPPORTED_MESSAGE = "view_image is not allowed because you do not support image inputs";

package/src/tools/web-search-tool.ts

@@ -1,5 +1,5 @@
 import type { ExtensionAPI, ExtensionContext, ToolDefinition } from "@mariozechner/pi-coding-agent";
-import { Type } from "@sinclair/typebox";
+import { Type } from "typebox";
 import { Box, Container, Text } from "@mariozechner/pi-tui";
 import { isOpenAICodexModel } from "../adapter/codex-model.ts";
 
@@ -8,7 +8,7 @@ const WEB_SEARCH_LOCAL_EXECUTION_MESSAGE =
 	"web_search is a native openai-codex provider tool and should not execute locally";
 export const WEB_SEARCH_SESSION_NOTE_TYPE = "codex-web-search-session-note";
 export const WEB_SEARCH_SESSION_NOTE_TEXT =
-	"Native OpenAI Codex web search is enabled for this session. Search runs silently and is not surfaced as a separate tool call.";
+	"Native OpenAI Codex web search is enabled for this session. Search activity is surfaced as merged foldable status messages instead of native tool-call rows.";
 const WEB_SEARCH_MULTIMODAL_CONTENT_TYPES = ["text", "image"] as const;
 
 const WEB_SEARCH_PARAMETERS = Type.Unsafe<Record<string, never>>({

package/src/tools/write-stdin-tool.ts

@@ -1,5 +1,5 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { Type } from "@sinclair/typebox";
+import { Type } from "typebox";
 import { Container, Text } from "@mariozechner/pi-tui";
 import { renderWriteStdinCall } from "./codex-rendering.ts";
 import type { ExecSessionManager, UnifiedExecResult } from "./exec-session-manager.ts";