diff --git a/litellm/llms/infinity/common_utils.py b/litellm/llms/infinity/common_utils.py index e23fe4a0d37..211d430847d 100644 --- a/litellm/llms/infinity/common_utils.py +++ b/litellm/llms/infinity/common_utils.py @@ -4,10 +4,17 @@ 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") + self.request = httpx.Request( + method="POST", url="https://github.com/michaelfeil/infinity" + ) self.response = httpx.Response(status_code=status_code, request=self.request) super().__init__( status_code=status_code, diff --git a/tests/test_litellm/llms/infinity/test_common_utils.py b/tests/test_litellm/llms/infinity/test_common_utils.py new file mode 100644 index 00000000000..895c8d3182c --- /dev/null +++ b/tests/test_litellm/llms/infinity/test_common_utils.py @@ -0,0 +1,9 @@ +from litellm.llms.infinity.common_utils import InfinityError + + +def test_infinity_error_default_headers_are_not_shared() -> None: + first_error = InfinityError(status_code=500, message="first") + second_error = InfinityError(status_code=500, message="second") + + assert first_error.headers is None + assert second_error.headers is None