From ebb7f545addc7edaa030a2eecd245170d5c9fd89 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:00:04 -0300 Subject: [PATCH] refactor: Update end_all_traces method to accept error parameter --- src/backend/base/langflow/graph/graph/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/graph/graph/base.py b/src/backend/base/langflow/graph/graph/base.py index b0fbc0ab2..73e0cc8f5 100644 --- a/src/backend/base/langflow/graph/graph/base.py +++ b/src/backend/base/langflow/graph/graph/base.py @@ -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]]: