From e04dff8154f0c8a5272234ed2ad4184ad77300f8 Mon Sep 17 00:00:00 2001
From: Masen Furer <m_github@0x26.net>
Date: Mon, 9 Dec 2024 17:03:07 -0800
Subject: [PATCH] use async_sessionmaker

---
 reflex/model.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/reflex/model.py b/reflex/model.py
index 314d91916..03b1cc828 100644
--- a/reflex/model.py
+++ b/reflex/model.py
@@ -25,7 +25,7 @@ from reflex.utils.compat import sqlmodel, sqlmodel_field_has_primary_key
 
 _ENGINE: dict[str, sqlalchemy.engine.Engine] = {}
 _ASYNC_ENGINE: dict[str, sqlalchemy.ext.asyncio.AsyncEngine] = {}
-_AsyncSessionLocal: dict[str | None, sqlalchemy.orm.sessionmaker] = {}
+_AsyncSessionLocal: dict[str | None, sqlalchemy.ext.asyncio.async_sessionmaker] = {}
 
 # Import AsyncSession _after_ reflex.utils.compat
 from sqlmodel.ext.asyncio.session import AsyncSession  # noqa: E402
@@ -532,11 +532,11 @@ def asession(url: str | None = None) -> AsyncSession:
     """
     global _AsyncSessionLocal
     if url not in _AsyncSessionLocal:
-        _AsyncSessionLocal[url] = sqlalchemy.orm.sessionmaker(
+        _AsyncSessionLocal[url] = sqlalchemy.ext.asyncio.async_sessionmaker(
+            bind=get_async_engine(url),
+            class_=AsyncSession,
             autocommit=False,
             autoflush=False,
-            bind=get_async_engine(url),  # type: ignore
-            class_=AsyncSession,
         )
     return _AsyncSessionLocal[url]()