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

fix(internal_user_endpoints.py): expose new 'internal_user_budget_duration' flag

Browse source 
Relevant to - https://github.com/BerriAI/litellm/issues/5106
This commit is contained in:
Krrish Dholakia 2024-08-08 13:05:03 -07:00
parent 8831f30e9d
commit 2776d9cb0d
3 changed files with 14 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

@ -91,6 +91,10 @@ async def new_user(
if litellm.max_internal_user_budget is not None:
data_json["max_budget"] = litellm.max_internal_user_budget
if "budget_duration" in data_json and data_json["budget_duration"] is None:
if litellm.internal_user_budget_duration is not None:
data_json["budget_duration"] = litellm.internal_user_budget_duration
response = await generate_key_helper_fn(request_type="user", **data_json)
# Admin UI Logic