From bf57cb8e43b53ff3473e9322d7ec03423c644a78 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 21 Jun 2024 23:32:51 -0300 Subject: [PATCH] refactor: Add child link dictionary to LangSmithTracer class --- src/backend/base/langflow/services/tracing/service.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/services/tracing/service.py b/src/backend/base/langflow/services/tracing/service.py index e9761b039..7f5b429eb 100644 --- a/src/backend/base/langflow/services/tracing/service.py +++ b/src/backend/base/langflow/services/tracing/service.py @@ -127,7 +127,7 @@ class TracingService(Service): self.logs_queue.put_nowait((self._add_log, (trace_name, log))) @asynccontextmanager - def trace_context( + async def trace_context( self, trace_name: str, trace_type: str, inputs: Dict[str, Any] = None, metadata: Dict[str, Any] = None ): self._start_traces(trace_name, trace_type, inputs, metadata) @@ -196,6 +196,7 @@ class LangSmithTracer: if metadata: child.add_metadata(raw_inputs) self._children[trace_name] = child + self._child_link = {} def _convert_to_langchain_types(self, io_dict: Dict[str, Any]): converted = {}