scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

Restore

Browse source
This commit is contained in:
Dharamendra Kumar 2026-02-26 13:31:37 -08:00 • committed by Sameer Kankute
parent d3946caa37
commit ca417ff2f9
1 changed files with 0 additions and 42 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
tests/local_testing/test_exceptions.py
View file

@ -1422,46 +1422,4 @@ async def test_exception_bubbling_up(sync_mode, stream_mode, model):
assert exc_info.value.type == "invalid_request_error"
def test_midstream_fallback_error_status_code_propagation():
"""
MidStreamFallbackError should preserve the original status code and keep
message/request/response fields consistent after super().__init__().
"""
import litellm
from httpx import Request, Response
# 1) Wrapping a 429 should preserve the 429 status
original_req = Request("POST", "https://api.openai.com/v1/chat/completions")
original_resp = Response(status_code=429, request=original_req)
rate_limit_error = litellm.RateLimitError(
message="Rate limit exceeded",
llm_provider="openai",
model="gpt-4o-mini",
response=original_resp,
)
midstream_error = litellm.exceptions.MidStreamFallbackError(
message="stream broke",
model="gpt-4o-mini",
llm_provider="openai",
original_exception=rate_limit_error,
)
assert midstream_error.status_code == 429
assert midstream_error.response.status_code == 429
assert str(midstream_error.response.request.url) == "https://openai.com/v1/"
assert midstream_error.message == "litellm.MidStreamFallbackError: stream broke"
assert midstream_error.args == ("litellm.MidStreamFallbackError: stream broke",)
# 2) With no original exception, should default to 503
midstream_fallback = litellm.exceptions.MidStreamFallbackError(
message="stream broke without original",
model="gpt-4o-mini",
llm_provider="openai",
original_exception=None,
)
assert midstream_fallback.status_code == 503
assert midstream_fallback.response.status_code == 503
assert str(midstream_fallback.response.request.url) == "https://openai.com/v1/"