pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-23 22:44:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 15aa68a342
commit bf5e7b343a
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)
else:
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
# 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:
if successor.id not in visited:
stack.append(successor.id)