From 211903073889ae7c0583a873f473088e907357b3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 28 Aug 2024 21:58:41 -0700
Subject: [PATCH] fix(team_endpoints.py): update to include the budget in the
 response

---
 litellm/proxy/management_endpoints/team_endpoints.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py
index 0b99b4c43..5b6885ecd 100644
--- a/litellm/proxy/management_endpoints/team_endpoints.py
+++ b/litellm/proxy/management_endpoints/team_endpoints.py
@@ -917,7 +917,8 @@ async def team_info(
 
         ## GET ALL MEMBERSHIPS ##
         team_memberships = await prisma_client.db.litellm_teammembership.find_many(
-            where={"team_id": team_id}
+            where={"team_id": team_id},
+            include={"litellm_budget_table": True},
         )
         return {
             "team_id": team_id,