diff --git a/litellm/exceptions.py b/litellm/exceptions.py index 197e64c75..0abe3e33a 100644 --- a/litellm/exceptions.py +++ b/litellm/exceptions.py @@ -543,7 +543,7 @@ class InternalServerError(openai.InternalServerError): # type: ignore class APIError(openai.APIError): # type: ignore def __init__( self, - status_code, + status_code: int, message, llm_provider, model, @@ -754,7 +754,7 @@ class MockException(openai.APIError): # used for testing def __init__( self, - status_code, + status_code: int, message, llm_provider, model, diff --git a/litellm/llms/anthropic.py b/litellm/llms/anthropic.py index 2221a3d73..e3bc595dd 100644 --- a/litellm/llms/anthropic.py +++ b/litellm/llms/anthropic.py @@ -72,7 +72,7 @@ class AnthropicConstants(Enum): class AnthropicError(Exception): - def __init__(self, status_code, message): + def __init__(self, status_code: int, message): self.status_code = status_code self.message: str = message self.request = httpx.Request(