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],