phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(http_handler.py): return original response headers

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-30 01:54:49 -08:00
parent 9c35a3b554
commit aee601d1d8
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/llms/custom_httpx/http_handler.py
View file

@ -78,6 +78,7 @@ class MaskedHTTPStatusError(httpx.HTTPStatusError):
response=httpx.Response( response=httpx.Response(
status_code=original_error.response.status_code, status_code=original_error.response.status_code,
content=original_error.response.content, content=original_error.response.content,
headers=original_error.response.headers,
), ),
) )
self.message = message self.message = message

4
tests/local_testing/test_exceptions.py
View file

@ -1146,7 +1146,9 @@ async def test_exception_with_headers_httpx(
except litellm.RateLimitError as e: except litellm.RateLimitError as e:
exception_raised = True exception_raised = True
assert e.litellm_response_headers is not None assert (
e.litellm_response_headers is not None
), "litellm_response_headers is None"
print("e.litellm_response_headers", e.litellm_response_headers) print("e.litellm_response_headers", e.litellm_response_headers)
assert int(e.litellm_response_headers["retry-after"]) == cooldown_time assert int(e.litellm_response_headers["retry-after"]) == cooldown_time