Browse Source

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

feiyang_deepnova 6 months ago
parent
commit
999d3f1539
1 changed files with 1 additions and 0 deletions
  1. 1 0
      api/core/workflow/nodes/answer/base_stream_processor.py

+ 1 - 0
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)