From a180e9e7c44c77a175a1ee12c99fb30587d8b2f6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 27 Jun 2024 15:20:51 -0300 Subject: [PATCH] chore: Update LangSmithTracer constructor parameter type to UUID --- src/backend/base/langflow/services/tracing/service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/services/tracing/service.py b/src/backend/base/langflow/services/tracing/service.py index fd96b3eb9..4bdf17fa8 100644 --- a/src/backend/base/langflow/services/tracing/service.py +++ b/src/backend/base/langflow/services/tracing/service.py @@ -182,7 +182,7 @@ class TracingService(Service): class LangSmithTracer(BaseTracer): - def __init__(self, trace_name: str, trace_type: str, project_name: str, trace_id: str): + def __init__(self, trace_name: str, trace_type: str, project_name: str, trace_id: UUID): from langsmith.run_trees import RunTree self.trace_name = trace_name @@ -293,7 +293,7 @@ class LangSmithTracer(BaseTracer): inputs: dict[str, Any], outputs: Dict[str, Any], error: str | None = None, - metadata: Optional[dict[str, Any]] = None, + metadata: dict[str, Any] | None = None, ): self._run_tree.add_metadata({"inputs": inputs, "metadata": metadata or {}}) self._run_tree.end(outputs=outputs, error=error)