fix retry logic to include 502, 503, 504 only among 5xx, remove 409

GooeyAI · Sep 11, 2024 · 9db5911 · 9db5911
1 parent 47964c0
commit 9db5911
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gooey/core/http_client.py b/src/gooey/core/http_client.py
@@ -86,8 +86,8 @@ def _retry_timeout(response: httpx.Response, retries: int) -> float:
 
 
 def _should_retry(response: httpx.Response) -> bool:
-    retriable_400s = [429, 408, 409]
-    return response.status_code >= 500 or response.status_code in retriable_400s
+    retry_on_codes = [429, 408, 502, 503, 504]
+    return response.status_code in retry_on_codes
 
 
 def remove_omit_from_dict(