phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Rename request header

Browse source
This commit is contained in:
Marc Abramowitz 2024-06-21 21:52:55 -07:00 • committed by Ishaan Jaff
parent efae9fa991
commit d14138dbd9
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/litellm_core_utils/redact_messages.py
View file

@ -28,12 +28,13 @@ def redact_message_input_output_from_logging(
Removes messages, prompts, input, response from logging. This modifies the data in-place
only redacts when litellm.turn_off_message_logging == True
"""
request_headers = litellm_logging_obj.model_call_details['litellm_params']['metadata']['headers']
# check if user opted out of logging message/response to callbacks
if litellm.turn_off_message_logging is not True:
if litellm.turn_off_message_logging is not True and request_headers.get('litellm-enable-message-redaction', False):
return result
request_headers = litellm_logging_obj.model_call_details['litellm_params']['metadata']['headers']
if request_headers and request_headers.get('litellm-turn-on-message-logging', False):
if request_headers and request_headers.get('litellm-disable-message-redaction', False):
return result
# remove messages, prompts, input, response from logging