aboutsummaryrefslogtreecommitdiff
path: root/R2R/r2r/base/abstractions/exception.py
blob: c76625a35e7f6e244dee97e9a9df3c567183dc53 (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from typing import Any, Optional


class R2RException(Exception):
    def __init__(
        self, message: str, status_code: int, detail: Optional[Any] = None
    ):
        self.message = message
        self.status_code = status_code
        super().__init__(self.message)


class R2RDocumentProcessingError(R2RException):
    def __init__(self, error_message, document_id):
        self.document_id = document_id
        super().__init__(error_message, 400, {"document_id": document_id})