Handle ConnectionError

This commit is contained in:
Paul Gauthier 2023-06-03 06:16:33 -07:00
parent 4f1e1181c4
commit 1882cf55db

View file

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