scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

reduce code duplication

Browse source
This commit is contained in:
Moritz Müller 2026-03-06 10:13:08 +01:00
parent 8440048569
commit e0ef9bc411
1 changed files with 16 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
litellm/litellm_core_utils/redact_messages.py
View file

@ -85,36 +85,22 @@ def perform_redaction(model_call_details: dict, result):
model_call_details["input"] = ""
# Redact streaming response
if (
model_call_details.get("stream", False) is True
and "complete_streaming_response" in model_call_details
):
_streaming_response = model_call_details["complete_streaming_response"]
if hasattr(_streaming_response, "choices"):
for choice in _streaming_response.choices:
_redact_choice_content(choice)
elif hasattr(_streaming_response, "output"):
_redact_responses_api_output(_streaming_response.output)
# Redact reasoning field in ResponsesAPIResponse
if hasattr(_streaming_response, "reasoning") and _streaming_response.reasoning is not None:
_streaming_response.reasoning = None
if (
model_call_details.get("stream", False) is True
and "async_complete_streaming_response" in model_call_details
):
_streaming_response = model_call_details["async_complete_streaming_response"]
if hasattr(_streaming_response, "choices"):
for choice in _streaming_response.choices:
_redact_choice_content(choice)
elif hasattr(_streaming_response, "output"):
_redact_responses_api_output(_streaming_response.output)
# Redact reasoning field in ResponsesAPIResponse
if (
hasattr(_streaming_response, "reasoning")
and _streaming_response.reasoning is not None
):
_streaming_response.reasoning = None
for _streaming_key in ("complete_streaming_response", "async_complete_streaming_response"):
if (
model_call_details.get("stream", False) is True
and _streaming_key in model_call_details
):
_streaming_response = model_call_details[_streaming_key]
if _streaming_response is None:
continue
if hasattr(_streaming_response, "choices"):
for choice in _streaming_response.choices:
_redact_choice_content(choice)
elif hasattr(_streaming_response, "output"):
_redact_responses_api_output(_streaming_response.output)
# Redact reasoning field in ResponsesAPIResponse
if hasattr(_streaming_response, "reasoning") and _streaming_response.reasoning is not None:
_streaming_response.reasoning = None
# Redact result
if result is not None: