litellm-proxy-extras 0.1.19__py3-none-any.whl → 0.2.0__py3-none-any.whl

litellm_proxy_extras/migrations/20250514142245_add_guardrails_table/migration.sql

@@ -0,0 +1,15 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_GuardrailsTable" (
+    "guardrail_id" TEXT NOT NULL,
+    "guardrail_name" TEXT NOT NULL,
+    "litellm_params" JSONB NOT NULL,
+    "guardrail_info" JSONB,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_GuardrailsTable_pkey" PRIMARY KEY ("guardrail_id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_GuardrailsTable_guardrail_name_key" ON "LiteLLM_GuardrailsTable"("guardrail_name");
+

litellm_proxy_extras/migrations/20250522223020_managed_object_table/migration.sql

@@ -0,0 +1,32 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_ManagedFileTable" ADD COLUMN     "created_by" TEXT,
+ADD COLUMN     "flat_model_file_ids" TEXT[] DEFAULT ARRAY[]::TEXT[],
+ADD COLUMN     "updated_by" TEXT;
+
+-- CreateTable
+CREATE TABLE "LiteLLM_ManagedObjectTable" (
+    "id" TEXT NOT NULL,
+    "unified_object_id" TEXT NOT NULL,
+    "model_object_id" TEXT NOT NULL,
+    "file_object" JSONB NOT NULL,
+    "file_purpose" TEXT NOT NULL,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "created_by" TEXT,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+    "updated_by" TEXT,
+
+    CONSTRAINT "LiteLLM_ManagedObjectTable_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_ManagedObjectTable_unified_object_id_key" ON "LiteLLM_ManagedObjectTable"("unified_object_id");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_ManagedObjectTable_model_object_id_key" ON "LiteLLM_ManagedObjectTable"("model_object_id");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_ManagedObjectTable_unified_object_id_idx" ON "LiteLLM_ManagedObjectTable"("unified_object_id");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_ManagedObjectTable_model_object_id_idx" ON "LiteLLM_ManagedObjectTable"("model_object_id");
+

litellm_proxy_extras/schema.prisma

@@ -453,13 +453,30 @@ model LiteLLM_ManagedFileTable {
   id String @id @default(uuid())
   unified_file_id String @unique // The base64 encoded unified file ID
   file_object Json // Stores the OpenAIFileObject
-  model_mappings Json // Stores the mapping of model_id -> provider_file_id
+  model_mappings Json 
+  flat_model_file_ids String[]  @default([]) // Flat list of model file id's - for faster querying of model id -> unified file id
   created_at DateTime @default(now())
+  created_by String? 
   updated_at DateTime @updatedAt
+  updated_by String?
 
   @@index([unified_file_id])
 }
 
+model LiteLLM_ManagedObjectTable { // for batches or finetuning jobs which use the 
+  id String @id @default(uuid())
+  unified_object_id String @unique // The base64 encoded unified file ID
+  model_object_id String @unique // the id returned by the backend API provider 
+  file_object Json // Stores the OpenAIFileObject
+  file_purpose String // either 'batch' or 'fine-tune' 
+  created_at DateTime @default(now())
+  created_by String?
+  updated_at DateTime @updatedAt
+  updated_by String? 
+
+  @@index([unified_object_id])
+  @@index([model_object_id])
+}
 
 model LiteLLM_ManagedVectorStoresTable {
   vector_store_id String @id
@@ -470,4 +487,14 @@ model LiteLLM_ManagedVectorStoresTable {
   created_at DateTime @default(now())
   updated_at DateTime @updatedAt
   litellm_credential_name String?
+}
+
+// Guardrails table for storing guardrail configurations
+model LiteLLM_GuardrailsTable {
+  guardrail_id String @id @default(uuid())
+  guardrail_name String @unique
+  litellm_params Json
+  guardrail_info Json?
+  created_at DateTime @default(now())
+  updated_at DateTime @updatedAt
 }

{litellm_proxy_extras-0.1.19.dist-info → litellm_proxy_extras-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: litellm-proxy-extras
-Version: 0.1.19
+Version: 0.2.0
 Summary: Additional files for the LiteLLM Proxy. Reduces the size of the main litellm package.
 Author: BerriAI
 Requires-Python: >=3.8, !=2.7.*, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*, !=3.6.*, !=3.7.*
@@ -29,7 +29,7 @@ pip install litellm-proxy-extras
 OR 
 
 ```bash
-pip install litellm[proxy] # installs litellm-proxy-extras and other proxy dependencies.
+pip install litellm[proxy] # installs litellm-proxy-extras and other proxy dependencies
 ```
 
 To use the migrations, run:

{litellm_proxy_extras-0.1.19.dist-info → litellm_proxy_extras-0.2.0.dist-info}/RECORD

@@ -19,10 +19,12 @@ litellm_proxy_extras/migrations/20250507184818_add_mcp_key_team_permission_mgmt/
 litellm_proxy_extras/migrations/20250508072103_add_status_to_spendlogs/migration.sql,sha256=V4mIhoi3n8sUzV8_YY9AA_wkgvk65m5ZOAU3_DtoUo8,77
 litellm_proxy_extras/migrations/20250509141545_use_big_int_for_daily_spend_tables/migration.sql,sha256=UPaAXsryGvVZT_SnOf-1yHu5bKPTVr2gP3zY64FKR7o,1331
 litellm_proxy_extras/migrations/20250510142544_add_session_id_index_spend_logs/migration.sql,sha256=M_y4xLnlGApiUIovLv0xA7lL5gJBZWrcFe7b5MSUOik,102
+litellm_proxy_extras/migrations/20250514142245_add_guardrails_table/migration.sql,sha256=mxBhXoedxkabrkn7YPiKn-OVmoXh6RsSEXY9a0-FkQk,506
+litellm_proxy_extras/migrations/20250522223020_managed_object_table/migration.sql,sha256=SbchBsESJa8M5O4R1iqvfgssX5StSrL85Umd2Drns_Y,1185
 litellm_proxy_extras/migrations/migration_lock.toml,sha256=HbF6jQUaoTYRBzZ1LF4fi37ZK26o6AMRL7viSXBHwhA,24
-litellm_proxy_extras/schema.prisma,sha256=k3TXUCJaMkZ1q_9Je2V0aoeYYw2osqyMFUTSxHRLfEI,18573
+litellm_proxy_extras/schema.prisma,sha256=AyzMzVmvCBs6XjbdjyDTazdAqVGVG1SzZFmhIvOxBZg,19530
 litellm_proxy_extras/utils.py,sha256=QImDkrfX_NGCp0tJ4T9IZpmXyzVZqey4jnVVEO0Hcrs,15050
-litellm_proxy_extras-0.1.19.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
-litellm_proxy_extras-0.1.19.dist-info/METADATA,sha256=hauDdaBeDu-id2S09n3YVOUleHcaYahlF9YPhob2ssM,1268
-litellm_proxy_extras-0.1.19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-litellm_proxy_extras-0.1.19.dist-info/RECORD,,
+litellm_proxy_extras-0.2.0.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
+litellm_proxy_extras-0.2.0.dist-info/METADATA,sha256=joZBCZ2gDwvwwlLiTz_m0-hDKVYe8hDp5lgGDtQCMpQ,1266
+litellm_proxy_extras-0.2.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+litellm_proxy_extras-0.2.0.dist-info/RECORD,,