From 3fbe7d1059fc76b25b06b914b02cee569b259304 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Mon, 30 Mar 2026 19:40:16 -0700
Subject: [PATCH] prisma_schema_sync

---
 .circleci/config.yml | 43 ++++++++++++++++++++++---------------------
 1 file changed, 22 insertions(+), 21 deletions(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 126794f3feb..f3edf703e13 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -3295,37 +3295,39 @@ jobs:
       - setup_google_dns
       - attach_workspace:
           at: ~/project
+      - run:
+          name: Install dockerize
+          command: |
+            sudo wget https://github.com/jwilder/dockerize/releases/download/v0.6.1/dockerize-linux-amd64-v0.6.1.tar.gz
+            sudo tar -C /usr/local/bin -xzvf dockerize-linux-amd64-v0.6.1.tar.gz
+            sudo rm dockerize-linux-amd64-v0.6.1.tar.gz
+      - run:
+          name: Start PostgreSQL Database
+          command: |
+            docker run -d \
+              --name postgres-db \
+              -e POSTGRES_USER=postgres \
+              -e POSTGRES_PASSWORD=postgres \
+              -e POSTGRES_DB=litellm_schema_sync \
+              -p 5432:5432 \
+              postgres:14
+      - run:
+          name: Wait for PostgreSQL to be ready
+          command: dockerize -wait tcp://localhost:5432 -timeout 1m
       - run:
           name: Load Docker Database Image
           command: |
             zstd -d litellm-docker-database.tar.zst --stdout | docker load
             docker images | grep litellm-docker-database
       - run:
-          name: Install Neon CLI
+          name: Run schema sync via prisma db push
           command: |
-            npm i -g neonctl
-      - run:
-          name: Install curl and dockerize
-          command: |
-            sudo apt-get update
-            sudo apt-get install -y curl
-            sudo wget https://github.com/jwilder/dockerize/releases/download/v0.6.1/dockerize-linux-amd64-v0.6.1.tar.gz
-            sudo tar -C /usr/local/bin -xzvf dockerize-linux-amd64-v0.6.1.tar.gz
-            sudo rm dockerize-linux-amd64-v0.6.1.tar.gz
-      - run:
-          name: Sync schema on base e2e database
-          command: |
-            BASE_DATABASE_URL=$(neon connection-string \
-              --project-id $NEON_PROJECT_ID \
-              --api-key $NEON_API_KEY \
-              --branch br-fancy-paper-ad1olsb3 \
-              --database-name yuneng-trial-db \
-              --role neondb_owner)
             docker run -d \
               -p 4000:4000 \
-              -e DATABASE_URL=$BASE_DATABASE_URL \
+              -e DATABASE_URL="postgresql://postgres:postgres@host.docker.internal:5432/litellm_schema_sync" \
               -e LITELLM_MASTER_KEY="sk-1234" \
               --name schema-sync \
+              --add-host=host.docker.internal:host-gateway \
               -v $(pwd)/litellm/proxy/example_config_yaml/simple_config.yaml:/app/config.yaml \
               litellm-docker-database:ci \
               --config /app/config.yaml \
@@ -3503,7 +3505,6 @@ workflows:
                 - main
                 - /litellm_.*/
       - prisma_schema_sync:
-          context: e2e_ui_tests
           requires:
             - build_docker_database_image
           filters: