feat: Use all_messages to get initial messages and rename chunks to messages

This commit is contained in:
Paul Gauthier (aider) 2024-08-26 15:14:40 -07:00
parent 9b70819de1
commit b53b1c95d5

View file

@ -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