[Backport release-1.17] feat: Add aio module and update mypy configuration to include aio files

[Backport release-1.17] feat: Add aio module and update mypy configuration to include aio files #2197