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

refactor: Update end_all_traces method to accept error parameter

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:00:04 -03:00
commit ebb7f545ad
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -233,14 +233,14 @@ class Graph:
self.tracing_service.set_run_name(name)
self.tracing_service.initialize_tracers()
def end_all_traces(self, outputs: dict[str, Any] | None = None):
async def end_all_traces(self, outputs: dict[str, Any] | None = None, error: str | None = None):
if not self.tracing_service:
return
self._end_time = datetime.now(timezone.utc)
if outputs is None:
outputs = {}
outputs |= self.metadata
self.tracing_service.end(outputs)
await self.tracing_service.end(outputs, error)
@property
def sorted_vertices_layers(self) -> List[List[str]]: