litellm-proxy-extras 0.1.6__tar.gz → 0.1.8__tar.gz

{litellm_proxy_extras-0.1.6 → litellm_proxy_extras-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: litellm-proxy-extras
-Version: 0.1.6
+Version: 0.1.8
 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.*
@@ -9,7 +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
 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.1.8/litellm_proxy_extras/migrations/20250411215431_add_managed_file_table/migration.sql

@@ -0,0 +1,18 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_ManagedFileTable" (
+    "id" TEXT NOT NULL,
+    "unified_file_id" TEXT NOT NULL,
+    "file_object" JSONB NOT NULL,
+    "model_mappings" JSONB NOT NULL,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_ManagedFileTable_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_ManagedFileTable_unified_file_id_key" ON "LiteLLM_ManagedFileTable"("unified_file_id");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_ManagedFileTable_unified_file_id_idx" ON "LiteLLM_ManagedFileTable"("unified_file_id");
+

litellm_proxy_extras-0.1.8/litellm_proxy_extras/migrations/20250415191926_add_daily_team_table/migration.sql

@@ -0,0 +1,36 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_DailyTeamSpend" (
+    "id" TEXT NOT NULL,
+    "team_id" TEXT NOT NULL,
+    "date" TEXT NOT NULL,
+    "api_key" TEXT NOT NULL,
+    "model" TEXT NOT NULL,
+    "model_group" TEXT,
+    "custom_llm_provider" TEXT,
+    "prompt_tokens" INTEGER NOT NULL DEFAULT 0,
+    "completion_tokens" INTEGER NOT NULL DEFAULT 0,
+    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
+    "api_requests" INTEGER NOT NULL DEFAULT 0,
+    "successful_requests" INTEGER NOT NULL DEFAULT 0,
+    "failed_requests" INTEGER NOT NULL DEFAULT 0,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_DailyTeamSpend_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyTeamSpend_date_idx" ON "LiteLLM_DailyTeamSpend"("date");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyTeamSpend_team_id_idx" ON "LiteLLM_DailyTeamSpend"("team_id");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyTeamSpend_api_key_idx" ON "LiteLLM_DailyTeamSpend"("api_key");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_DailyTeamSpend_model_idx" ON "LiteLLM_DailyTeamSpend"("model");
+
+-- 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");
+

{litellm_proxy_extras-0.1.6 → litellm_proxy_extras-0.1.8}/litellm_proxy_extras/schema.prisma

@@ -355,3 +355,14 @@ enum JobStatus {
   INACTIVE
 }
 
+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
+  created_at DateTime @default(now())
+  updated_at DateTime @updatedAt
+
+  @@index([unified_file_id])
+}
+

litellm_proxy_extras-0.1.8/litellm_proxy_extras/utils.py

@@ -0,0 +1,181 @@
+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
+
+
+def str_to_bool(value: Optional[str]) -> bool:
+    if value is None:
+        return False
+    return value.lower() in ("true", "1", "t", "y", "yes")
+
+
+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:
+        """
+        Set up the database using either prisma migrate or prisma db push
+        Uses migrations from litellm-proxy-extras package
+
+        Args:
+            schema_path (str): Path to the Prisma schema file
+            use_migrate (bool): Whether to use prisma migrate instead of db push
+
+        Returns:
+            bool: True if setup was successful, False otherwise
+        """
+        use_migrate = str_to_bool(os.getenv("USE_PRISMA_MIGRATE")) or use_migrate
+        for attempt in range(4):
+            original_dir = os.getcwd()
+            migrations_dir = ProxyExtrasDBManager._get_prisma_dir()
+            os.chdir(migrations_dir)
+
+            try:
+                if use_migrate:
+                    logger.info("Running prisma migrate deploy")
+                    try:
+                        # Set migrations directory for Prisma
+                        result = subprocess.run(
+                            ["prisma", "migrate", "deploy"],
+                            timeout=60,
+                            check=True,
+                            capture_output=True,
+                            text=True,
+                        )
+                        logger.info(f"prisma migrate deploy stdout: {result.stdout}")
+
+                        logger.info("prisma migrate deploy completed")
+                        return True
+                    except subprocess.CalledProcessError as e:
+                        logger.info(f"prisma db error: {e.stderr}, e: {e.stdout}")
+                        if (
+                            "P3005" in e.stderr
+                            and "database schema is not empty" in e.stderr
+                        ):
+                            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(
+                        ["prisma", "db", "push", "--accept-data-loss"],
+                        timeout=60,
+                        check=True,
+                    )
+                    return True
+            except subprocess.TimeoutExpired:
+                logger.info(f"Attempt {attempt + 1} timed out")
+                time.sleep(random.randrange(5, 15))
+            except subprocess.CalledProcessError as e:
+                attempts_left = 3 - attempt
+                retry_msg = (
+                    f" Retrying... ({attempts_left} attempts left)"
+                    if attempts_left > 0
+                    else ""
+                )
+                logger.info(f"The process failed to execute. Details: {e}.{retry_msg}")
+                time.sleep(random.randrange(5, 15))
+            finally:
+                os.chdir(original_dir)
+                pass
+        return False

