litellm-proxy-extras 0.2.25__py3-none-any.whl → 0.2.27__py3-none-any.whl

litellm_proxy_extras/migrations/20250918083359_drop_spec_version_column_from_mcp_table/migration.sql

@@ -5,4 +5,4 @@
 
 */
 -- AlterTable
-ALTER TABLE "public"."LiteLLM_MCPServerTable" DROP COLUMN "spec_version";
+ALTER TABLE "LiteLLM_MCPServerTable" DROP COLUMN "spec_version";

litellm_proxy_extras/migrations/20251006143948_add_mcp_tool_permissions/migration.sql

@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_ObjectPermissionTable" ADD COLUMN     "mcp_tool_permissions" JSONB;
+

litellm_proxy_extras/migrations/20251011084309_add_tag_table/migration.sql

@@ -0,0 +1,18 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_TagTable" (
+    "tag_name" TEXT NOT NULL,
+    "description" TEXT,
+    "models" TEXT[],
+    "model_info" JSONB,
+    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
+    "budget_id" TEXT,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "created_by" TEXT,
+    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+
+    CONSTRAINT "LiteLLM_TagTable_pkey" PRIMARY KEY ("tag_name")
+);
+
+-- AddForeignKey
+ALTER TABLE "LiteLLM_TagTable" ADD CONSTRAINT "LiteLLM_TagTable_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE SET NULL ON UPDATE CASCADE;
+

litellm_proxy_extras/schema.prisma

@@ -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
+  tags LiteLLM_TagTable[] // multiple tags can have the same budget
   team_membership LiteLLM_TeamMembership[] // budgets of Users within a Team 
   organization_membership LiteLLM_OrganizationMembership[] // budgets of Users within a Organization 
 }
