dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 02:05:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: add cacheable_messages() method to ChatChunks class

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-26 15:13:37 -07:00
parent 8ffbf96151
commit 9231154871
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
aider/coders/chat_chunks.py
View file

@ -51,3 +51,10 @@ class ChatChunks:
content["cache_control"] = {"type": "ephemeral"} content["cache_control"] = {"type": "ephemeral"}
messages[-1]["content"] = [content] 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)):
if isinstance(message.get('content'), list) and message['content'][0].get('cache_control'):
return chunks[:len(chunks)-i]
return chunks