refactor(web): 优化流程节点的定义和实现

service-web/client/src/components/flow/node/LlmNode.tsx

@@ -1,51 +1,61 @@
 import type {NodeProps} from '@xyflow/react'
 import {Tag} from 'antd'
-import AmisNode, {inputsFormColumns, outputsFormColumns} from './AmisNode.tsx'
 import React from 'react'
+import {useContextStore} from '../store/ContextStore.ts'
+import {useDataStore} from '../store/DataStore.ts'
+import {useFlowStore} from '../store/FlowStore.ts'
+import AmisNode, {inputsFormColumns, NormalNodeHandler, outputsFormColumns} from './AmisNode.tsx'
 
 const modelMap: Record<string, string> = {
   qwen3: 'Qwen3',
   deepseek: 'Deepseek',
 }
 
-const LlmNode = (props: NodeProps) => AmisNode({
-  nodeProps: props,
-  type: 'normal',
-  defaultNodeName: '大模型',
-  defaultNodeDescription: '使用大模型对话',
-  extraNodeDescription: nodeData => {
-    const model = nodeData?.model as string | undefined
-    return model
-      ? <div className="mt-2 flex justify-between">
-        <span>大模型</span>
-        <Tag className="m-0" color="blue">{modelMap[model]}</Tag>
-      </div>
-      : <></>
-  },
-  columnSchema: (props) => [
-    ...inputsFormColumns(props),
-    {
-      type: 'divider',
-    },
-    {
-      type: 'select',
-      name: 'model',
-      label: '大模型',
-      required: true,
-      selectFirst: true,
-      options: Object.keys(modelMap).map(key => ({label: modelMap[key], value: key})),
-    },
-    {
-      type: 'textarea',
-      name: 'systemPrompt',
-      label: '系统提示词',
-      required: true,
-    },
-    {
-      type: 'divider',
-    },
-    ...outputsFormColumns(false, true, {text: {type: 'string'}}),
-  ],
-})
+const LlmNode = (props: NodeProps) => {
+  const {getNodes, getEdges} = useFlowStore()
+  const {getData} = useDataStore()
+  const {getInputSchema} = useContextStore()
+  return (
+    <AmisNode
+      nodeProps={props}
+      defaultNodeName="大模型"
+      defaultNodeDescription="使用大模型对话"
+      extraNodeDescription={nodeData => {
+        const model = nodeData?.model as string | undefined
+        return model
+          ? <div className="mt-2 flex justify-between">
+            <span>大模型</span>
+            <Tag className="m-0" color="blue">{modelMap[model]}</Tag>
+          </div>
+          : <></>
+      }}
+      columnSchema={() => [
+        ...inputsFormColumns(props.id, getInputSchema(), getNodes(), getEdges(), getData()),
+        {
+          type: 'divider',
+        },
+        {
+          type: 'select',
+          name: 'model',
+          label: '大模型',
+          required: true,
+          selectFirst: true,
+          options: Object.keys(modelMap).map(key => ({label: modelMap[key], value: key})),
+        },
+        {
+          type: 'textarea',
+          name: 'systemPrompt',
+          label: '系统提示词',
+          required: true,
+        },
+        {
+          type: 'divider',
+        },
+        ...outputsFormColumns(false, true, {text: {type: 'string'}}),
+      ]}
+      handler={<NormalNodeHandler/>}
+    />
+  )
+}
 
 export default React.memo(LlmNode)