scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-16 23:41:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix _strip_base64_from_messages

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-31 17:59:13 -07:00
parent 2fe2704a21
commit 51349cdc99
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/custom_logger.py
View file

@ -592,7 +592,7 @@ class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callbac
• Recursively redact inline base64 blobs in *any* string field, at any depth.
"""
raw_messages: Any = payload.get("messages", [])
messages: list[Any] = raw_messages if isinstance(raw_messages, list) else []
messages: List[Any] = raw_messages if isinstance(raw_messages, list) else []
verbose_logger.debug(f"[CustomLogger] Stripping base64 from {len(messages)} messages")
if messages:
@ -644,14 +644,14 @@ class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callbac
ctype = content.get("type")
return not (isinstance(ctype, str) and ctype != "text")
def _process_messages(self, messages: list[Any], max_depth: int = DEFAULT_MAX_RECURSE_DEPTH_SENSITIVE_DATA_MASKER) -> List[Dict[str, Any]]:
def _process_messages(self, messages: List[Any], max_depth: int = DEFAULT_MAX_RECURSE_DEPTH_SENSITIVE_DATA_MASKER) -> List[Dict[str, Any]]:
filtered_messages: List[Dict[str, Any]] = []
for msg in messages:
if not isinstance(msg, dict):
continue
contents: Any = msg.get("content")
if isinstance(contents, list):
cleaned: list[Any] = []
cleaned: List[Any] = []
for c in contents:
if self._should_keep_content(content=c):
cleaned.append(self._redact_base64(value=c, max_depth=max_depth))