From f01602b5703f0daad62b781dec91fce72cf784d9 Mon Sep 17 00:00:00 2001
From: takatost <takatost@users.noreply.github.com>
Date: Sat, 14 Sep 2024 14:02:09 +0800
Subject: [PATCH] fix(workflow): the answer node after the iteration node
 containing the answer was output prematurely (#8419)

---
 api/core/workflow/nodes/answer/answer_stream_generate_router.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/api/core/workflow/nodes/answer/answer_stream_generate_router.py b/api/core/workflow/nodes/answer/answer_stream_generate_router.py
index 5e6de8fb15..bbd1f88867 100644
--- a/api/core/workflow/nodes/answer/answer_stream_generate_router.py
+++ b/api/core/workflow/nodes/answer/answer_stream_generate_router.py
@@ -152,6 +152,7 @@ class AnswerStreamGeneratorRouter:
                 NodeType.ANSWER.value,
                 NodeType.IF_ELSE.value,
                 NodeType.QUESTION_CLASSIFIER.value,
+                NodeType.ITERATION.value,
             }:
                 answer_dependencies[answer_node_id].append(source_node_id)
             else: