From 6d9e3e0e8d36d1905c5f0add7a348070040f8b20 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaan@berri.ai>
Date: Fri, 24 Apr 2026 18:19:26 -0700
Subject: [PATCH] schema: add model_spend Json column to LiteLLM_TeamMembership
 for per-model spend tracking

---
 litellm/proxy/schema.prisma | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma
index 34686148ce0..e83ef330c9b 100644
--- a/litellm/proxy/schema.prisma
+++ b/litellm/proxy/schema.prisma
@@ -613,11 +613,12 @@ model LiteLLM_UserNotifications {
 
 model LiteLLM_TeamMembership {
   // Use this table to track the Internal User's Spend within a Team + Set Budgets, rpm limits for the user within the team
-  user_id    String
-  team_id    String
-  spend      Float    @default(0.0)
-  total_spend Float @default(0.0)
-  budget_id String?
+  user_id     String
+  team_id     String
+  spend       Float    @default(0.0)
+  total_spend Float    @default(0.0)
+  budget_id   String?
+  model_spend Json     @default("{}")
   litellm_budget_table LiteLLM_BudgetTable?   @relation(fields: [budget_id], references: [budget_id])
   @@id([user_id, team_id])
 }