implemented accepts_multi_system_msgs

This commit is contained in:
Paul Gauthier 2024-04-30 07:57:29 -07:00
parent 2d16ee16ac
commit d38a38f0dd
2 changed files with 16 additions and 3 deletions

View file

@ -547,11 +547,21 @@ class Coder:
total_tokens = 0
messages += self.cur_messages
messages[-1]["content"] += "\n\n" + self.fmt_system_prompt(self.gpt_prompts.system_reminder)
final = messages[-1]
# Add the reminder prompt if we still have room to include it.
# if total_tokens < self.main_model.info.get("max_input_tokens", 0):
# messages += reminder_message
if total_tokens < self.main_model.info.get("max_input_tokens", 0):
if self.main_model.accepts_multi_system_msgs:
messages += reminder_message
elif final["role"] == "user":
# stuff it into the user message
new_content = (
final["content"]
+ "\n\n"
+ self.fmt_system_prompt(self.gpt_prompts.system_reminder)
)
messages[-1] = dict(role=final["role"], content=new_content)
return messages