dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 10:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Handle ConnectionError

Browse source
This commit is contained in:
Paul Gauthier 2023-06-03 06:16:33 -07:00
parent 4f1e1181c4
commit 1882cf55db
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aider/coder.py
View file

@ -8,6 +8,7 @@ from pathlib import Path
import git import git
import openai import openai
import requests
from openai.error import RateLimitError from openai.error import RateLimitError
from rich.console import Console from rich.console import Console
from rich.live import Live from rich.live import Live
@ -407,10 +408,13 @@ class Coder:
) )
break break
except RateLimitError as err: except RateLimitError as err:
retry_after = 1
self.io.tool_error(f"RateLimitError: {err}") self.io.tool_error(f"RateLimitError: {err}")
self.io.tool_error(f"Retry in {retry_after} seconds.") except requests.exceptions.ConnectionError as err:
time.sleep(retry_after) self.io.tool_error(f"ConnectionError: {err}")
retry_after = 1
self.io.tool_error(f"Retry in {retry_after} seconds.")
time.sleep(retry_after)
self.show_send_output(completion, silent) self.show_send_output(completion, silent)
except KeyboardInterrupt: except KeyboardInterrupt: