@earendil-works/pi-radius-work 0.1.5 → 0.1.7

package/README.md

@@ -46,7 +46,9 @@ Set `GOOGLE_WORKSPACE_TOKENS_DIR` to use another token directory. Set `GOOGLE_WO
 
 ## Image generation
 
-The `image_generation` tool accepts a text prompt and optional base64-encoded input images. It can generate a new image or edit supplied images. The tool is only available while the current model uses the `radius` or `radius-dev` provider.
+The `image_generation` tool accepts a text prompt, an optional output path, and optional base64-encoded input images. It can generate a new image or edit supplied images. Generated image bytes are written directly to disk instead of being added to the conversation context. Relative output paths are resolved from Pi's working directory; when no path is supplied, the tool creates `generated-image-<tool-call-id>.<extension>` there. If the gateway returns multiple images, the filenames receive numeric suffixes.
+
+The tool result contains only the saved file paths and small metadata. The tool is only available while the current model uses the `radius` or `radius-dev` provider.
 
 The gateway defaults to `https://radius.pi.dev`. Development and API-key overrides use `PI_GATEWAY`, `PI_GATEWAY_API_KEY`, `PI_DEV_GATEWAY`, and `PI_DEV_GATEWAY_API_KEY`.
 

package/extensions/radius-image-generation.ts