{litellm_proxy_extras-0.1.6 → litellm_proxy_extras-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm-proxy-extras"
-version = "0.1.6"
+version = "0.1.8"
 description = "Additional files for the LiteLLM Proxy. Reduces the size of the main litellm package."
 authors = ["BerriAI"]
 readme = "README.md"
@@ -22,7 +22,7 @@ requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.commitizen]
-version = "0.1.6"
+version = "0.1.8"
 version_files = [
     "pyproject.toml:version",
     "../requirements.txt:litellm-proxy-extras==",

litellm_proxy_extras-0.1.6/litellm_proxy_extras/migrations/20250412075123_team_member_permissions/migration.sql

@@ -1,411 +0,0 @@
--- CreateEnum
-CREATE TYPE "JobStatus" AS ENUM ('ACTIVE', 'INACTIVE');
-
--- CreateTable
-CREATE TABLE "LiteLLM_BudgetTable" (
-    "budget_id" TEXT NOT NULL,
-    "max_budget" DOUBLE PRECISION,
-    "soft_budget" DOUBLE PRECISION,
-    "max_parallel_requests" INTEGER,
-    "tpm_limit" BIGINT,
-    "rpm_limit" BIGINT,
-    "model_max_budget" JSONB,
-    "budget_duration" TEXT,
-    "budget_reset_at" TIMESTAMP(3),
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "created_by" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_by" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_BudgetTable_pkey" PRIMARY KEY ("budget_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_CredentialsTable" (
-    "credential_id" TEXT NOT NULL,
-    "credential_name" TEXT NOT NULL,
-    "credential_values" JSONB NOT NULL,
-    "credential_info" JSONB,
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "created_by" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_by" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_CredentialsTable_pkey" PRIMARY KEY ("credential_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_ProxyModelTable" (
-    "model_id" TEXT NOT NULL,
-    "model_name" TEXT NOT NULL,
-    "litellm_params" JSONB NOT NULL,
-    "model_info" JSONB,
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "created_by" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_by" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_ProxyModelTable_pkey" PRIMARY KEY ("model_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_OrganizationTable" (
-    "organization_id" TEXT NOT NULL,
-    "organization_alias" TEXT NOT NULL,
-    "budget_id" TEXT NOT NULL,
-    "metadata" JSONB NOT NULL DEFAULT '{}',
-    "models" TEXT[],
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "model_spend" JSONB NOT NULL DEFAULT '{}',
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "created_by" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_by" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_OrganizationTable_pkey" PRIMARY KEY ("organization_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_ModelTable" (
-    "id" SERIAL NOT NULL,
-    "aliases" JSONB,
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "created_by" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_by" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_ModelTable_pkey" PRIMARY KEY ("id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_TeamTable" (
-    "team_id" TEXT NOT NULL,
-    "team_alias" TEXT,
-    "organization_id" TEXT,
-    "admins" TEXT[],
-    "members" TEXT[],
-    "members_with_roles" JSONB NOT NULL DEFAULT '{}',
-    "metadata" JSONB NOT NULL DEFAULT '{}',
-    "max_budget" DOUBLE PRECISION,
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "models" TEXT[],
-    "max_parallel_requests" INTEGER,
-    "tpm_limit" BIGINT,
-    "rpm_limit" BIGINT,
-    "budget_duration" TEXT,
-    "budget_reset_at" TIMESTAMP(3),
-    "blocked" BOOLEAN NOT NULL DEFAULT false,
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "model_spend" JSONB NOT NULL DEFAULT '{}',
-    "model_max_budget" JSONB NOT NULL DEFAULT '{}',
-    "team_member_permissions" TEXT[] DEFAULT ARRAY[]::TEXT[],
-    "model_id" INTEGER,
-
-    CONSTRAINT "LiteLLM_TeamTable_pkey" PRIMARY KEY ("team_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_UserTable" (
-    "user_id" TEXT NOT NULL,
-    "user_alias" TEXT,
-    "team_id" TEXT,
-    "sso_user_id" TEXT,
-    "organization_id" TEXT,
-    "password" TEXT,
-    "teams" TEXT[] DEFAULT ARRAY[]::TEXT[],
-    "user_role" TEXT,
-    "max_budget" DOUBLE PRECISION,
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "user_email" TEXT,
-    "models" TEXT[],
-    "metadata" JSONB NOT NULL DEFAULT '{}',
-    "max_parallel_requests" INTEGER,
-    "tpm_limit" BIGINT,
-    "rpm_limit" BIGINT,
-    "budget_duration" TEXT,
-    "budget_reset_at" TIMESTAMP(3),
-    "allowed_cache_controls" TEXT[] DEFAULT ARRAY[]::TEXT[],
-    "model_spend" JSONB NOT NULL DEFAULT '{}',
-    "model_max_budget" JSONB NOT NULL DEFAULT '{}',
-    "created_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
-    "updated_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
-
-    CONSTRAINT "LiteLLM_UserTable_pkey" PRIMARY KEY ("user_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_VerificationToken" (
-    "token" TEXT NOT NULL,
-    "key_name" TEXT,
-    "key_alias" TEXT,
-    "soft_budget_cooldown" BOOLEAN NOT NULL DEFAULT false,
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "expires" TIMESTAMP(3),
-    "models" TEXT[],
-    "aliases" JSONB NOT NULL DEFAULT '{}',
-    "config" JSONB NOT NULL DEFAULT '{}',
-    "user_id" TEXT,
-    "team_id" TEXT,
-    "permissions" JSONB NOT NULL DEFAULT '{}',
-    "max_parallel_requests" INTEGER,
-    "metadata" JSONB NOT NULL DEFAULT '{}',
-    "blocked" BOOLEAN,
-    "tpm_limit" BIGINT,
-    "rpm_limit" BIGINT,
-    "max_budget" DOUBLE PRECISION,
-    "budget_duration" TEXT,
-    "budget_reset_at" TIMESTAMP(3),
-    "allowed_cache_controls" TEXT[] DEFAULT ARRAY[]::TEXT[],
-    "model_spend" JSONB NOT NULL DEFAULT '{}',
-    "model_max_budget" JSONB NOT NULL DEFAULT '{}',
-    "budget_id" TEXT,
-    "organization_id" TEXT,
-    "created_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
-    "created_by" TEXT,
-    "updated_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
-    "updated_by" TEXT,
-
-    CONSTRAINT "LiteLLM_VerificationToken_pkey" PRIMARY KEY ("token")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_EndUserTable" (
-    "user_id" TEXT NOT NULL,
-    "alias" TEXT,
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "allowed_model_region" TEXT,
-    "default_model" TEXT,
-    "budget_id" TEXT,
-    "blocked" BOOLEAN NOT NULL DEFAULT false,
-
-    CONSTRAINT "LiteLLM_EndUserTable_pkey" PRIMARY KEY ("user_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_Config" (
-    "param_name" TEXT NOT NULL,
-    "param_value" JSONB,
-
-    CONSTRAINT "LiteLLM_Config_pkey" PRIMARY KEY ("param_name")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_SpendLogs" (
-    "request_id" TEXT NOT NULL,
-    "call_type" TEXT NOT NULL,
-    "api_key" TEXT NOT NULL DEFAULT '',
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "total_tokens" INTEGER NOT NULL DEFAULT 0,
-    "prompt_tokens" INTEGER NOT NULL DEFAULT 0,
-    "completion_tokens" INTEGER NOT NULL DEFAULT 0,
-    "startTime" TIMESTAMP(3) NOT NULL,
-    "endTime" TIMESTAMP(3) NOT NULL,
-    "completionStartTime" TIMESTAMP(3),
-    "model" TEXT NOT NULL DEFAULT '',
-    "model_id" TEXT DEFAULT '',
-    "model_group" TEXT DEFAULT '',
-    "custom_llm_provider" TEXT DEFAULT '',
-    "api_base" TEXT DEFAULT '',
-    "user" TEXT DEFAULT '',
-    "metadata" JSONB DEFAULT '{}',
-    "cache_hit" TEXT DEFAULT '',
-    "cache_key" TEXT DEFAULT '',
-    "request_tags" JSONB DEFAULT '[]',
-    "team_id" TEXT,
-    "end_user" TEXT,
-    "requester_ip_address" TEXT,
-    "messages" JSONB DEFAULT '{}',
-    "response" JSONB DEFAULT '{}',
-
-    CONSTRAINT "LiteLLM_SpendLogs_pkey" PRIMARY KEY ("request_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_ErrorLogs" (
-    "request_id" TEXT NOT NULL,
-    "startTime" TIMESTAMP(3) NOT NULL,
-    "endTime" TIMESTAMP(3) NOT NULL,
-    "api_base" TEXT NOT NULL DEFAULT '',
-    "model_group" TEXT NOT NULL DEFAULT '',
-    "litellm_model_name" TEXT NOT NULL DEFAULT '',
-    "model_id" TEXT NOT NULL DEFAULT '',
-    "request_kwargs" JSONB NOT NULL DEFAULT '{}',
-    "exception_type" TEXT NOT NULL DEFAULT '',
-    "exception_string" TEXT NOT NULL DEFAULT '',
-    "status_code" TEXT NOT NULL DEFAULT '',
-
-    CONSTRAINT "LiteLLM_ErrorLogs_pkey" PRIMARY KEY ("request_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_UserNotifications" (
-    "request_id" TEXT NOT NULL,
-    "user_id" TEXT NOT NULL,
-    "models" TEXT[],
-    "justification" TEXT NOT NULL,
-    "status" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_UserNotifications_pkey" PRIMARY KEY ("request_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_TeamMembership" (
-    "user_id" TEXT NOT NULL,
-    "team_id" TEXT NOT NULL,
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "budget_id" TEXT,
-
-    CONSTRAINT "LiteLLM_TeamMembership_pkey" PRIMARY KEY ("user_id","team_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_OrganizationMembership" (
-    "user_id" TEXT NOT NULL,
-    "organization_id" TEXT NOT NULL,
-    "user_role" TEXT,
-    "spend" DOUBLE PRECISION DEFAULT 0.0,
-    "budget_id" TEXT,
-    "created_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
-    "updated_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
-
-    CONSTRAINT "LiteLLM_OrganizationMembership_pkey" PRIMARY KEY ("user_id","organization_id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_InvitationLink" (
-    "id" TEXT NOT NULL,
-    "user_id" TEXT NOT NULL,
-    "is_accepted" BOOLEAN NOT NULL DEFAULT false,
-    "accepted_at" TIMESTAMP(3),
-    "expires_at" TIMESTAMP(3) NOT NULL,
-    "created_at" TIMESTAMP(3) NOT NULL,
-    "created_by" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL,
-    "updated_by" TEXT NOT NULL,
-
-    CONSTRAINT "LiteLLM_InvitationLink_pkey" PRIMARY KEY ("id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_AuditLog" (
-    "id" TEXT NOT NULL,
-    "updated_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "changed_by" TEXT NOT NULL DEFAULT '',
-    "changed_by_api_key" TEXT NOT NULL DEFAULT '',
-    "action" TEXT NOT NULL,
-    "table_name" TEXT NOT NULL,
-    "object_id" TEXT NOT NULL,
-    "before_value" JSONB,
-    "updated_values" JSONB,
-
-    CONSTRAINT "LiteLLM_AuditLog_pkey" PRIMARY KEY ("id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_DailyUserSpend" (
-    "id" TEXT NOT NULL,
-    "user_id" TEXT NOT NULL,
-    "date" TEXT NOT NULL,
-    "api_key" TEXT NOT NULL,
-    "model" TEXT NOT NULL,
-    "model_group" TEXT,
-    "custom_llm_provider" TEXT,
-    "prompt_tokens" INTEGER NOT NULL DEFAULT 0,
-    "completion_tokens" INTEGER NOT NULL DEFAULT 0,
-    "spend" DOUBLE PRECISION NOT NULL DEFAULT 0.0,
-    "api_requests" INTEGER NOT NULL DEFAULT 0,
-    "successful_requests" INTEGER NOT NULL DEFAULT 0,
-    "failed_requests" INTEGER NOT NULL DEFAULT 0,
-    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "updated_at" TIMESTAMP(3) NOT NULL,
-
-    CONSTRAINT "LiteLLM_DailyUserSpend_pkey" PRIMARY KEY ("id")
-);
-
--- CreateTable
-CREATE TABLE "LiteLLM_CronJob" (
-    "cronjob_id" TEXT NOT NULL,
-    "pod_id" TEXT NOT NULL,
-    "status" "JobStatus" NOT NULL DEFAULT 'INACTIVE',
-    "last_updated" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
-    "ttl" TIMESTAMP(3) NOT NULL,
-
-    CONSTRAINT "LiteLLM_CronJob_pkey" PRIMARY KEY ("cronjob_id")
-);
-
--- CreateIndex
-CREATE UNIQUE INDEX "LiteLLM_CredentialsTable_credential_name_key" ON "LiteLLM_CredentialsTable"("credential_name");
-
--- CreateIndex
-CREATE UNIQUE INDEX "LiteLLM_TeamTable_model_id_key" ON "LiteLLM_TeamTable"("model_id");
-
--- CreateIndex
-CREATE UNIQUE INDEX "LiteLLM_UserTable_sso_user_id_key" ON "LiteLLM_UserTable"("sso_user_id");
-
--- CreateIndex
-CREATE INDEX "LiteLLM_SpendLogs_startTime_idx" ON "LiteLLM_SpendLogs"("startTime");
-
--- CreateIndex
-CREATE INDEX "LiteLLM_SpendLogs_end_user_idx" ON "LiteLLM_SpendLogs"("end_user");
-
--- CreateIndex
-CREATE UNIQUE INDEX "LiteLLM_OrganizationMembership_user_id_organization_id_key" ON "LiteLLM_OrganizationMembership"("user_id", "organization_id");
-
--- CreateIndex
-CREATE INDEX "LiteLLM_DailyUserSpend_date_idx" ON "LiteLLM_DailyUserSpend"("date");
-
--- CreateIndex
-CREATE INDEX "LiteLLM_DailyUserSpend_user_id_idx" ON "LiteLLM_DailyUserSpend"("user_id");
-
--- CreateIndex
-CREATE INDEX "LiteLLM_DailyUserSpend_api_key_idx" ON "LiteLLM_DailyUserSpend"("api_key");
-
--- CreateIndex
-CREATE INDEX "LiteLLM_DailyUserSpend_model_idx" ON "LiteLLM_DailyUserSpend"("model");
-
--- 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");
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_OrganizationTable" ADD CONSTRAINT "LiteLLM_OrganizationTable_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE RESTRICT ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_TeamTable" ADD CONSTRAINT "LiteLLM_TeamTable_organization_id_fkey" FOREIGN KEY ("organization_id") REFERENCES "LiteLLM_OrganizationTable"("organization_id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_TeamTable" ADD CONSTRAINT "LiteLLM_TeamTable_model_id_fkey" FOREIGN KEY ("model_id") REFERENCES "LiteLLM_ModelTable"("id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_UserTable" ADD CONSTRAINT "LiteLLM_UserTable_organization_id_fkey" FOREIGN KEY ("organization_id") REFERENCES "LiteLLM_OrganizationTable"("organization_id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_VerificationToken" ADD CONSTRAINT "LiteLLM_VerificationToken_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_VerificationToken" ADD CONSTRAINT "LiteLLM_VerificationToken_organization_id_fkey" FOREIGN KEY ("organization_id") REFERENCES "LiteLLM_OrganizationTable"("organization_id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_EndUserTable" ADD CONSTRAINT "LiteLLM_EndUserTable_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- 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;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_OrganizationMembership" ADD CONSTRAINT "LiteLLM_OrganizationMembership_user_id_fkey" FOREIGN KEY ("user_id") REFERENCES "LiteLLM_UserTable"("user_id") ON DELETE RESTRICT ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_OrganizationMembership" ADD CONSTRAINT "LiteLLM_OrganizationMembership_organization_id_fkey" FOREIGN KEY ("organization_id") REFERENCES "LiteLLM_OrganizationTable"("organization_id") ON DELETE RESTRICT ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_OrganizationMembership" ADD CONSTRAINT "LiteLLM_OrganizationMembership_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE SET NULL ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_InvitationLink" ADD CONSTRAINT "LiteLLM_InvitationLink_user_id_fkey" FOREIGN KEY ("user_id") REFERENCES "LiteLLM_UserTable"("user_id") ON DELETE RESTRICT ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_InvitationLink" ADD CONSTRAINT "LiteLLM_InvitationLink_created_by_fkey" FOREIGN KEY ("created_by") REFERENCES "LiteLLM_UserTable"("user_id") ON DELETE RESTRICT ON UPDATE CASCADE;
-
--- AddForeignKey
-ALTER TABLE "LiteLLM_InvitationLink" ADD CONSTRAINT "LiteLLM_InvitationLink_updated_by_fkey" FOREIGN KEY ("updated_by") REFERENCES "LiteLLM_UserTable"("user_id") ON DELETE RESTRICT ON UPDATE CASCADE;
-

litellm_proxy_extras-0.1.6/litellm_proxy_extras/utils.py

@@ -1,83 +0,0 @@
-import os
-import random
-import subprocess
-import time
-from typing import Optional
-
-from litellm_proxy_extras._logging import logger
-
-
-def str_to_bool(value: Optional[str]) -> bool:
-    if value is None:
-        return False
-    return value.lower() in ("true", "1", "t", "y", "yes")
-
-
-class ProxyExtrasDBManager:
-    @staticmethod
-    def setup_database(schema_path: str, use_migrate: bool = False) -> bool:
-        """
-        Set up the database using either prisma migrate or prisma db push
-        Uses migrations from litellm-proxy-extras package
-
-        Args:
-            schema_path (str): Path to the Prisma schema file
-            use_migrate (bool): Whether to use prisma migrate instead of db push
-
-        Returns:
-            bool: True if setup was successful, False otherwise
-        """
-        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__)
-            os.chdir(migrations_dir)
-
-            try:
-                if use_migrate:
-                    logger.info("Running prisma migrate deploy")
-                    try:
-                        # Set migrations directory for Prisma
-                        result = subprocess.run(
-                            ["prisma", "migrate", "deploy"],
-                            timeout=60,
-                            check=True,
-                            capture_output=True,
-                            text=True,
-                        )
-                        logger.info(f"prisma migrate deploy stdout: {result.stdout}")
-
-                        logger.info("prisma migrate deploy completed")
-                        return True
-                    except subprocess.CalledProcessError as e:
-                        logger.info(f"prisma db error: {e.stderr}, e: {e.stdout}")
-                        if (
-                            "P3005" in e.stderr
-                            and "database schema is not empty" in e.stderr
-                        ):
-                            logger.info("Error: Database schema is not empty")
-                            return False
-                else:
-                    # Use prisma db push with increased timeout
-                    subprocess.run(
-                        ["prisma", "db", "push", "--accept-data-loss"],
-                        timeout=60,
-                        check=True,
-                    )
-                    return True
-            except subprocess.TimeoutExpired:
-                logger.info(f"Attempt {attempt + 1} timed out")
-                time.sleep(random.randrange(5, 15))
-            except subprocess.CalledProcessError as e:
-                attempts_left = 3 - attempt
-                retry_msg = (
-                    f" Retrying... ({attempts_left} attempts left)"
-                    if attempts_left > 0
-                    else ""
-                )
-                logger.info(f"The process failed to execute. Details: {e}.{retry_msg}")
-                time.sleep(random.randrange(5, 15))
-            finally:
-                os.chdir(original_dir)
-                pass
-        return False