From f82852fdd0786a29a772db098c398ff685a77ef6 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Sat, 8 Nov 2025 13:30:03 -0800
Subject: [PATCH] redact_message_input_output_from_custom_logger

---
 litellm/litellm_core_utils/redact_messages.py | 33 ++++++++++++-------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/litellm/litellm_core_utils/redact_messages.py b/litellm/litellm_core_utils/redact_messages.py
index 849cb20cdc5..17f3ceb7716 100644
--- a/litellm/litellm_core_utils/redact_messages.py
+++ b/litellm/litellm_core_utils/redact_messages.py
@@ -7,17 +7,17 @@
 #
 #  Thank you users! We ❤️ you! - Krrish & Ishaan
 
+import asyncio
 import copy
 from typing import TYPE_CHECKING, Any, Optional
 
 import litellm
 from litellm.integrations.custom_logger import CustomLogger
-from litellm.secret_managers.main import str_to_bool
-from litellm.types.utils import StandardCallbackDynamicParams
 from litellm.litellm_core_utils.core_helpers import (
     get_metadata_variable_name_from_kwargs,
 )
-import asyncio
+from litellm.secret_managers.main import str_to_bool
+from litellm.types.utils import StandardCallbackDynamicParams
 
 if TYPE_CHECKING:
     from litellm.litellm_core_utils.litellm_logging import (
@@ -42,43 +42,52 @@ def redact_message_input_output_from_custom_logger(
 
 def _redact_choice_content(choice):
     """Helper to redact content in a choice (message or delta)."""
+    from litellm.types.utils import LiteLLMCommonStrings
+    redacted_str = LiteLLMCommonStrings.redacted_by_litellm.value
+    
     if isinstance(choice, litellm.Choices):
-        choice.message.content = "redacted-by-litellm"
+        choice.message.content = redacted_str
         if hasattr(choice.message, "reasoning_content"):
-            choice.message.reasoning_content = "redacted-by-litellm"
+            choice.message.reasoning_content = redacted_str
         if hasattr(choice.message, "thinking_blocks"):
             choice.message.thinking_blocks = None
     elif isinstance(choice, litellm.utils.StreamingChoices):
-        choice.delta.content = "redacted-by-litellm"
+        choice.delta.content = redacted_str
         if hasattr(choice.delta, "reasoning_content"):
-            choice.delta.reasoning_content = "redacted-by-litellm"
+            choice.delta.reasoning_content = redacted_str
         if hasattr(choice.delta, "thinking_blocks"):
             choice.delta.thinking_blocks = None
 
 
 def _redact_responses_api_output(output_items):
     """Helper to redact ResponsesAPIResponse output items."""
+    from litellm.types.utils import LiteLLMCommonStrings
+    redacted_str = LiteLLMCommonStrings.redacted_by_litellm.value
+    
     for output_item in output_items:
         if hasattr(output_item, "content") and isinstance(output_item.content, list):
             for content_part in output_item.content:
                 if hasattr(content_part, "text"):
-                    content_part.text = "redacted-by-litellm"
+                    content_part.text = redacted_str
         
         # Redact reasoning items in output array
         if hasattr(output_item, "type") and output_item.type == "reasoning":
             if hasattr(output_item, "summary") and isinstance(output_item.summary, list):
                 for summary_item in output_item.summary:
                     if hasattr(summary_item, "text"):
-                        summary_item.text = "redacted-by-litellm"
+                        summary_item.text = redacted_str
 
 
 def perform_redaction(model_call_details: dict, result):
     """
     Performs the actual redaction on the logging object and result.
     """
+    from litellm.types.utils import LiteLLMCommonStrings
+    redacted_str = LiteLLMCommonStrings.redacted_by_litellm.value
+    
     # Redact model_call_details
     model_call_details["messages"] = [
-        {"role": "user", "content": "redacted-by-litellm"}
+        {"role": "user", "content": redacted_str}
     ]
     model_call_details["prompt"] = ""
     model_call_details["input"] = ""
@@ -106,7 +115,7 @@ def perform_redaction(model_call_details: dict, result):
             hasattr(result, '__aiter__') or  # async generator
             hasattr(result, '__anext__')):   # async iterator
             # For async objects, return a simple redacted response without deepcopy
-            return {"text": "redacted-by-litellm"}
+            return {"text": redacted_str}
         
         _result = copy.deepcopy(result)
         if isinstance(_result, litellm.ModelResponse):
@@ -123,7 +132,7 @@ def perform_redaction(model_call_details: dict, result):
             if hasattr(_result, "data") and _result.data is not None:
                 _result.data = []
         else:
-            return {"text": "redacted-by-litellm"}
+            return {"text": redacted_str}
         return _result