Update to use async_sessionmaker

grillazz · Aug 2, 2023 · dfa5510 · dfa5510
1 parent 9622a47
commit dfa5510
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app/database.py b/app/database.py
@@ -1,7 +1,8 @@
 from collections.abc import AsyncGenerator
 
 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.ext.asyncio import async_sessionmaker
+
 
 from app import config
 from app.utils.logging import AppLogger
@@ -17,7 +18,7 @@
 
 # expire_on_commit=False will prevent attributes from being expired
 # after commit.
-AsyncSessionFactory = sessionmaker(engine, autoflush=False, expire_on_commit=False, class_=AsyncSession)
+AsyncSessionFactory = async_sessionmaker(engine, autoflush=False, expire_on_commit=False, class_=AsyncSession)
 
 
 # Dependency