From 325df8d62aaaaa8d079ed2269b781dcdfcd0202a Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Tue, 10 Mar 2026 09:45:28 +0530 Subject: [PATCH] Fix logging tests --- litellm/litellm_core_utils/redact_messages.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/litellm/litellm_core_utils/redact_messages.py b/litellm/litellm_core_utils/redact_messages.py index ad68f3851a..ddeb24d04a 100644 --- a/litellm/litellm_core_utils/redact_messages.py +++ b/litellm/litellm_core_utils/redact_messages.py @@ -123,6 +123,16 @@ def perform_redaction(model_call_details: dict, result): elif isinstance(_result, litellm.EmbeddingResponse): if hasattr(_result, "data") and _result.data is not None: _result.data = [] + elif isinstance(_result, dict) and "choices" in _result: + # ModelResponse.model_dump() returns dict - redact choices in place + if isinstance(_result.get("choices"), list) and len(_result["choices"]) > 0: + choice = _result["choices"][0] + if isinstance(choice, dict) and "message" in choice: + msg = choice["message"] + if isinstance(msg, dict) and "content" in msg: + msg["content"] = "redacted-by-litellm" + if isinstance(msg, dict) and "audio" in msg: + msg["audio"] = None else: return {"text": "redacted-by-litellm"} return _result