scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix(proxy): clear budget_duration and budget_reset_at when set to null on key update

Browse source
This commit is contained in:
PRABHU KIRAN VANDRANKI 2026-05-19 19:17:15 -04:00
parent a72414a061
commit 05542d9b60
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1842,6 +1842,10 @@ async def prepare_key_update_data(
key_reset_at = get_budget_reset_time(budget_duration=budget_duration)
non_default_values["budget_reset_at"] = key_reset_at
non_default_values["budget_duration"] = budget_duration
else:
# budget_duration explicitly set to null: clear the field and reset time
non_default_values["budget_duration"] = None
non_default_values["budget_reset_at"] = None
if "budget_limits" in non_default_values and non_default_values["budget_limits"]:
from litellm.proxy.common_utils.timezone_utils import get_budget_reset_time