From 1882cf55db78be1ea52581be886badb7c5224562 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Sat, 3 Jun 2023 06:16:33 -0700
Subject: [PATCH] Handle ConnectionError

---
 aider/coder.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/aider/coder.py b/aider/coder.py
index c8007f33b..69ba19a14 100755
--- a/aider/coder.py
+++ b/aider/coder.py
@@ -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,10 +408,13 @@ class Coder:
                     )
                     break
                 except RateLimitError as err:
-                    retry_after = 1
                     self.io.tool_error(f"RateLimitError: {err}")
-                    self.io.tool_error(f"Retry in {retry_after} seconds.")
-                    time.sleep(retry_after)
+                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)
 
             self.show_send_output(completion, silent)
         except KeyboardInterrupt: