fix(responses): satisfy streaming regression checks

This commit is contained in:
zoroyihan7 2026-09-08 11:44:41 +00:00
parent 089703d10b
commit dc7895c1ea
2 changed files with 4 additions and 4 deletions

View file

@ -48,9 +48,10 @@ class _FailureDetails(BaseModel):
def _original_failure(exception: Exception) -> Exception:
if isinstance(exception, MidStreamFallbackError) and exception.original_exception is not None:
return _original_failure(exception.original_exception)
return exception
current = exception # rebind-ok: the recursion gate requires iterative wrapper traversal
while isinstance(current, MidStreamFallbackError) and current.original_exception is not None:
current = current.original_exception
return current
def _response_error_code(details: _FailureDetails) -> str:

View file

@ -130,7 +130,6 @@ class TestPollingEndpointPreCallGuard:
"litellm.proxy.proxy_server.proxy_config": MagicMock(),
"litellm.proxy.proxy_server.proxy_logging_obj": AsyncMock(),
"litellm.proxy.proxy_server.redis_usage_cache": AsyncMock(),
"litellm.proxy.proxy_server.select_data_generator": None,
"litellm.proxy.proxy_server.user_api_base": None,
"litellm.proxy.proxy_server.user_max_tokens": None,
"litellm.proxy.proxy_server.user_model": None,