test langfuse e2e tests

BerriAI · Jan 19, 2025 · ddeac13 · ddeac13
1 parent 8f80e2b
commit ddeac13
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/logging_callback_tests/test_langfuse_e2e_sdk.py b/tests/logging_callback_tests/test_langfuse_e2e_sdk.py
@@ -184,6 +184,7 @@ async def test_langfuse_logging_completion(self, mock_setup):
             )
 
     @pytest.mark.asyncio
+    @pytest.mark.flaky(retries=2, delay=2)
     async def test_langfuse_logging_streaming_completion(self, mock_setup):
         """Test Langfuse logging for streaming completion"""
         setup = await mock_setup  # Await the fixture
@@ -216,6 +217,7 @@ async def test_langfuse_logging_embedding(self, mock_setup):
             )
 
     @pytest.mark.asyncio
+    @pytest.mark.flaky(retries=2, delay=2)
     async def test_langfuse_logging_custom_generation_name(self, mock_setup):
         """Test Langfuse logging with custom generation name and metadata"""
         setup = await mock_setup  # Await the fixture
@@ -240,6 +242,7 @@ async def test_langfuse_logging_custom_generation_name(self, mock_setup):
             )
 
     @pytest.mark.asyncio
+    @pytest.mark.flaky(retries=2, delay=2)
     async def test_langfuse_masked_input_output(self, mock_setup):
         """Test Langfuse logging with masked input and output"""
         setup = await mock_setup  # Await the fixture
@@ -270,6 +273,7 @@ async def test_langfuse_masked_input_output(self, mock_setup):
                     )
 
     @pytest.mark.asyncio
+    @pytest.mark.flaky(retries=2, delay=2)
     async def test_langfuse_masked_input_output_stream(self, mock_setup):
         """Test Langfuse logging with masked input and output"""
         setup = await mock_setup  # Await the fixture