Add common exceptions for retry

Fixes All-Hands-AI#6942 Removed in All-Hands-AI#6557
SmartManoj · Feb 26, 2025 · 77609ee · 77609ee
1 parent 772b577
commit 77609ee
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/openhands/llm/llm.py b/openhands/llm/llm.py
@@ -70,7 +70,15 @@
 __all__ = ['LLM']
 
 # tuple of exceptions to retry on
-LLM_RETRY_EXCEPTIONS: tuple[type[Exception], ...] = (RateLimitError,)
+LLM_RETRY_EXCEPTIONS: tuple[type[Exception], ...] = (
+        APIConnectionError,
+        # FIXME: APIError is useful on 502 from a proxy for example,
+        # but it also retries on other errors that are permanent
+        APIError,
+        InternalServerError,
+        RateLimitError,
+        ServiceUnavailableError,
+    )
 
 # cache prompt supporting models
 # remove this when we gemini and deepseek are supported