From f6e9c9bc708c6681e18061962d0b4639fd58b95f Mon Sep 17 00:00:00 2001
From: lanyuanxiaoyao <lanyuanxiaoyao@gmail.com>
Date: Sun, 13 Jul 2025 22:24:40 +0800
Subject: [PATCH] =?UTF-8?q?fix(web):=20=E4=BF=AE=E5=A4=8D=E6=A8=A1?=
 =?UTF-8?q?=E6=9D=BF=E8=8A=82=E7=82=B9=E6=B2=A1=E6=9C=89=E8=BE=93=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/components/flow/node/TemplateNode.tsx | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/service-web/client/src/components/flow/node/TemplateNode.tsx b/service-web/client/src/components/flow/node/TemplateNode.tsx
index bbe243b..22123d6 100644
--- a/service-web/client/src/components/flow/node/TemplateNode.tsx
+++ b/service-web/client/src/components/flow/node/TemplateNode.tsx
@@ -1,10 +1,10 @@
 import type {NodeProps} from '@xyflow/react'
 import {Tag} from 'antd'
-import React, {useCallback} from 'react'
+import React, {useCallback, useEffect} from 'react'
 import {useContextStore} from '../store/ContextStore.ts'
 import {useDataStore} from '../store/DataStore.ts'
 import {useFlowStore} from '../store/FlowStore.ts'
-import AmisNode, {EndNodeHandler, inputsFormColumns} from './AmisNode.tsx'
+import AmisNode, {inputsFormColumns, NormalNodeHandler, outputsFormColumns} from './AmisNode.tsx'
 
 const typeMap: Record<string, string> = {
   default: '默认',
@@ -15,11 +15,24 @@ const typeMap: Record<string, string> = {
 
 const TemplateNode = (props: NodeProps) => {
   const {getNodes, getEdges} = useFlowStore()
-  const {getData, getDataById} = useDataStore()
+  const {getData, getDataById, mergeDataById} = useDataStore()
   const {getInputSchema} = useContextStore()
 
   const nodeData = getDataById(props.id)
 
+  useEffect(() => {
+    mergeDataById(
+      props.id,
+      {
+        outputs: {
+          text: {
+            type: 'string',
+          },
+        },
+      },
+    )
+  }, [props.id])
+
   const columnsSchema = useCallback(
     () => [
       ...inputsFormColumns(props.id, getInputSchema(), getNodes(), getEdges(), getData()),
@@ -55,6 +68,7 @@ const TemplateNode = (props: NodeProps) => {
           min_height: 500,
         },
       },
+      ...outputsFormColumns(false, true),
     ],
     [props.id],
   )
@@ -71,7 +85,7 @@ const TemplateNode = (props: NodeProps) => {
           : <></>
       }
       columnSchema={columnsSchema}
-      handler={<EndNodeHandler/>}
+      handler={<NormalNodeHandler/>}
     />
   )
 }