added ServiceUnavailableError for retries

This commit is contained in:
Paul Gauthier 2023-06-24 19:54:18 -07:00
parent 1acc5e04d0
commit 8dcf819cb5

View file

@ -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."),
)