fix: 图片理解节点可以作为结束节点

--bug=1049088 --user=刘瑞斌【应用编排】图片理解模型应该可以作为结束节点 https://www.tapd.cn/57709429/s/1612298
1Panel-dev · Nov 19, 2024 · a28b55b · a28b55b
1 parent 56a09d7
commit a28b55b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/apps/application/flow/workflow_manage.py b/apps/application/flow/workflow_manage.py
@@ -52,7 +52,7 @@ def __init__(self, _id: str, _type: str, x: int, y: int, properties: dict, **kwa
             self.__setattr__(keyword, kwargs.get(keyword))
 
 
-end_nodes = ['ai-chat-node', 'reply-node', 'function-node', 'function-lib-node', 'application-node']
+end_nodes = ['ai-chat-node', 'reply-node', 'function-node', 'function-lib-node', 'application-node', 'image-understand-node']
 
 
 class Flow:

diff --git a/ui/src/workflow/common/validate.ts b/ui/src/workflow/common/validate.ts
@@ -5,6 +5,7 @@ const end_nodes: Array<string> = [
   WorkflowType.Reply,
   WorkflowType.FunctionLib,
   WorkflowType.FunctionLibCustom,
+  WorkflowType.ImageUnderstandNode,
   WorkflowType.Application
 ]
 export class WorkFlowInstance {