diff --git a/litellm/llms/infinity/common_utils.py b/litellm/llms/infinity/common_utils.py index e23fe4a0d37..657bbd8bfd3 100644 --- a/litellm/llms/infinity/common_utils.py +++ b/litellm/llms/infinity/common_utils.py @@ -4,7 +4,7 @@ from litellm.llms.base_llm.chat.transformation import BaseLLMException class InfinityError(BaseLLMException): - def __init__(self, status_code: int, message: str, headers: dict | httpx.Headers = {}): + def __init__(self, status_code: int, message: str, headers: dict | httpx.Headers | None = None): self.status_code = status_code self.message = message self.request = httpx.Request(method="POST", url="https://github.com/michaelfeil/infinity") @@ -15,4 +15,4 @@ class InfinityError(BaseLLMException): request=self.request, response=self.response, headers=headers, - ) # Call the base class constructor with the parameters it needs + ) # Call the base class constructor with the parameters it needs \ No newline at end of file