Added APIError

This commit is contained in:
Paul Gauthier 2023-06-24 20:04:49 -07:00
parent 8dcf819cb5
commit fe3f1c194d

View file

@ -11,7 +11,7 @@ import backoff
import git import git
import openai import openai
import requests import requests
from openai.error import RateLimitError, ServiceUnavailableError from openai.error import APIError, RateLimitError, ServiceUnavailableError
from rich.console import Console from rich.console import Console
from rich.live import Live from rich.live import Live
from rich.markdown import Markdown from rich.markdown import Markdown
@ -494,7 +494,7 @@ class Coder:
@backoff.on_exception( @backoff.on_exception(
backoff.expo, backoff.expo,
(ServiceUnavailableError, RateLimitError, requests.exceptions.ConnectionError), (APIError, ServiceUnavailableError, RateLimitError, requests.exceptions.ConnectionError),
max_tries=5, max_tries=5,
on_backoff=lambda details: print(f"Retry in {details['wait']} seconds."), on_backoff=lambda details: print(f"Retry in {details['wait']} seconds."),
) )