From 7c79cc06acbdd6c45efaeba5978a151005f5bf4b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 3 Sep 2024 08:40:22 -0700 Subject: [PATCH] fix(internal_user_endpoints.py): set budget_reset_at for /user/update --- .../proxy/management_endpoints/internal_user_endpoints.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/litellm/proxy/management_endpoints/internal_user_endpoints.py b/litellm/proxy/management_endpoints/internal_user_endpoints.py index 76ffc9089d9..2a359f92e01 100644 --- a/litellm/proxy/management_endpoints/internal_user_endpoints.py +++ b/litellm/proxy/management_endpoints/internal_user_endpoints.py @@ -529,6 +529,13 @@ async def user_update( non_default_values["budget_duration"] = ( litellm.internal_user_budget_duration ) + duration_s = _duration_in_seconds( + duration=non_default_values["budget_duration"] + ) + user_reset_at = datetime.now(timezone.utc) + timedelta( + seconds=duration_s + ) + non_default_values["budget_reset_at"] = user_reset_at ## ADD USER, IF NEW ## verbose_proxy_logger.debug("/user/update: Received data = %s", data)