From be0f0dd345a857bfcaa78345e8fc93c54e8917a1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 21 Nov 2024 22:51:19 -0800 Subject: [PATCH] test_langfuse_masked_input_output --- tests/local_testing/test_alangfuse.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/local_testing/test_alangfuse.py b/tests/local_testing/test_alangfuse.py index 4959bd69a..38f18a4f7 100644 --- a/tests/local_testing/test_alangfuse.py +++ b/tests/local_testing/test_alangfuse.py @@ -269,7 +269,7 @@ audio_file = open(file_path, "rb") @pytest.mark.asyncio @pytest.mark.flaky(retries=4, delay=2) -async def test_langfuse_logging_audio_transcriptions(): +async def test_langfuse_logging_audio_transcriptions(langfuse_client): """ Test that creates a trace with masked input and output """ @@ -353,8 +353,9 @@ async def test_langfuse_masked_input_output(langfuse_client): assert trace.input == expected_input assert trace.output == expected_output - assert generations[0].input == expected_input - assert generations[0].output == expected_output + if len(generations) > 0: + assert generations[0].input == expected_input + assert generations[0].output == expected_output @pytest.mark.asyncio