fix(infinity): avoid shared default error headers

This commit is contained in:
Ricky-7-Yan 2026-08-31 02:19:35 +08:00
parent 4ba8517134
commit d724636365
2 changed files with 18 additions and 2 deletions

View file

@ -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,

View file

@ -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