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"