pycharm中Asyncconcontextmanager的类型提示

Question

我正在使用Pycharm Professional 2022.1。 我有一种异步方法在异步环境中使用数据库来管理会话

@contextlib.asynccontextmanager
async def begin_transaction(self) -> AsyncIterator[Session]:
    _session: Optional[Session] = None
    try:
        await self.session_lock.acquire()
        _session: Session = Session()
        yield _session
    finally:
        if _session is not None:
            _session.commit()
            _session.close()

        if self.session_lock.locked():
            self.session_lock.release()

，但是当我在异步语句中使用此方法时，Pycharm无法识别类型：

net/5zozl.png“ rel =“ nofollow noreferrer”>

那么，您是否有任何技巧可以键入提示a的输出变量，而无需手动指定pycharm上的语句？