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

Fix datetime.datetime.utcnow DeprecationWarning

Browse source 
Eliminates these warning when running tests:

```
$ cd litellm/tests
pytest test_key_generate_prisma.py -x -vv
...
====================================================================== warnings summary =======================================================================
...
test_key_generate_prisma.py::test_generate_and_call_with_expired_key
test_key_generate_prisma.py::test_key_with_no_permissions
  /Users/abramowi/Code/OpenSource/litellm/litellm/proxy/proxy_server.py:2934: DeprecationWarning: datetime.datetime.utcnow() is deprecated and scheduled for removal in a future version. Use timezone-aware objects to represent datetimes in UTC: datetime.datetime.now(datetime.UTC).
    expires = datetime.utcnow() + timedelta(seconds=duration_s)
...
```
This commit is contained in:
Marc Abramowitz 2024-05-16 11:56:02 -07:00
parent 5310706d1d
commit 4af6638be6
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_server.py
View file

@ -2863,19 +2863,19 @@ async def generate_key_helper_fn(
expires = None expires = None
else: else:
duration_s = _duration_in_seconds(duration=duration) duration_s = _duration_in_seconds(duration=duration)
expires = datetime.utcnow() + timedelta(seconds=duration_s) expires = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
if key_budget_duration is None: # one-time budget if key_budget_duration is None: # one-time budget
key_reset_at = None key_reset_at = None
else: else:
duration_s = _duration_in_seconds(duration=key_budget_duration) duration_s = _duration_in_seconds(duration=key_budget_duration)
key_reset_at = datetime.utcnow() + timedelta(seconds=duration_s) key_reset_at = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
if budget_duration is None: # one-time budget if budget_duration is None: # one-time budget
reset_at = None reset_at = None
else: else:
duration_s = _duration_in_seconds(duration=budget_duration) duration_s = _duration_in_seconds(duration=budget_duration)
reset_at = datetime.utcnow() + timedelta(seconds=duration_s) reset_at = datetime.now(timezone.utc) + timedelta(seconds=duration_s)
aliases_json = json.dumps(aliases) aliases_json = json.dumps(aliases)
config_json = json.dumps(config) config_json = json.dumps(config)
@ -4974,7 +4974,7 @@ async def update_key_fn(request: Request, data: UpdateKeyRequest):
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) duration_s = _duration_in_seconds(duration=duration)
expires = datetime.utcnow() + timedelta(seconds=duration_s) expires = datetime.datetime.now(timezone.utc) + timedelta(seconds=duration_s)
non_default_values["expires"] = expires non_default_values["expires"] = expires
response = await prisma_client.update_data( response = await prisma_client.update_data(