diff --git a/database.py b/database.py index 770383a..12e4ae4 100644 --- a/database.py +++ b/database.py @@ -3,9 +3,11 @@ from sqlalchemy.pool import NullPool from sqlalchemy.orm import sessionmaker, declarative_base from config import settings +connect_args = {"check_same_thread": False} if settings.database_url.startswith("sqlite") else {} + engine = create_engine( settings.database_url, - connect_args={"check_same_thread": False}, + connect_args=connect_args, poolclass=NullPool ) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)