Retry api calls to gemini on network read error

This commit is contained in:
Debanjum
2025-07-06 09:25:00 -07:00
parent 2ec39d295d
commit afa810e552

View File

@@ -76,7 +76,11 @@ def _is_retryable_error(exception: BaseException) -> bool:
if isinstance(exception, gerrors.APIError):
return exception.code in [429, 502, 503, 504]
# client errors
if isinstance(exception, httpx.TimeoutException) or isinstance(exception, httpx.NetworkError):
if (
isinstance(exception, httpx.TimeoutException)
or isinstance(exception, httpx.NetworkError)
or isinstance(exception, httpx.ReadError)
):
return True
# validation errors
if isinstance(exception, ValueError):