From efae9fa9911c44a5eeea29af705955ed660828fd Mon Sep 17 00:00:00 2001 From: Marc Abramowitz Date: Fri, 21 Jun 2024 20:21:19 -0700 Subject: [PATCH] Turn on message logging via request header --- litellm/litellm_core_utils/redact_messages.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/litellm_core_utils/redact_messages.py b/litellm/litellm_core_utils/redact_messages.py index 8f270d8bec..91f340cb80 100644 --- a/litellm/litellm_core_utils/redact_messages.py +++ b/litellm/litellm_core_utils/redact_messages.py @@ -32,6 +32,10 @@ def redact_message_input_output_from_logging( if litellm.turn_off_message_logging is not True: 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): + return result + # remove messages, prompts, input, response from logging litellm_logging_obj.model_call_details["messages"] = [ {"role": "user", "content": "redacted-by-litellm"}