mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-27 16:10:30 +08:00
Andreas Köpf
ead51ff423
* only ranking assistant replies by default * add tasks/availability endpoint allow to specify desired task * move rank_prompter_replies option to TreeManagerConfiguration * fix type annotation * remove desired_task_type from _random_task_selection() * fix typo * Convert query_tree_size to sqlachemy, return 'full' text-labeling tasks if they were explicitly requested
87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
from enum import IntEnum
|
|
from http import HTTPStatus
|
|
|
|
|
|
class OasstErrorCode(IntEnum):
|
|
"""
|
|
Error codes of the Open-Assistant backend API.
|
|
|
|
Ranges:
|
|
0-1000: general errors
|
|
1000-2000: tasks endpoint
|
|
2000-3000: prompt_repository, task_repository, user_repository
|
|
3000-4000: external resources
|
|
"""
|
|
|
|
# 0-1000: general errors
|
|
GENERIC_ERROR = 0
|
|
DATABASE_URI_NOT_SET = 1
|
|
API_CLIENT_NOT_AUTHORIZED = 2
|
|
ROOT_TOKEN_NOT_AUTHORIZED = 3
|
|
DATABASE_MAX_RETRIES_EXHAUSTED = 4
|
|
TOO_MANY_REQUESTS = 429
|
|
|
|
SERVER_ERROR0 = 500
|
|
SERVER_ERROR1 = 501
|
|
|
|
# 1000-2000: tasks endpoint
|
|
TASK_INVALID_REQUEST_TYPE = 1000
|
|
TASK_ACK_FAILED = 1001
|
|
TASK_NACK_FAILED = 1002
|
|
TASK_INVALID_RESPONSE_TYPE = 1003
|
|
TASK_INTERACTION_REQUEST_FAILED = 1004
|
|
TASK_GENERATION_FAILED = 1005
|
|
TASK_REQUESTED_TYPE_NOT_AVAILABLE = 1006
|
|
TASK_AVAILABILITY_QUERY_FAILED = 1007
|
|
|
|
# 2000-3000: prompt_repository
|
|
INVALID_FRONTEND_MESSAGE_ID = 2000
|
|
MESSAGE_NOT_FOUND = 2001
|
|
RATING_OUT_OF_RANGE = 2002
|
|
INVALID_RANKING_VALUE = 2003
|
|
INVALID_TASK_TYPE = 2004
|
|
USER_NOT_SPECIFIED = 2005
|
|
NO_MESSAGE_TREE_FOUND = 2006
|
|
NO_REPLIES_FOUND = 2007
|
|
INVALID_MESSAGE = 2008
|
|
BROKEN_CONVERSATION = 2009
|
|
TREE_NOT_IN_GROWING_STATE = 2010
|
|
CORRUPT_RANKING_RESULT = 2011
|
|
AUTH_AND_USERNAME_REQUIRED = 2012
|
|
|
|
TEXT_LABELS_WRONG_MESSAGE_ID = 2050
|
|
TEXT_LABELS_INVALID_LABEL = 2051
|
|
TEXT_LABELS_MANDATORY_LABEL_MISSING = 2052
|
|
|
|
TASK_NOT_FOUND = 2100
|
|
TASK_EXPIRED = 2101
|
|
TASK_PAYLOAD_TYPE_MISMATCH = 2102
|
|
TASK_ALREADY_UPDATED = 2103
|
|
TASK_NOT_ACK = 2104
|
|
TASK_ALREADY_DONE = 2105
|
|
TASK_NOT_COLLECTIVE = 2106
|
|
TASK_NOT_ASSIGNED_TO_USER = 2106
|
|
TASK_UNEXPECTED_PAYLOAD_TYPE_ = 2107
|
|
USER_NOT_FOUND = 2200
|
|
|
|
# 3000-4000: external resources
|
|
HUGGINGFACE_API_ERROR = 3001
|
|
|
|
|
|
class OasstError(Exception):
|
|
"""Base class for Open-Assistant exceptions."""
|
|
|
|
message: str
|
|
error_code: int
|
|
http_status_code: HTTPStatus
|
|
|
|
def __init__(self, message: str, error_code: OasstErrorCode, http_status_code: HTTPStatus = HTTPStatus.BAD_REQUEST):
|
|
super().__init__(message, error_code, http_status_code) # make excetpion picklable (fill args member)
|
|
self.message = message
|
|
self.error_code = error_code
|
|
self.http_status_code = http_status_code
|
|
|
|
def __repr__(self) -> str:
|
|
class_name = self.__class__.__name__
|
|
return f'{class_name}(message="{self.message}", error_code={self.error_code}, http_status_code={self.http_status_code})'
|