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

fix: update condition to run end_all_traces (#2707)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-07-15 13:13:49 -03:00 • committed by GitHub
commit 537e358b65
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -260,7 +260,7 @@ async def build_vertex(
if graph.stop_vertex and graph.stop_vertex in next_runnable_vertices: if graph.stop_vertex and graph.stop_vertex in next_runnable_vertices:
next_runnable_vertices = [graph.stop_vertex] next_runnable_vertices = [graph.stop_vertex]
if not graph.run_manager.vertices_to_run and not next_runnable_vertices: if graph.run_manager.all_predecessors_are_fulfilled() and not next_runnable_vertices:
background_tasks.add_task(graph.end_all_traces) background_tasks.add_task(graph.end_all_traces)
build_response = VertexBuildResponse( build_response = VertexBuildResponse(

2
src/backend/base/langflow/graph/graph/base.py
View file

@ -539,6 +539,8 @@ class Graph:
"""Marks a vertex in the graph.""" """Marks a vertex in the graph."""
vertex = self.get_vertex(vertex_id) vertex = self.get_vertex(vertex_id)
vertex.set_state(state) vertex.set_state(state)
if state == VertexStates.INACTIVE:
self.run_manager.remove_from_predecessors(vertex_id)
def mark_branch(self, vertex_id: str, state: str, visited: Optional[set] = None, output_name: Optional[str] = None): def mark_branch(self, vertex_id: str, state: str, visited: Optional[set] = None, output_name: Optional[str] = None):
"""Marks a branch of the graph.""" """Marks a branch of the graph."""

3
src/backend/base/langflow/graph/graph/runnable_vertices_manager.py
View file

@ -41,6 +41,9 @@ class RunnableVerticesManager:
self.run_predecessors = state["run_predecessors"] self.run_predecessors = state["run_predecessors"]
self.vertices_to_run = state["vertices_to_run"] self.vertices_to_run = state["vertices_to_run"]
def all_predecessors_are_fulfilled(self) -> bool:
return all(not value for value in self.run_predecessors.values())
def update_run_state(self, run_predecessors: dict, vertices_to_run: set): def update_run_state(self, run_predecessors: dict, vertices_to_run: set):
self.run_predecessors.update(run_predecessors) self.run_predecessors.update(run_predecessors)
self.vertices_to_run.update(vertices_to_run) self.vertices_to_run.update(vertices_to_run)