test_langfuse_masked_input_output

This commit is contained in:
Ishaan Jaff 2024-11-21 22:51:19 -08:00
parent 027967d260
commit be0f0dd345

View file

@ -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