@@ -156,6 +157,7 @@ model LiteLLM_ObjectPermissionTable {
   object_permission_id  String         @id @default(uuid())
   mcp_servers           String[]       @default([])
   mcp_access_groups     String[]       @default([])
+  mcp_tool_permissions  Json?          // Tool-level permissions for MCP servers. Format: {"server_id": ["tool_name_1", "tool_name_2"]}
   vector_stores         String[]       @default([])
   teams                 LiteLLM_TeamTable[]
   verification_tokens   LiteLLM_VerificationToken[]
@@ -244,6 +246,20 @@ model LiteLLM_EndUserTable {
   blocked Boolean @default(false)
 }
 
+// Track tags with budgets and spend
+model LiteLLM_TagTable {
+  tag_name String @id
+  description String?
+  models String[]
+  model_info Json? // maps model_id to model_name
+  spend Float @default(0.0)
+  budget_id String?
+  litellm_budget_table LiteLLM_BudgetTable? @relation(fields: [budget_id], references: [budget_id])
+  created_at DateTime @default(now()) @map("created_at")
+  created_by String?
+  updated_at DateTime @default(now()) @updatedAt @map("updated_at")
+}
+
 // store proxy config.yaml
 model LiteLLM_Config {
   param_name String @id

litellm_proxy_extras/utils.py

@@ -131,7 +131,9 @@ class ProxyExtrasDBManager:
         )
 
     @staticmethod
-    def _resolve_all_migrations(migrations_dir: str, schema_path: str):
+    def _resolve_all_migrations(
+        migrations_dir: str, schema_path: str, mark_all_applied: bool = True
+    ):
         """
         1. Compare the current database state to schema.prisma and generate a migration for the diff.
         2. Run prisma migrate deploy to apply any pending migrations.
@@ -210,6 +212,8 @@ class ProxyExtrasDBManager:
             logger.warning("Migration diff application timed out.")
 
         # 3. Mark all migrations as applied
+        if not mark_all_applied:
+            return
         migration_names = ProxyExtrasDBManager._get_migration_names(migrations_dir)
         logger.info(f"Resolving {len(migration_names)} migrations")
         for migration_name in migration_names:
@@ -263,6 +267,13 @@ class ProxyExtrasDBManager:
                         logger.info(f"prisma migrate deploy stdout: {result.stdout}")
 
                         logger.info("prisma migrate deploy completed")
+
+                        # Run sanity check to ensure DB matches schema
+                        logger.info("Running post-migration sanity check...")
+                        ProxyExtrasDBManager._resolve_all_migrations(
+                            migrations_dir, schema_path, mark_all_applied=False
+                        )
+                        logger.info("✅ Post-migration sanity check completed")
                         return True
                     except subprocess.CalledProcessError as e:
                         logger.info(f"prisma db error: {e.stderr}, e: {e.stdout}")

{litellm_proxy_extras-0.2.25.dist-info → litellm_proxy_extras-0.2.27.dist-info}/METADATA

@@ -1,8 +1,7 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: litellm-proxy-extras
-Version: 0.2.25
+Version: 0.2.27
 Summary: Additional files for the LiteLLM Proxy. Reduces the size of the main litellm package.
-License-File: LICENSE
 Author: BerriAI
 Requires-Python: >=3.8, !=2.7.*, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*, !=3.6.*, !=3.7.*
 Classifier: Programming Language :: Python :: 3
@@ -10,8 +9,6 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
 Project-URL: Documentation, https://docs.litellm.ai
 Project-URL: Homepage, https://litellm.ai
 Project-URL: Repository, https://github.com/BerriAI/litellm

{litellm_proxy_extras-0.2.25.dist-info → litellm_proxy_extras-0.2.27.dist-info}/RECORD

@@ -34,14 +34,16 @@ litellm_proxy_extras/migrations/20250711220620_add_stdio_mcp/migration.sql,sha25
 litellm_proxy_extras/migrations/20250718125714_add_litellm_params_to_vector_stores/migration.sql,sha256=yZfez4-HKbZeRQs4E6QhrMA7wuyrBQrYzTrW9S1HwcY,101
 litellm_proxy_extras/migrations/20250802162330_prompt_table/migration.sql,sha256=rssp_rCzwY5veFUJeGSOoGOvck3IZ9sL86RuHzQJ-s0,452
 litellm_proxy_extras/migrations/20250806095134_rename_alias_to_server_name_mcp_table/migration.sql,sha256=Bge3bWO96jFhVzdjHX5sD7SzCLnrrYMq_6ilzJFAQ6w,465
-litellm_proxy_extras/migrations/20250918083359_drop_spec_version_column_from_mcp_table/migration.sql,sha256=7Q2E0OSK4wZXi_GeNR9qF-SbQh1OHvJG-A-02nQxBgU,240
+litellm_proxy_extras/migrations/20250918083359_drop_spec_version_column_from_mcp_table/migration.sql,sha256=hmAt0uvPXq9Nvtf92HqRhrGTcBxEFWwxOjsG2zTCtg4,231
 litellm_proxy_extras/migrations/20250926194702_unnamed_migration/migration.sql,sha256=YCdSPn3yPse6HoZGW9lFEqy49xOf4yuJH_qrf8lrJIE,294
 litellm_proxy_extras/migrations/20251003165142_add_allowed_tools_to_mcp/migration.sql,sha256=D8olxfIAbERAzYLI6ldT40KUB_3t9CeRgl4n2DT45lU,115
 litellm_proxy_extras/migrations/20251003190954_extra_headers_to_mcp_table/migration.sql,sha256=TEi08V1ZQAondOxdSIP6MAU2P5nueSpVvhtmdoSGp-c,115
+litellm_proxy_extras/migrations/20251006143948_add_mcp_tool_permissions/migration.sql,sha256=RFISpCSAb9YawwQHBdkap9RBn-NrBcmhflOTyUJuljk,104
+litellm_proxy_extras/migrations/20251011084309_add_tag_table/migration.sql,sha256=z1fiUODOMNByeSMzIBgs3Ee0DuDBVwJBJEZQV8eNXVk,653
 litellm_proxy_extras/migrations/migration_lock.toml,sha256=HbF6jQUaoTYRBzZ1LF4fi37ZK26o6AMRL7viSXBHwhA,24
-litellm_proxy_extras/schema.prisma,sha256=rjfGAzGkr7_tsI-9qhBGcoNmUqOv_jf_QgD_MQN-JS0,21549
-litellm_proxy_extras/utils.py,sha256=zcUGsFa7lI9pbC8Sj6GQitAYULuKq89vXBGxm7R87rg,15175
-litellm_proxy_extras-0.2.25.dist-info/METADATA,sha256=lGvhwdi-y5DwXgB6VQd5NuQZsIcpx344KoAwe5C7G94,1391
-litellm_proxy_extras-0.2.25.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-litellm_proxy_extras-0.2.25.dist-info/licenses/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
-litellm_proxy_extras-0.2.25.dist-info/RECORD,,
+litellm_proxy_extras/schema.prisma,sha256=NVHRxWZIpj9neb-I3zqB0k9aBOCoEWLMcHbvRpErXbY,22224
+litellm_proxy_extras/utils.py,sha256=2g3sHuOMy_9sjze2Kw42lBXcS6ustYBygd4vaiJ6Kkk,15679
+litellm_proxy_extras-0.2.27.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
+litellm_proxy_extras-0.2.27.dist-info/METADATA,sha256=GdGRU2C4h_Zm12ohfrqb4a1MlfjAtzvr26cFyGaKx9g,1267
+litellm_proxy_extras-0.2.27.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+litellm_proxy_extras-0.2.27.dist-info/RECORD,,

{litellm_proxy_extras-0.2.25.dist-info → litellm_proxy_extras-0.2.27.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.1
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any