From c20630b77d5221000c163020e2c08b152b2cbffa Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 9 Jul 2024 12:03:55 -0300 Subject: [PATCH] fix: add vertices in the first layer to the vertices being run (#2594) --- src/backend/base/langflow/api/v1/chat.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/base/langflow/api/v1/chat.py b/src/backend/base/langflow/api/v1/chat.py index 7e933d1df..61155c43a 100644 --- a/src/backend/base/langflow/api/v1/chat.py +++ b/src/backend/base/langflow/api/v1/chat.py @@ -106,6 +106,9 @@ async def retrieve_vertices_order( else: first_layer = graph.sort_vertices() + for vertex_id in first_layer: + graph.run_manager.add_to_vertices_being_run(vertex_id) + # Now vertices is a list of lists # We need to get the id of each vertex # and return the same structure but only with the ids