From 466fff4aaf411eee1e4d3dc3ee233bbc5eed8483 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 16 Jan 2024 17:35:39 -0800 Subject: [PATCH] (fix) when budget, spend is None comparison --- litellm/proxy/proxy_server.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 353c3157db..1d0245c51c 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -348,10 +348,11 @@ async def user_api_key_auth( 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 - if user_current_spend > user_max_budget: - raise Exception( - f"ExceededBudget: User {valid_token.user_id} has exceeded their budget. Current spend: {user_current_spend}; Max Budget: {user_max_budget}" - ) + if user_max_budget is not None and user_current_spend is not None: + if user_current_spend > user_max_budget: + raise Exception( + f"ExceededBudget: User {valid_token.user_id} has exceeded their budget. Current spend: {user_current_spend}; Max Budget: {user_max_budget}" + ) # Check 3. If token is expired if valid_token.expires is not None: