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

fix(_logging.py): fix logging

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-19 09:34:07 -07:00
parent c0e5586137
commit b461905745
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/_logging.py
View file

@ -59,8 +59,11 @@ class SensitiveDataFilter(logging.Filter):
if not hasattr(record, "msg") or not record.msg: if not hasattr(record, "msg") or not record.msg:
return True return True
# Convert message to string if it's not already # If the message is a format string with args, we need to format it first
msg = str(record.msg) if record.args:
msg = record.msg % record.args
else:
msg = str(record.msg)
key_pattern = r'["\']?([^"\':\s]+)["\']?\s*[:=]' key_pattern = r'["\']?([^"\':\s]+)["\']?\s*[:=]'
keys = re.findall(key_pattern, msg) keys = re.findall(key_pattern, msg)
@ -86,7 +89,9 @@ class SensitiveDataFilter(logging.Filter):
pattern = f"\"{key}\":\\s*'[^']*'" pattern = f"\"{key}\":\\s*'[^']*'"
msg = re.sub(pattern, f"\"{key}\": 'REDACTED'", msg) msg = re.sub(pattern, f"\"{key}\": 'REDACTED'", msg)
# Set the message and clear args since we've already formatted it
record.msg = msg record.msg = msg
record.args = None
return True return True