Update app/db.py

app/db.py

@@ -55,6 +55,18 @@ async def get_user_db(session: AsyncSession = Depends(get_async_session)):
     yield SQLAlchemyUserDatabase(session, User)
 
 
+async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
+    async with async_session() as session:
+        assert isinstance(session, AsyncSession)
+        yield session
+
+
+async def connect() -> None:
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all, checkfirst=True)
+
+
+"""
 async def connect() -> None:
     async with engine.begin() as conn:
         await conn.run_sync(Base.metadata.create_all, checkfirst=True)
@@ -63,3 +75,4 @@ async def connect() -> None:
 async def disconnect() -> None:
     if engine:
         await engine.dispose()
+"""