From eff241f9f281f3c55dd0d415d2026884af1fe791 Mon Sep 17 00:00:00 2001
From: Henry Li <henry1943@163.com>
Date: Thu, 29 Jan 2026 12:51:04 +0800
Subject: [PATCH] feat: enable images in content

---
 backend/src/agents/lead_agent/prompt.py                       | 1 +
 frontend/src/components/workspace/messages/message-group.tsx  | 1 -
 .../src/components/workspace/messages/message-list-item.tsx   | 4 ++--
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/backend/src/agents/lead_agent/prompt.py b/backend/src/agents/lead_agent/prompt.py
index 4248d4c..13f7acf 100644
--- a/backend/src/agents/lead_agent/prompt.py
+++ b/backend/src/agents/lead_agent/prompt.py
@@ -128,6 +128,7 @@ You have access to skills that provide optimized workflows for specific tasks. E
 - Progressive Loading: Load resources incrementally as referenced in skills
 - Output Files: Final deliverables must be in `/mnt/user-data/outputs`
 - Clarity: Be direct and helpful, avoid unnecessary meta-commentary
+- Including Images and Mermaid: Images and Mermaid diagrams are always welcomed in the Markdown format, and you're encouraged to use `![Image Description](image_path)\n\n` or "```mermaid" to display images in response or Markdown files
 - Multi-task: Better utilize parallel tool calling to call multiple tools at one time for better performance
 - Language Consistency: Keep using the same language as user's
 - Always Respond: Your thinking is internal. You MUST always provide a visible response to the user after thinking.
diff --git a/frontend/src/components/workspace/messages/message-group.tsx b/frontend/src/components/workspace/messages/message-group.tsx
index 60a7f9a..5afc301 100644
--- a/frontend/src/components/workspace/messages/message-group.tsx
+++ b/frontend/src/components/workspace/messages/message-group.tsx
@@ -233,7 +233,6 @@ function ToolCall({
         }[];
       }
     )?.results;
-    console.info("image_search", results);
     return (
       <ChainOfThoughtStep key={id} label={label} icon={SearchIcon}>
         {Array.isArray(results) && (
diff --git a/frontend/src/components/workspace/messages/message-list-item.tsx b/frontend/src/components/workspace/messages/message-list-item.tsx
index 3caf6bd..cd184bf 100644
--- a/frontend/src/components/workspace/messages/message-list-item.tsx
+++ b/frontend/src/components/workspace/messages/message-list-item.tsx
@@ -87,7 +87,7 @@ function MessageContent_({
             if (typeof src !== "string") {
               return (
                 <img
-                  className="max-w-full overflow-hidden rounded-lg"
+                  className="max-w-[90%] overflow-hidden rounded-lg"
                   src={src}
                   alt={alt}
                 />
@@ -100,7 +100,7 @@ function MessageContent_({
             return (
               <a href={url} target="_blank" rel="noopener noreferrer">
                 <img
-                  className="max-w-full overflow-hidden rounded-lg"
+                  className="max-w-[90%] overflow-hidden rounded-lg"
                   src={url}
                   alt={alt}
                 />