From fe3f1c194d8e24c81249aa76e03656754415931c Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sat, 24 Jun 2023 20:04:49 -0700 Subject: [PATCH] Added APIError --- 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 46293560e..1852d8d80 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, ServiceUnavailableError +from openai.error import APIError, 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, - (ServiceUnavailableError, RateLimitError, requests.exceptions.ConnectionError), + (APIError, ServiceUnavailableError, RateLimitError, requests.exceptions.ConnectionError), max_tries=5, on_backoff=lambda details: print(f"Retry in {details['wait']} seconds."), )