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

fix(base.py): only add successors if is_start (#2513)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-07-04 09:25:16 -03:00 • committed by GitHub
commit bf5e7b343a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1236,9 +1236,11 @@ class Graph:
stack.append(successor.id) stack.append(successor.id)
else: else:
excluded.add(successor.id) excluded.add(successor.id)
elif current_id not in stop_predecessors: elif current_id not in stop_predecessors and is_start:
# If the current vertex is not the target vertex, we should add all its successors # If the current vertex is not the target vertex, we should add all its successors
# to the stack if they are not in visited # to the stack if they are not in visited
# If we are starting from the beginning, we should add all successors
for successor in current_vertex.successors: for successor in current_vertex.successors:
if successor.id not in visited: if successor.id not in visited:
stack.append(successor.id) stack.append(successor.id)