models/db.py

from sqlmodel import SQLModel, create_engine, Session

from config.settings import settings

engine = create_engine(settings.database_url, echo=False)

def init_db():
    SQLModel.metadata.create_all(engine)

def get_session():
    return Session(engine)