From 952dbb9eb7a4422045a1275009f3905eae16970b Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 21 Nov 2024 22:59:36 -0800 Subject: [PATCH] test_langfuse_masked_input_output --- tests/local_testing/test_alangfuse.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/local_testing/test_alangfuse.py b/tests/local_testing/test_alangfuse.py index 47531c20f..1728b8feb 100644 --- a/tests/local_testing/test_alangfuse.py +++ b/tests/local_testing/test_alangfuse.py @@ -304,7 +304,6 @@ async def test_langfuse_logging_audio_transcriptions(langfuse_client): @pytest.mark.asyncio -@pytest.mark.flaky(retries=12, delay=2) async def test_langfuse_masked_input_output(langfuse_client): """ Test that creates a trace with masked input and output @@ -340,11 +339,11 @@ async def test_langfuse_masked_input_output(langfuse_client): reversed(langfuse_client.get_generations(trace_id=_unique_trace_name).data) ) - assert expected_input in trace.input - assert expected_output in trace.output + assert expected_input in str(trace.input) + assert expected_output in str(trace.output) if len(generations) > 0: - assert expected_input in generations[0].input - assert expected_output in generations[0].output + assert expected_input in str(generations[0].input) + assert expected_output in str(generations[0].output) @pytest.mark.asyncio