scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-02 02:11:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

feat: migration for LiteLLM_AgentVMConfig BYOC AWS settings table

Browse source 
Picked up by `prisma migrate deploy` on proxy startup (the
`litellm-proxy-extras` package bundles its own migration directory).
LIT-2878
This commit is contained in:
Ishaan Jaff 2026-05-06 15:26:01 -07:00
parent b65ebb586b
commit 2f6aac4d86
No known key found for this signature in database
1 changed files with 28 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
litellm-proxy-extras/litellm_proxy_extras/migrations/20260506220000_add_agent_vm_config/migration.sql Normal file
View file

@ -0,0 +1,28 @@
-- Adds `LiteLLM_AgentVMConfig` for BYOC AWS settings keyed by team.
-- Used by `litellm/proxy/agent_session_endpoints/vm_providers/ec2.py` to
-- provision per-session EC2 instances in the customer's AWS account.
--
-- The `aws_creds_enc` column is a JSON blob whose individual fields are
-- encrypted via `encrypt_value_helper` (libsodium SecretBox), so partial DB
-- exports never expose the secret. Schema fields match
-- `litellm/proxy/agent_session_endpoints/vm_providers/base.Ec2Config`.
CREATE TABLE IF NOT EXISTS "LiteLLM_AgentVMConfig" (
"team_id" TEXT NOT NULL,
"aws_creds_enc" TEXT,
"region" TEXT,
"subnet_id" TEXT,
"security_group_id" TEXT,
"iam_instance_profile" TEXT,
"ami_id" TEXT,
"instance_type" TEXT,
"use_spot" BOOLEAN NOT NULL DEFAULT true,
"warm_pool_size" INTEGER NOT NULL DEFAULT 0,
"network_access" JSONB,
"created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
"created_by" TEXT NOT NULL,
"updated_at" TIMESTAMP(3) NOT NULL,
"updated_by" TEXT NOT NULL,
CONSTRAINT "LiteLLM_AgentVMConfig_pkey" PRIMARY KEY ("team_id")
);