fix: Update update_cur_messages method in WholeFileCoder class

This commit is contained in:
Paul Gauthier 2024-08-22 12:45:50 -07:00 committed by Paul Gauthier (aider)
parent e0bab8505a
commit 100e0b9856

View file

@ -9,17 +9,10 @@ from .wholefile_prompts import WholeFilePrompts
class WholeFileCoder(Coder): class WholeFileCoder(Coder):
"""A coder that operates on entire files for code modifications.""" """A coder that operates on entire files for code modifications."""
edit_format = "whole" edit_format = "whole"
gpt_prompts = WholeFilePrompts() gpt_prompts = WholeFilePrompts()
def update_cur_messages(self, edited):
if edited:
self.cur_messages += [
dict(role="assistant", content=self.gpt_prompts.redacted_edit_message)
]
else:
self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
def render_incremental_response(self, final): def render_incremental_response(self, final):
try: try:
return self.get_edits(mode="diff") return self.get_edits(mode="diff")