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:59:36 -08:00
parent 366a6895e2
commit 952dbb9eb7
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/local_testing/test_alangfuse.py
View file

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