Python: Gather a dictionary of asyncio Task instances while preserving keys

gather_dict.py

async def gather_dict(tasks: dict):
    async def mark(key, coro):
        return key, await coro

    return {
        key: result
        for key, result in await gather(
            *(mark(key, coro) for key, coro in tasks.items())
        )
    }

Type hints with any types:

import asyncio
from collections.abc import Awaitable, Mapping
from typing import TypeVar

_K = TypeVar("_K")
_V = TypeVar("_V")


async def gather_dict(tasks: Mapping[_K, Awaitable[_V]]) -> dict[_K, _V]:
    async def inner(key: _K, coro: Awaitable[_V]) -> tuple[_K, _V]:
        return key, await coro

    pair_tasks = (inner(key, coro) for key, coro in tasks.items())
    return dict(await asyncio.gather(*pair_tasks))

Variation for nested dicts/lists

import asyncio
from collections.abc import Awaitable, Mapping
from typing import Any, TypeVar

_K = TypeVar("_K")
_V = TypeVar("_V")

async def gather_dict(tasks: Mapping[_K, Awaitable[_V] | Mapping | list]) -> dict[_K, _V]:
    async def resolve(value):
        if isinstance(value, Mapping):
            return await gather_dict(value)
        elif isinstance(value, list):
            return [await resolve(v) for v in value]
        elif inspect.isawaitable(value):
            return await value
        else:
            return value

    async def inner(key: _K, value) -> tuple[_K, Any]:
        return key, await resolve(value)

    pair_tasks = (inner(key, value) for key, value in tasks.items())
    return dict(await asyncio.gather(*pair_tasks))