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

test(test_key_generate_prisma.py): add unit testing for global proxy budget

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-24 17:43:01 -08:00
parent 8518e2e2d1
commit 7a8f10c1c0
3 changed files with 182 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/proxy/utils.py
View file

@ -454,11 +454,10 @@ class PrismaClient:
# Execute the raw query
# The asterisk before `user_id_list` unpacks the list into separate arguments
response = await self.db.query_raw(sql_query)
return response
elif table_name == "user" and query_type == "find_all":
response = await self.db.litellm_usertable.find_many( # type: ignore
order={"spend": "desc"},
)
elif query_type == "find_all":
response = await self.db.litellm_usertable.find_many( # type: ignore
order={"spend": "desc"},
)
return response
elif table_name == "spend":
verbose_proxy_logger.debug(