diff --git a/project/app/db.py b/project/app/db.py index 95b7692..d894394 100644 --- a/project/app/db.py +++ b/project/app/db.py @@ -1,14 +1,14 @@ import os -from sqlmodel import SQLModel, create_engine -from sqlmodel.ext.asyncio.session import AsyncSession, AsyncEngine +from sqlmodel import SQLModel +from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, AsyncEngine from sqlalchemy.orm import sessionmaker DATABASE_URL = os.environ.get("DATABASE_URL") -engine = AsyncEngine(create_engine(DATABASE_URL, echo=True, future=True)) +engine = create_async_engine(DATABASE_URL, echo=True, future=True) async def init_db(): async with engine.begin() as conn: @@ -21,4 +21,4 @@ async def get_session() -> AsyncSession: engine, class_=AsyncSession, expire_on_commit=False ) async with async_session() as session: - yield session \ No newline at end of file + yield session