From b2728791871d95061049fd215fdfdc5bdb45893e Mon Sep 17 00:00:00 2001
From: gitea_admin <admin@druppie.local>
Date: Thu, 11 Jun 2026 10:33:18 +0000
Subject: [PATCH] Automated commit

---
 app/database.py | 36 ++++++++++--------------------------
 1 file changed, 10 insertions(+), 26 deletions(-)

diff --git a/app/database.py b/app/database.py
index 1db343e..ecfacf6 100644
--- a/app/database.py
+++ b/app/database.py
@@ -1,19 +1,21 @@
 import logging
-from sqlalchemy import create_engine, text
+from sqlalchemy import create_engine
 from sqlalchemy.orm import DeclarativeBase, sessionmaker
-
 from app.config import settings
 
 logger = logging.getLogger(__name__)
-
-engine = create_engine(settings.database_url)
+engine = create_engine(
+    settings.database_url,
+    pool_pre_ping=True,
+    pool_size=5,
+    max_overflow=10,
+    pool_recycle=1800,
+)
 SessionLocal = sessionmaker(bind=engine)
 
-
 class Base(DeclarativeBase):
     pass
 
-
 def get_db():
     db = SessionLocal()
     try:
@@ -21,28 +23,10 @@ def get_db():
     finally:
         db.close()
 
-
 def init_db():
-    """Create tables if they don't exist.
-
-    Uses an advisory lock to prevent race conditions when multiple
-    gunicorn workers start simultaneously.
-    """
     import app.models  # noqa: F401
-    import app.rag  # noqa: F401 — register vector_* tables (pgvector)
     try:
-        import app.chat  # noqa: F401 — register chat models
+        import app.chat  # noqa: F401
     except ImportError:
         pass
-
-    with engine.connect() as conn:
-        # PostgreSQL advisory lock to prevent concurrent CREATE TABLE
-        try:
-            conn.execute(text("SELECT pg_advisory_lock(42)"))
-            conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
-            Base.metadata.create_all(bind=engine)
-            conn.execute(text("SELECT pg_advisory_unlock(42)"))
-            conn.commit()
-        except Exception:
-            # Fallback for non-PostgreSQL (e.g. SQLite in tests)
-            Base.metadata.create_all(bind=engine)
+    Base.metadata.create_all(bind=engine)