From aeb0b7cad65750591fc929a7d013d8e9bd7fe660 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaan@berri.ai>
Date: Fri, 24 Apr 2026 18:19:37 -0700
Subject: [PATCH] feat: handle TEAM_MEMBER_MODEL entity type in spend update
 queue aggregation

---
 litellm/proxy/db/db_transaction_queue/spend_update_queue.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/litellm/proxy/db/db_transaction_queue/spend_update_queue.py b/litellm/proxy/db/db_transaction_queue/spend_update_queue.py
index 8100a1e8a12..7b6adc79cd2 100644
--- a/litellm/proxy/db/db_transaction_queue/spend_update_queue.py
+++ b/litellm/proxy/db/db_transaction_queue/spend_update_queue.py
@@ -143,6 +143,7 @@ class SpendUpdateQueue(BaseUpdateQueue):
             key_list_transactions={},
             team_list_transactions={},
             team_member_list_transactions={},
+            team_member_model_list_transactions={},
             org_list_transactions={},
             tag_list_transactions={},
             agent_list_transactions={},
@@ -155,6 +156,7 @@ class SpendUpdateQueue(BaseUpdateQueue):
             Litellm_EntityType.KEY: "key_list_transactions",
             Litellm_EntityType.TEAM: "team_list_transactions",
             Litellm_EntityType.TEAM_MEMBER: "team_member_list_transactions",
+            Litellm_EntityType.TEAM_MEMBER_MODEL: "team_member_model_list_transactions",
             Litellm_EntityType.ORGANIZATION: "org_list_transactions",
             Litellm_EntityType.TAG: "tag_list_transactions",
             Litellm_EntityType.AGENT: "agent_list_transactions",
@@ -201,6 +203,10 @@ class SpendUpdateQueue(BaseUpdateQueue):
                 transactions_dict = db_spend_update_transactions[
                     "team_member_list_transactions"
                 ]
+            elif dict_key == "team_member_model_list_transactions":
+                transactions_dict = db_spend_update_transactions[
+                    "team_member_model_list_transactions"
+                ]
             elif dict_key == "org_list_transactions":
                 transactions_dict = db_spend_update_transactions[
                     "org_list_transactions"