This commit is contained in:
Paul Gauthier 2023-07-22 16:09:57 -03:00
parent 69aea3e745
commit e5ffb78224

View file

@ -33,7 +33,7 @@ class ChatSummary:
num = num // 2
# we want the head to end with an assistant msg
if messages[num]["role"] == "assistant":
while messages[num]["role"] == "assistant" and num < len(messages) - 1:
num += 1
head = messages[:num]