klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Retry api calls to gemini on network read error

Browse Source
This commit is contained in:
Debanjum
2025-07-06 09:25:00 -07:00
parent 2ec39d295d
commit afa810e552
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/khoj/processor/conversation/google/utils.py
View File

@@ -76,7 +76,11 @@ def _is_retryable_error(exception: BaseException) -> bool:
if isinstance(exception, gerrors.APIError): if isinstance(exception, gerrors.APIError):
return exception.code in [429, 502, 503, 504] return exception.code in [429, 502, 503, 504]
# client errors # 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 return True
# validation errors # validation errors
if isinstance(exception, ValueError): if isinstance(exception, ValueError):