jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add successors to stack if not visited

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-22 15:35:50 -03:00
commit d0401fec45
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/graph/graph/base.py
View file

@ -911,6 +911,12 @@ class Graph:
stack.append(successor.id) stack.append(successor.id)
else: else:
excluded.add(successor.id) excluded.add(successor.id)
else:
# If the current vertex is not the target vertex, we should add all its successors
# to the stack if they are not in visited
for successor in current_vertex.successors:
if successor.id not in visited:
stack.append(successor.id)
# Filter the original graph's vertices and edges to keep only those in `visited` # Filter the original graph's vertices and edges to keep only those in `visited`
vertices_to_keep = [self.get_vertex(vid) for vid in visited] vertices_to_keep = [self.get_vertex(vid) for vid in visited]