Another day, another error to handle.

plasma-umass · Mar 29, 2024 · 95b54f0 · 95b54f0
1 parent c27694e
commit 95b54f0
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, litellm.APIError) as e:
+        except (openai.RateLimitError, openai.APITimeoutError, litellm.APIError, litellm.exceptions.ServiceUnavailableError) as e:
 
             # This message usually indicates out of money in account
             if 'You exceeded your current quota' in str(e):