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 total_tokens = 0
messages += self.cur_messages 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. # Add the reminder prompt if we still have room to include it.
# if total_tokens < self.main_model.info.get("max_input_tokens", 0): if total_tokens < self.main_model.info.get("max_input_tokens", 0):
# messages += reminder_message 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 return messages

View file

@ -235,6 +235,9 @@ class Model:
self.send_undo_reply = True self.send_undo_reply = True
return # <-- return # <--
if "gpt-3.5" in model or "gpt-4" in model:
self.accepts_multi_system_msgs = True
# use the defaults # use the defaults
if self.edit_format == "diff": if self.edit_format == "diff":
self.use_repo_map = True self.use_repo_map = True