dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 06:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

stop passing content to update_cur_messages

Browse source
This commit is contained in:
Paul Gauthier 2023-07-20 09:41:38 -03:00
parent 993bfa6e4c
commit 6cb8f4e762
4 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

7
aider/coders/base_coder.py
View file

@ -532,11 +532,10 @@ class Coder:
edited, edit_error = self.apply_updates() edited, edit_error = self.apply_updates()
if edit_error: if edit_error:
self.update_cur_messages(content, set()) self.update_cur_messages(set())
return edit_error return edit_error
# TODO: this shouldn't use content, should use self.partial_.... self.update_cur_messages(edited)
self.update_cur_messages(content, edited)
if edited: if edited:
if self.repo and self.auto_commits and not self.dry_run: if self.repo and self.auto_commits and not self.dry_run:
@ -552,7 +551,7 @@ class Coder:
if add_rel_files_message: if add_rel_files_message:
return add_rel_files_message return add_rel_files_message
def update_cur_messages(self, content, edited): def update_cur_messages(self, edited):
if self.partial_response_content: if self.partial_response_content:
self.cur_messages += [dict(role="assistant", content=self.partial_response_content)] self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
if self.partial_response_function_call: if self.partial_response_function_call:

4
aider/coders/single_wholefile_func_coder.py
View file

@ -34,13 +34,13 @@ class SingleWholeFileFunctionCoder(Coder):
self.gpt_prompts = SingleWholeFileFunctionPrompts() self.gpt_prompts = SingleWholeFileFunctionPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def update_cur_messages(self, content, edited): def update_cur_messages(self, edited):
if edited: if edited:
self.cur_messages += [ self.cur_messages += [
dict(role="assistant", content=self.gpt_prompts.redacted_edit_message) dict(role="assistant", content=self.gpt_prompts.redacted_edit_message)
] ]
else: else:
self.cur_messages += [dict(role="assistant", content=content)] self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
def get_context_from_history(self, history): def get_context_from_history(self, history):
context = "" context = ""

4
aider/coders/wholefile_coder.py
View file

@ -12,13 +12,13 @@ class WholeFileCoder(Coder):
self.gpt_prompts = WholeFilePrompts() self.gpt_prompts = WholeFilePrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def update_cur_messages(self, content, edited): def update_cur_messages(self, edited):
if edited: if edited:
self.cur_messages += [ self.cur_messages += [
dict(role="assistant", content=self.gpt_prompts.redacted_edit_message) dict(role="assistant", content=self.gpt_prompts.redacted_edit_message)
] ]
else: else:
self.cur_messages += [dict(role="assistant", content=content)] self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
def get_context_from_history(self, history): def get_context_from_history(self, history):
context = "" context = ""

4
aider/coders/wholefile_func_coder.py
View file

@ -47,13 +47,13 @@ class WholeFileFunctionCoder(Coder):
self.gpt_prompts = WholeFileFunctionPrompts() self.gpt_prompts = WholeFileFunctionPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def update_cur_messages(self, content, edited): def update_cur_messages(self, edited):
if edited: if edited:
self.cur_messages += [ self.cur_messages += [
dict(role="assistant", content=self.gpt_prompts.redacted_edit_message) dict(role="assistant", content=self.gpt_prompts.redacted_edit_message)
] ]
else: else:
self.cur_messages += [dict(role="assistant", content=content)] self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
def get_context_from_history(self, history): def get_context_from_history(self, history):
context = "" context = ""