phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) raise exception budget_duration is set and max_budget is Not

Browse source
This commit is contained in:
ishaan-jaff 2024-01-25 11:32:05 -08:00
parent 3ef2afb0e4
commit 126b87e3fa
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/proxy_server.py
View file

@ -1651,11 +1651,12 @@ async def startup_event():
user_id="default_user_id",
)
if (
prisma_client is not None
and litellm.max_budget > 0
and litellm.budget_duration is not None
):
if prisma_client is not None and litellm.max_budget > 0:
if litellm.budget_duration is None:
raise Exception(
"budget_duration not set on Proxy. budget_duration is required to use max_budget."
)
# add proxy budget to db in the user table
await generate_key_helper_fn(
user_id=litellm_proxy_budget_name,