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

Expand the tail to ensure head ends with assistant msg

Browse source
This commit is contained in:
Paul Gauthier 2023-07-25 06:58:41 -03:00
parent 4b41eb7525
commit f0ec6cb36a
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
aider/history.py
View file

@ -40,7 +40,7 @@ class ChatSummary:
half_max_tokens = self.max_tokens // 2 half_max_tokens = self.max_tokens // 2
# Iterate over the messages in reverse order # Iterate over the messages in reverse order
for i in range(len(sized)-1, -1, -1): for i in range(len(sized) - 1, -1, -1):
tokens, _msg = sized[i] tokens, _msg = sized[i]
if tail_tokens + tokens < half_max_tokens: if tail_tokens + tokens < half_max_tokens:
tail_tokens += tokens tail_tokens += tokens
@ -49,8 +49,8 @@ class ChatSummary:
break break
# Ensure the head ends with an assistant message # Ensure the head ends with an assistant message
while messages[split_index - 1]["role"] != "assistant" and split_index < len(messages): while messages[split_index - 1]["role"] != "assistant" and split_index > 1:
split_index += 1 split_index -= 1
head = messages[:split_index] head = messages[:split_index]
tail = messages[split_index:] tail = messages[split_index:]