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

refactor: Update set_run_id method to accept only uuid.UUID or None as run_id parameter

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 00:38:01 -03:00
commit e59966bff8
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -213,7 +213,7 @@ class Graph:
raise ValueError("Run ID not set")
return self._run_id
def set_run_id(self, run_id: str | uuid.UUID | None = None):
def set_run_id(self, run_id: uuid.UUID | None = None):
"""
Sets the ID of the current run.
@ -223,10 +223,10 @@ class Graph:
if run_id is None:
run_id = uuid.uuid4()
run_id = str(run_id)
run_id_str = str(run_id)
for vertex in self.vertices:
self.state_manager.subscribe(run_id, vertex.update_graph_state)
self._run_id = run_id
self.state_manager.subscribe(run_id_str, vertex.update_graph_state)
self._run_id = run_id_str
if self.tracing_service:
self.tracing_service.set_run_id(run_id)