Files
litellm/tests/test_litellm/test_exception_exports.py
T

32 lines
1.0 KiB
Python

"""
Test that all standard HTTP error exceptions are exported from litellm.__init__.
"""
import litellm
def test_permission_denied_error_is_exported():
"""PermissionDeniedError (403) should be accessible as litellm.PermissionDeniedError."""
assert hasattr(litellm, "PermissionDeniedError")
assert litellm.PermissionDeniedError is not None
def test_all_http_error_exceptions_exported():
"""All standard HTTP error exceptions should be accessible at module level."""
expected_exceptions = [
"BadRequestError", # 400
"AuthenticationError", # 401
"PermissionDeniedError", # 403
"NotFoundError", # 404
"Timeout", # 408
"UnprocessableEntityError", # 422
"RateLimitError", # 429
"InternalServerError", # 500
"BadGatewayError", # 502
"ServiceUnavailableError", # 503
]
for exc_name in expected_exceptions:
assert hasattr(
litellm, exc_name
), f"litellm.{exc_name} is not exported from litellm.__init__"