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

♻️ (base.py): Refactor add_trace and end_trace methods to use Union type for optional arguments to improve code readability and type safety

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

7
src/backend/base/langflow/services/tracing/base.py
View file

@ -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