Support data URLs for image prompts

Author: benbrandt

src/CodexAcpClient.ts

@@ -703,7 +703,7 @@ function buildPromptItems(prompt: acp.ContentBlock[]): UserInput[] {
             case "text":
                 return {type: "text", text: block.text, text_elements: []};
             case "image": {
-                const url = shouldUseImageUri(block.uri) ? block.uri : imageDataUrl(block);
+                const url = isSupportedImageUrl(block.uri) ? block.uri : imageDataUrl(block);
                 return {type: "image", url};
             }
             case "resource_link":
@@ -733,13 +733,13 @@ function imageDataUrl(block: acp.ContentBlock & { type: "image" }): string {
     return `data:${block.mimeType};base64,${block.data}`;
 }
 
-function shouldUseImageUri(uri: string | null | undefined): uri is string {
+function isSupportedImageUrl(uri: string | null | undefined): uri is string {
     if (!uri) {
         return false;
     }
     try {
         const protocol = new URL(uri).protocol;
-        return protocol === "http:" || protocol === "https:";
+        return protocol === "http:" || protocol === "https:" || protocol === "data:";
     } catch {
         return false;
     }

src/__tests__/CodexACPAgent/CodexAcpClient.test.ts

@@ -1430,6 +1430,24 @@ describe('ACP server test', { timeout: 40_000 }, () => {
         }));
     });
 
+    it ('should use inline image data when image uri is client-internal', async () => {
+        const { mockFixture, turnStartSpy } = setupPromptFixture({
+            supportedInputModalities: ["text", "image"],
+        });
+
+        const prompt: acp.ContentBlock[] = [
+            { type: "image", mimeType: "image/png", data: "abc123", uri: "zed:///agent/pasted-image?name=Image" },
+        ];
+
+        await mockFixture.getCodexAcpAgent().prompt({ sessionId: "id", prompt });
+
+        expect(turnStartSpy).toHaveBeenCalledWith(expect.objectContaining({
+            input: [
+                { type: "image", url: "data:image/png;base64,abc123" },
+            ]
+        }));
+    });
+
     it ('should show rate limits from multiple sources in status', async () => {
         const rateLimits: RateLimitsMap = new Map();
         rateLimits.set("limit-1", {