From 999d3f1539f49ebb9a7e9b7353acfd5988db6be3 Mon Sep 17 00:00:00 2001 From: feiyang_deepnova <736320652@qq.com> Date: Wed, 23 Oct 2024 01:20:02 +0800 Subject: [PATCH] fix: add downstream nodes of this branch (#9640) --- api/core/workflow/nodes/answer/base_stream_processor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/core/workflow/nodes/answer/base_stream_processor.py b/api/core/workflow/nodes/answer/base_stream_processor.py index 36c3fe180a..52d0358c76 100644 --- a/api/core/workflow/nodes/answer/base_stream_processor.py +++ b/api/core/workflow/nodes/answer/base_stream_processor.py @@ -41,6 +41,7 @@ class StreamProcessor(ABC): continue else: unreachable_first_node_ids.append(edge.target_node_id) + unreachable_first_node_ids.extend(self._fetch_node_ids_in_reachable_branch(edge.target_node_id)) for node_id in unreachable_first_node_ids: self._remove_node_ids_in_unreachable_branch(node_id, reachable_node_ids)