def __init__(self, session: AsyncSession):
        """Initialize the UnitOfWork with a database session.

        Args:
        ----
            session (AsyncSession): The database session.

        """
        self.session = session
        self._committed = False
        self._rolledback = False
Initialize the UnitOfWork with a database session.

        Args:
        ----
            session (AsyncSession): The database session.

        
backend/airweave/db/unit_of_work.py