From 8dcf819cb57c648b0fd9609f68f9a8c07c4f885e Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sat, 24 Jun 2023 19:54:18 -0700 Subject: [PATCH] added ServiceUnavailableError for retries --- aider/coders/base_coder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 77463f102..46293560e 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -11,7 +11,7 @@ import backoff import git import openai import requests -from openai.error import RateLimitError +from openai.error import RateLimitError, ServiceUnavailableError from rich.console import Console from rich.live import Live from rich.markdown import Markdown @@ -494,7 +494,7 @@ class Coder: @backoff.on_exception( backoff.expo, - (RateLimitError, requests.exceptions.ConnectionError), + (ServiceUnavailableError, RateLimitError, requests.exceptions.ConnectionError), max_tries=5, on_backoff=lambda details: print(f"Retry in {details['wait']} seconds."), )