phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_organizations.py): add testing for /organization/new endpoint

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-02 12:13:54 -08:00
parent eedd446a6a
commit 6fb19c5d42
3 changed files with 53 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
schema.prisma
View file

@ -21,11 +21,12 @@ model LiteLLM_BudgetTable {
created_by String
updated_at DateTime @default(now()) @updatedAt @map("updated_at")
updated_by String
organization LiteLLM_OrganizationTable[] // multiple orgs can have the same budget
}
model LiteLLM_OrganizationTable {
organization_id String @id @default(uuid())
organization_alias String?
organization_alias String
budget_id String
metadata Json @default("{}")
models String[]
@ -35,7 +36,8 @@ model LiteLLM_OrganizationTable {
created_by String
updated_at DateTime @default(now()) @updatedAt @map("updated_at")
updated_by String
litellm_budget_table LiteLLM_BudgetTable @relation(fields: [budget_id], references: [budget_id])
litellm_budget_table LiteLLM_BudgetTable? @relation(fields: [budget_id], references: [budget_id])
teams LiteLLM_TeamTable[]
}
// Assign prod keys to groups, not individuals
@ -59,7 +61,7 @@ model LiteLLM_TeamTable {
updated_at DateTime @default(now()) @updatedAt @map("updated_at")
model_spend Json @default("{}")
model_max_budget Json @default("{}")
litellm_organization_table LiteLLM_OrganizationTable @relation(fields: [organization_id], references: [organization_id])
litellm_organization_table LiteLLM_OrganizationTable? @relation(fields: [organization_id], references: [organization_id])
}
// Track spend, rate limit, budget Users