diff --git a/taskiq_faststream/broker.py b/taskiq_faststream/broker.py
index 4ed4e7e..34b6b2a 100644
--- a/taskiq_faststream/broker.py
+++ b/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],