From 0117159da0651915119c319c86c91b3afaba1544 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Wed, 11 Jan 2023 13:56:21 +0100 Subject: [PATCH] returning 503 Service Unavailable when requested task type is not available --- backend/oasst_backend/tree_manager.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/oasst_backend/tree_manager.py b/backend/oasst_backend/tree_manager.py index 16dc4d16..fd7c1c67 100644 --- a/backend/oasst_backend/tree_manager.py +++ b/backend/oasst_backend/tree_manager.py @@ -1,5 +1,6 @@ import random from enum import Enum +from http import HTTPStatus from typing import Optional, Tuple from uuid import UUID @@ -217,14 +218,16 @@ class TreeManager: if task_type == TaskType.NONE: raise OasstError( f"No tasks of type '{desired_task_type.value}' are currently available.", - OasstErrorCode.TASK_REQUESTED_TYPE_UNAVAILABLE, + OasstErrorCode.TASK_REQUESTED_TYPE_NOT_AVAILABLE, + HTTPStatus.SERVICE_UNAVAILABLE, ) if task_role != TaskRole.ANY: # Todo: Allow role specific message selection... raise OasstError( f"No tasks of type '{desired_task_type.value}' are currently available.", - OasstErrorCode.TASK_REQUESTED_TYPE_UNAVAILABLE, + OasstErrorCode.TASK_REQUESTED_TYPE_NOT_AVAILABLE, + HTTPStatus.SERVICE_UNAVAILABLE, ) message_tree_id = None