From d0c27c1fd57a01226681dc6bf74868aec6b8eeb8 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Thu, 7 May 2026 10:06:55 -0700
Subject: [PATCH] feat(v2 managed agents): hand-written migration for
 managed_agent tables

---
 .../migration.sql                             | 42 +++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 litellm-proxy-extras/litellm_proxy_extras/migrations/20260507120000_add_managed_agent_tables/migration.sql

diff --git a/litellm-proxy-extras/litellm_proxy_extras/migrations/20260507120000_add_managed_agent_tables/migration.sql b/litellm-proxy-extras/litellm_proxy_extras/migrations/20260507120000_add_managed_agent_tables/migration.sql
new file mode 100644
index 00000000000..2e535797915
--- /dev/null
+++ b/litellm-proxy-extras/litellm_proxy_extras/migrations/20260507120000_add_managed_agent_tables/migration.sql
@@ -0,0 +1,42 @@
+-- CreateTable
+CREATE TABLE IF NOT EXISTS "LiteLLM_ManagedAgent" (
+    "id" TEXT NOT NULL,
+    "name" TEXT NOT NULL,
+    "config" JSONB NOT NULL,
+    "created_by" TEXT,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+
+    CONSTRAINT "LiteLLM_ManagedAgent_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateTable
+CREATE TABLE IF NOT EXISTS "LiteLLM_ManagedAgentSession" (
+    "id" TEXT NOT NULL,
+    "agent_id" TEXT NOT NULL,
+    "sandbox_type" TEXT NOT NULL,
+    "sandbox_size" TEXT NOT NULL,
+    "sandbox_timeout_minutes" INTEGER NOT NULL,
+    "sandbox_idle_timeout_minutes" INTEGER NOT NULL,
+    "sandbox_image" TEXT,
+    "sandbox_url" TEXT,
+    "sandbox_metadata" JSONB,
+    "status" TEXT NOT NULL DEFAULT 'provisioning',
+    "repos" JSONB,
+    "env_vars" JSONB,
+    "created_by" TEXT,
+    "created_at" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3) NOT NULL,
+    "terminated_at" TIMESTAMP(3),
+
+    CONSTRAINT "LiteLLM_ManagedAgentSession_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE INDEX IF NOT EXISTS "LiteLLM_ManagedAgentSession_agent_id_idx" ON "LiteLLM_ManagedAgentSession"("agent_id");
+
+-- CreateIndex
+CREATE INDEX IF NOT EXISTS "LiteLLM_ManagedAgentSession_created_by_idx" ON "LiteLLM_ManagedAgentSession"("created_by");
+
+-- CreateIndex
+CREATE INDEX IF NOT EXISTS "LiteLLM_ManagedAgentSession_status_idx" ON "LiteLLM_ManagedAgentSession"("status");