diff --git a/fastapi_users/types.py b/fastapi_users/types.py index 39b37ca6..6fbe8959 100644 --- a/fastapi_users/types.py +++ b/fastapi_users/types.py @@ -1,7 +1,13 @@ -from typing import Callable, Coroutine, TypeVar, Union +from typing import AsyncGenerator, Callable, Coroutine, Generator, TypeVar, Union RETURN_TYPE = TypeVar("RETURN_TYPE") DependencyCallable = Callable[ - ..., Union[RETURN_TYPE, Coroutine[None, None, RETURN_TYPE]] + ..., + Union[ + RETURN_TYPE, + Coroutine[None, None, RETURN_TYPE], + AsyncGenerator[RETURN_TYPE, None], + Generator[RETURN_TYPE, None, None], + ], ]