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

fix(key_management_endpoints.py): if budget duration set, set budget_reset_at

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-24 15:02:03 -07:00
parent dc3b39ca71
commit b5c5ed2209
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -333,6 +333,13 @@ async def update_key_fn(
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(
duration=non_default_values["budget_duration"]
)
key_reset_at = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
non_default_values["budget_reset_at"] = key_reset_at
response = await prisma_client.update_data(
token=key, data={**non_default_values, "token": key}
)