feat(db): add session management and database connection setup

core/config.py

@@ -8,6 +8,15 @@ class Settings(BaseSettings):
     APP_NAME: str = "Makhfi AI"
     APP_VERSION: str = "0.1.0"
 
+    # --- Supabase Configuration ---
+    SUPABASE_KEY: str
+    SUPABASE_URL: str
+    SUPABASE_JWT_SECRET: str
+
+    # --- Database Configuration ---
+    DATABASE_URL: str
+    DB_CONNECT_ARGS: dict = {"sslmode": "require"}
+    
     # --- API Keys ---
     GOOGLE_API_KEY: str
     GROQ_API_KEY: str

db/models/__init__.py

@@ -0,0 +1,4 @@
+from db.models.chat import Chat
+from db.models.message import Message
+
+__all__ = ["Chat", "Message"]

db/session.py

@@ -0,0 +1,26 @@
+from sqlmodel import SQLModel, create_engine, Session
+
+from db.models import Chat, Message
+from core.config import get_settings
+
+settings = get_settings()
+
+engine = create_engine(
+    settings.DATABASE_URL,
+    pool_pre_ping=True,
+    connect_args=settings.DB_CONNECT_ARGS
+)
+
+def get_db():
+    """
+    Dependency function to get a database session.
+    Ensures the session is always closed after the request.
+    """
+    with Session(engine) as session:
+        yield session
+
+def create_db_and_tables():
+    """
+    Utility function to create database tables.
+    """
+    SQLModel.metadata.create_all(engine)