forked from phoenix/litellm-mirror
test_langfuse_masked_input_output
This commit is contained in:
parent
027967d260
commit
be0f0dd345
1 changed files with 4 additions and 3 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue