push

Author: hassiebp

tests/test_decorators.py

@@ -1311,7 +1311,7 @@ def level_1_function(*args, **kwargs):
 @pytest.mark.asyncio
 async def test_multiproject_async_context_propagation_basic():
     """Test that nested async decorated functions inherit langfuse_public_key from parent in multi-project setup"""
-    LangfuseResourceManager.reset()
+    LangfuseResourceManager._instances.pop("fake", None)
     client1 = Langfuse()  # Reads from environment
     Langfuse(public_key="pk-test-project2", secret_key="sk-test-project2")
 
@@ -1378,7 +1378,7 @@ async def async_level_1_function(*args, **kwargs):
 @pytest.mark.asyncio
 async def test_multiproject_mixed_sync_async_context_propagation():
     """Test context propagation between sync and async decorated functions in multi-project setup"""
-    LangfuseResourceManager.reset()
+    LangfuseResourceManager._instances.pop("fake", None)
     client1 = Langfuse()  # Reads from environment
     Langfuse(public_key="pk-test-project2", secret_key="sk-test-project2")
 
@@ -1447,7 +1447,7 @@ async def async_level_1_function(*args, **kwargs):
 @pytest.mark.asyncio
 async def test_multiproject_concurrent_async_context_isolation():
     """Test that concurrent async executions don't interfere with each other's context in multi-project setup"""
-    LangfuseResourceManager.reset()
+    LangfuseResourceManager._instances.pop("fake", None)
     client1 = Langfuse()  # Reads from environment
     Langfuse(public_key="pk-test-project2", secret_key="sk-test-project2")
 
@@ -1535,7 +1535,7 @@ async def async_level_1_function(task_id, *args, **kwargs):
 @pytest.mark.asyncio
 async def test_multiproject_async_generator_context_propagation():
     """Test context propagation with async generators in multi-project setup"""
-    LangfuseResourceManager.reset()
+    LangfuseResourceManager._instances.pop("fake", None)
     client1 = Langfuse()  # Reads from environment
     Langfuse(public_key="pk-test-project2", secret_key="sk-test-project2")
 
@@ -1597,7 +1597,7 @@ async def async_consumer_function():
 @pytest.mark.asyncio
 async def test_multiproject_async_context_exception_handling():
     """Test that async context is properly restored even when exceptions occur in multi-project setup"""
-    LangfuseResourceManager.reset()
+    LangfuseResourceManager._instances.pop("fake", None)
     client1 = Langfuse()  # Reads from environment
     Langfuse(public_key="pk-test-project2", secret_key="sk-test-project2")