From 5fb46cf32ecb835341b402a2f7995665fada8637 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 3 Apr 2024 20:17:34 -0700 Subject: [PATCH] build(schema.prisma): add proxy model table to prisma schema --- litellm/proxy/schema.prisma | 14 +++++++++++++- schema.prisma | 12 ++++++++++++ 2 files changed, 25 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma index 5ce0670b1..529ae7f2b 100644 --- a/litellm/proxy/schema.prisma +++ b/litellm/proxy/schema.prisma @@ -27,6 +27,18 @@ model LiteLLM_BudgetTable { end_users LiteLLM_EndUserTable[] // multiple end-users can have the same budget } +// Models on proxy +model LiteLLM_ProxyModelTable { + model_id String @id @default(uuid()) + model_name String + litellm_params Json + model_info Json? + created_at DateTime @default(now()) @map("created_at") + created_by String + updated_at DateTime @default(now()) @updatedAt @map("updated_at") + updated_by String +} + model LiteLLM_OrganizationTable { organization_id String @id @default(uuid()) organization_alias String @@ -174,4 +186,4 @@ model LiteLLM_UserNotifications { models String[] justification String status String // approved, disapproved, pending -} +} \ No newline at end of file diff --git a/schema.prisma b/schema.prisma index 6e2400a12..529ae7f2b 100644 --- a/schema.prisma +++ b/schema.prisma @@ -27,6 +27,18 @@ model LiteLLM_BudgetTable { end_users LiteLLM_EndUserTable[] // multiple end-users can have the same budget } +// Models on proxy +model LiteLLM_ProxyModelTable { + model_id String @id @default(uuid()) + model_name String + litellm_params Json + model_info Json? + created_at DateTime @default(now()) @map("created_at") + created_by String + updated_at DateTime @default(now()) @updatedAt @map("updated_at") + updated_by String +} + model LiteLLM_OrganizationTable { organization_id String @id @default(uuid()) organization_alias String