From 2dee3532ff1bec2b060c119f84d26dd62fb425d0 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Fri, 24 Oct 2025 11:59:08 +0800
Subject: [PATCH] feat: Variable assignment nodes support ending nodes (#4247)

---
 apps/application/flow/common.py    | 3 ++-
 ui/src/workflow/common/validate.ts | 7 ++++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/apps/application/flow/common.py b/apps/application/flow/common.py
index 98f706fde..2ba944875 100644
--- a/apps/application/flow/common.py
+++ b/apps/application/flow/common.py
@@ -20,7 +20,8 @@ from models_provider.tools import get_model_credential
 from tools.models.tool import Tool
 
 end_nodes = ['ai-chat-node', 'reply-node', 'function-node', 'function-lib-node', 'application-node',
-             'image-understand-node', 'speech-to-text-node', 'text-to-speech-node', 'image-generate-node']
+             'image-understand-node', 'speech-to-text-node', 'text-to-speech-node', 'image-generate-node',
+             'variable-assign-node']
 
 
 class Answer:
diff --git a/ui/src/workflow/common/validate.ts b/ui/src/workflow/common/validate.ts
index 3d781af9e..50e7e94e9 100644
--- a/ui/src/workflow/common/validate.ts
+++ b/ui/src/workflow/common/validate.ts
@@ -1,6 +1,6 @@
-import {WorkflowType, WorkflowMode} from '@/enums/application'
+import { WorkflowType, WorkflowMode } from '@/enums/application'
 
-import {t} from '@/locales'
+import { t } from '@/locales'
 
 const end_nodes: Array<string> = [
   WorkflowType.AiChat,
@@ -18,7 +18,8 @@ const end_nodes: Array<string> = [
   WorkflowType.LoopBodyNode,
   WorkflowType.LoopNode,
   WorkflowType.LoopBreakNode,
-  WorkflowType.VideoUnderstandNode
+  WorkflowType.VideoUnderstandNode,
+  WorkflowType.VariableAssignNode,
 ]
 
 const loop_end_nodes: Array<string> = [