migrate more tests

Author: hassiebp

langfuse/_client/client.py

@@ -168,7 +168,7 @@ def __init__(
 
         if debug:
             logging.basicConfig(
-                format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+                format="%asctime)s - %(name)s - %(levelname)s - %(message)s"
             )
             langfuse_logger.setLevel(logging.DEBUG)
 

langfuse/logger.py

@@ -15,6 +15,7 @@
 
 # Create the main Langfuse logger
 langfuse_logger = logging.getLogger("langfuse")
+langfuse_logger.setLevel(logging.WARNING)
 
 # Configure httpx logger to reduce noise from HTTP requests
 httpx_logger = logging.getLogger("httpx")

langfuse/media.py

@@ -5,10 +5,13 @@
 import logging
 import os
 import re
-from typing import Any, Literal, Optional, Tuple, TypeVar, cast
+from typing import TYPE_CHECKING, Any, Literal, Optional, Tuple, TypeVar, cast
 
 import requests
 
+if TYPE_CHECKING:
+    from langfuse._client.client import Langfuse
+
 from langfuse.api import MediaContentType
 from langfuse.types import ParsedMediaReference
 
@@ -211,7 +214,7 @@ def _parse_base64_data_uri(
     def resolve_media_references(
         *,
         obj: T,
-        langfuse_client: Any,
+        langfuse_client: "Langfuse",
         resolve_with: Literal["base64_data_uri"],
         max_depth: int = 10,
         content_fetch_timeout_seconds: int = 10,
@@ -275,9 +278,9 @@ def traverse(obj: Any, depth: int) -> Any:
                         parsed_media_reference = LangfuseMedia.parse_reference_string(
                             reference_string
                         )
-                        media_data = langfuse_client.fetch_media(
+                        media_data = langfuse_client.api.media.get(
                             parsed_media_reference["media_id"]
-                        ).data
+                        )
                         media_content = requests.get(
                             media_data.url, timeout=content_fetch_timeout_seconds
                         )

tests/test_json.py

@@ -1,19 +1,19 @@
 import builtins
-from dataclasses import dataclass
 import importlib
 import json
-from datetime import datetime, timezone, date
-from unittest.mock import patch
 import uuid
-from bson import ObjectId
+from dataclasses import dataclass
+from datetime import date, datetime, timezone
+from unittest.mock import patch
 
 import pytest
+from bson import ObjectId
 from langchain.schema.messages import HumanMessage
 from pydantic import BaseModel
 
 import langfuse
+from langfuse._utils.serializer import EventSerializer
 from langfuse.api.resources.commons.types.observation_level import ObservationLevel
-from langfuse.serializer import EventSerializer
 
 
 class TestModel(BaseModel):

tests/test_logger.py

@@ -1,7 +1,7 @@
 import os
 
 from langfuse import Langfuse
-from langfuse.callback import CallbackHandler
+from langfuse.logger import langfuse_logger
 
 """
 Level	Numeric value
@@ -12,65 +12,22 @@
 """
 
 
-def test_via_env():
-    os.environ["LANGFUSE_DEBUG"] = "True"
-
-    langfuse = Langfuse()
-
-    assert langfuse.log.level == 10
+def test_default_langfuse():
+    Langfuse()
 
-    os.environ.pop("LANGFUSE_DEBUG")
+    assert langfuse_logger.level == 30
 
 
-def test_via_env_callback():
+def test_via_env():
     os.environ["LANGFUSE_DEBUG"] = "True"
 
-    callback = CallbackHandler()
+    Langfuse()
+
+    assert langfuse_logger.level == 10
 
-    assert callback.log.level == 10
-    assert callback.langfuse.log.level == 10
     os.environ.pop("LANGFUSE_DEBUG")
 
 
 def test_debug_langfuse():
-    langfuse = Langfuse(debug=True)
-    assert langfuse.log.level == 10
-
-
-def test_default_langfuse():
-    langfuse = Langfuse()
-    assert langfuse.log.level == 30
-
-
-def test_default_langfuse_callback():
-    callback = CallbackHandler()
-    assert callback.log.level == 30
-    assert callback.log.level == 30
-    assert callback.langfuse.log.level == 30
-
-
-def test_debug_langfuse_callback():
-    callback = CallbackHandler(debug=True)
-    assert callback.log.level == 10
-    assert callback.log.level == 10
-    assert callback.langfuse.log.level == 10
-
-
-def test_default_langfuse_trace_callback():
-    langfuse = Langfuse()
-    trace = langfuse.trace(name="test")
-    callback = trace.getNewHandler()
-
-    assert callback.log.level == 30
-    assert callback.log.level == 30
-    assert callback.trace.log.level == 30
-
-
-def test_debug_langfuse_trace_callback():
-    langfuse = Langfuse(debug=True)
-    trace = langfuse.trace(name="test")
-    callback = trace.getNewHandler()
-
-    assert callback.log.level == 10
-    assert callback.log.level == 10
-    assert callback.trace.log.level == 10
+    Langfuse(debug=True)
+    assert langfuse_logger.level == 10

tests/test_media.py

@@ -4,7 +4,7 @@
 
 import pytest
 
-from langfuse.client import Langfuse
+from langfuse._client.client import Langfuse
 from langfuse.media import LangfuseMedia
 from tests.utils import get_api
 
@@ -124,7 +124,7 @@ def test_replace_media_reference_string_in_object():
     mock_trace_name = f"test-trace-with-audio-{uuid4()}"
     base64_audio = base64.b64encode(mock_audio_bytes).decode()
 
-    trace = langfuse.trace(
+    span = langfuse.start_span(
         name=mock_trace_name,
         metadata={
             "context": {
@@ -133,40 +133,38 @@ def test_replace_media_reference_string_in_object():
                 )
             }
         },
-    )
+    ).end()
 
     langfuse.flush()
 
     # Verify media reference string format
-    fetched_trace = get_api().trace.get(trace.id)
-    media_ref = fetched_trace.metadata["context"]["nested"]
+    fetched_trace = get_api().trace.get(span.trace_id)
+    media_ref = fetched_trace.observations[0].metadata["context"]["nested"]
     assert re.match(
         r"^@@@langfuseMedia:type=audio/wav\|id=.+\|source=base64_data_uri@@@$",
         media_ref,
     )
 
     # Resolve media references back to base64
-    resolved_trace = langfuse.resolve_media_references(
-        obj=fetched_trace, resolve_with="base64_data_uri"
+    resolved_obs = langfuse.resolve_media_references(
+        obj=fetched_trace.observations[0], resolve_with="base64_data_uri"
     )
 
     # Verify resolved base64 matches original
     expected_base64 = f"data:audio/wav;base64,{base64_audio}"
-    assert resolved_trace["metadata"]["context"]["nested"] == expected_base64
+    assert resolved_obs["metadata"]["context"]["nested"] == expected_base64
 
     # Create second trace reusing the media reference
-    trace2 = langfuse.trace(
+    span2 = langfuse.start_span(
         name=f"2-{mock_trace_name}",
-        metadata={
-            "context": {"nested": resolved_trace["metadata"]["context"]["nested"]}
-        },
-    )
+        metadata={"context": {"nested": resolved_obs["metadata"]["context"]["nested"]}},
+    ).end()
 
     langfuse.flush()
 
     # Verify second trace has same media reference
-    fetched_trace2 = get_api().trace.get(trace2.id)
+    fetched_trace2 = get_api().trace.get(span2.trace_id)
     assert (
-        fetched_trace2.metadata["context"]["nested"]
-        == fetched_trace.metadata["context"]["nested"]
+        fetched_trace2.observations[0].metadata["context"]["nested"]
+        == fetched_trace.observations[0].metadata["context"]["nested"]
     )