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

test(test_users.py): test budgets with resets

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-24 15:30:30 -08:00
parent e471157d57
commit bb7705b494
3 changed files with 39 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/utils.py
View file

@ -384,7 +384,9 @@ class PrismaClient:
)
if response is not None:
# for prisma we need to cast the expires time to str
if isinstance(response.expires, datetime):
if response.expires is not None and isinstance(
response.expires, datetime
):
response.expires = response.expires.isoformat()
elif query_type == "find_all" and user_id is not None:
response = await self.db.litellm_verificationtoken.find_many(