diff --git a/backend/common/core/db.py b/backend/common/core/db.py index 37eff29b..dd6c9781 100644 --- a/backend/common/core/db.py +++ b/backend/common/core/db.py @@ -1,14 +1,15 @@ from sqlmodel import Session, create_engine, SQLModel - from common.core.config import settings +engine = create_engine(str(settings.SQLALCHEMY_DATABASE_URI), pool_size=20, max_overflow=30, pool_recycle=3600, + pool_pre_ping=True) -engine = create_engine(str(settings.SQLALCHEMY_DATABASE_URI)) def get_session(): with Session(engine) as session: yield session + def init_db(): SQLModel.metadata.create_all(engine)