Skip to content

Commit 2dee353

Browse files
authored
feat: Variable assignment nodes support ending nodes (#4247)
1 parent a66e636 commit 2dee353

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

apps/application/flow/common.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@
2020
from tools.models.tool import Tool
2121

2222
end_nodes = ['ai-chat-node', 'reply-node', 'function-node', 'function-lib-node', 'application-node',
23-
'image-understand-node', 'speech-to-text-node', 'text-to-speech-node', 'image-generate-node']
23+
'image-understand-node', 'speech-to-text-node', 'text-to-speech-node', 'image-generate-node',
24+
'variable-assign-node']
2425

2526

2627
class Answer:

ui/src/workflow/common/validate.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
import {WorkflowType, WorkflowMode} from '@/enums/application'
1+
import { WorkflowType, WorkflowMode } from '@/enums/application'
22

3-
import {t} from '@/locales'
3+
import { t } from '@/locales'
44

55
const end_nodes: Array<string> = [
66
WorkflowType.AiChat,
@@ -18,7 +18,8 @@ const end_nodes: Array<string> = [
1818
WorkflowType.LoopBodyNode,
1919
WorkflowType.LoopNode,
2020
WorkflowType.LoopBreakNode,
21-
WorkflowType.VideoUnderstandNode
21+
WorkflowType.VideoUnderstandNode,
22+
WorkflowType.VariableAssignNode,
2223
]
2324

2425
const loop_end_nodes: Array<string> = [

0 commit comments

Comments
 (0)