From 6221031314bda5b2d6e2f3fd7499cbd5463d50b0 Mon Sep 17 00:00:00 2001 From: Pavel Kirilin Date: Wed, 5 Mar 2025 03:13:45 +0100 Subject: [PATCH 1/2] Fixed types for coroutine funcs. --- taskiq_pipelines/pipeliner.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/taskiq_pipelines/pipeliner.py b/taskiq_pipelines/pipeliner.py index 44562ac..8112987 100644 --- a/taskiq_pipelines/pipeliner.py +++ b/taskiq_pipelines/pipeliner.py @@ -1,3 +1,4 @@ +from types import CoroutineType from typing import ( Any, Coroutine, @@ -69,7 +70,9 @@ def call_next( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, _T2]], + AsyncKicker[Any, CoroutineType[Any, Any, _T2]], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], + AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, _T2]], ], param_name: Union[Optional[str], Literal[-1]] = None, **additional_kwargs: Any, @@ -128,7 +131,9 @@ def call_after( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, _T2]], + AsyncKicker[Any, CoroutineType[Any, Any, _T2]], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], + AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, _T2]], ], **additional_kwargs: Any, ) -> "Pipeline[_FuncParams, _T2]": ... @@ -182,7 +187,9 @@ def map( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, _T2]], + AsyncKicker[Any, CoroutineType[Any, Any, _T2]], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], + AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, _T2]], ], param_name: Optional[str] = None, skip_errors: bool = False, @@ -251,7 +258,9 @@ def filter( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, bool]], + AsyncKicker[Any, CoroutineType[Any, Any, bool]], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, bool]], + AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, bool]], ], param_name: Optional[str] = None, skip_errors: bool = False, From 03e1a5b1fb185e84ca90f2414c321bb4b306a5b0 Mon Sep 17 00:00:00 2001 From: Pavel Kirilin Date: Wed, 5 Mar 2025 03:16:42 +0100 Subject: [PATCH 2/2] Fixed errors. --- taskiq_pipelines/pipeliner.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/taskiq_pipelines/pipeliner.py b/taskiq_pipelines/pipeliner.py index 8112987..f7c3eb8 100644 --- a/taskiq_pipelines/pipeliner.py +++ b/taskiq_pipelines/pipeliner.py @@ -70,9 +70,9 @@ def call_next( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, _T2]], - AsyncKicker[Any, CoroutineType[Any, Any, _T2]], + AsyncKicker[Any, "CoroutineType[Any, Any, _T2]"], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], - AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, _T2]], + AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, _T2]"], ], param_name: Union[Optional[str], Literal[-1]] = None, **additional_kwargs: Any, @@ -131,9 +131,9 @@ def call_after( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, _T2]], - AsyncKicker[Any, CoroutineType[Any, Any, _T2]], + AsyncKicker[Any, "CoroutineType[Any, Any, _T2]"], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], - AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, _T2]], + AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, _T2]"], ], **additional_kwargs: Any, ) -> "Pipeline[_FuncParams, _T2]": ... @@ -187,9 +187,9 @@ def map( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, _T2]], - AsyncKicker[Any, CoroutineType[Any, Any, _T2]], + AsyncKicker[Any, "CoroutineType[Any, Any, _T2]"], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], - AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, _T2]], + AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, _T2]"], ], param_name: Optional[str] = None, skip_errors: bool = False, @@ -258,9 +258,9 @@ def filter( self: "Pipeline[_FuncParams, _ReturnType]", task: Union[ AsyncKicker[Any, Coroutine[Any, Any, bool]], - AsyncKicker[Any, CoroutineType[Any, Any, bool]], + AsyncKicker[Any, "CoroutineType[Any, Any, bool]"], AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, bool]], - AsyncTaskiqDecoratedTask[Any, CoroutineType[Any, Any, bool]], + AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, bool]"], ], param_name: Optional[str] = None, skip_errors: bool = False,