litellm-proxy-extras 0.4.3__py3-none-any.whl

litellm_proxy_extras/migrations/20250509141545_use_big_int_for_daily_spend_tables/migration.sql

@@ -0,0 +1,27 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyTagSpend" ALTER COLUMN "prompt_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "completion_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "cache_read_input_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "cache_creation_input_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "api_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "successful_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "failed_requests" SET DATA TYPE BIGINT;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyTeamSpend" ALTER COLUMN "prompt_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "completion_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "api_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "successful_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "failed_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "cache_creation_input_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "cache_read_input_tokens" SET DATA TYPE BIGINT;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyUserSpend" ALTER COLUMN "prompt_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "completion_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "api_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "failed_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "successful_requests" SET DATA TYPE BIGINT,
+ALTER COLUMN "cache_creation_input_tokens" SET DATA TYPE BIGINT,
+ALTER COLUMN "cache_read_input_tokens" SET DATA TYPE BIGINT;
+

litellm_proxy_extras/migrations/20250510142544_add_session_id_index_spend_logs/migration.sql

@@ -0,0 +1,3 @@
+-- CreateIndex
+CREATE INDEX "LiteLLM_SpendLogs_session_id_idx" ON "LiteLLM_SpendLogs"("session_id");
+

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/migrations/20250526154401_allow_null_entity_id/migration.sql

@@ -0,0 +1,9 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyTagSpend" ALTER COLUMN "tag" DROP NOT NULL;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyTeamSpend" ALTER COLUMN "team_id" DROP NOT NULL;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_DailyUserSpend" ALTER COLUMN "user_id" DROP NOT NULL;
+

litellm_proxy_extras/migrations/20250528185438_add_vector_stores_to_object_permissions/migration.sql

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

litellm_proxy_extras/migrations/20250603210143_cascade_budget_changes/migration.sql

@@ -0,0 +1,6 @@
+-- DropForeignKey
+ALTER TABLE "LiteLLM_TeamMembership" DROP CONSTRAINT "LiteLLM_TeamMembership_budget_id_fkey";
+
+-- AddForeignKey
+ALTER TABLE "LiteLLM_TeamMembership" ADD CONSTRAINT "LiteLLM_TeamMembership_budget_id_fkey" FOREIGN KEY ("budget_id") REFERENCES "LiteLLM_BudgetTable"("budget_id") ON DELETE CASCADE ON UPDATE CASCADE;
+

litellm_proxy_extras/migrations/20250618225828_add_health_check_table/migration.sql

@@ -0,0 +1,28 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_HealthCheckTable" (
+    "health_check_id" TEXT NOT NULL,
+    "model_name" TEXT NOT NULL,
+    "model_id" TEXT,
+    "status" TEXT NOT NULL,
+    "healthy_count" INTEGER NOT NULL DEFAULT 0,
+    "unhealthy_count" INTEGER NOT NULL DEFAULT 0,
+    "error_message" TEXT,
+    "response_time_ms" DOUBLE PRECISION,
+    "details" JSONB,
+    "checked_by" TEXT,
+    "checked_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_HealthCheckTable_pkey" PRIMARY KEY ("health_check_id")
+);
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_HealthCheckTable_model_name_idx" ON "LiteLLM_HealthCheckTable"("model_name");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_HealthCheckTable_checked_at_idx" ON "LiteLLM_HealthCheckTable"("checked_at");
+
+-- CreateIndex
+CREATE INDEX "LiteLLM_HealthCheckTable_status_idx" ON "LiteLLM_HealthCheckTable"("status");
+

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/migrations/20250802162330_prompt_table/migration.sql

@@ -0,0 +1,15 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_PromptTable" (
+    "id" TEXT NOT NULL,
+    "prompt_id" TEXT NOT NULL,
+    "litellm_params" JSONB NOT NULL,
+    "prompt_info" JSONB,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_PromptTable_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_PromptTable_prompt_id_key" ON "LiteLLM_PromptTable"("prompt_id");
+

litellm_proxy_extras/migrations/20250806095134_rename_alias_to_server_name_mcp_table/migration.sql

