From ff3ecb36d4ba128def67ccbefa732e162efcf260 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Tue, 9 Dec 2025 18:17:41 +0800
Subject: [PATCH] fix: Knowledge Base Workflow - The icon of the data source
 tool node in the variable selection drop-down box does not correspond to it
 (#4475)

---
 ui/src/workflow/common/app-node.ts           | 8 ++++++--
 ui/src/workflow/icons/tool-lib-node-icon.vue | 5 ++++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/ui/src/workflow/common/app-node.ts b/ui/src/workflow/common/app-node.ts
index 8b9bedc0a..1894d1831 100644
--- a/ui/src/workflow/common/app-node.ts
+++ b/ui/src/workflow/common/app-node.ts
@@ -93,13 +93,17 @@ class AppNode extends HtmlResize.view {
         children: globalFields,
       })
     }
-    result.push({
+    const value: any = {
       value: this.props.model.id,
       icon: this.props.model.properties.node_data?.icon,
       label: this.props.model.properties.stepName,
       type: this.props.model.type,
       children: this.props.model.properties?.config?.fields || [],
-    })
+    }
+    if (this.props.model.properties.kind) {
+      value['kind'] = this.props.model.properties.kind
+    }
+    result.push(value)
     return result
   }
   get_up_node_field_dict(contain_self: boolean, use_cache: boolean) {
diff --git a/ui/src/workflow/icons/tool-lib-node-icon.vue b/ui/src/workflow/icons/tool-lib-node-icon.vue
index 027b4e308..a208ec38b 100644
--- a/ui/src/workflow/icons/tool-lib-node-icon.vue
+++ b/ui/src/workflow/icons/tool-lib-node-icon.vue
@@ -8,7 +8,9 @@
   >
     <img :src="item?.icon" alt="" />
   </el-avatar>
-
+  <el-avatar v-else-if="item?.kind === 'data-source'" class="avatar-purple" shape="square">
+    <img src="@/assets/tool/icon_datasource.svg" style="width: 58%" alt="" />
+  </el-avatar>
   <el-avatar v-else-if="item?.tool_type === 'DATA_SOURCE'" class="avatar-purple" shape="square">
     <img src="@/assets/tool/icon_datasource.svg" style="width: 58%" alt="" />
   </el-avatar>
@@ -23,6 +25,7 @@ const props = defineProps<{
     name: string
     icon: string
     tool_type: string
+    kind?: string
   }
 }>()
 </script>