db bug

backend/core/config.py

@@ -1,19 +1,19 @@
-# pydantic-settings: env vars, constants (TOP_K, etc.)
+import os
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
 class Settings(BaseSettings):
-    DATABASE_URL: str
-    GROQ_API_KEY: str
-    UPLOAD_DIR: str = "./uploads"
-    FASTEMBED_CACHE_PATH: str = "../fastembed_cache"
-    SECRET_KEY: str
+    DATABASE_URL: str = ""
+    GROQ_API_KEY: str = ""
+    UPLOAD_DIR: str = "/tmp/uploads" if os.environ.get("VERCEL") else "./uploads"
+    FASTEMBED_CACHE_PATH: str = "/tmp/fastembed_cache" if os.environ.get("VERCEL") else "../fastembed_cache"
+    SECRET_KEY: str = "default_secret_key_change_in_production_32bytes"
     TOP_K: int = 5
     CHUNK_SIZE: int = 500
     CHUNK_OVERLAP: int = 50
     MAX_FILE_SIZE: int = 10 * 1024 * 1024  # 10MB
     # Comma-separated list of allowed CORS origins (e.g., "http://localhost:5173,https://example.com")
-    ALLOWED_ORIGINS: str = "http://localhost:5173,http://localhost:80"
+    ALLOWED_ORIGINS: str = "*"
     # Number of uvicorn workers for production
     UVICORN_WORKERS: int = 4
     # Database connection pool settings
@@ -25,11 +25,10 @@ class Settings(BaseSettings):
     @property
     def allowed_origins_list(self) -> list[str]:
         """Parse ALLOWED_ORIGINS comma-separated string into a list."""
+        if self.ALLOWED_ORIGINS == "*":
+            return ["*"]
         return [origin.strip() for origin in self.ALLOWED_ORIGINS.split(",") if origin.strip()]
 
-    def model_post_init(self, __context: object) -> None:
-        if not self.SECRET_KEY:
-            raise ValueError("SECRET_KEY environment variable must be set when auth is enabled")
 
 
 settings = Settings(_env_file=".env", _env_file_encoding="utf-8")  # type: ignore[call-arg]

backend/db/init.py

@@ -11,27 +11,43 @@ logger = logging.getLogger(__name__)
 
 
 async def ensure_database_exists() -> None:
-    base_url, db_name = settings.DATABASE_URL.rsplit("/", 1)
-    admin_engine = create_async_engine(
-        f"{base_url}/postgres",
-        isolation_level="AUTOCOMMIT",
-    )
+    if not settings.DATABASE_URL or "/" not in settings.DATABASE_URL:
+        logger.warning("DATABASE_URL is not set or invalid.")
+        return
     try:
-        async with admin_engine.connect() as conn:
-            exists = await conn.scalar(
-                text("SELECT 1 FROM pg_database WHERE datname = :name"),
-                {"name": db_name},
-            )
-            if not exists:
-                await conn.execute(text(f'CREATE DATABASE "{db_name}"'))
-                logger.info("Created database '%s'.", db_name)
-    finally:
-        await admin_engine.dispose()
+        base_url, db_name = settings.DATABASE_URL.rsplit("/", 1)
+        # Strip query parameters if present (e.g. ?sslmode=require)
+        if "?" in db_name:
+            db_name = db_name.split("?")[0]
+        admin_engine = create_async_engine(
+            f"{base_url}/postgres",
+            isolation_level="AUTOCOMMIT",
+        )
+        try:
+            async with admin_engine.connect() as conn:
+                exists = await conn.scalar(
+                    text("SELECT 1 FROM pg_database WHERE datname = :name"),
+                    {"name": db_name},
+                )
+                if not exists:
+                    await conn.execute(text(f'CREATE DATABASE "{db_name}"'))
+                    logger.info("Created database '%s'.", db_name)
+        finally:
+            await admin_engine.dispose()
+    except Exception as e:
+        logger.warning("Could not auto-create database (managed cloud DBs usually pre-create databases): %s", e)
 
 
 async def init_database() -> None:
+    if not settings.DATABASE_URL:
+        logger.warning("DATABASE_URL is not configured.")
+        return
     await ensure_database_exists()
-    async with engine.begin() as conn:
-        await conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
-        await conn.run_sync(Base.metadata.create_all)
-    logger.info("Database schema is ready.")
+    try:
+        async with engine.begin() as conn:
+            await conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
+            await conn.run_sync(Base.metadata.create_all)
+        logger.info("Database schema is ready.")
+    except Exception as e:
+        logger.error("Database initialization failed: %s", e)
+

backend/db/session.py

@@ -2,11 +2,11 @@ from typing import AsyncGenerator
 from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession
 from core.config import settings
 
+db_url = settings.DATABASE_URL if settings.DATABASE_URL else "sqlite+aiosqlite:///:memory:"
+
 engine = create_async_engine(
-    settings.DATABASE_URL,
+    db_url,
     echo=False,  # Disable SQL echo in production
-    pool_size=settings.DB_POOL_SIZE,
-    max_overflow=settings.DB_MAX_OVERFLOW,
     pool_pre_ping=True,
 )