phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(feat) fix bug where several tokens created in litellm v token

Browse source
This commit is contained in:
ishaan-jaff 2024-02-22 21:33:43 -08:00
parent 21e1c2dc21
commit 86d10d33a3
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/proxy_server.py
View file

@ -1864,6 +1864,11 @@ async def generate_key_helper_fn(
table_name="user", table_name="user",
update_key_values=update_key_values, 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 ## CREATE KEY
verbose_proxy_logger.debug(f"prisma_client: Creating Key={key_data}") verbose_proxy_logger.debug(f"prisma_client: Creating Key={key_data}")