File size: 750 Bytes
a1d960f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import http

from typing import Optional

from fastapi import HTTPException, status


class HfApiException(Exception):
    def __init__(
        self,
        status_code: int,
        detail: Optional[str] = None,
    ) -> None:
        if detail is None:
            self.detail = http.HTTPStatus(status_code).phrase
        else:
            self.detail = detail
        self.status_code = status_code

    def __repr__(self) -> str:
        class_name = self.__class__.__name__
        return f"{class_name}(status_code={self.status_code!r}, detail={self.detail!r})"

    def __str__(self) -> str:
        return self.__repr__()


INVALID_API_KEY_ERROR = HfApiException(
    status_code=status.HTTP_403_FORBIDDEN,
    detail="Invalid API Key",
)