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'))