From 9ac8a60f5ff408ff4f30b4d2b2c44be4a211b91b Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Wed, 7 Jun 2023 12:42:43 -0700
Subject: [PATCH] Finish support for gpt-4-32k

---
 aider/coder.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/aider/coder.py b/aider/coder.py
index 5fd58e190..9fc78c3be 100755
--- a/aider/coder.py
+++ b/aider/coder.py
@@ -106,7 +106,7 @@ class Coder:
             self.io.tool_output("Not using git.")
             self.find_common_root()
 
-        if main_model == models.GPT4:
+        if main_model != models.GPT35:
             rm_io = io if self.verbose else None
             self.repo_map = RepoMap(
                 map_tokens,
@@ -313,7 +313,7 @@ class Coder:
         ]
 
         main_sys = self.gpt_prompts.main_system
-        if self.main_model == models.GPT4:
+        if self.main_model != models.GPT35:
             main_sys += "\n" + self.gpt_prompts.system_reminder
 
         messages = [
@@ -342,7 +342,7 @@ class Coder:
         if edit_error:
             return edit_error
 
-        if self.main_model == models.GPT4 or (self.main_model == models.GPT35 and not edited):
+        if self.main_model != models.GPT35 or not edited:
             # Don't add assistant messages to the history if they contain "edits"
             # Because those edits are actually fully copies of the file!
             # That wastes too much context window.
@@ -778,7 +778,7 @@ class Coder:
         return set(self.get_all_relative_files()) - set(self.get_inchat_relative_files())
 
     def apply_updates(self, content):
-        if self.main_model == models.GPT4:
+        if self.main_model in (models.GPT4, models.GPT4_32k):
             method = self.update_files_gpt4
         elif self.main_model == models.GPT35:
             method = self.update_files_gpt35