dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 22:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

added ServiceUnavailableError for retries

Browse source
This commit is contained in:
Paul Gauthier 2023-06-24 19:54:18 -07:00
parent 1acc5e04d0
commit 8dcf819cb5
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aider/coders/base_coder.py
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 from openai.error import 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,
(RateLimitError, requests.exceptions.ConnectionError), (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."),
) )