From 9f1882eccaff7edc4bc6a66534b0438a0e097a16 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Thu, 1 Jun 2023 13:14:56 +0200 Subject: [PATCH] Fix --- src/core/taskiq_middlewares.py | 2 +- src/core/taskiq_worker.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/core/taskiq_middlewares.py b/src/core/taskiq_middlewares.py index adf0141..0a04778 100644 --- a/src/core/taskiq_middlewares.py +++ b/src/core/taskiq_middlewares.py @@ -7,7 +7,7 @@ from taskiq.result import TaskiqResult from taskiq_dependencies.dependency import Dependency -class FastAPIREtryMiddleware(SimpleRetryMiddleware): +class FastAPIRetryMiddleware(SimpleRetryMiddleware): @staticmethod def _remove_depends( task_func: Any, message_kwargs: dict[str, Any] diff --git a/src/core/taskiq_worker.py b/src/core/taskiq_worker.py index 95f9055..71e1630 100644 --- a/src/core/taskiq_worker.py +++ b/src/core/taskiq_worker.py @@ -2,7 +2,7 @@ import taskiq_fastapi from taskiq_redis import ListQueueBroker, RedisAsyncResultBackend from core.config import REDIS_URL -from core.taskiq_middlewares import FastAPIREtryMiddleware +from core.taskiq_middlewares import FastAPIRetryMiddleware broker = ( @@ -10,7 +10,7 @@ broker = ( .with_result_backend( RedisAsyncResultBackend(redis_url=REDIS_URL, result_ex_time=5 * 60) ) - .with_middlewares(FastAPIREtryMiddleware()) + .with_middlewares(FastAPIRetryMiddleware()) )