From 63d99abf41c3fd836af562a9d4bcbd2509261dbc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ren=C3=A9=20Pfitzner?= <pfitzner.rene@gmail.com>
Date: Sat, 21 Jun 2025 13:37:21 +0100
Subject: [PATCH] Change default db pooling

---
 backend/open_webui/env.py         | 8 +++-----
 backend/open_webui/internal/db.py | 6 +++++-
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py
index 7061fd252..b3eca9856 100644
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@@ -272,15 +272,13 @@ if "postgres://" in DATABASE_URL:
 
 DATABASE_SCHEMA = os.environ.get("DATABASE_SCHEMA", None)
 
-DATABASE_POOL_SIZE = os.environ.get("DATABASE_POOL_SIZE", 0)
+DATABASE_POOL_SIZE = os.environ.get("DATABASE_POOL_SIZE", None)
 
-if DATABASE_POOL_SIZE == "":
-    DATABASE_POOL_SIZE = 0
-else:
+if DATABASE_POOL_SIZE != None:
     try:
         DATABASE_POOL_SIZE = int(DATABASE_POOL_SIZE)
     except Exception:
-        DATABASE_POOL_SIZE = 0
+        DATABASE_POOL_SIZE = None
 
 DATABASE_POOL_MAX_OVERFLOW = os.environ.get("DATABASE_POOL_MAX_OVERFLOW", 0)
 
diff --git a/backend/open_webui/internal/db.py b/backend/open_webui/internal/db.py
index f024354a6..c3d729711 100644
--- a/backend/open_webui/internal/db.py
+++ b/backend/open_webui/internal/db.py
@@ -94,10 +94,14 @@ else:
             pool_pre_ping=True,
             poolclass=QueuePool,
         )
-    else:
+    elif DATABASE_POOL_SIZE == 0:
         engine = create_engine(
             SQLALCHEMY_DATABASE_URL, pool_pre_ping=True, poolclass=NullPool
         )
+    else:
+        engine = create_engine(
+            SQLALCHEMY_DATABASE_URL, pool_pre_ping=True
+        )
 
 
 SessionLocal = sessionmaker(