mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 22:34:59 +00:00
retry on Timeout
This commit is contained in:
parent
039adfd346
commit
bc7610cdda
2 changed files with 11 additions and 5 deletions
|
@ -12,7 +12,7 @@ import backoff
|
|||
import git
|
||||
import openai
|
||||
import requests
|
||||
from openai.error import APIError, RateLimitError, ServiceUnavailableError
|
||||
from openai.error import APIError, RateLimitError, ServiceUnavailableError, Timeout
|
||||
from rich.console import Console, Text
|
||||
from rich.live import Live
|
||||
from rich.markdown import Markdown
|
||||
|
@ -565,7 +565,13 @@ class Coder:
|
|||
|
||||
@backoff.on_exception(
|
||||
backoff.expo,
|
||||
(APIError, ServiceUnavailableError, RateLimitError, requests.exceptions.ConnectionError),
|
||||
(
|
||||
Timeout,
|
||||
APIError,
|
||||
ServiceUnavailableError,
|
||||
RateLimitError,
|
||||
requests.exceptions.ConnectionError,
|
||||
),
|
||||
max_tries=5,
|
||||
on_backoff=lambda details: print(f"Retry in {details['wait']} seconds."),
|
||||
)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue