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:54:18 -08:00
parent be0f0dd345
commit 366a6895e2
1 changed files with 6 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/local_testing/test_alangfuse.py
View file

@ -326,20 +326,9 @@ async def test_langfuse_masked_input_output(langfuse_client):
mock_response="This is a test response",
)
print(response)
expected_input = (
"redacted-by-litellm"
if mask_value
else {"messages": [{"content": "This is a test", "role": "user"}]}
)
expected_input = "redacted-by-litellm" if mask_value else "This is a test"
expected_output = (
"redacted-by-litellm"
if mask_value
else {
"content": "This is a test response",
"role": "assistant",
"function_call": None,
"tool_calls": None,
}
"redacted-by-litellm" if mask_value else "This is a test response"
)
langfuse_client.flush()
await asyncio.sleep(30)
@ -351,11 +340,11 @@ async def test_langfuse_masked_input_output(langfuse_client):
reversed(langfuse_client.get_generations(trace_id=_unique_trace_name).data)
)
assert trace.input == expected_input
assert trace.output == expected_output
assert expected_input in trace.input
assert expected_output in trace.output
if len(generations) > 0:
assert generations[0].input == expected_input
assert generations[0].output == expected_output
assert expected_input in generations[0].input
assert expected_output in generations[0].output
@pytest.mark.asyncio