From fe35ba3cdc44930d52ca8124c8150a1f294d2bc5 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Fri, 29 Aug 2025 17:12:02 +0800
Subject: [PATCH] fix: update MCP tool icon rendering in index.vue

---
 ui/src/workflow/nodes/ai-chat-node/index.vue | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/ui/src/workflow/nodes/ai-chat-node/index.vue b/ui/src/workflow/nodes/ai-chat-node/index.vue
index e5ed9234e..2b8e23509 100644
--- a/ui/src/workflow/nodes/ai-chat-node/index.vue
+++ b/ui/src/workflow/nodes/ai-chat-node/index.vue
@@ -139,11 +139,20 @@
         >
           <div class="flex-between border border-r-6 white-bg mb-4" style="padding: 5px 8px">
             <div class="flex align-center" style="line-height: 20px">
-              <ToolIcon type="MCP" class="mr-8" :size="20" />
+              <el-avatar
+                v-if="relatedObject(mcpToolSelectOptions, chat_data.mcp_tool_id, 'id')?.icon"
+                shape="square"
+                :size="20"
+                style="background: none"
+                class="mr-8"
+              >
+                <img :src="resetUrl(relatedObject(mcpToolSelectOptions, chat_data.mcp_tool_id, 'id')?.icon)" alt="" />
+              </el-avatar>
+              <ToolIcon v-else type="MCP" class="mr-8" :size="20" />
 
               <div
                 class="ellipsis"
-                :title="relatedObject(toolSelectOptions, chat_data.mcp_tool_id, 'id')?.name"
+                :title="relatedObject(mcpToolSelectOptions, chat_data.mcp_tool_id, 'id')?.name"
               >
                 {{
                   relatedObject(mcpToolSelectOptions, chat_data.mcp_tool_id, 'id')?.name ||