retry TimeoutError in google sheets api

GooeyAI · Feb 13, 2024 · eee1578 · eee1578
1 parent 6e22901
commit eee1578
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/recipes/DocExtract.py b/recipes/DocExtract.py
@@ -487,9 +487,12 @@ def process_source(
 
 
 def google_api_should_retry(e: Exception) -> bool:
-    return isinstance(e, HttpError) and (
-        e.resp.status in (408, 429) or e.resp.status > 500
-    )
+    from googleapiclient.errors import HttpError
+
+    return (
+        isinstance(e, HttpError)
+        and (e.resp.status in (408, 429) or e.resp.status > 500)
+    ) or isinstance(e, TimeoutError)
 
 
 @retry_if(google_api_should_retry)