Handle overload errors in Anthropic.

plasma-umass · Mar 28, 2024 · 04a2e3f · 04a2e3f
1 parent 23e2b8b
commit 04a2e3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/coverup/coverup.py b/src/coverup/coverup.py
@@ -432,7 +432,7 @@ async def do_chat(seg: CodeSegment, completion: dict) -> str:
 
             return await litellm.acreate(**completion)
 
-        except (openai.RateLimitError, openai.APITimeoutError) as e:
+        except (openai.RateLimitError, openai.APITimeoutError, litellm.APIError) as e:
 
             # This message usually indicates out of money in account
             if 'You exceeded your current quota' in str(e):