Skip to content

Commit

Permalink
fix(base.py): only add successors if is_start (#2513)
Browse files Browse the repository at this point in the history
  • Loading branch information
ogabrielluiz authored Jul 4, 2024
1 parent 15aa68a commit bf5e7b3
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/backend/base/langflow/graph/graph/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -1236,9 +1236,11 @@ def get_successors(vertex, recursive=True):
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)
Expand Down

0 comments on commit bf5e7b3

Please sign in to comment.