litellm-proxy-extras 0.2.5__py3-none-any.whl → 0.2.11__py3-none-any.whl

litellm_proxy_extras/migrations/20250625145206_cascade_budget_and_loosen_managed_file_json/migration.sql

@@ -0,0 +1,9 @@
+-- DropForeignKey
+ALTER TABLE "LiteLLM_TeamMembership" DROP CONSTRAINT "LiteLLM_TeamMembership_budget_id_fkey";
+
+-- AlterTable
+ALTER TABLE "LiteLLM_ManagedFileTable" ALTER COLUMN "file_object" DROP NOT NULL;
+
+-- AddForeignKey
+ALTER TABLE "LiteLLM_TeamMembership" ADD CONSTRAINT "LiteLLM_TeamMembership_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE SET NULL ON UPDATE CASCADE;
+

litellm_proxy_extras/migrations/20250625213625_add_status_to_managed_object_table/migration.sql

@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_ManagedObjectTable" ADD COLUMN     "status" TEXT;
+

litellm_proxy_extras/migrations/20250707212517_add_mcp_info_column_mcp_servers/migration.sql

@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_MCPServerTable" ADD COLUMN     "mcp_info" JSONB DEFAULT '{}';
+

litellm_proxy_extras/migrations/20250707230009_add_mcp_namespaced_tool_name/migration.sql

@@ -0,0 +1,42 @@
+-- DropIndex
+DROP INDEX "LiteLLM_DailyTagSpend_tag_date_api_key_model_custom_llm_pro_key";
+
+-- DropIndex
+DROP INDEX "LiteLLM_DailyTeamSpend_team_id_date_api_key_model_custom_ll_key";
+
+-- DropIndex
+DROP INDEX "LiteLLM_DailyUserSpend_user_id_date_api_key_model_custom_ll_key";
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyTagSpend" ADD COLUMN     "mcp_namespaced_tool_name" TEXT,
+ALTER COLUMN "model" DROP NOT NULL;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyTeamSpend" ADD COLUMN     "mcp_namespaced_tool_name" TEXT,
+ALTER COLUMN "model" DROP NOT NULL;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyUserSpend" ADD COLUMN     "mcp_namespaced_tool_name" TEXT,
+ALTER COLUMN "model" DROP NOT NULL;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_SpendLogs" ADD COLUMN     "mcp_namespaced_tool_name" TEXT;
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyTagSpend_mcp_namespaced_tool_name_idx" ON "LiteLLM_DailyTagSpend"("mcp_namespaced_tool_name");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_DailyTagSpend_tag_date_api_key_model_custom_llm_pro_key" ON "LiteLLM_DailyTagSpend"("tag", "date", "api_key", "model", "custom_llm_provider", "mcp_namespaced_tool_name");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyTeamSpend_mcp_namespaced_tool_name_idx" ON "LiteLLM_DailyTeamSpend"("mcp_namespaced_tool_name");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_DailyTeamSpend_team_id_date_api_key_model_custom_ll_key" ON "LiteLLM_DailyTeamSpend"("team_id", "date", "api_key", "model", "custom_llm_provider", "mcp_namespaced_tool_name");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyUserSpend_mcp_namespaced_tool_name_idx" ON "LiteLLM_DailyUserSpend"("mcp_namespaced_tool_name");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_DailyUserSpend_user_id_date_api_key_model_custom_ll_key" ON "LiteLLM_DailyUserSpend"("user_id", "date", "api_key", "model", "custom_llm_provider", "mcp_namespaced_tool_name");
+

litellm_proxy_extras/migrations/20250711220620_add_stdio_mcp/migration.sql

@@ -0,0 +1,10 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_MCPServerTable" ADD COLUMN     "args" TEXT[] DEFAULT ARRAY[]::TEXT[],
+ADD COLUMN     "command" TEXT,
+ADD COLUMN     "env" JSONB DEFAULT '{}',
+ADD COLUMN     "mcp_access_groups" TEXT[],
+ALTER COLUMN "url" DROP NOT NULL;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_ObjectPermissionTable" ADD COLUMN     "mcp_access_groups" TEXT[] DEFAULT ARRAY[]::TEXT[];
+

litellm_proxy_extras/migrations/20250718125714_add_litellm_params_to_vector_stores/migration.sql

@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_ManagedVectorStoresTable" ADD COLUMN     "litellm_params" JSONB;
+

litellm_proxy_extras/schema.prisma

