From 7b50d00a637a409df7e63617bd8cfde4b221e56e Mon Sep 17 00:00:00 2001
From: Matt Bertrand <mrbertrand@gmail.com>
Date: Sun, 9 Feb 2025 15:50:45 -0500
Subject: [PATCH] Fix LiteLLM database reset issue

---
 config/litellm_config.yml        | 1 +
 config/postgres/init-litellm.sql | 2 ++
 docker-compose.services.yml      | 2 +-
 env/backend.env                  | 2 +-
 env/backend.local.example.env    | 1 +
 5 files changed, 6 insertions(+), 2 deletions(-)
 create mode 100644 config/postgres/init-litellm.sql

diff --git a/config/litellm_config.yml b/config/litellm_config.yml
index f970a37..dfa4545 100644
--- a/config/litellm_config.yml
+++ b/config/litellm_config.yml
@@ -10,6 +10,7 @@ model_list:
 
 general_settings:
   master_key: os.environ/LITELLM_MASTER_KEY
+  database_url: os.environ/DATABASE_URL
 
 litellm_settings:
   # The following should set default customer budgets, but they are
diff --git a/config/postgres/init-litellm.sql b/config/postgres/init-litellm.sql
new file mode 100644
index 0000000..b1e3deb
--- /dev/null
+++ b/config/postgres/init-litellm.sql
@@ -0,0 +1,2 @@
+CREATE DATABASE litellm;
+GRANT ALL PRIVILEGES ON DATABASE litellm TO postgres;
diff --git a/docker-compose.services.yml b/docker-compose.services.yml
index 6a602a9..980c715 100644
--- a/docker-compose.services.yml
+++ b/docker-compose.services.yml
@@ -76,7 +76,7 @@ services:
       - path: .env
         required: false
     environment:
-      - DATABASE_URL=${DATABASE_URL:-postgres://postgres:postgres@db:5433/litellm}
+      - DATABASE_URL=${LITELLM_DATABASE_URL:-postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@db:${PGPORT}/litellm}
       - OPENAI_API_KEY=${OPENAI_API_KEY}
       - LITELLM_MASTER_KEY=${AI_PROXY_AUTH_TOKEN}
       - LITELLM_SALT_KEY=${AI_PROXY_AUTH_TOKEN}
diff --git a/env/backend.env b/env/backend.env
index 5c91879..726ce4e 100644
--- a/env/backend.env
+++ b/env/backend.env
@@ -45,4 +45,4 @@ KEYCLOAK_DISCOVERY_URL=http://${KEYCLOAK_SVC_HOSTNAME}:${KEYCLOAK_PORT}/realms/o
 KEYCLOAK_SCOPES="openid profile ol-profile"
 
 # Channels settings
-REDIS_DOMAIN=redis://redis:6379/0
\ No newline at end of file
+REDIS_DOMAIN=redis://redis:6379/0
diff --git a/env/backend.local.example.env b/env/backend.local.example.env
index 2e9e22b..d515751 100644
--- a/env/backend.local.example.env
+++ b/env/backend.local.example.env
@@ -47,6 +47,7 @@
 # OPENAI_API_KEY=
 # AI_DEFAULT_RECOMMENDATION_MODEL=
 # AI_DEFAULT_SYLLABUS_MODEL=
+# LITELLM_DATABASE_URL=postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@db:${PGPORT}/litellm
 
 #AWS settings
 # AWS_DEFAULT_REGION=