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

Fix issue with next_vertices_ids in build_vertex function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-03 18:10:43 -03:00
commit 2861391e6e
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -124,7 +124,7 @@ async def build_vertex(
artifacts = vertex.artifacts artifacts = vertex.artifacts
else: else:
raise ValueError(f"No result found for vertex {vertex_id}") raise ValueError(f"No result found for vertex {vertex_id}")
next_vertices_ids = vertex.successors_ids
result_data_response = ResultDataResponse(**result_dict.model_dump()) result_data_response = ResultDataResponse(**result_dict.model_dump())
except Exception as exc: except Exception as exc:
@ -160,6 +160,13 @@ async def build_vertex(
graph.reset_inactive_vertices() graph.reset_inactive_vertices()
chat_service.set_cache(flow_id, graph) chat_service.set_cache(flow_id, graph)
# graph.stop_vertex tells us if the user asked
# to stop the build of the graph at a certain vertex
# if it is in next_vertices_ids, we need to remove other
# vertices from next_vertices_ids
if graph.stop_vertex and graph.stop_vertex in next_vertices_ids:
next_vertices_ids = [graph.stop_vertex]
build_response = VertexBuildResponse( build_response = VertexBuildResponse(
next_vertices_ids=next_vertices_ids, next_vertices_ids=next_vertices_ids,
inactive_vertices=inactive_vertices, inactive_vertices=inactive_vertices,