ref: Use AsyncSession in some tests (#5151)

2024-12-08 02:09:43 +01:00 · 2024-12-08 02:09:43 +01:00 · 624a2dde5d
commit 624a2dde5d
parent f97a326023
7 changed files with 53 additions and 47 deletions
--- a/src/backend/base/langflow/services/database/utils.py
+++ b/src/backend/base/langflow/services/database/utils.py
@ -74,7 +74,7 @@ def session_getter(db_service: DatabaseService):
@asynccontextmanager
 async def async_session_getter(db_service: DatabaseService):
    try:
-        session = AsyncSession(db_service.async_engine)
+        session = AsyncSession(db_service.async_engine, expire_on_commit=False)
        yield session
    except Exception:
        logger.exception("Session rollback because of exception")