From 8151f4b415664e976449c2d41e440aee4db51beb Mon Sep 17 00:00:00 2001 From: Yann Combarnous Date: Mon, 5 May 2025 10:21:37 +0200 Subject: [PATCH 1/3] feat: add typing for generators in tasks --- taskiq_faststream/broker.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/taskiq_faststream/broker.py b/taskiq_faststream/broker.py index 4ed4e7e..c28699b 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]], + typing.Callable[[], typing.AsyncGenerator[SendableMessage]], ] = None, *, schedule: list[ScheduledTask], From 3aa40bfde6ba03174abf615f50ada82a8c9ed7a2 Mon Sep 17 00:00:00 2001 From: Pastukhov Nikita Date: Mon, 5 May 2025 18:33:53 +0300 Subject: [PATCH 2/3] Update broker.py --- taskiq_faststream/broker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/taskiq_faststream/broker.py b/taskiq_faststream/broker.py index c28699b..cee1606 100644 --- a/taskiq_faststream/broker.py +++ b/taskiq_faststream/broker.py @@ -73,8 +73,8 @@ def task( # type: ignore[override] SendableMessage, typing.Callable[[], SendableMessage], typing.Callable[[], typing.Awaitable[SendableMessage]], - typing.Callable[[], typing.Generator[SendableMessage]], - typing.Callable[[], typing.AsyncGenerator[SendableMessage]], + typing.Callable[[], typing.Generator[SendableMessage, None, None]], + typing.Callable[[], typing.AsyncGenerator[SendableMessage, None, None]], ] = None, *, schedule: list[ScheduledTask], From 2a0df3c9b64c7943b7485c9598e69628dc0152ed Mon Sep 17 00:00:00 2001 From: Pastukhov Nikita Date: Mon, 5 May 2025 18:34:57 +0300 Subject: [PATCH 3/3] Update broker.py --- taskiq_faststream/broker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/taskiq_faststream/broker.py b/taskiq_faststream/broker.py index cee1606..34b6b2a 100644 --- a/taskiq_faststream/broker.py +++ b/taskiq_faststream/broker.py @@ -74,7 +74,7 @@ def task( # type: ignore[override] typing.Callable[[], SendableMessage], typing.Callable[[], typing.Awaitable[SendableMessage]], typing.Callable[[], typing.Generator[SendableMessage, None, None]], - typing.Callable[[], typing.AsyncGenerator[SendableMessage, None, None]], + typing.Callable[[], typing.AsyncGenerator[SendableMessage, None]], ] = None, *, schedule: list[ScheduledTask],