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

2 years ago · 999d3f1539
parent a7ee51e5d8
commit 999d3f1539
1 changed files with 1 additions and 0 deletions
--- 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)