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

Refactor build_vertex function to reset both inactivated and activated vertices

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-01 23:56:35 -03:00
commit 94a00c0c5e
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -154,13 +154,15 @@ async def build_vertex(
result_data_response.timedelta = timedelta
vertex.add_build_time(timedelta)
inactivated_vertices = None
if graph.inactivated_vertices:
inactivated_vertices = list(graph.inactivated_vertices)
graph.reset_inactivated_vertices()
inactivated_vertices = list(graph.inactivated_vertices)
graph.reset_inactivated_vertices()
activated_vertices = list(graph.activated_vertices)
graph.reset_activated_vertices()
chat_service.set_cache(flow_id, graph)
build_response = VertexBuildResponse(
inactivated_vertices=inactivated_vertices,
activated_vertices=activated_vertices,
valid=valid,
params=params,
id=vertex.id,