From 6dac15a71279de2de9a6c63f16b7715c034ee3dc Mon Sep 17 00:00:00 2001
From: Andrew Stout <stout01@users.noreply.github.com>
Date: Sat, 16 Aug 2025 17:40:06 -0500
Subject: [PATCH] Update db schema after updating litellm

---
 ct/litellm.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ct/litellm.sh b/ct/litellm.sh
index 4c5d7c9d..e196542b 100644
--- a/ct/litellm.sh
+++ b/ct/litellm.sh
@@ -39,6 +39,10 @@ function update_script() {
     msg_info "Updating $APP"
     $STD "$VENV_PATH/bin/python" -m pip install --upgrade litellm[proxy] prisma
 
+    msg_info "Updating DB Schema"
+    uv --directory=/opt/litellm run litellm --config /opt/litellm/litellm.yaml --use_prisma_db_push --skip_server_startup
+    msg_ok "DB Schema Updated"
+
     msg_info "Starting ${APP}"
     systemctl start litellm.service
     msg_ok "Started ${APP}"