File size: 2,404 Bytes
6172a47 | 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 | """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,
)
|