Spaces:
Running
Running
File size: 3,103 Bytes
0157ac7 | 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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | """Unified exception hierarchy for providers."""
from typing import Any
class ProviderError(Exception):
"""Base exception for all provider errors."""
def __init__(
self,
message: str,
status_code: int = 500,
error_type: str = "api_error",
raw_error: Any = None,
):
super().__init__(message)
self.message = message
self.status_code = status_code
self.error_type = error_type
self.raw_error = raw_error
def to_anthropic_format(self) -> dict:
"""Convert to Anthropic-compatible error response."""
return {
"type": "error",
"error": {
"type": self.error_type,
"message": self.message,
},
}
class AuthenticationError(ProviderError):
"""Raised when API key is invalid or missing."""
def __init__(self, message: str, raw_error: Any = None):
super().__init__(
message,
status_code=401,
error_type="authentication_error",
raw_error=raw_error,
)
class InvalidRequestError(ProviderError):
"""Raised when the request parameters are invalid."""
def __init__(self, message: str, raw_error: Any = None):
super().__init__(
message,
status_code=400,
error_type="invalid_request_error",
raw_error=raw_error,
)
class RateLimitError(ProviderError):
"""Raised when rate limit is exceeded."""
def __init__(self, message: str, raw_error: Any = None):
super().__init__(
message,
status_code=429,
error_type="rate_limit_error",
raw_error=raw_error,
)
class OverloadedError(ProviderError):
"""Raised when the provider is overloaded."""
def __init__(self, message: str, raw_error: Any = None):
super().__init__(
message,
status_code=529,
error_type="overloaded_error",
raw_error=raw_error,
)
class APIError(ProviderError):
"""Raised when the provider returns a generic API error."""
def __init__(self, message: str, status_code: int = 500, raw_error: Any = None):
super().__init__(
message,
status_code=status_code,
error_type="api_error",
raw_error=raw_error,
)
class UnknownProviderTypeError(InvalidRequestError):
"""Raised when ``provider_id`` is not registered in the provider map."""
def __init__(self, message: str) -> None:
super().__init__(message)
class ServiceUnavailableError(ProviderError):
"""Raised when the server is not ready (e.g. app lifespan did not wire state)."""
def __init__(self, message: str, raw_error: Any = None):
super().__init__(
message,
status_code=503,
error_type="api_error",
raw_error=raw_error,
)
class ModelListResponseError(ServiceUnavailableError):
"""Raised when a provider model-list response cannot be parsed safely."""
|