From 5b9920449ded5718e8e5f1234c7374c0610784d2 Mon Sep 17 00:00:00 2001
From: v-zhangjc9 <v-zhangjc9.gd@chinatelecom.cn>
Date: Sat, 12 Jul 2025 20:46:12 +0800
Subject: [PATCH] =?UTF-8?q?feat(web):=20=E5=A2=9E=E5=8A=A0=E8=8A=82?=
 =?UTF-8?q?=E7=82=B9=E5=9B=BE=E6=A0=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service-web/client/src/components/flow/FlowEditor.tsx   | 2 +-
 service-web/client/src/components/flow/NodeRegistry.tsx | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/service-web/client/src/components/flow/FlowEditor.tsx b/service-web/client/src/components/flow/FlowEditor.tsx
index cd72673..0cf292f 100644
--- a/service-web/client/src/components/flow/FlowEditor.tsx
+++ b/service-web/client/src/components/flow/FlowEditor.tsx
@@ -93,7 +93,7 @@ function FlowEditor(props: FlowEditorProps) {
                 type: 'group',
                 label: group,
                 children: NodeRegistry.filter(i => isEqual(group, i.group))
-                  .map(i => ({key: i.key, label: i.name}))
+                  .map(i => ({key: i.key, label: i.name, icon: i.icon})),
               })),
             onClick: ({key}) => {
               try {
diff --git a/service-web/client/src/components/flow/NodeRegistry.tsx b/service-web/client/src/components/flow/NodeRegistry.tsx
index 3f280bd..8ad892a 100644
--- a/service-web/client/src/components/flow/NodeRegistry.tsx
+++ b/service-web/client/src/components/flow/NodeRegistry.tsx
@@ -1,4 +1,5 @@
 import {has, isEmpty} from 'licia'
+import type {JSX} from 'react'
 import {getAllIncomerNodeOutputVariables} from './Helper.tsx'
 import CodeNode from './node/CodeNode.tsx'
 import KnowledgeNode from './node/KnowledgeNode.tsx'
@@ -34,6 +35,7 @@ type NodeDefine = {
   key: string,
   group: string,
   name: string,
+  icon: JSX.Element,
   description: string,
   component: any,
   checkers: NodeChecker[],
@@ -44,6 +46,7 @@ export const NodeRegistry: NodeDefine[] = [
     key: 'llm-node',
     group: '普通节点',
     name: '大模型',
+    icon: <i className="fa fa-message"/>,
     description: '使用大模型对话',
     component: LlmNode,
     checkers: [inputVariableChecker],
@@ -52,6 +55,7 @@ export const NodeRegistry: NodeDefine[] = [
     key: 'knowledge-node',
     group: '普通节点',
     name: '知识库',
+    icon: <i className="fa fa-book-bookmark"/>,
     description: '',
     component: KnowledgeNode,
     checkers: [inputVariableChecker],
@@ -60,6 +64,7 @@ export const NodeRegistry: NodeDefine[] = [
     key: 'code-node',
     group: '普通节点',
     name: '代码执行',
+    icon: <i className="fa fa-code"/>,
     description: '执行自定义的处理代码',
     component: CodeNode,
     checkers: [inputVariableChecker],
@@ -68,6 +73,7 @@ export const NodeRegistry: NodeDefine[] = [
     key: 'switch-node',
     group: '逻辑节点',
     name: '分支',
+    icon: <i className="fa fa-code-fork"/>,
     description: '根据不同的情况前往不同的分支',
     component: SwitchNode,
     checkers: [],
@@ -76,6 +82,7 @@ export const NodeRegistry: NodeDefine[] = [
     key: 'output-node',
     group: '输出节点',
     name: '输出',
+    icon: <i className="fa fa-file"/>,
     description: '定义输出变量',
     component: OutputNode,
     checkers: [inputVariableChecker],