From 69131a423e56282fbe555ad8cab71e21e6ea4a2d Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Thu, 22 Aug 2024 12:33:42 -0700
Subject: [PATCH] fix: Import ChatChunks class from separate file

---
 aider/coders/base_coder.py | 50 +-------------------------------------
 1 file changed, 1 insertion(+), 49 deletions(-)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index ca8b02244..42bc679c2 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -61,55 +61,7 @@ all_fences = [
 ]
 
 
-@dataclass
-class ChatChunks:
-    system: List = field(default_factory=list)
-    examples: List = field(default_factory=list)
-    done: List = field(default_factory=list)
-    repo: List = field(default_factory=list)
-    readonly_files: List = field(default_factory=list)
-    chat_files: List = field(default_factory=list)
-    cur: List = field(default_factory=list)
-    reminder: List = field(default_factory=list)
-
-    def all_messages(self):
-        return (
-            self.system
-            + self.examples
-            + self.readonly_files
-            + self.repo
-            + self.done
-            + self.chat_files
-            + self.cur
-            + self.reminder
-        )
-
-    def add_cache_control_headers(self):
-        if self.examples:
-            self.add_cache_control(self.examples)
-        else:
-            self.add_cache_control(self.system)
-
-        if self.readonly_files:
-            self.add_cache_control(self.readonly_files)
-        else:
-            self.add_cache_control(self.repo)
-
-        self.add_cache_control(self.chat_files)
-
-    def add_cache_control(self, messages):
-        if not messages:
-            return
-
-        content = messages[-1]["content"]
-        if type(content) is str:
-            content = dict(
-                type="text",
-                text=content,
-            )
-        content["cache_control"] = {"type": "ephemeral"}
-
-        messages[-1]["content"] = [content]
+from .chat_chunks import ChatChunks
 
 
 class Coder: