From f7ec116c263a905a66c9ad6dec7348b445bee486 Mon Sep 17 00:00:00 2001
From: Henry Li <henry1943@163.com>
Date: Thu, 29 Jan 2026 09:19:43 +0800
Subject: [PATCH] feat: enhance search_image

---
 .../workspace/messages/message-group.tsx      | 31 ++++++++++++++-----
 1 file changed, 23 insertions(+), 8 deletions(-)

diff --git a/frontend/src/components/workspace/messages/message-group.tsx b/frontend/src/components/workspace/messages/message-group.tsx
index e2cfcba..e0bc49c 100644
--- a/frontend/src/components/workspace/messages/message-group.tsx
+++ b/frontend/src/components/workspace/messages/message-group.tsx
@@ -222,23 +222,38 @@ function ToolCall({
     if (typeof args.query === "string") {
       label = t.toolCalls.searchForRelatedImagesFor(args.query);
     }
+    const results = (
+      result as {
+        results: {
+          source_url: string;
+          thumbnail_url: string;
+          image_url: string;
+          title: string;
+        }[];
+      }
+    )?.results;
     return (
       <ChainOfThoughtStep key={id} label={label} icon={SearchIcon}>
-        {Array.isArray(result) && (
+        {Array.isArray(results) && (
           <ChainOfThoughtSearchResults>
-            {result.map((item) => (
-              <ChainOfThoughtSearchResult key={item.url}>
-                <a href={item.url} target="_blank" rel="noreferrer">
+            {Array.isArray(results) &&
+              results.map((item) => (
+                <a
+                  className="size-24 overflow-hidden rounded-lg object-cover"
+                  key={item.image_url}
+                  href={item.source_url}
+                  target="_blank"
+                  rel="noreferrer"
+                >
                   <img
-                    className="size-24 rounded-md object-cover"
-                    src={item.url}
+                    className="size-24 object-cover"
+                    src={item.thumbnail_url}
                     alt={item.title}
                     width={100}
                     height={100}
                   />
                 </a>
-              </ChainOfThoughtSearchResult>
-            ))}
+              ))}
           </ChainOfThoughtSearchResults>
         )}
       </ChainOfThoughtStep>