From a28b55b6ad0f12aaacf1b642aac69e6ac04767ef Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Tue, 19 Nov 2024 15:51:19 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=9B=BE=E7=89=87=E7=90=86=E8=A7=A3?=
 =?UTF-8?q?=E8=8A=82=E7=82=B9=E5=8F=AF=E4=BB=A5=E4=BD=9C=E4=B8=BA=E7=BB=93?=
 =?UTF-8?q?=E6=9D=9F=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

--bug=1049088 --user=刘瑞斌 【应用编排】图片理解模型应该可以作为结束节点 https://www.tapd.cn/57709429/s/1612298
---
 apps/application/flow/workflow_manage.py | 2 +-
 ui/src/workflow/common/validate.ts       | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/application/flow/workflow_manage.py b/apps/application/flow/workflow_manage.py
index d36a58b1f..b05bdea83 100644
--- a/apps/application/flow/workflow_manage.py
+++ b/apps/application/flow/workflow_manage.py
@@ -52,7 +52,7 @@ class Node:
             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
index 86f7d8af3..bbe96de3f 100644
--- 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 {