Merge pull request #84 from yann-combarnous/feature/generator-typings

feat: add typing for generators in tasks

Author: Lancetnik

taskiq_faststream/broker.py

@@ -73,6 +73,8 @@ def task(  # type: ignore[override]
             SendableMessage,
             typing.Callable[[], SendableMessage],
             typing.Callable[[], typing.Awaitable[SendableMessage]],
+            typing.Callable[[], typing.Generator[SendableMessage, None, None]],
+            typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]],
         ] = None,
         *,
         schedule: list[ScheduledTask],