@@ -0,0 +1,10 @@
+-- Migration for existing tables: rename alias to server_name if upgrading
+DO $$
+BEGIN
+    IF EXISTS (SELECT 1 FROM information_schema.columns WHERE table_name = 'LiteLLM_MCPServerTable' AND column_name = 'alias') THEN
+        ALTER TABLE "LiteLLM_MCPServerTable" RENAME COLUMN "alias" TO "server_name";
+    END IF;
+END $$;
+
+-- Migration for existing tables: add alias column if upgrading
+ALTER TABLE "LiteLLM_MCPServerTable" ADD COLUMN IF NOT EXISTS "alias" TEXT; 

litellm_proxy_extras/migrations/20250918083359_drop_spec_version_column_from_mcp_table/migration.sql

@@ -0,0 +1,8 @@
+/*
+  Warnings:
+
+  - You are about to drop the column `spec_version` on the `LiteLLM_MCPServerTable` table. All the data in the column will be lost.
+
+*/
+-- AlterTable
+ALTER TABLE "LiteLLM_MCPServerTable" DROP COLUMN "spec_version";

litellm_proxy_extras/migrations/20250926194702_unnamed_migration/migration.sql

@@ -0,0 +1,7 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_VerificationToken" ADD COLUMN     "auto_rotate" BOOLEAN DEFAULT false,
+ADD COLUMN     "key_rotation_at" TIMESTAMP(3),
+ADD COLUMN     "last_rotation_at" TIMESTAMP(3),
+ADD COLUMN     "rotation_count" INTEGER DEFAULT 0,
+ADD COLUMN     "rotation_interval" TEXT;
+

litellm_proxy_extras/migrations/20251003165142_add_allowed_tools_to_mcp/migration.sql

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

litellm_proxy_extras/migrations/20251003190954_extra_headers_to_mcp_table/migration.sql

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

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/migrations/20251023141814_add_search_tool_table/migration.sql

@@ -0,0 +1,15 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_SearchToolsTable" (
+    "search_tool_id" TEXT NOT NULL,
+    "search_tool_name" TEXT NOT NULL,
+    "litellm_params" JSONB NOT NULL,
+    "search_tool_info" JSONB,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_SearchToolsTable_pkey" PRIMARY KEY ("search_tool_id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_SearchToolsTable_search_tool_name_key" ON "LiteLLM_SearchToolsTable"("search_tool_name");
+

litellm_proxy_extras/migrations/20251031181430_add_cache_config_table/migration.sql

@@ -0,0 +1,20 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_SSOConfig" (
+    "id" TEXT NOT NULL DEFAULT 'sso_config',
+    "sso_settings" JSONB NOT NULL,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_SSOConfig_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateTable
+CREATE TABLE "LiteLLM_CacheConfig" (
+    "id" TEXT NOT NULL DEFAULT 'cache_config',
+    "cache_settings" JSONB NOT NULL,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_CacheConfig_pkey" PRIMARY KEY ("id")
+);
+

litellm_proxy_extras/migrations/20251101131415_add_managed_vector_store_index_table/migration.sql

@@ -0,0 +1,17 @@
+-- CreateTable
+CREATE TABLE "LiteLLM_ManagedVectorStoreIndexTable" (
+    "id" TEXT NOT NULL,
+    "index_name" TEXT NOT NULL,
+    "litellm_params" JSONB NOT NULL,
+    "index_info" JSONB,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "created_by" TEXT,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+    "updated_by" TEXT,
+
+    CONSTRAINT "LiteLLM_ManagedVectorStoreIndexTable_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "LiteLLM_ManagedVectorStoreIndexTable_index_name_key" ON "LiteLLM_ManagedVectorStoreIndexTable"("index_name");
+

litellm_proxy_extras/migrations/20251103072422_add_static_headers/migration.sql

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

litellm_proxy_extras/migrations/20251104220043_add_credentials_to_mcp_servers/migration.sql

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

litellm_proxy_extras/migrations/migration_lock.toml

@@ -0,0 +1 @@
+provider = "postgresql"