phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(proxy_server.py): fix /key/update endpoint to update key duration

Browse source 
also adds a test for this to our ci/cd
This commit is contained in:
Krrish Dholakia 2024-03-29 21:47:10 -07:00
parent 68a1fa36cf
commit af2eabba91
2 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/proxy_server.py
View file

@ -4189,6 +4189,13 @@ async def update_key_fn(request: Request, data: UpdateKeyRequest):
0,
): # models default to [], spend defaults to 0, we should not reset these values
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.utcnow() + timedelta(seconds=duration_s)
non_default_values["expires"] = expires
response = await prisma_client.update_data(
token=key, data={**non_default_values, "token": key}
)