diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py index dc2ade387b..2acacbb961 100644 --- a/litellm/tests/test_alangfuse.py +++ b/litellm/tests/test_alangfuse.py @@ -218,7 +218,9 @@ async def test_langfuse_logging_without_request_response(stream, langfuse_client _trace_data = trace.data - assert _trace_data[0].input == {"messages": "redacted-by-litellm"} + assert _trace_data[0].input == { + "messages": [{"content": "redacted-by-litellm", "role": "user"}] + } assert _trace_data[0].output == { "role": "assistant", "content": "redacted-by-litellm",