From b53b1c95d55c6167c43f408059ec4c61d58f0463 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 26 Aug 2024 15:14:40 -0700 Subject: [PATCH] feat: Use `all_messages` to get initial messages and rename `chunks` to `messages` --- aider/coders/chat_chunks.py | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/aider/coders/chat_chunks.py b/aider/coders/chat_chunks.py index 4e294bc27..881e781f2 100644 --- a/aider/coders/chat_chunks.py +++ b/aider/coders/chat_chunks.py @@ -53,17 +53,10 @@ class ChatChunks: messages[-1]["content"] = [content] def cacheable_messages(self): - chunks = ( - self.system - + self.examples - + self.readonly_files - + self.repo - + self.done - + self.chat_files - ) - for i, message in enumerate(reversed(chunks)): + messages = self.all_messages() + for i, message in enumerate(reversed(messages)): if isinstance(message.get("content"), list) and message["content"][0].get( "cache_control" ): - return chunks[: len(chunks) - i] - return chunks + return messages[: len(messages) - i] + return messages