tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-02 10:21:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix logging tests

Browse Source
This commit is contained in:
Sameer Kankute
2026-03-10 09:45:28 +05:30
parent 30b82c3a0c
commit 325df8d62a
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/litellm_core_utils/redact_messages.py
+10
View File
@@ -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