removed debug conditionals and addressed clean_headers

This commit is contained in:
shivam 2026-02-14 19:52:01 -08:00
parent f7c3c9964b
commit 4e66d82127
2 changed files with 38 additions and 34 deletions

View file

@ -901,31 +901,29 @@ async def add_litellm_data_to_request( # noqa: PLR0915
## Cache Controls
headers = request.headers
if verbose_proxy_logger.isEnabledFor(logging.DEBUG):
verbose_proxy_logger.debug(
"Request Headers: %s",
clean_headers(
headers,
litellm_key_header_name=(
general_settings.get("litellm_key_header_name")
if general_settings is not None
else None
),
verbose_proxy_logger.debug(
"Request Headers: %s",
clean_headers(
headers,
litellm_key_header_name=(
general_settings.get("litellm_key_header_name")
if general_settings is not None
else None
),
)
),
)
cache_control_header = headers.get("Cache-Control", None)
if cache_control_header:
cache_dict = parse_cache_control(cache_control_header)
data["ttl"] = cache_dict.get("s-maxage")
if verbose_proxy_logger.isEnabledFor(logging.DEBUG):
verbose_proxy_logger.debug(
"receiving data: %s",
{
k: (_mask_secret_fields_for_logging(v) if k == "secret_fields" else v)
for k, v in data.items()
},
)
verbose_proxy_logger.debug(
"receiving data: %s",
{
k: (_mask_secret_fields_for_logging(v) if k == "secret_fields" else v)
for k, v in data.items()
},
)
# Parse metadata if it's a string (e.g., from multipart/form-data)
if "metadata" in data and data["metadata"] is not None:
@ -1152,14 +1150,13 @@ async def add_litellm_data_to_request( # noqa: PLR0915
user_api_key_dict=user_api_key_dict,
)
if verbose_proxy_logger.isEnabledFor(logging.DEBUG):
verbose_proxy_logger.debug(
"[PROXY] returned data from litellm_pre_call_utils: %s",
{
k: (_mask_secret_fields_for_logging(v) if k == "secret_fields" else v)
for k, v in data.items()
},
)
verbose_proxy_logger.debug(
"[PROXY] returned data from litellm_pre_call_utils: %s",
{
k: (_mask_secret_fields_for_logging(v) if k == "secret_fields" else v)
for k, v in data.items()
},
)
## ENFORCED PARAMS CHECK
# loop through each enforced param

View file

@ -9842,22 +9842,29 @@ async def async_queue_request(
data["proxy_server_request"] = {
"url": str(request.url),
"method": request.method,
"headers": clean_headers(
"headers": dict(request.headers),
"body": copy.copy(data), # use copy instead of deepcopy
}
if verbose_proxy_logger.isEnabledFor(logging.DEBUG):
# Use clean_headers only for logging to avoid logging JWT/auth tokens
# while preserving all headers (including MCP headers) in the stored data
_safe_headers = clean_headers(
request.headers,
litellm_key_header_name=(
general_settings.get("litellm_key_header_name")
if general_settings is not None
else None
),
),
"body": copy.copy(data), # use copy instead of deepcopy
}
if verbose_proxy_logger.isEnabledFor(logging.DEBUG):
)
verbose_proxy_logger.debug(
"receiving data: %s",
{
k: (_mask_secret_fields_for_logging(v) if k == "secret_fields" else v)
k: (
_mask_secret_fields_for_logging(v) if k == "secret_fields"
else {**v, "headers": _safe_headers} if k == "proxy_server_request"
else v
)
for k, v in data.items()
},
)