From cd6f28cdbc91f7415196a1e0e354160f3103fbc1 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Sun, 21 May 2023 01:59:17 +0200 Subject: [PATCH] Fix --- src/core/taskiq_middlewares.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/core/taskiq_middlewares.py b/src/core/taskiq_middlewares.py index 219cf9d..adf0141 100644 --- a/src/core/taskiq_middlewares.py +++ b/src/core/taskiq_middlewares.py @@ -14,6 +14,8 @@ class FastAPIREtryMiddleware(SimpleRetryMiddleware): ) -> dict[str, Any]: sig = signature(task_func) + keys_to_remove = [] + for key in message_kwargs.keys(): param = sig.parameters.get(key, None) @@ -21,7 +23,10 @@ class FastAPIREtryMiddleware(SimpleRetryMiddleware): continue if isinstance(param.default, Dependency): - message_kwargs.pop(key) + keys_to_remove.append(key) + + for key in keys_to_remove: + message_kwargs.pop(key) return message_kwargs