mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-30 16:40:05 +08:00
Andreas Köpf
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
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
|
|
"""
|
|
|
|
# 0-1000: general errors
|
|
GENERIC_ERROR = 0
|
|
DATABASE_URI_NOT_SET = 1
|
|
API_CLIENT_NOT_AUTHORIZED = 2
|
|
|
|
# 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
|
|
|
|
# 2000-3000: prompt_repository
|
|
INVALID_POST_ID = 2000
|
|
POST_NOT_FOUND = 2001
|
|
RATING_OUT_OF_RANGE = 2002
|
|
INVALID_RANKING_VALUE = 2003
|
|
INVALID_TASK_TYPE = 2004
|
|
USER_NOT_SPECIFIED = 2005
|
|
NO_THREADS_FOUND = 2006
|
|
WORK_PACKAGE_NOT_FOUND = 2100
|
|
WORK_PACKAGE_EXPIRED = 2101
|
|
WORK_PACKAGE_PAYLOAD_TYPE_MISMATCH = 2102
|
|
WORK_PACKAGE_ALREADY_UPDATED = 2103
|
|
WORK_PACKAGE_NOT_ACK = 2104
|
|
WORK_PACKAGE_ALREADY_DONE = 2105
|
|
|
|
|
|
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})'
|