From 81c528f6ce678dfb59f7f56e64211ae20c9d9b21 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 25 Jan 2024 11:39:57 -0800 Subject: [PATCH] (fix) raise correct error when proxy crossed budget --- litellm/proxy/proxy_server.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c2c15d7e9..43e2ec5f5 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -415,9 +415,11 @@ async def user_api_key_auth( user_info=_user, ) ) + + _user_id = _user.get("user_id", 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}" + f"ExceededBudget: User {_user_id} has exceeded their budget. Current spend: {user_current_spend}; Max Budget: {user_max_budget}" ) else: # Token exists, not expired now check if its in budget for the user