diff --git a/ui/src/ui/chat/grouped-render.ts b/ui/src/ui/chat/grouped-render.ts index 4a9ccec14..a12029bb6 100644 --- a/ui/src/ui/chat/grouped-render.ts +++ b/ui/src/ui/chat/grouped-render.ts @@ -39,6 +39,14 @@ function extractImages(message: unknown): ImageBlock[] { ? data : `data:${mediaType};base64,${data}`; images.push({ url }); + } else if (typeof b.data === "string" && typeof b.mimeType === "string") { + // Handle tool result format: { type: "image", data: "base64...", mimeType: "image/png" } + const data = b.data as string; + const mimeType = b.mimeType as string; + const url = data.startsWith("data:") + ? data + : `data:${mimeType};base64,${data}`; + images.push({ url }); } else if (typeof b.url === "string") { images.push({ url: b.url }); }