jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add time_elapsed variable to track the elapsed time during stream_build()

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-13 18:10:48 -03:00
commit d243cfd98f
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/api/v1/chat.py
View file

@ -142,7 +142,7 @@ async def stream_build(
number_of_nodes = len(graph.vertices) number_of_nodes = len(graph.vertices)
update_build_status(cache_service, flow_id, BuildStatus.IN_PROGRESS) update_build_status(cache_service, flow_id, BuildStatus.IN_PROGRESS)
time_elapsed = ""
try: try:
user_id = flow_cache["user_id"] user_id = flow_cache["user_id"]
except KeyError: except KeyError:
@ -175,8 +175,10 @@ async def stream_build(
logger.exception(exc) logger.exception(exc)
params = str(exc) params = str(exc)
valid = False valid = False
time_elapsed = format_elapsed_time(time.perf_counter() - start_time)
update_build_status(cache_service, flow_id, BuildStatus.FAILURE) update_build_status(cache_service, flow_id, BuildStatus.FAILURE)
vertex_id = vertex.parent_node_id if vertex.parent_is_top_level else vertex.id vertex_id = vertex.parent_node_id if vertex.parent_is_top_level else vertex.id
if vertex_id in graph.top_level_vertices: if vertex_id in graph.top_level_vertices:
response = { response = {