pi-openai-toolkit 0.17.2 → 0.18.0

package/README.md

@@ -168,7 +168,7 @@ Toolkit does not fall back between routes. See [search configuration](docs/confi
 
 ### Generate an image
 
-Requires a Responses session and may incur provider charges. Enable it with an image model your provider supports:
+Requires an eligible Responses-capable session and may incur provider charges. The local Pi tool sends a separate Images API request to the current provider (`/images/generations` or `/images/edits`; Codex uses `/codex/images/...`) rather than adding an image tool to the conversation request. Enable it with an image model your provider supports:
 
 ```json
 {
@@ -183,7 +183,7 @@ Requires a Responses session and may incur provider charges. Enable it with an i
 }
 ```
 
-Ask Pi to generate an image or edit explicitly supplied local references using `openai_generate_image`. The default model must be in `allowedModels`; image settings are global. See [image configuration](docs/configuration.md#images).
+Ask Pi to generate an image or edit explicitly supplied local references using `openai_generate_image`. The selected image model is sent in the independent request; the active conversation model remains the routing/auth model. Set the tool's optional `model` to another exact `allowedModels` entry, or use `batchSize` (1-10) for concurrent independent variants. Batch calls save canonical artifacts and cannot use a single-file `outputPath`; image settings are global. See [image configuration](docs/configuration.md#images).
 
 ### Review tool calls automatically
 

package/README.zh.md

@@ -168,7 +168,7 @@ Toolkit 不会在路由之间自动回退。后端要求见[搜索配置](docs/c
 
 ### 生成图片
 
-需要 Responses 会话，并可能产生服务商费用。选择服务商支持的生图模型后启用：
+需要符合条件的 Responses-capable 会话，并可能产生服务商费用。Pi 工具会向当前服务商发送独立的 Images API 请求（`/images/generations` 或 `/images/edits`；Codex 使用 `/codex/images/...`），不会把生图 tool 加入普通对话请求。选择服务商支持的生图模型后启用：
 
 ```json
 {
@@ -183,7 +183,7 @@ Toolkit 不会在路由之间自动回退。后端要求见[搜索配置](docs/c
 }
 ```
 
-让 Pi 通过 `openai_generate_image` 生成图片，或编辑明确提供的本地参考图。默认模型必须在 `allowedModels` 中；生图设置全局生效。详见[生图配置](docs/configuration.md#images)。
+让 Pi 通过 `openai_generate_image` 生成图片，或编辑明确提供的本地参考图。选中的生图模型会写入独立请求；当前对话模型仍只负责路由和认证。工具的可选 `model` 必须精确匹配 `allowedModels`，也可以使用 `batchSize`（1-10）并发生成多个独立变体。批量调用会保存 canonical artifacts，不能同时指定单文件 `outputPath`；生图设置全局生效。详见[生图配置](docs/configuration.md#images)。
 
 ### 启用工具调用自动审查
 

package/docs/configuration.md

@@ -104,7 +104,7 @@ There is no v2 search master switch. `local` and `unmanaged` differ in payload c
 
 Image settings are global-only under `defaults.imageGeneration`. `enabled` defaults to `false`. `defaultModel` defaults to `"gpt-image-2.5"`; `allowedModels` defaults to `["gpt-image-2.5"]`.
 
-These are bare output-model IDs, not active-session model keys. Each normalized ID must contain 1-256 characters. The allowed list must remain nonempty after normalization and contain the explicit default. Order has no default-selection meaning. A caller's one-call `model` must be a member; policy and membership are checked before auth, local-reference preparation/upload, and paid dispatch. The active session remains the Responses routing model. Config does not grant permission to upload arbitrary local files or imply that a provider supports a particular image model.
+These are bare output-model IDs, not active-session model keys. Each normalized ID must contain 1-256 characters. The allowed list must remain nonempty after normalization and contain the explicit default. A caller's one-call `model` must be a member; policy and membership are checked before auth, local-reference preparation/upload, and paid dispatch. The active session remains the routing/auth model, while the selected image model is sent in a separate Images API request. Generation uses `/images/generations`; edits use `/images/edits` (Codex uses the corresponding `/codex/images/...` paths). The wrapper never injects a Responses `image_generation` tool. `batchSize` is a tool argument from 1-10; each item is an independent `n: 1` request, partial success preserves saved artifacts, and batch calls cannot use a single-file `outputPath`. Config does not grant permission to upload arbitrary local files or imply that a provider supports a particular image model.
 
 ---
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-openai-toolkit",
-  "version": "0.17.2",
+  "version": "0.18.0",
   "private": false,
   "type": "module",
   "engines": {

package/src/codex-astra/extension.ts

@@ -1,6 +1,5 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { ASTRA_MODEL_ID } from "../types";
-import { CODEX_CLIENT_VERSION } from "../responses-headers";
 import {
 	getEffortControlState,
 	planStableEffort,
@@ -101,20 +100,6 @@ export function registerCodexAstraExtension(pi: ExtensionAPI): void {
 			return undefined;
 		}
 	});
-
-	pi.on("before_provider_headers", (event, ctx) => {
-		try {
-			const model = ctx.model;
-			if (!model || model.api !== "openai-codex-responses") return;
-			// Live requests need the same backend version gate as our synthetic
-			// ones: an old or missing `version` never reaches gated SKUs.
-			if (!event.headers.version) {
-				event.headers.version = CODEX_CLIENT_VERSION;
-			}
-		} catch {
-			// Header injection is best-effort.
-		}
-	});
 }
 
 export default function codexAstraExtension(pi: ExtensionAPI): void {

package/src/context-management/codex-provider.ts

@@ -146,7 +146,6 @@ export function codexContextProviderHeaders(
 		headers.set("ChatGPT-Account-ID", provider.accountId);
 		headers.set("originator", CODEX_TOOL_ORIGINATOR);
 		headers.set("User-Agent", userAgent(CODEX_TOOL_VERSION));
-		headers.set("version", CODEX_TOOL_VERSION);
 	} else {
 		headers.set("Authorization", `Bearer ${provider.apiKey}`);
 		headers.delete("ChatGPT-Account-ID");

package/src/image-generation/client.ts

@@ -1,10 +1,11 @@
 import { buildResponsesRequestHeaders } from "../responses-headers";
-import type { ResponsesRuntime } from "../runtime";
+import { normalizeBaseUrl, type ResponsesRuntime } from "../runtime";
 import {
 	extractProviderErrorMessage,
-	parseImageGenerationResponse,
+	parseImagesApiResponse,
 	type ImageGenerationRequestBody,
 } from "./protocol";
+import type { PreparedReferenceImage } from "./types";
 import {
 	IMAGE_REQUEST_TIMEOUT_MS,
 	MAX_IMAGE_ERROR_BYTES,
@@ -39,8 +40,62 @@ export type ImageGenerationFetch = (
 	init?: RequestInit,
 ) => Promise<Response>;
 
-function buildRequestHeaders(runtime: ResponsesRuntime): Headers {
-	return buildResponsesRequestHeaders(runtime, { accept: "application/json" });
+function buildRequestHeaders(runtime: ResponsesRuntime, contentType: string | null = "application/json"): Headers {
+	return buildResponsesRequestHeaders(runtime, { accept: "application/json", contentType });
+}
+
+function buildImagesEndpoint(runtime: ResponsesRuntime, action: ImageGenerationRequestBody["action"]): string {
+	const baseUrl = normalizeBaseUrl(runtime.baseUrl);
+	if (!baseUrl) throw new Error("Image generation requires a provider base URL.");
+	if (baseUrl.endsWith("/responses") || baseUrl.endsWith("/codex/responses")) {
+		throw new Error("Image generation requires the provider base URL, not a resolved Responses URL.");
+	}
+	const suffix = runtime.api === "openai-codex-responses" ? "codex/images" : "images";
+	const baseAlreadyIncludesCodex = runtime.api === "openai-codex-responses" && baseUrl.endsWith("/codex");
+	return `${baseUrl}/${baseAlreadyIncludesCodex ? "images" : suffix}/${action === "edit" ? "edits" : "generations"}`;
+}
+
+function dataUrl(reference: PreparedReferenceImage): string {
+	return `data:${reference.mimeType};base64,${reference.bytes.toString("base64")}`;
+}
+
+function jsonRequestBody(runtime: ResponsesRuntime, body: ImageGenerationRequestBody): Record<string, unknown> {
+	const common = {
+		model: body.model,
+		prompt: body.prompt,
+		n: body.n,
+		size: body.size,
+		quality: body.quality,
+	};
+	if (body.action === "edit") {
+		return {
+			...common,
+			images: body.references.map((reference) => ({ image_url: dataUrl(reference) })),
+		};
+	}
+	return runtime.api === "openai-codex-responses"
+		? common
+		: { ...common, output_format: "png", response_format: "b64_json" };
+}
+
+function multipartRequestBody(body: ImageGenerationRequestBody): FormData {
+	const form = new FormData();
+	form.append("model", body.model);
+	form.append("prompt", body.prompt);
+	form.append("n", String(body.n));
+	form.append("size", body.size);
+	form.append("quality", body.quality);
+	form.append("output_format", "png");
+	form.append("response_format", "b64_json");
+	for (const [index, reference] of body.references.entries()) {
+		const extension = reference.mimeType === "image/jpeg" ? "jpg" : reference.mimeType.slice("image/".length);
+		form.append(
+			"image",
+			new Blob([new Uint8Array(reference.bytes)], { type: reference.mimeType }),
+			`reference-${index + 1}.${extension}`,
+		);
+	}
+	return form;
 }
 
 async function readBoundedBody(
@@ -141,12 +196,16 @@ export async function requestGeneratedImage(args: {
 		? AbortSignal.any([args.signal, timeoutSignal])
 		: timeoutSignal;
 
+	const multipart = args.body.action === "edit" && args.runtime.api === "openai-responses";
+	const transportBody: BodyInit = multipart
+		? multipartRequestBody(args.body)
+		: JSON.stringify(jsonRequestBody(args.runtime, args.body));
 	let response: Response;
 	try {
-		response = await fetchFn(args.runtime.responsesUrl, {
+		response = await fetchFn(buildImagesEndpoint(args.runtime, args.body.action), {
 			method: "POST",
-			headers: buildRequestHeaders(args.runtime),
-			body: JSON.stringify(args.body),
+			headers: buildRequestHeaders(args.runtime, multipart ? null : "application/json"),
+			body: transportBody,
 			signal,
 		});
 	} catch (error) {
@@ -230,7 +289,7 @@ export async function requestGeneratedImage(args: {
 		};
 	}
 
-	const parsed = parseImageGenerationResponse(payload);
+	const parsed = parseImagesApiResponse(payload);
 	if (!parsed.ok) {
 		return {
 			ok: false,
@@ -244,5 +303,8 @@ export async function requestGeneratedImage(args: {
 
 export const _clientTest = {
 	buildRequestHeaders,
+	buildImagesEndpoint,
+	jsonRequestBody,
+	multipartRequestBody,
 	readBoundedBody,
 };

package/src/image-generation/eligibility.ts

@@ -2,9 +2,9 @@ import type { ImageGenerationConfig } from "../types";
 import { IMAGE_GENERATION_CAPABLE_APIS, type ImageGenerationModel } from "./types";
 
 /**
- * The hosted `image_generation` tool is a capability of the Responses API rather than of a
- * specific model, so support is decided by API alone. `enabled` is the single opt-in switch
- * and defaults to off because every successful provider request may incur a charge.
+ * The independent Images API is exposed only for Responses-capable runtimes rather than for a
+ * specific conversation model. `enabled` is the single opt-in switch and defaults to off because
+ * every successful provider request may incur a charge.
  */
 export function isImageGenerationEnabledForModel(
 	model: ImageGenerationModel | undefined,

package/src/image-generation/extension.ts

@@ -9,13 +9,14 @@ import {
 	IMAGE_GENERATION_QUALITIES,
 	IMAGE_GENERATION_SIZES,
 	IMAGE_GENERATION_TOOL_NAME,
+	MAX_IMAGE_BATCH_SIZE,
 	MAX_IMAGE_MODEL_ID_CHARS,
 	MAX_IMAGE_PATH_CHARS,
 	MAX_IMAGE_PROMPT_CHARS,
 	MAX_REFERENCE_IMAGE_COUNT,
 	ImageGenerationError,
 	sanitizeImageDiagnostic,
-	type ImageGenerationDetails,
+	type ImageGenerationToolDetails,
 } from "./types";
 
 const registeredApis = new WeakSet<object>();
@@ -82,6 +83,18 @@ const GenerateImageParameters = Type.Object(
 				},
 			),
 		),
+		batchSize: Type.Optional(
+			Type.Union(
+				[
+					Type.Null(),
+					Type.Integer({ minimum: 1, maximum: MAX_IMAGE_BATCH_SIZE }),
+				],
+				{
+					description:
+						"Run this many independent image requests concurrently (1-10). Batch requests use generated artifacts and cannot set outputPath.",
+				},
+			),
+		),
 	},
 	{ additionalProperties: false },
 );
@@ -111,12 +124,12 @@ export function registerImageGenerationExtension(
 	if (registeredApis.has(pi)) return;
 	registeredApis.add(pi);
 
-	pi.registerTool<typeof GenerateImageParameters, ImageGenerationDetails, Record<string, never>>({
+	pi.registerTool<typeof GenerateImageParameters, ImageGenerationToolDetails, Record<string, never>>({
 		name: IMAGE_GENERATION_TOOL_NAME,
 		label: "OpenAI Generate Image",
 		description:
-			"Generate a PNG image, or edit from one to five user-approved local reference images, through the current Responses-capable model and a configured hosted image_generation model. This is a paid provider operation.",
-		promptSnippet: "Generate or edit PNG images through the current Responses-capable model and a configured imageGeneration.allowedModels entry.",
+			"Generate PNG images, or edit from one to five user-approved local reference images, through an independent Images API request using a configured image model. This is a paid provider operation; batchSize runs independent requests concurrently.",
+		promptSnippet: "Generate or edit PNG images through an independent Images API request and a configured imageGeneration.allowedModels entry.",
 		promptGuidelines: [
 			"Use openai_generate_image when the user explicitly asks to create, draw, render, or edit a raster image and the active model speaks a Responses API.",
 			"Do not call openai_generate_image speculatively: it consumes the user's provider or gateway image quota.",
@@ -124,6 +137,7 @@ export function registerImageGenerationExtension(
 			"Set referenceImagePaths to null unless the user explicitly identified local files; never invent paths or placeholder strings, and remember upload requires user approval.",
 			"Set outputPath to null unless the user explicitly asks for a destination; never invent a destination or placeholder string, and otherwise use the default Pi agent artifact.",
 			"Set model to null unless the user named an image model configured in imageGeneration.allowedModels; never invent or guess a model id, and remember an unlisted model fails before the paid request.",
+			"Use batchSize only when the user asks for multiple variants; each batch item is an independent paid request, and batchSize cannot be combined with outputPath.",
 			"Do not substitute Python, browser automation, shell scripts, or unrelated image tools for an eligible image request.",
 		],
 		parameters: GenerateImageParameters,

package/src/image-generation/protocol.ts

@@ -4,6 +4,7 @@ import {
 	IMAGE_GENERATION_QUALITIES,
 	IMAGE_GENERATION_SIZES,
 	MAX_GENERATED_IMAGE_BYTES,
+	MAX_IMAGE_BATCH_SIZE,
 	MAX_IMAGE_DIAGNOSTIC_CHARS,
 	MAX_IMAGE_DIMENSION,
 	MAX_IMAGE_IDENTIFIER_CHARS,
@@ -26,26 +27,13 @@ const BASE64_PATTERN = /^(?:[A-Za-z0-9+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/
 const DATA_URL_PATTERN = /^data:image\/[a-z0-9.+-]+;base64,(.*)$/is;
 
 export type ImageGenerationRequestBody = {
+	action: NormalizedGenerateImageParams["action"];
 	model: string;
-	store: false;
-	stream: false;
-	parallel_tool_calls: false;
-	input: Array<{
-		role: "user";
-		content: Array<
-			| { type: "input_text"; text: string }
-			| { type: "input_image"; image_url: string; detail: "auto" }
-		>;
-	}>;
-	tools: Array<{
-		type: "image_generation";
-		model: string;
-		action: "generate" | "edit";
-		size: NormalizedGenerateImageParams["size"];
-		quality: NormalizedGenerateImageParams["quality"];
-		output_format: "png";
-	}>;
-	tool_choice: { type: "image_generation" };
+	prompt: string;
+	n: 1;
+	size: NormalizedGenerateImageParams["size"];
+	quality: NormalizedGenerateImageParams["quality"];
+	references: readonly PreparedReferenceImage[];
 };
 
 export type ParsedImageResponseResult =
@@ -137,6 +125,22 @@ function normalizeRequestedModel(value: unknown): string | undefined {
 	return trimmed;
 }
 
+function normalizeBatchSize(value: unknown): number {
+	if (value === null || value === undefined) return 1;
+	if (
+		typeof value !== "number" ||
+		!Number.isInteger(value) ||
+		value < 1 ||
+		value > MAX_IMAGE_BATCH_SIZE
+	) {
+		throw new ImageGenerationError(
+			"invalid-parameters",
+			`batchSize must be an integer from 1 to ${MAX_IMAGE_BATCH_SIZE}.`,
+		);
+	}
+	return value;
+}
+
 /**
  * Configured model ids are only trusted as far as the shared bounds: unusable entries are
  * dropped, and an empty or malformed list falls back to the shipped default model.
@@ -153,7 +157,7 @@ function usableImageModels(models: readonly string[] | undefined): string[] {
 }
 
 /**
- * Resolve the nested `image_generation` model for one call. An omitted request selects the
+ * Resolve the independent Images API model for one call. An omitted request selects the
  * explicit default when supplied (v2), or the first configured model for legacy callers.
  * An explicit request must match a configured id exactly before any paid dispatch.
  */
@@ -213,6 +217,13 @@ export function normalizeGenerateImageParams(
 
 	const referenceImagePaths = normalizePathList(params.referenceImagePaths);
 	const outputPath = normalizeOutputPath(params.outputPath);
+	const batchSize = normalizeBatchSize(params.batchSize);
+	if (batchSize > 1 && outputPath !== undefined) {
+		throw new ImageGenerationError(
+			"invalid-parameters",
+			"outputPath cannot be used when batchSize is greater than 1; use the generated artifacts instead.",
+		);
+	}
 
 	return {
 		prompt,
@@ -221,12 +232,12 @@ export function normalizeGenerateImageParams(
 		size,
 		quality,
 		model: normalizeRequestedModel(params.model),
+		batchSize,
 		action: referenceImagePaths.length > 0 ? "edit" : "generate",
 	};
 }
 
 export function buildImageGenerationRequest(args: {
-	routingModel: string;
 	imageModel: string;
 	params: NormalizedGenerateImageParams;
 	references: readonly PreparedReferenceImage[];
@@ -238,34 +249,14 @@ export function buildImageGenerationRequest(args: {
 			"A configured image generation model is required.",
 		);
 	}
-	const content: ImageGenerationRequestBody["input"][number]["content"] = [
-		{ type: "input_text", text: args.params.prompt },
-	];
-	for (const reference of args.references) {
-		content.push({
-			type: "input_image",
-			image_url: `data:${reference.mimeType};base64,${reference.bytes.toString("base64")}`,
-			detail: "auto",
-		});
-	}
-
 	return {
-		model: args.routingModel,
-		store: false,
-		stream: false,
-		parallel_tool_calls: false,
-		input: [{ role: "user", content }],
-		tools: [
-			{
-				type: "image_generation",
-				model: imageModel,
-				action: args.params.action,
-				size: args.params.size,
-				quality: args.params.quality,
-				output_format: "png",
-			},
-		],
-		tool_choice: { type: "image_generation" },
+		action: args.params.action,
+		model: imageModel,
+		prompt: args.params.prompt,
+		n: 1,
+		size: args.params.size,
+		quality: args.params.quality,
+		references: args.references,
 	};
 }
 
@@ -390,7 +381,7 @@ export function extractProviderErrorMessage(value: unknown, fallback: string): s
 	return fallback;
 }
 
-export function parseImageGenerationResponse(value: unknown): ParsedImageResponseResult {
+export function parseImagesApiResponse(value: unknown): ParsedImageResponseResult {
 	if (!isRecord(value)) {
 		return {
 			ok: false,
@@ -398,75 +389,47 @@ export function parseImageGenerationResponse(value: unknown): ParsedImageRespons
 			errorMessage: "Image generation returned a non-object response.",
 		};
 	}
-
-	if (value.status !== "completed") {
-		return {
-			ok: false,
-			reason: "request-rejected",
-			errorMessage: extractProviderErrorMessage(
-				value,
-				`Image generation did not complete (status: ${String(value.status ?? "unknown")}).`,
-			),
-		};
-	}
-	if (!Array.isArray(value.output)) {
+	if (!Array.isArray(value.data)) {
 		return {
 			ok: false,
 			reason: "malformed-response",
-			errorMessage: "Image generation response did not contain an output array.",
+			errorMessage: "Image generation response did not contain a data array.",
 		};
 	}
-
-	const calls = new Map<string, { result: string; revisedPrompt?: string }>();
-	for (let index = 0; index < value.output.length; index += 1) {
-		const item = value.output[index];
-		if (!isRecord(item) || item.type !== "image_generation_call") continue;
-		if (item.status !== "completed") continue;
-		const result = typeof item.result === "string"
-			? item.result
-			: typeof item.b64_json === "string"
-				? item.b64_json
-				: undefined;
-		if (!result?.trim()) continue;
-		const id = normalizedOptionalString(item.id, MAX_IMAGE_IDENTIFIER_CHARS) ?? `image_generation_${index}`;
-		const revisedPrompt = normalizedOptionalString(item.revised_prompt, MAX_IMAGE_DIAGNOSTIC_CHARS);
-		const existing = calls.get(id);
-		if (existing && existing.result !== result) {
-			return {
-				ok: false,
-				reason: "malformed-response",
-				errorMessage: "Image generation returned conflicting results for one image call.",
-			};
-		}
-		calls.set(id, { result, revisedPrompt: revisedPrompt ?? existing?.revisedPrompt });
-	}
-
-	if (calls.size === 0) {
+	if (value.data.length === 0) {
 		return {
 			ok: false,
 			reason: "no-image",
 			errorMessage: "Image generation completed without a usable image result.",
 		};
 	}
-	if (calls.size !== 1) {
+	if (value.data.length !== 1) {
 		return {
 			ok: false,
 			reason: "malformed-response",
-			errorMessage: "Image generation returned more than one image result for a single-image request.",
+			errorMessage: "Image generation returned more than one image for a single request.",
 		};
 	}
 
-	const [imageCallId, call] = [...calls.entries()][0]!;
-	const decoded = decodeGeneratedPng(call.result);
+	const item = value.data[0];
+	if (!isRecord(item) || typeof item.b64_json !== "string" || !item.b64_json.trim()) {
+		return {
+			ok: false,
+			reason: "no-image",
+			errorMessage: "Image generation response did not contain base64 image data.",
+		};
+	}
+	const decoded = decodeGeneratedPng(item.b64_json);
 	if (!decoded.ok) return decoded;
 
 	return {
 		ok: true,
 		image: {
 			bytes: decoded.bytes,
-			imageCallId,
+			imageCallId:
+				normalizedOptionalString(item.id, MAX_IMAGE_IDENTIFIER_CHARS) ?? "image_generation_0",
 			responseId: normalizedOptionalString(value.id, MAX_IMAGE_IDENTIFIER_CHARS),
-			revisedPrompt: call.revisedPrompt,
+			revisedPrompt: normalizedOptionalString(item.revised_prompt, MAX_IMAGE_DIAGNOSTIC_CHARS),
 			width: decoded.width,
 			height: decoded.height,
 		},

package/src/image-generation/render.ts

@@ -21,7 +21,7 @@ import {
 	MAX_GENERATED_IMAGE_BYTES,
 	isImageGenerationDetails,
 	type GenerateImageParams,
-	type ImageGenerationDetails,
+	type ImageGenerationToolDetails,
 } from "./types";
 
 export type ImageGenerationRenderState = Record<string, never>;
@@ -54,7 +54,7 @@ function isPathInsideOrEqual(root: string, candidate: string): boolean {
 	return rel === "" || (!rel.startsWith(`..${sep}`) && rel !== ".." && !isAbsolute(rel));
 }
 
-function resultText(result: AgentToolResult<ImageGenerationDetails>): string {
+function resultText(result: AgentToolResult<ImageGenerationToolDetails>): string {
 	return result.content
 		.filter((block): block is { type: "text"; text: string } => block.type === "text")
 		.map((block) => block.text)
@@ -148,7 +148,7 @@ function textOnly(args: {
 export function createImageGenerationResultRenderer(
 	deps: ImageRendererDependencies = DEFAULT_RENDER_DEPS,
 ): (
-	result: AgentToolResult<ImageGenerationDetails>,
+	result: AgentToolResult<ImageGenerationToolDetails>,
 	options: ToolRenderResultOptions,
 	theme: Theme,
 	context: ImageGenerationRenderContext,