From afa810e5522a1512c9bdd24220aca09756026a80 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Sun, 6 Jul 2025 09:25:00 -0700 Subject: [PATCH] Retry api calls to gemini on network read error --- src/khoj/processor/conversation/google/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/conversation/google/utils.py b/src/khoj/processor/conversation/google/utils.py index 8105c014..f2b0dc84 100644 --- a/src/khoj/processor/conversation/google/utils.py +++ b/src/khoj/processor/conversation/google/utils.py @@ -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):