@@ -1,6 +1,10 @@
+import { mkdir, writeFile } from "node:fs/promises";
+import { dirname, extname, resolve } from "node:path";
 import { Type } from "@earendil-works/pi-ai";
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { Box, Spacer, Text } from "@earendil-works/pi-tui";
+import {
+  type ExtensionAPI,
+  withFileMutationQueue,
+} from "@earendil-works/pi-coding-agent";
 
 type GatewayImageGenerationConfig = {
   endpoint: string;
@@ -25,12 +29,19 @@ type GatewayProvider = {
 
 type ImageGenerationToolInput = {
   prompt: string;
+  output_path?: string;
   input_images?: Array<{
     mediaType: string;
     data: string;
   }>;
 };
 
+type SavedImage = {
+  path: string;
+  mediaType: string;
+  bytes: number;
+};
+
 type GatewayImageGenerationResponse = {
   kind: "radius#image_generation";
   model: string;
@@ -49,7 +60,6 @@ const DEV_GATEWAY_PROVIDER_ID = "radius-dev";
 const GATEWAY_API_KEY_ENV = "PI_GATEWAY_API_KEY";
 const DEV_GATEWAY_API_KEY_ENV = "PI_DEV_GATEWAY_API_KEY";
 const IMAGE_GENERATION_TOOL_NAME = "image_generation";
-const IMAGE_GENERATION_MESSAGE_TYPE = "radius-image-generation";
 
 export default async function (pi: ExtensionAPI) {
   const providers: GatewayProvider[] = [
@@ -73,36 +83,22 @@ export default async function (pi: ExtensionAPI) {
     });
   }
 
-  pi.registerMessageRenderer(IMAGE_GENERATION_MESSAGE_TYPE, (message, _options, theme) => {
-    const details = message.details as GatewayImageGenerationResponse | undefined;
-    const box = new Box(1, 1, (t) => theme.bg("customMessageBg", t));
-    box.addChild(
-      new Text(theme.fg("customMessageLabel", "\x1b[1m[image_generation]\x1b[22m"), 0, 0),
-    );
-    box.addChild(new Spacer(1));
-    box.addChild(new Text(theme.fg("customMessageText", message.content as string), 0, 0));
-    if (details?.images?.[0]) {
-      const first = details.images[0];
-      const match = /^data:[^;]+;base64,(.*)$/s.exec(first.dataUrl);
-      if (match?.[1]) {
-        box.addChild(new Spacer(1));
-        box.addChild(
-          new Text(theme.fg("customMessageText", `[image attached: ${first.mediaType}]`), 0, 0),
-        );
-      }
-    }
-    return box;
-  });
-
   pi.registerTool({
     name: IMAGE_GENERATION_TOOL_NAME,
     label: "Image Generation",
     description:
-      "Generate or edit an image through Radius using Gemini 2.5 Flash Image via OpenRouter.",
-    promptSnippet: "Generate or edit an image through Radius",
+      "Generate or edit an image through Radius using Gemini 2.5 Flash Image via OpenRouter. Generated images are written to disk and the tool returns their paths.",
+    promptSnippet: "Generate or edit an image through Radius and save it to disk",
     parameters: Type.Object(
       {
         prompt: Type.String({ description: "Image generation prompt" }),
+        output_path: Type.Optional(
+          Type.String({
+            minLength: 1,
+            description:
+              "File path for the generated image, relative to the working directory unless absolute. For multiple images, a numeric suffix is added. Defaults to generated-image-<tool-call-id>.<extension>.",
+          }),
+        ),
         input_images: Type.Optional(
           Type.Array(
             Type.Object(
@@ -118,7 +114,7 @@ export default async function (pi: ExtensionAPI) {
       { additionalProperties: false },
     ),
     async execute(
-      _toolCallId: string,
+      toolCallId: string,
       params: ImageGenerationToolInput,
       signal: AbortSignal | undefined,
       _onUpdate: unknown,
@@ -147,28 +143,95 @@ export default async function (pi: ExtensionAPI) {
         throw new Error(`${response.status} ${response.statusText}: ${await response.text()}`);
       }
       const payload = (await response.json()) as GatewayImageGenerationResponse;
-      const summary = `Generated ${payload.images.length} image(s) with ${payload.model}.`;
-      pi.sendMessage({
-        customType: IMAGE_GENERATION_MESSAGE_TYPE,
-        content: summary,
-        display: true,
-        details: payload,
-      });
-      const content: Array<any> = [{ type: "text", text: JSON.stringify(payload, null, 2) }];
-      for (const image of payload.images) {
-        const match = /^data:[^;]+;base64,(.*)$/s.exec(image.dataUrl);
-        if (match?.[1]) {
-          content.push({ type: "image", data: match[1], mimeType: image.mediaType });
-        }
-      }
+      const images = await saveGeneratedImages(payload, params.output_path, toolCallId, ctx.cwd);
+      const summary = [
+        `Generated ${images.length} ${images.length === 1 ? "image" : "images"} with ${payload.model}.`,
+        ...images.map((image) => `Saved ${image.mediaType} (${image.bytes} bytes) to ${image.path}`),
+      ].join("\n");
       return {
-        content,
-        details: { provider: provider.id, ...payload },
+        content: [{ type: "text", text: summary }],
+        details: {
+          provider: provider.id,
+          model: payload.model,
+          images,
+        },
       };
     },
   });
 }
 
+async function saveGeneratedImages(
+  payload: GatewayImageGenerationResponse,
+  requestedPath: string | undefined,
+  toolCallId: string,
+  cwd: string,
+): Promise<SavedImage[]> {
+  if (payload.images.length === 0) {
+    throw new Error("Radius returned no generated images");
+  }
+
+  const basePath = requestedPath
+    ? resolve(cwd, requestedPath.replace(/^@/u, ""))
+    : undefined;
+  const safeToolCallId = toolCallId.replace(/[^a-zA-Z0-9_-]/gu, "-").slice(0, 80) || Date.now();
+  const savedImages: SavedImage[] = [];
+
+  for (const [index, image] of payload.images.entries()) {
+    const data = decodeImageDataUrl(image.dataUrl);
+    const path = basePath
+      ? addImageIndex(basePath, index, payload.images.length)
+      : resolve(
+          cwd,
+          `generated-image-${safeToolCallId}${payload.images.length > 1 ? `-${index + 1}` : ""}${extensionForMediaType(image.mediaType)}`,
+        );
+
+    await withFileMutationQueue(path, async () => {
+      await mkdir(dirname(path), { recursive: true });
+      await writeFile(path, data);
+    });
+    savedImages.push({ path, mediaType: image.mediaType, bytes: data.length });
+  }
+
+  return savedImages;
+}
+
+function decodeImageDataUrl(dataUrl: string): Buffer {
+  const match = /^data:[^;,]+;base64,(.*)$/su.exec(dataUrl);
+  if (!match?.[1]) {
+    throw new Error("Radius returned an invalid image data URL");
+  }
+  return Buffer.from(match[1], "base64");
+}
+
+function addImageIndex(path: string, index: number, total: number): string {
+  if (total === 1) {
+    return path;
+  }
+  const extension = extname(path);
+  const stem = extension ? path.slice(0, -extension.length) : path;
+  return `${stem}-${index + 1}${extension}`;
+}
+
+function extensionForMediaType(mediaType: string): string {
+  switch (mediaType.toLowerCase()) {
+    case "image/avif":
+      return ".avif";
+    case "image/gif":
+      return ".gif";
+    case "image/jpeg":
+    case "image/jpg":
+      return ".jpg";
+    case "image/png":
+      return ".png";
+    case "image/svg+xml":
+      return ".svg";
+    case "image/webp":
+      return ".webp";
+    default:
+      return ".img";
+  }
+}
+
 function selectProvider(providers: GatewayProvider[], ctx: any): GatewayProvider {
   const provider = providers.find((candidate) => candidate.id === ctx.model?.provider);
   if (!provider) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@earendil-works/pi-radius-work",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "Radius Google Workspace and image generation extensions for Pi",
   "keywords": [
     "pi-package"