scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix: guard int() cast in MidStreamFallbackError to handle non-numeric status_code

Browse source
This commit is contained in:
PRABHU KIRAN VANDRANKI 2026-05-19 19:26:25 -04:00
parent a72414a061
commit ed610b85b5
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/exceptions.py
View file

@ -955,7 +955,10 @@ class MidStreamFallbackError(ServiceUnavailableError): # type: ignore
is_pre_first_chunk: bool = False,
):
original_status = getattr(original_exception, "status_code", None)
self.status_code = int(original_status) if original_status is not None else 503
try:
self.status_code = int(original_status) if original_status is not None else 503
except (ValueError, TypeError):
self.status_code = 503
self.message = f"litellm.MidStreamFallbackError: {message}"
self.model = model
self.llm_provider = llm_provider
@ -997,7 +1000,10 @@ class MidStreamFallbackError(ServiceUnavailableError): # type: ignore
)
# Restore the propagated status and original response/request objects
self.status_code = int(original_status) if original_status is not None else 503
try:
self.status_code = int(original_status) if original_status is not None else 503
except (ValueError, TypeError):
self.status_code = 503
self.response = _saved_response
self.request = _saved_request
self.message = _saved_message