litellm-proxy-extras 0.1.4__py3-none-any.whl → 0.1.7__py3-none-any.whl

litellm_proxy_extras/migrations/20250412081753_team_member_permissions/migration.sql

@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_TeamTable" ADD COLUMN     "team_member_permissions" TEXT[] DEFAULT ARRAY[]::TEXT[];
+

litellm_proxy_extras/schema.prisma

@@ -106,6 +106,7 @@ model LiteLLM_TeamTable {
     updated_at    DateTime               @default(now()) @updatedAt @map("updated_at")
     model_spend      Json @default("{}")
     model_max_budget Json @default("{}")
+    team_member_permissions String[] @default([])
     model_id Int? @unique // id for LiteLLM_ModelTable -> stores team-level model aliases
     litellm_organization_table LiteLLM_OrganizationTable?   @relation(fields: [organization_id], references: [organization_id])
     litellm_model_table LiteLLM_ModelTable? @relation(fields: [model_id], references: [id])

litellm_proxy_extras/utils.py

@@ -1,7 +1,9 @@
+import glob
 import os
 import random
 import subprocess
 import time
+from pathlib import Path
 from typing import Optional
 
 from litellm_proxy_extras._logging import logger
@@ -14,6 +16,94 @@ def str_to_bool(value: Optional[str]) -> bool:
 
 
 class ProxyExtrasDBManager:
+    @staticmethod
+    def _get_prisma_dir() -> str:
+        """Get the path to the migrations directory"""
+        migrations_dir = os.path.dirname(__file__)
+        return migrations_dir
+
+    @staticmethod
+    def _create_baseline_migration(schema_path: str) -> bool:
+        """Create a baseline migration for an existing database"""
+        prisma_dir = ProxyExtrasDBManager._get_prisma_dir()
+        prisma_dir_path = Path(prisma_dir)
+        init_dir = prisma_dir_path / "migrations" / "0_init"
+
+        # Create migrations/0_init directory
+        init_dir.mkdir(parents=True, exist_ok=True)
+
+        # Generate migration SQL file
+        migration_file = init_dir / "migration.sql"
+
+        try:
+            # Generate migration diff with increased timeout
+            subprocess.run(
+                [
+                    "prisma",
+                    "migrate",
+                    "diff",
+                    "--from-empty",
+                    "--to-schema-datamodel",
+                    str(schema_path),
+                    "--script",
+                ],
+                stdout=open(migration_file, "w"),
+                check=True,
+                timeout=30,
+            )  # 30 second timeout
+
+            # Mark migration as applied with increased timeout
+            subprocess.run(
+                [
+                    "prisma",
+                    "migrate",
+                    "resolve",
+                    "--applied",
+                    "0_init",
+                ],
+                check=True,
+                timeout=30,
+            )
+
+            return True
+        except subprocess.TimeoutExpired:
+            logger.warning(
+                "Migration timed out - the database might be under heavy load."
+            )
+            return False
+        except subprocess.CalledProcessError as e:
+            logger.warning(f"Error creating baseline migration: {e}")
+            return False
+
+    @staticmethod
+    def _get_migration_names(migrations_dir: str) -> list:
+        """Get all migration directory names from the migrations folder"""
+        migration_paths = glob.glob(f"{migrations_dir}/migrations/*/migration.sql")
+        logger.info(f"Found {len(migration_paths)} migrations at {migrations_dir}")
+        return [Path(p).parent.name for p in migration_paths]
+
+    @staticmethod
+    def _resolve_all_migrations(migrations_dir: str):
+        """Mark all existing migrations as applied"""
+        migration_names = ProxyExtrasDBManager._get_migration_names(migrations_dir)
+        logger.info(f"Resolving {len(migration_names)} migrations")
+        for migration_name in migration_names:
+            try:
+                logger.info(f"Resolving migration: {migration_name}")
+                subprocess.run(
+                    ["prisma", "migrate", "resolve", "--applied", migration_name],
+                    timeout=60,
+                    check=True,
+                    capture_output=True,
+                    text=True,
+                )
+                logger.debug(f"Resolved migration: {migration_name}")
+            except subprocess.CalledProcessError as e:
+                if "is already recorded as applied in the database." not in e.stderr:
+                    logger.warning(
+                        f"Failed to resolve migration {migration_name}: {e.stderr}"
+                    )
+
     @staticmethod
     def setup_database(schema_path: str, use_migrate: bool = False) -> bool:
         """
@@ -30,7 +120,7 @@ class ProxyExtrasDBManager:
         use_migrate = str_to_bool(os.getenv("USE_PRISMA_MIGRATE")) or use_migrate
         for attempt in range(4):
             original_dir = os.getcwd()
-            migrations_dir = os.path.dirname(__file__)
+            migrations_dir = ProxyExtrasDBManager._get_prisma_dir()
             os.chdir(migrations_dir)
 
             try:
@@ -55,8 +145,16 @@ class ProxyExtrasDBManager:
                             "P3005" in e.stderr
                             and "database schema is not empty" in e.stderr
                         ):
-                            logger.info("Error: Database schema is not empty")
-                            return False
+                            logger.info(
+                                "Database schema is not empty, creating baseline migration"
+                            )
+                            ProxyExtrasDBManager._create_baseline_migration(schema_path)
+                            logger.info(
+                                "Baseline migration created, resolving all migrations"
+                            )
+                            ProxyExtrasDBManager._resolve_all_migrations(migrations_dir)
+                            logger.info("✅ All migrations resolved.")
+                            return True
                 else:
                     # Use prisma db push with increased timeout
                     subprocess.run(

{litellm_proxy_extras-0.1.4.dist-info → litellm_proxy_extras-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: litellm-proxy-extras
-Version: 0.1.4
+Version: 0.1.7
 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.1.4.dist-info → litellm_proxy_extras-0.1.7.dist-info}/RECORD

@@ -6,10 +6,11 @@ litellm_proxy_extras/migrations/20250327180120_add_api_requests_to_daily_user_ta
 litellm_proxy_extras/migrations/20250329084805_new_cron_job_table/migration.sql,sha256=eZNDwrzKtWFXkTqOKb9JS4hzum1dI-VTXvMqzhryfxc,404
 litellm_proxy_extras/migrations/20250331215456_track_success_and_failed_requests_daily_agg_table/migration.sql,sha256=tyeLY6u8KFyw71osCBM-sdjhIgvHoFCK88cIx8dExNY,178
 litellm_proxy_extras/migrations/20250411215431_add_managed_file_table/migration.sql,sha256=Yu2K37Q90LDhxsFo_64sH0PXdSQ3sHs45Lqzxv2t_20,625
+litellm_proxy_extras/migrations/20250412081753_team_member_permissions/migration.sql,sha256=v3vDx5lb6SLCzXCe_A2NZj7zzmucRXM08aQun_G_MkE,120
 litellm_proxy_extras/migrations/migration_lock.toml,sha256=HbF6jQUaoTYRBzZ1LF4fi37ZK26o6AMRL7viSXBHwhA,24
-litellm_proxy_extras/schema.prisma,sha256=jlH_P8bI3FAOdEqEHSNqT0cePe74M3p_EbXPRBvzKnw,14529
-litellm_proxy_extras/utils.py,sha256=G4FaWA01jwQ4482CWtFERKp0BosCwxsuqyv7bAIk7xg,3203
-litellm_proxy_extras-0.1.4.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
-litellm_proxy_extras-0.1.4.dist-info/METADATA,sha256=ZHFn7yKeWsl5m5VRG6LjPjE0vW-EemPM67kvOLC_ky4,1267
-litellm_proxy_extras-0.1.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-litellm_proxy_extras-0.1.4.dist-info/RECORD,,
+litellm_proxy_extras/schema.prisma,sha256=AJs-oTl488I539rej87DQw73l2x28IhLNGw2gZ6T8AA,14579
+litellm_proxy_extras/utils.py,sha256=2CQEBosLTjXpGEBwx3sGYglSD4QNy609iV4Ppo9AJdo,7087
+litellm_proxy_extras-0.1.7.dist-info/LICENSE,sha256=sXDWv46INd01fgEWgdsCj01R4vsOqJIFj1bgH7ObgnM,1419
+litellm_proxy_extras-0.1.7.dist-info/METADATA,sha256=OEh02aKnCi3sXljyI2d8IrzEH0IFyTBUrSHOgSE-ECM,1267
+litellm_proxy_extras-0.1.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+litellm_proxy_extras-0.1.7.dist-info/RECORD,,