From 5895772485eb97d8eaf1eb0a3a4508956672e52a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 8 Jul 2024 20:37:07 -0300 Subject: [PATCH] fix: add vertices in the first layer to the vertices being run --- 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 7e933d1dfd52..61155c43a165 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