From d243cfd98fe2e9e5f55a7761628cb599b1ea6129 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 13 Dec 2023 18:10:48 -0300 Subject: [PATCH] Add time_elapsed variable to track the elapsed time during stream_build() --- src/backend/langflow/api/v1/chat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py index 26765623a..c034c2676 100644 --- a/src/backend/langflow/api/v1/chat.py +++ b/src/backend/langflow/api/v1/chat.py @@ -142,7 +142,7 @@ async def stream_build( number_of_nodes = len(graph.vertices) update_build_status(cache_service, flow_id, BuildStatus.IN_PROGRESS) - + time_elapsed = "" try: user_id = flow_cache["user_id"] except KeyError: @@ -175,8 +175,10 @@ async def stream_build( logger.exception(exc) params = str(exc) valid = False + time_elapsed = format_elapsed_time(time.perf_counter() - start_time) update_build_status(cache_service, flow_id, BuildStatus.FAILURE) + vertex_id = vertex.parent_node_id if vertex.parent_is_top_level else vertex.id if vertex_id in graph.top_level_vertices: response = {