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

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

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