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

Put back 2nd reminder

Browse source
This commit is contained in:
Paul Gauthier 2023-11-14 10:46:16 -08:00
parent 26a340e21f
commit e0fea58276
1 changed files with 10 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

23
aider/coders/base_coder.py
View file

@ -429,22 +429,19 @@ class Coder:
messages += self.done_messages
messages += self.get_files_messages()
# reminder_message = [
# dict(role="system", content=self.fmt_system_reminder()),
# ]
reminder_message = [
dict(role="system", content=self.fmt_system_reminder()),
]
# messages_tokens = self.main_model.token_count(messages)
# reminder_tokens = self.main_model.token_count(reminder_message)
# cur_tokens = self.main_model.token_count(self.cur_messages)
messages_tokens = self.main_model.token_count(messages)
reminder_tokens = self.main_model.token_count(reminder_message)
cur_tokens = self.main_model.token_count(self.cur_messages)
# total_tokens = messages_tokens + reminder_tokens + cur_tokens
total_tokens = messages_tokens + reminder_tokens + cur_tokens
# Add the reminder prompt if there's a lot of context since the original
# system prompt, and we still have room to include it.
# THRESH = 2048
# if messages_tokens > THRESH and total_tokens < self.main_model.max_context_tokens:
# messages += reminder_message
# Add the reminder prompt if we still have room to include it.
if total_tokens < self.main_model.max_context_tokens:
messages += reminder_message
messages += self.cur_messages