Add summary prefix

This commit is contained in:
Paul Gauthier 2023-07-22 12:11:01 -03:00
parent c26917851f
commit 69aea3e745
3 changed files with 7 additions and 2 deletions

View file

@ -15,6 +15,7 @@ class ChatSummary:
def summarize(self, messages):
num = len(messages)
dump(num)
if num < 2:
return messages
@ -66,6 +67,8 @@ class ChatSummary:
]
summary = simple_send_with_retries(model=models.GPT35.name, messages=messages)
summary = prompts.summary_prefix + summary
dump(summary)
return [dict(role="user", content=summary)]