phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +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 30a8071bf1
commit f148094d18
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(