From e007bb65b5cd1a4b4b1d0b1fe0126fa9f67f017e Mon Sep 17 00:00:00 2001 From: kvadros Date: Mon, 7 Oct 2024 08:35:28 +0400 Subject: [PATCH] Proxy: include customer budget in responses (#5977) --- litellm/proxy/proxy_server.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 9e214c6ca..d2684ec4c 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -6029,7 +6029,8 @@ async def end_user_info( ) user_info = await prisma_client.db.litellm_endusertable.find_first( - where={"user_id": end_user_id} + where={"user_id": end_user_id}, + include={"litellm_budget_table": True} ) if user_info is None: @@ -6271,7 +6272,9 @@ async def list_team( detail={"error": CommonProxyErrors.db_not_connected_error.value}, ) - response = await prisma_client.db.litellm_endusertable.find_many() + response = await prisma_client.db.litellm_endusertable.find_many( + include={"litellm_budget_table": True} + ) returned_response: List[LiteLLM_EndUserTable] = [] for item in response: