Catch and appropriately retry *all* litellm exceptions #598

Aider-AI · May 9, 2024 · 728a629 · 728a629
1 parent b158e1c
commit 728a629
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/aider/sendchat.py b/aider/sendchat.py
@@ -5,12 +5,12 @@
 import httpx
 import openai
 
-# from diskcache import Cache
-from openai import APIConnectionError, InternalServerError, RateLimitError
-
 from aider.dump import dump  # noqa: F401
 from aider.litellm import litellm
 
+# from diskcache import Cache
+
+
 CACHE_PATH = "~/.aider.send.cache.v1"
 CACHE = None
 # CACHE = Cache(CACHE_PATH)
@@ -26,12 +26,13 @@ def should_giveup(e):
 @backoff.on_exception(
     backoff.expo,
     (
-        InternalServerError,
-        RateLimitError,
-        APIConnectionError,
         httpx.ConnectError,
         httpx.RemoteProtocolError,
+        litellm.exceptions.APIConnectionError,
+        litellm.exceptions.APIError,
+        litellm.exceptions.RateLimitError,
         litellm.exceptions.ServiceUnavailableError,
+        litellm.exceptions.Timeout,
     ),
     giveup=should_giveup,
     max_time=60,