refac

aider/coder.py

@@ -391,22 +391,15 @@ class Coder:
 
         return prompts.added_files.format(fnames=", ".join(mentioned_rel_fnames))
 
-    def send(self, messages, model=None, silent=False):
-        if not model:
-            model = self.main_model
-
-        self.resp = ""
-        interrupted = False
-        try:
+    def send_with_retries(self, model, messages):
         while True:
             try:
-                    completion = openai.ChatCompletion.create(
+                return openai.ChatCompletion.create(
                     model=model,
                     messages=messages,
                     temperature=0,
                     stream=True,
                 )
-                    break
             except RateLimitError as err:
                 self.io.tool_error(f"RateLimitError: {err}")
             except requests.exceptions.ConnectionError as err:
@@ -416,6 +409,14 @@ class Coder:
             self.io.tool_error(f"Retry in {retry_after} seconds.")
             time.sleep(retry_after)
 
+    def send(self, messages, model=None, silent=False):
+        if not model:
+            model = self.main_model
+
+        self.resp = ""
+        interrupted = False
+        try:
+            completion = self.send_with_retries(model, messages)
             self.show_send_output(completion, silent)
         except KeyboardInterrupt:
             interrupted = True