From ddeac13ca7e74b8e56d17ef888b99f963870a636 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 18 Jan 2025 18:47:48 -0800 Subject: [PATCH] test langfuse e2e tests --- tests/logging_callback_tests/test_langfuse_e2e_sdk.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/logging_callback_tests/test_langfuse_e2e_sdk.py b/tests/logging_callback_tests/test_langfuse_e2e_sdk.py index fc9329adef23..76ddc014d067 100644 --- 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