From d37ae29a3b280e61d2708d5e8deac5c9cd0f7d73 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 27 Jun 2024 15:20:20 -0300 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(base.py):=20Refactor=20ad?= =?UTF-8?q?d=5Ftrace=20and=20end=5Ftrace=20methods=20to=20use=20Union=20ty?= =?UTF-8?q?pe=20for=20optional=20arguments=20to=20improve=20code=20readabi?= =?UTF-8?q?lity=20and=20type=20safety?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/services/tracing/base.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/services/tracing/base.py b/src/backend/base/langflow/services/tracing/base.py index 3d855436d..5b4605382 100644 --- a/src/backend/base/langflow/services/tracing/base.py +++ b/src/backend/base/langflow/services/tracing/base.py @@ -1,5 +1,6 @@ from abc import ABC, abstractmethod from typing import Any, Dict +from uuid import UUID class BaseTracer(ABC): @@ -12,11 +13,13 @@ class BaseTracer(ABC): raise NotImplementedError @abstractmethod - def add_trace(self, trace_name: str, trace_type: str, inputs: Dict[str, Any], metadata: Dict[str, Any] = None): + def add_trace( + self, trace_name: str, trace_type: str, inputs: Dict[str, Any], metadata: Dict[str, Any] | None = None + ): raise NotImplementedError @abstractmethod - def end_trace(self, trace_name: str, outputs: Dict[str, Any] = None, error: str = None): + def end_trace(self, trace_name: str, outputs: Dict[str, Any] | None = None, error: str | None = None): raise NotImplementedError @abstractmethod