diff --git a/backend/session.py b/backend/session.py index 62c7ca7..09f9525 100644 --- a/backend/session.py +++ b/backend/session.py @@ -1,15 +1,13 @@ from typing import AsyncGenerator -from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession -from sqlalchemy.orm import sessionmaker +from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker from .config import PG_DATABASE, PG_HOST, PG_PASSWORD, PG_LOGIN DATABASE_URL = f"postgresql+asyncpg://{PG_LOGIN}:{PG_PASSWORD}@{PG_HOST}/{PG_DATABASE}" engine = create_async_engine(DATABASE_URL) -session_maker = sessionmaker( +session_maker = async_sessionmaker( engine, - class_=AsyncSession, expire_on_commit=False, autocommit=False, autoflush=False,