From 1a97a80c519ab57413293fd45849613838ed0a8e Mon Sep 17 00:00:00 2001 From: kayoch1n Date: Wed, 3 Sep 2025 11:18:57 +0800 Subject: [PATCH] Format code --- litellm/proxy/common_utils/callback_utils.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/common_utils/callback_utils.py b/litellm/proxy/common_utils/callback_utils.py index 46c9afef3cc..d52592952bc 100644 --- a/litellm/proxy/common_utils/callback_utils.py +++ b/litellm/proxy/common_utils/callback_utils.py @@ -316,21 +316,27 @@ def get_remaining_tokens_and_requests_from_request_data(data: Dict) -> Dict[str, headers = {} _metadata = data.get("metadata", None) or {} model_group = get_model_group_from_request_data(data) - + # The h11 package considers "/" or ":" invalid and raise a LocalProtocolError - h11_model_group_name = model_group.replace('/', '-').replace(':', '-') + h11_model_group_name = ( + model_group.replace("/", "-").replace(":", "-") if model_group else None + ) # Remaining Requests remaining_requests_variable_name = f"litellm-key-remaining-requests-{model_group}" remaining_requests = _metadata.get(remaining_requests_variable_name, None) if remaining_requests: - headers[f"x-litellm-key-remaining-requests-{h11_model_group_name}"] = remaining_requests + headers[f"x-litellm-key-remaining-requests-{h11_model_group_name}"] = ( + remaining_requests + ) # Remaining Tokens remaining_tokens_variable_name = f"litellm-key-remaining-tokens-{model_group}" remaining_tokens = _metadata.get(remaining_tokens_variable_name, None) if remaining_tokens: - headers[f"x-litellm-key-remaining-tokens-{h11_model_group_name}"] = remaining_tokens + headers[f"x-litellm-key-remaining-tokens-{h11_model_group_name}"] = ( + remaining_tokens + ) return headers