From 428762542c372b4b910f6130b695d8f47ca0bfe9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 10 Sep 2024 08:04:18 -0700 Subject: [PATCH] fix regen keys when no duration is passed --- .../management_endpoints/key_management_endpoints.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index ec17358e154..3173e38864c 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -293,12 +293,12 @@ async def prepare_key_update_data( continue if v is not None and v not in ([], {}, 0): non_default_values[k] = v - if "duration" in non_default_values: duration = non_default_values.pop("duration") - duration_s = _duration_in_seconds(duration=duration) - expires = datetime.now(timezone.utc) + timedelta(seconds=duration_s) - non_default_values["expires"] = expires + if duration and (isinstance(duration, str)) and len(duration) > 0: + duration_s = _duration_in_seconds(duration=duration) + expires = datetime.now(timezone.utc) + timedelta(seconds=duration_s) + non_default_values["expires"] = expires if "budget_duration" in non_default_values: duration_s = _duration_in_seconds(