Merge pull request #101 from levinotik/patch-1

Update to use async_sessionmaker
grillazz · Aug 2, 2023 · 2c1d006 · 2c1d006
2 parents 9622a47 + 4c8d783
commit 2c1d006
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/database.py b/app/database.py
@@ -1,7 +1,7 @@
 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 +17,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,)
 
 
 # Dependency