Argument of type "PrefectFuture[int, Sync]" cannot be assigned to parameter "i" of type "int" in function "__call__"

Description

pyright error:

basic_flow.py:12:18 - error: Argument of type "PrefectFuture[int, Sync]" cannot be assigned to parameter "i" of type "int" in function "__call__"
    "PrefectFuture[int, Sync]" is incompatible with "int" (reportGeneralTypeIssues)

mypy error:

basic_flow.py:12: error: No overload variant of "__call__" of "Task" matches argument type "PrefectFuture[None, Literal[False]]"
basic_flow.py:12: note: Possible overload variants:
basic_flow.py:12: note:     def __call__(self, i: int) -> PrefectFuture[None, Literal[False]]
basic_flow.py:12: note:     def __call__(self, i: int) -> Awaitable[PrefectFuture[<nothing>, Literal[True]]]

Reproduction / Example

basic_flow.py:

from prefect import flow, get_run_logger, task
from prefect.futures import PrefectFuture
from prefect.utilities.asyncio import Sync


@flow
def add_flow(i: int) -> PrefectFuture[int, Sync]:
    # result is PrefectFuture[int, Sync]
    result = add_one(i)

    # passing the result future will resolve to its int value
    print_result(result)    # <--- line 12: pyright/mypy errors here
    return result

@task
def add_one(i: int) -> int:
    return i + 1

@task
def print_result(i: int) -> None:
    logger = get_run_logger()
    logger.info(f"print_result: {i=}")

if __name__ == "__main__":
    add_flow(1)

prefect 2.0b8

[madkinsz]

The comment above still stands:

To support this, we’d need to update the input signature of your task to make each argument a Union[Future[T], T]. I do not believe this is possible with Python parameter specs yet. We may be able to write a mypy plugin to make this feasible, but pyright does not support plugins.

[madkinsz]

We automatically coerce futures into values for you, but that kind of magic isn’t supported by type checkers. I’m not sure this is something we can reasonably address. To support this, we’d need to update the input signature of your task to make each argument a Union[Future[T], T]. I do not believe this is possible with Python parameter specs yet. We may be able to write a mypy plugin to make this feasible, but pyright does not support plugins.