@@ -155,8 +155,8 @@ model LiteLLM_UserTable {
 model LiteLLM_ObjectPermissionTable {
   object_permission_id  String         @id @default(uuid())
   mcp_servers           String[]       @default([])
+  mcp_access_groups     String[]       @default([])
   vector_stores         String[]       @default([])
-  
   teams                 LiteLLM_TeamTable[]
   verification_tokens   LiteLLM_VerificationToken[]
   organizations         LiteLLM_OrganizationTable[]
@@ -168,7 +168,7 @@ model LiteLLM_MCPServerTable {
   server_id    String         @id @default(uuid())
   alias        String?
   description  String?
-  url          String
+  url          String?
   transport    String         @default("sse")
   spec_version String         @default("2025-03-26")
   auth_type    String?        
@@ -176,6 +176,12 @@ model LiteLLM_MCPServerTable {
   created_by   String?
   updated_at   DateTime?      @default(now()) @updatedAt @map("updated_at")
   updated_by   String?
+  mcp_info     Json?          @default("{}")
+  mcp_access_groups String[]
+  // Stdio-specific fields
+  command      String?
+  args         String[]       @default([])
+  env          Json?          @default("{}")
 }
 
 // Generate Tokens for Proxy
@@ -262,6 +268,7 @@ model LiteLLM_SpendLogs {
   response            Json?     @default("{}")
   session_id          String?
   status              String?
+  mcp_namespaced_tool_name String? 
   proxy_server_request Json?     @default("{}")
   @@index([startTime])
   @@index([end_user])
@@ -360,9 +367,10 @@ model LiteLLM_DailyUserSpend {
   user_id             String? 
   date                String
   api_key             String   
-  model               String   
+  model               String?
   model_group         String?  
-  custom_llm_provider String?  
+  custom_llm_provider String?
+  mcp_namespaced_tool_name String?
   prompt_tokens       BigInt      @default(0)
   completion_tokens   BigInt      @default(0)
   cache_read_input_tokens     BigInt      @default(0)
@@ -374,11 +382,12 @@ model LiteLLM_DailyUserSpend {
   created_at          DateTime @default(now())
   updated_at          DateTime @updatedAt
 
-  @@unique([user_id, date, api_key, model, custom_llm_provider])
+  @@unique([user_id, date, api_key, model, custom_llm_provider, mcp_namespaced_tool_name])
   @@index([date])
   @@index([user_id])
   @@index([api_key])
   @@index([model])
+  @@index([mcp_namespaced_tool_name])
 }
 
 // Track daily team spend metrics per model and key
@@ -387,9 +396,10 @@ model LiteLLM_DailyTeamSpend {
   team_id             String?
   date                String
   api_key             String   
-  model               String   
+  model               String?
   model_group         String?  
-  custom_llm_provider String?  
+  custom_llm_provider String?
+  mcp_namespaced_tool_name String?
   prompt_tokens       BigInt      @default(0)
   completion_tokens   BigInt      @default(0)
   cache_read_input_tokens     BigInt      @default(0)
@@ -401,11 +411,12 @@ model LiteLLM_DailyTeamSpend {
   created_at          DateTime @default(now())
   updated_at          DateTime @updatedAt
 
-  @@unique([team_id, date, api_key, model, custom_llm_provider])
+  @@unique([team_id, date, api_key, model, custom_llm_provider, mcp_namespaced_tool_name])
   @@index([date])
   @@index([team_id])
   @@index([api_key])
   @@index([model])
+  @@index([mcp_namespaced_tool_name])
 }
 
 // Track daily team spend metrics per model and key
@@ -414,9 +425,10 @@ model LiteLLM_DailyTagSpend {
   tag                 String?   
   date                String
   api_key             String   
-  model               String   
+  model               String?
   model_group         String?  
-  custom_llm_provider String?  
+  custom_llm_provider String?
+  mcp_namespaced_tool_name String?
   prompt_tokens       BigInt      @default(0)
   completion_tokens   BigInt      @default(0)
   cache_read_input_tokens     BigInt      @default(0)
@@ -428,11 +440,12 @@ model LiteLLM_DailyTagSpend {
   created_at          DateTime @default(now())
   updated_at          DateTime @updatedAt
 
-  @@unique([tag, date, api_key, model, custom_llm_provider])
+  @@unique([tag, date, api_key, model, custom_llm_provider, mcp_namespaced_tool_name])
   @@index([date])
   @@index([tag])
   @@index([api_key])
   @@index([model])
+  @@index([mcp_namespaced_tool_name])
 }
 
 
@@ -453,8 +466,8 @@ enum JobStatus {
 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 
+  file_object Json? // Stores the OpenAIFileObject
+  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? 
@@ -469,7 +482,8 @@ model LiteLLM_ManagedObjectTable { // for batches or finetuning jobs which use t
   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' 
+  file_purpose String // either 'batch' or 'fine-tune'
+  status String? // check if batch cost has been tracked  
   created_at DateTime @default(now())
   created_by String?
   updated_at DateTime @updatedAt
@@ -488,6 +502,7 @@ model LiteLLM_ManagedVectorStoresTable {
   created_at DateTime @default(now())
   updated_at DateTime @updatedAt
   litellm_credential_name String?
+  litellm_params Json?
 }
 
 // Guardrails table for storing guardrail configurations

{litellm_proxy_extras-0.2.5.dist-info → litellm_proxy_extras-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: litellm-proxy-extras
-Version: 0.2.5
+Version: 0.2.11
 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.*

{litellm_proxy_extras-0.2.5.dist-info → litellm_proxy_extras-0.2.11.dist-info}/RECORD

@@ -25,10 +25,16 @@ litellm_proxy_extras/migrations/20250526154401_allow_null_entity_id/migration.sq
 litellm_proxy_extras/migrations/20250528185438_add_vector_stores_to_object_permissions/migration.sql,sha256=CMoueiGIy2-_7Ik7ktzGCRPfhBH43Y7nJkAvfxcYnI0,122
 litellm_proxy_extras/migrations/20250603210143_cascade_budget_changes/migration.sql,sha256=LMUCGnK6z6Bh_mKikp5_0Xs_EnsoPMb-dTY-sJBnXVg,332
 litellm_proxy_extras/migrations/20250618225828_add_health_check_table/migration.sql,sha256=FpLvgnxHKp3Jb-USx-EWOGeVCXDpSznGBloqlWAm9bg,980
+litellm_proxy_extras/migrations/20250625145206_cascade_budget_and_loosen_managed_file_json/migration.sql,sha256=AjRZvl6dlMp9YRosTy0Iaa6ZqMXAydFY7FlzOp-T09I,429
+litellm_proxy_extras/migrations/20250625213625_add_status_to_managed_object_table/migration.sql,sha256=XcFDSGnTIOue1laLzJL062Mv6mI2bu05KCOgrzvWQK8,86
+litellm_proxy_extras/migrations/20250707212517_add_mcp_info_column_mcp_servers/migration.sql,sha256=2r7drloJuEDy-x9_fsQXB613YJ7gg6yIsL5MhFEv-Nk,98
+litellm_proxy_extras/migrations/20250707230009_add_mcp_namespaced_tool_name/migration.sql,sha256=L973Q5PV5pFcpuZzyOzo6aFNK02EZK-KAHfcyjtPAIU,1854
+litellm_proxy_extras/migrations/20250711220620_add_stdio_mcp/migration.sql,sha256=t3telBGnxzo8kXQYnyR69vgxcPYf9ysoWt1ySRHnJgE,381
+litellm_proxy_extras/migrations/20250718125714_add_litellm_params_to_vector_stores/migration.sql,sha256=yZfez4-HKbZeRQs4E6QhrMA7wuyrBQrYzTrW9S1HwcY,101
 litellm_proxy_extras/migrations/migration_lock.toml,sha256=HbF6jQUaoTYRBzZ1LF4fi37ZK26o6AMRL7viSXBHwhA,24
-litellm_proxy_extras/schema.prisma,sha256=WTTSbxUIE5ZG12WKh6wodpuc8u7P3ethIhFLsrDO-lA,20054
+litellm_proxy_extras/schema.prisma,sha256=sKIAhiHqvCFkEre8iHTaQY61MKE4SyMjnzJKZBhccU0,20718
 litellm_proxy_extras/utils.py,sha256=QImDkrfX_NGCp0tJ4T9IZpmXyzVZqey4jnVVEO0Hcrs,15050
-litellm_proxy_extras-0.2.5.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
-litellm_proxy_extras-0.2.5.dist-info/METADATA,sha256=Mvu_t_YfK3xhTwLixStEqvSd1JyjHTFDqFKm0EVmKdU,1266
-litellm_proxy_extras-0.2.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-litellm_proxy_extras-0.2.5.dist-info/RECORD,,
+litellm_proxy_extras-0.2.11.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
+litellm_proxy_extras-0.2.11.dist-info/METADATA,sha256=bTt-shIFFCjyOXTK5G8EusYBhuan6xPIfPvdi5X2DYU,1267
+litellm_proxy_extras-0.2.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+litellm_proxy_extras-0.2.11.dist-info/RECORD,,