From 8ee3573d9a9c03c71dc373edc685dd22f4880497 Mon Sep 17 00:00:00 2001 From: Egor Matveev Date: Sat, 28 Dec 2024 13:37:07 +0300 Subject: [PATCH] fix --- app/routers/take.py | 8 ++++---- app/utils/configurator.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/app/routers/take.py b/app/routers/take.py index d4e3d4f..cd11f04 100644 --- a/app/routers/take.py +++ b/app/routers/take.py @@ -5,7 +5,7 @@ import pydantic import typing from app.storage.mongo import tasks -from app.utils.configurator import configurator +# from app.utils.configurator import configurator DEFAULT_RETRY_AFTER = 0.2 @@ -30,6 +30,6 @@ async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response: task = await tasks.take_task(queue) if not task: return Response(task=None) - retry_after_config = configurator.get_config('retry_after') - retry_after = retry_after_config.get(queue) or retry_after_config.get('default') or DEFAULT_RETRY_AFTER - return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload), retry_after=retry_after) + # retry_after_config = configurator.get_config('retry_after') + # retry_after = retry_after_config.get(queue) or retry_after_config.get('default') or DEFAULT_RETRY_AFTER + return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload), retry_after=DEFAULT_RETRY_AFTER) diff --git a/app/utils/configurator.py b/app/utils/configurator.py index 062ff71..88bddf7 100644 --- a/app/utils/configurator.py +++ b/app/utils/configurator.py @@ -85,4 +85,4 @@ class ConfiguratorClient: return self.experiment_storage[name] -configurator = ConfiguratorClient('queues', os.getenv('STAGE', 'local')) +# configurator = ConfiguratorClient('queues', os.getenv('STAGE', 'local'))