From 79c885956a225bb429e976b0b1f7460e998643f4 Mon Sep 17 00:00:00 2001
From: Mario Trangoni <mjtrangoni@gmail.com>
Date: Tue, 6 May 2025 17:36:48 +0200
Subject: [PATCH] feat: Split DATABASE_URL configuration to make it more robust
 #13433

Signed-off-by: Mario Trangoni <mjtrangoni@gmail.com>
---
 backend/open_webui/env.py | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py
index 59557349e..e4a747bee 100644
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@@ -262,7 +262,41 @@ if os.path.exists(f"{DATA_DIR}/ollama.db"):
 else:
     pass
 
-DATABASE_URL = os.environ.get("DATABASE_URL", f"sqlite:///{DATA_DIR}/webui.db")
+# Get all Database related environment variables
+DB_TYPE = os.environ.get("DATABASE_TYPE", "sqlite")
+DB_USER = os.environ.get("DATABASE_USER", "openwebui")
+DB_PASSWORD = os.environ.get("DATABASE_PASSWORD", "")
+DB_HOST = os.environ.get("DATABASE_HOST", "localhost")
+DB_PORT = os.environ.get("DATABASE_PORT", "5432")
+DB_NAME = os.environ.get("DATABASE_NAME", "openwebui")
+DB_DEFAULT_URL = f"sqlite:///{DATA_DIR}/webui.db"
+DB_URL = os.environ.get("DATABASE_URL", DB_DEFAULT_URL)
+
+# Define the database URL
+DATABASE_URL = ""
+
+# Construct the database URL based on the database type
+if DB_URL and DB_URL != DB_DEFAULT_URL:
+    # emulate the legacy default if DATABASE_URL environment variable is set
+    DATABASE_URL = DB_URL
+elif DB_TYPE == "sqlite":
+    # Ensure that SQLite uses the default
+    DATABASE_URL = DB_DEFAULT_URL
+elif DB_TYPE == "postgresql":
+    # Check if essential variables are set
+    if not (DB_USER and DB_NAME):
+        raise ValueError("DATABASE_USER and DATABASE_NAME must be set for PostgreSQL.")
+    if not DB_PASSWORD:
+        DATABASE_URL = (
+            f"postgresql://{DB_USER}@{DB_HOST}:{DB_PORT}/{DB_NAME}"
+        )
+    else:
+        DATABASE_URL = (
+                f"postgresql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}"
+        )
+else:
+    raise ValueError(f"Unsupported database type: {DB_TYPE}. Supported types "
+                     "are: sqlite, postgresql")
 
 # Replace the postgres:// with postgresql://
 if "postgres://" in DATABASE_URL: