💎 [Feature] New HfApiException: INVALID_API_KEY_ERROR

networks/exceptions.py

@@ -0,0 +1,31 @@
+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",
+)