push

Author: hassiebp

langfuse/_client/resource_manager.py

@@ -241,6 +241,10 @@ def _initialize_instance(
             f"media_threads={media_upload_thread_count or 1}"
         )
 
+    @classmethod
+    def reset(cls):
+        cls._instances.clear()
+
     def add_score_task(self, event: dict):
         try:
             # Sample scores with the same sampler that is used for tracing

tests/test_otel.py

@@ -57,6 +57,7 @@ def cleanup_otel(self):
         original_provider = trace_api.get_tracer_provider()
         yield
         trace_api.set_tracer_provider(original_provider)
+        LangfuseResourceManager.reset()
 
     @pytest.fixture
     def memory_exporter(self):
@@ -875,6 +876,7 @@ def test_sampling(self, monkeypatch, tracer_provider, mock_processor_init):
         # Restore the original provider
         trace_api.set_tracer_provider(original_provider)
 
+    @pytest.mark.skip("Calling shutdown will pollute the global context")
     def test_shutdown_and_flush(self, langfuse_client, memory_exporter):
         """Test shutdown and flush operations."""
         # Create a span without ending it
@@ -2375,7 +2377,7 @@ def mock_process_media(*, data, span, field):
             span._process_media_in_attribute = original_process
 
 
-class TestOtelIdGeneration:
+class TestOtelIdGeneration(TestOTelBase):
     """Tests for trace_id and observation_id generation with and without seeds."""
 
     @pytest.fixture