scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

feat: handle TEAM_MEMBER_MODEL entity type in spend update queue aggregation

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-24 18:19:37 -07:00
parent 890652e27e
commit aeb0b7cad6
No known key found for this signature in database
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/db/db_transaction_queue/spend_update_queue.py
View file

@ -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"