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

🐛 (chat.py): fix issue where end_all_traces method was not being called correctly with the required arguments

Browse source 
💡 (chat.py): improve error handling by passing error message to end_all_traces method and ensure it is called with the correct arguments
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 17:59:39 -03:00
commit 3cba1380de
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -202,7 +202,7 @@ async def build_vertex(
logs = {output_label: Log(message=message, type="error")} logs = {output_label: Log(message=message, type="error")}
result_data_response = ResultDataResponse(results={}, logs=logs) result_data_response = ResultDataResponse(results={}, logs=logs)
artifacts = {} artifacts = {}
graph.end_all_traces(outputs=message) background_tasks.add_task(graph.end_all_traces, error=message["errorMessage"])
# If there's an error building the vertex # If there's an error building the vertex
# we need to clear the cache # we need to clear the cache
await chat_service.clear_cache(flow_id_str) await chat_service.clear_cache(flow_id_str)
@ -240,7 +240,7 @@ async def build_vertex(
next_runnable_vertices = [graph.stop_vertex] next_runnable_vertices = [graph.stop_vertex]
if not next_runnable_vertices: if not next_runnable_vertices:
graph.end_all_traces() background_tasks.add_task(graph.end_all_traces)
build_response = VertexBuildResponse( build_response = VertexBuildResponse(
inactivated_vertices=inactivated_vertices, inactivated_vertices=inactivated_vertices,