feat - include litellm_budget table when getting end_user

2024-05-22 13:30:08 -07:00 · 2024-05-22 13:30:08 -07:00 · eac7e70dca
commit eac7e70dca
parent 9609df16d3
1 changed files with 2 additions and 1 deletions
--- a/litellm/proxy/auth/auth_checks.py
+++ b/litellm/proxy/auth/auth_checks.py
@ -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: