Skip to content

Commit 8151f4b

Browse files
feat: add typing for generators in tasks
1 parent d642e6a commit 8151f4b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

taskiq_faststream/broker.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,8 @@ def task( # type: ignore[override]
7373
SendableMessage,
7474
typing.Callable[[], SendableMessage],
7575
typing.Callable[[], typing.Awaitable[SendableMessage]],
76+
typing.Callable[[], typing.Generator[SendableMessage]],
77+
typing.Callable[[], typing.AsyncGenerator[SendableMessage]],
7678
] = None,
7779
*,
7880
schedule: list[ScheduledTask],

0 commit comments

Comments
 (0)