fix: add downstream nodes of this branch (langgenius#9640)

AceDataCloud · Nov 9, 2024 · 0d3a076 · 0d3a076
1 parent a23bf33
commit 0d3a076
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/api/core/workflow/nodes/answer/base_stream_processor.py b/api/core/workflow/nodes/answer/base_stream_processor.py
@@ -41,6 +41,7 @@ def _remove_unreachable_nodes(self, event: NodeRunSucceededEvent) -> None:
                     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)