phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(internal_user_endpoints.py): respect 'max_user_budget' for new internal user's

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-07 18:50:40 -07:00
parent 639f84d609
commit b7e31638fd
2 changed files with 40 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/management_endpoints/internal_user_endpoints.py
View file

@ -87,6 +87,10 @@ async def new_user(
"user" # only create a user, don't create key if 'auto_create_key' set to False
)
if "max_budget" in data_json and data_json["max_budget"] is None:
if litellm.max_user_budget is not None:
data_json["max_budget"] = litellm.max_user_budget
response = await generate_key_helper_fn(request_type="user", **data_json)
# Admin UI Logic