From 63f18e7163789f69cc045818d8a38735b8613e85 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 24 Jan 2024 13:01:37 -0800 Subject: [PATCH] (fix) use get_attr for valid_token --- litellm/proxy/proxy_server.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index ca58371f454..06cca2712a3 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -383,8 +383,9 @@ async def user_api_key_auth( # Token exists, not expired now check if its in budget for the user if valid_token.spend is not None and valid_token.user_id is not None: - user_max_budget = user_id_information.max_budget - user_current_spend = user_id_information.spend + user_max_budget = getattr(user_id_information, "max_budget", None) + user_current_spend = getattr(user_id_information, "spend", None) + if user_max_budget is not None and user_current_spend is not None: if user_current_spend > user_max_budget: raise Exception(