scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

feat - include litellm_budget table when getting end_user

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-22 13:30:08 -07:00
parent 9609df16d3
commit eac7e70dca
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/auth/auth_checks.py
View file

@ -219,7 +219,8 @@ async def get_end_user_object(
# else, check db
try:
response = await prisma_client.db.litellm_endusertable.find_unique(
where={"user_id": end_user_id}
where={"user_id": end_user_id},
include={"litellm_budget_table": True},
)
if response is None: