Przeglądaj źródła

fix wrongly remove reset nodes (#20880)

Co-authored-by: zhuqingchao <zhuqingchao@xiaomi.com>
tags/1.5.0
ZalterCitty 4 miesięcy temu
rodzic
commit
6325129761
No account linked to committer's email address

+ 2
- 1
api/core/workflow/nodes/answer/base_stream_processor.py Wyświetl plik

@@ -57,7 +57,6 @@ class StreamProcessor(ABC):

# The branch_identify parameter is added to ensure that
# only nodes in the correct logical branch are included.
reachable_node_ids.append(edge.target_node_id)
ids = self._fetch_node_ids_in_reachable_branch(edge.target_node_id, run_result.edge_source_handle)
reachable_node_ids.extend(ids)
else:
@@ -74,6 +73,8 @@ class StreamProcessor(ABC):
self._remove_node_ids_in_unreachable_branch(node_id, reachable_node_ids)

def _fetch_node_ids_in_reachable_branch(self, node_id: str, branch_identify: Optional[str] = None) -> list[str]:
if node_id not in self.rest_node_ids:
self.rest_node_ids.append(node_id)
node_ids = []
for edge in self.graph.edge_mapping.get(node_id, []):
if edge.target_node_id == self.graph.root_node_id:

Ładowanie…
Anuluj
Zapisz