From 6e31bcf5a7e453a6472566e55572ab9162b625af Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 3 Jan 2025 10:02:10 -0800 Subject: [PATCH] fix - access metadata (#7523) --- litellm/proxy/hooks/model_max_budget_limiter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/hooks/model_max_budget_limiter.py b/litellm/proxy/hooks/model_max_budget_limiter.py index 3befca85163..ac02c915366 100644 --- a/litellm/proxy/hooks/model_max_budget_limiter.py +++ b/litellm/proxy/hooks/model_max_budget_limiter.py @@ -155,8 +155,8 @@ class _PROXY_VirtualKeyModelMaxBudgetLimiter(RouterBudgetLimiting): if standard_logging_payload is None: raise ValueError("standard_logging_payload is required") - _litellm_params = kwargs.get("litellm_params", {}) - _metadata = _litellm_params.get("metadata", {}) + _litellm_params: dict = kwargs.get("litellm_params", {}) or {} + _metadata: dict = _litellm_params.get("metadata", {}) or {} user_api_key_model_max_budget: Optional[dict] = _metadata.get( "user_api_key_model_max_budget", None )