From 40d51d1293c29e1833855c104e693aabac21f074 Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Sat, 21 Feb 2026 03:29:18 +0800
Subject: [PATCH] fix(agent-ui): show avatar in canvas mode and remove
 duplicate loading copy

---
 components/canvas/CanvasChatPanel.tsx   | 18 ++++++++++++++++++
 components/image-generation-loading.tsx |  4 ----
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/components/canvas/CanvasChatPanel.tsx b/components/canvas/CanvasChatPanel.tsx
index 2bfb06ce8..f35f182f7 100644
--- a/components/canvas/CanvasChatPanel.tsx
+++ b/components/canvas/CanvasChatPanel.tsx
@@ -6060,6 +6060,24 @@ export function CanvasChatPanel({ chatId, initialMessages = [], onClose, onOpenF
             </div>
 
             <div className="flex items-center gap-2 flex-shrink-0">
+              {activeComposerMode === "agent" ? (
+                <div
+                  className="h-8 w-8 overflow-hidden rounded-full border border-[#cfd4dc] bg-white flex items-center justify-center"
+                  title={selectDisplayModel?.displayName || "Agent模式"}
+                  aria-label={selectDisplayModel?.displayName || "Agent模式"}
+                >
+                  {selectDisplayModel?.avatarUrl ? (
+                    <img
+                      src={selectDisplayModel.avatarUrl}
+                      alt={selectDisplayModel.displayName || "Agent模式"}
+                      className="h-full w-full object-cover"
+                    />
+                  ) : (
+                    <Sparkles className="h-4 w-4 text-[#8b95a7]" />
+                  )}
+                </div>
+              ) : null}
+
               {activeComposerMode === "image" ? (
                 <Button
                   variant="ghost"
diff --git a/components/image-generation-loading.tsx b/components/image-generation-loading.tsx
index f8468a23e..fdde2239e 100644
--- a/components/image-generation-loading.tsx
+++ b/components/image-generation-loading.tsx
@@ -61,10 +61,6 @@ export const ImageGenerationLoading = memo(
             <span className="font-medium">开始生成</span>
           </div>
 
-          <div className="text-[15px] leading-7 text-[#3a4352]">
-            现在开始生成设计图。
-          </div>
-
           <div className="flex items-center gap-2 text-[#a8b1bf]">
             <CircleDot className="h-3.5 w-3.5 fill-current" />
             <span className="text-[14px]">{resolvedModelName}</span>