From 86d10d33a30abae83269b2c95a1ad0ef6bec7f7f Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 22 Feb 2024 21:33:43 -0800 Subject: [PATCH] (feat) fix bug where several tokens created in litellm v token --- litellm/proxy/proxy_server.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 7a1632fcd..5d6c5115e 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1864,6 +1864,11 @@ async def generate_key_helper_fn( table_name="user", update_key_values=update_key_values, ) + if user_id == litellm_proxy_budget_name: + # do not create a key for litellm_proxy_budget_name + # we only need to ensure this exists in the user table + # the LiteLLM_VerificationToken table will increase in size if we don't do this check + return key_data ## CREATE KEY verbose_proxy_logger.debug(f"prisma_client: Creating Key={key_data}")