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

langgenius · Oct 22, 2024 · 999d3f1 · 999d3f1
1 parent a7ee51e
commit 999d3f1
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)