push

Author: hassiebp

langfuse/_client/attributes.py

@@ -19,8 +19,8 @@
     ObservationTypeSpanLike,
 )
 from langfuse._utils.serializer import EventSerializer
+from langfuse.api import MapValue, SpanLevel
 from langfuse.model import PromptClient
-from langfuse.types import MapValue, SpanLevel
 
 
 class LangfuseOtelSpanAttributes:

langfuse/_client/span.py

@@ -51,8 +51,8 @@
     ObservationTypeSpanLike,
     get_observation_types_list,
 )
+from langfuse.api import MapValue, ScoreDataType, SpanLevel
 from langfuse.logger import langfuse_logger
-from langfuse.types import MapValue, ScoreDataType, SpanLevel
 
 # Factory mapping for observation classes
 # Note: "event" is handled separately due to special instantiation logic
@@ -273,7 +273,9 @@ def score(
         name: str,
         value: float,
         score_id: Optional[str] = None,
-        data_type: Optional[Literal["NUMERIC", "BOOLEAN"]] = None,
+        data_type: Optional[
+            Literal[ScoreDataType.NUMERIC, ScoreDataType.BOOLEAN]
+        ] = None,
         comment: Optional[str] = None,
         config_id: Optional[str] = None,
         timestamp: Optional[datetime] = None,
@@ -286,7 +288,9 @@ def score(
         name: str,
         value: str,
         score_id: Optional[str] = None,
-        data_type: Optional[Literal["CATEGORICAL"]] = "CATEGORICAL",
+        data_type: Optional[
+            Literal[ScoreDataType.CATEGORICAL]
+        ] = ScoreDataType.CATEGORICAL,
         comment: Optional[str] = None,
         config_id: Optional[str] = None,
         timestamp: Optional[datetime] = None,
@@ -351,7 +355,9 @@ def score_trace(
         name: str,
         value: float,
         score_id: Optional[str] = None,
-        data_type: Optional[Literal["NUMERIC", "BOOLEAN"]] = None,
+        data_type: Optional[
+            Literal[ScoreDataType.NUMERIC, ScoreDataType.BOOLEAN]
+        ] = None,
         comment: Optional[str] = None,
         config_id: Optional[str] = None,
         timestamp: Optional[datetime] = None,
@@ -364,7 +370,9 @@ def score_trace(
         name: str,
         value: str,
         score_id: Optional[str] = None,
-        data_type: Optional[Literal["CATEGORICAL"]] = "CATEGORICAL",
+        data_type: Optional[
+            Literal[ScoreDataType.CATEGORICAL]
+        ] = ScoreDataType.CATEGORICAL,
         comment: Optional[str] = None,
         config_id: Optional[str] = None,
         timestamp: Optional[datetime] = None,

langfuse/types.py

@@ -17,13 +17,10 @@ def my_evaluator(*, output: str, **kwargs) -> Evaluation:
     ```
 """
 
-from datetime import datetime
 from typing import (
     Any,
     Dict,
-    List,
     Literal,
-    Optional,
     Protocol,
     TypedDict,
 )
@@ -34,40 +31,13 @@ def my_evaluator(*, output: str, **kwargs) -> Evaluation:
     from typing_extensions import NotRequired
 
 
-from langfuse.api import MapValue, MediaContentType, UsageDetails
-from langfuse.model import PromptClient
+from langfuse.api import MediaContentType
 
 SpanLevel = Literal["DEBUG", "DEFAULT", "WARNING", "ERROR"]
 
 ScoreDataType = Literal["NUMERIC", "CATEGORICAL", "BOOLEAN"]
 
 
-class TraceMetadata(TypedDict):
-    name: Optional[str]
-    user_id: Optional[str]
-    session_id: Optional[str]
-    version: Optional[str]
-    release: Optional[str]
-    metadata: Optional[Any]
-    tags: Optional[List[str]]
-    public: Optional[bool]
-
-
-class ObservationParams(TraceMetadata, TypedDict):
-    input: Optional[Any]
-    output: Optional[Any]
-    level: Optional[SpanLevel]
-    status_message: Optional[str]
-    start_time: Optional[datetime]
-    end_time: Optional[datetime]
-    completion_start_time: Optional[datetime]
-    model: Optional[str]
-    model_parameters: Optional[Dict[str, MapValue]]
-    usage_details: Optional[UsageDetails]
-    cost_details: Optional[Dict[str, float]]
-    prompt: Optional[PromptClient]
-
-
 class MaskFunction(Protocol):
     """A function that masks data.
 
@@ -103,8 +73,6 @@ class TraceContext(TypedDict):
 __all__ = [
     "SpanLevel",
     "ScoreDataType",
-    "TraceMetadata",
-    "ObservationParams",
     "MaskFunction",
     "ParsedMediaReference",
     "TraceContext",