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

fix regen keys when no duration is passed

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-10 08:04:18 -07:00
parent 43cd657ac5
commit 428762542c
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -293,12 +293,12 @@ async def prepare_key_update_data(
continue continue
if v is not None and v not in ([], {}, 0): if v is not None and v not in ([], {}, 0):
non_default_values[k] = v non_default_values[k] = v
if "duration" in non_default_values: if "duration" in non_default_values:
duration = non_default_values.pop("duration") duration = non_default_values.pop("duration")
duration_s = _duration_in_seconds(duration=duration) if duration and (isinstance(duration, str)) and len(duration) > 0:
expires = datetime.now(timezone.utc) + timedelta(seconds=duration_s) duration_s = _duration_in_seconds(duration=duration)
non_default_values["expires"] = expires expires = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
non_default_values["expires"] = expires
if "budget_duration" in non_default_values: if "budget_duration" in non_default_values:
duration_s = _duration_in_seconds( duration_s = _duration_in_seconds(