scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

Incorporate Greptile review- None guard before mapping initial_budget_reset_at, echo stored value in response

Browse source
This commit is contained in:
Aayush Malviya 2026-03-06 12:08:18 +05:30
parent 28bf548ea8
commit 1a635ff79e
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -99,8 +99,10 @@ def _update_internal_new_user_params(data_json: dict, data: NewUserRequest) -> d
data_json.pop("teams", None) # handled separately
if "initial_budget_reset_at" in data_json:
if "initial_budget_reset_at" in data_json and data_json.get("initial_budget_reset_at") is not None:
data_json["budget_reset_at"] = data_json.pop("initial_budget_reset_at")
else:
data_json.pop("initial_budget_reset_at", None)
return data_json
@ -427,8 +429,8 @@ async def new_user(
response_dict["key"] = response.get("token", "")
if data.initial_budget_reset_at is not None:
response_dict["initial_budget_reset_at"] = data.initial_budget_reset_at
if data.initial_budget_reset_at is not None and response.get("budget_reset_at") is not None:
response_dict["initial_budget_reset_at"] = response.get("budget_reset_at")
new_user_response = NewUserResponse(**response_dict)