feat - create budgets when team/member_add

This commit is contained in:
Ishaan Jaff 2024-05-22 17:16:02 -07:00
parent 31fc6d79af
commit 50461eb22c
3 changed files with 35 additions and 7 deletions

View file

@ -25,6 +25,7 @@ model LiteLLM_BudgetTable {
organization LiteLLM_OrganizationTable[] // multiple orgs can have the same budget
keys LiteLLM_VerificationToken[] // multiple keys can have the same budget
end_users LiteLLM_EndUserTable[] // multiple end-users can have the same budget
team_membership LiteLLM_TeamMembership[] // budgets of Users within a Team
}
// Models on proxy
@ -208,4 +209,14 @@ model LiteLLM_UserNotifications {
models String[]
justification String
status String // approved, disapproved, pending
}
}
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)
budget_id String?
litellm_budget_table LiteLLM_BudgetTable? @relation(fields: [budget_id], references: [budget_id])
@@id([user_id, team_id])
}