fix tests

Author: hassiebp

.github/workflows/ci.yml

@@ -155,7 +155,7 @@ jobs:
       - name: Run the automated tests
         run: |
           python --version
-          poetry run pytest -n auto -s -v --log-cli-level=INFO
+          poetry run pytest -n auto --dist loadfile -s -v --log-cli-level=INFO
 
   all-tests-passed:
     # This allows us to have a branch protection rule for tests and deploys with matrix

tests/test_core_sdk.py

@@ -923,6 +923,7 @@ def test_create_span_and_generation():
 
     # Create initial span
     span = langfuse.start_span(name="span")
+    sleep(0.1)
     # Get trace ID for later use
     trace_id = span.trace_id
     # End the span

tests/test_decorators.py

@@ -11,7 +11,6 @@
 from langfuse import get_client, observe
 from langfuse.langchain import CallbackHandler
 from langfuse.media import LangfuseMedia
-from langfuse.openai import AsyncOpenAI
 from tests.utils import get_api
 
 mock_metadata = {"key": "metadata"}
@@ -804,6 +803,8 @@ async def main_async(**kwargs):
 
 @pytest.mark.asyncio
 async def test_async_nested_openai_chat_stream():
+    from langfuse.openai import AsyncOpenAI
+
     mock_name = "test_async_nested_openai_chat_stream"
     langfuse = get_client()
     mock_trace_id = langfuse.create_trace_id()

tests/test_langchain_integration.py

@@ -11,7 +11,6 @@
 from .utils import create_uuid
 
 
-# to avoid the instanciation of langfuse in side langfuse.openai.
 def _is_streaming_response(response):
     return isinstance(response, types.GeneratorType) or isinstance(
         response, types.AsyncGeneratorType

tests/test_media.py

@@ -61,16 +61,13 @@ def test_content_sha256_hash():
 
 def test_reference_string():
     media = LangfuseMedia(content_bytes=SAMPLE_JPEG_BYTES, content_type="image/jpeg")
-    # Reference string should be None initially as media_id is not set
-    assert media._reference_string is None
 
-    # Set media_id
-    media._media_id = "test-id"
+    media._media_id = "MwoGlsMS6lW8ijWeRyZKfD"
     reference = media._reference_string
-    assert reference is not None
-    assert "test-id" in reference
-    assert "image/jpeg" in reference
-    assert "bytes" in reference
+    assert (
+        reference
+        == "@@@langfuseMedia:type=image/jpeg|id=MwoGlsMS6lW8ijWeRyZKfD|source=bytes@@@"
+    )
 
 
 def test_parse_reference_string():