scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

refactor(logging): move the redacted-record sentinel and attribute name into constants
Some checks failed
ai-gateway image / ai-gateway release image (push) Has been cancelled
Details

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-17 05:22:26 +00:00
parent 55cf4c43ed
commit e5d6cc8a6b
2 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/_logging.py
View file

@ -12,6 +12,8 @@ from urllib.parse import unquote
import litellm
from litellm.constants import (
LITELLM_REDACTED_RECORD_ATTR,
LITELLM_REDACTED_RECORD_STAMP,
LITELLM_TRUNCATED_PAYLOAD_FIELD,
LITELLM_TRUNCATION_STDOUT_SAFEGUARD_NOTE,
MAX_BASE64_LENGTH_STDOUT_LOG,
@ -79,13 +81,11 @@ def _redact_structured_value(key: str | None, value: str) -> str:
return redact_structured_value(key, value)
_REDACTED_RECORD_ATTR: Final = "litellm_redacted"
_REDACTED_STAMP: Final = object()
_UNREDACTED_SCALAR_TYPES: Final = (bool, int, float, type(None))
def _is_redacted(record: logging.LogRecord) -> bool:
return getattr(record, _REDACTED_RECORD_ATTR, None) is _REDACTED_STAMP
return getattr(record, LITELLM_REDACTED_RECORD_ATTR, None) is LITELLM_REDACTED_RECORD_STAMP
def _scrubbing_changed_nothing(scrubbed: object, original: object) -> bool:
@ -194,7 +194,7 @@ class SecretRedactionFilter(logging.Filter):
elif not isinstance(value, _UNREDACTED_SCALAR_TYPES):
setattr(record, key, _redact_extra_value(key, value))
setattr(record, _REDACTED_RECORD_ATTR, _REDACTED_STAMP)
setattr(record, LITELLM_REDACTED_RECORD_ATTR, LITELLM_REDACTED_RECORD_STAMP)
return True
@ -562,7 +562,7 @@ def _get_standard_record_attrs() -> frozenset:
_STANDARD_RECORD_ATTRS: Final = _get_standard_record_attrs()
_NON_EXTRA_RECORD_ATTRS: Final = _STANDARD_RECORD_ATTRS | {_REDACTED_RECORD_ATTR}
_NON_EXTRA_RECORD_ATTRS: Final = _STANDARD_RECORD_ATTRS | {LITELLM_REDACTED_RECORD_ATTR}
# CorrelationContextFilter is the only legitimate source for these two JSON fields;
# see JsonFormatter.format() for why they're excluded from the generic message-content

2
litellm/constants.py
View file

@ -1510,6 +1510,8 @@ LITELLM_TRUNCATION_STDOUT_SAFEGUARD_NOTE: Final = (
"Full, untruncated data is logged to logging callbacks (OTEL, Datadog, etc.) and at DEBUG level. "
"To increase the truncation limit, set `MAX_STRING_LENGTH_STDOUT_LOG` in your env."
)
LITELLM_REDACTED_RECORD_ATTR: Final = "litellm_redacted"
LITELLM_REDACTED_RECORD_STAMP: Final[object] = object()
########################### LiteLLM Proxy Specific Constants ###########################
########################################################################################