phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test_langfuse_masked_input_output

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-21 22:51:19 -08:00
parent aff120b34f
commit a5df4f1a81
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/local_testing/test_alangfuse.py
View file

@ -263,7 +263,7 @@ audio_file = open(file_path, "rb")
@pytest.mark.asyncio @pytest.mark.asyncio
@pytest.mark.flaky(retries=4, delay=2) @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 Test that creates a trace with masked input and output
""" """
@ -347,6 +347,7 @@ async def test_langfuse_masked_input_output(langfuse_client):
assert trace.input == expected_input assert trace.input == expected_input
assert trace.output == expected_output assert trace.output == expected_output
if len(generations) > 0:
assert generations[0].input == expected_input assert generations[0].input == expected_input
assert generations[0].output == expected_output assert generations[0].output == expected_output