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: ,
description: '使用大模型对话',
component: LlmNode,
checkers: [inputVariableChecker],
@@ -52,6 +55,7 @@ export const NodeRegistry: NodeDefine[] = [
key: 'knowledge-node',
group: '普通节点',
name: '知识库',
+ icon: ,
description: '',
component: KnowledgeNode,
checkers: [inputVariableChecker],
@@ -60,6 +64,7 @@ export const NodeRegistry: NodeDefine[] = [
key: 'code-node',
group: '普通节点',
name: '代码执行',
+ icon: ,
description: '执行自定义的处理代码',
component: CodeNode,
checkers: [inputVariableChecker],
@@ -68,6 +73,7 @@ export const NodeRegistry: NodeDefine[] = [
key: 'switch-node',
group: '逻辑节点',
name: '分支',
+ icon: ,
description: '根据不同的情况前往不同的分支',
component: SwitchNode,
checkers: [],
@@ -76,6 +82,7 @@ export const NodeRegistry: NodeDefine[] = [
key: 'output-node',
group: '输出节点',
name: '输出',
+ icon: ,
description: '定义输出变量',
component: OutputNode,
checkers: [inputVariableChecker],