import asyncio
import random

from typing import Literal, reveal_type


async def get_result() -> Literal["A", "B", "C"]:
    return random.choice(["A", "B", "C"])


# correct inference:
async def process_one():
    task = asyncio.create_task(get_result())

    reveal_type(task)  # `Task[Literal["A", "B", "C"]]`
    result = await task
    reveal_type(result)  # `Literal["A", "B", "C"]`


# incorrect inference:
async def process_many():
    callbacks = [asyncio.create_task(get_result()) for _ in range(5)]
    reveal_type(callbacks)  # `list[Unknown | Task[str]]`
    for callback in callbacks:
        reveal_type(callback)  # `Unknown | Task[str]`
        value = await callback
        reveal_type(value)  # `Unknown | str`