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

cleanup

Browse source
This commit is contained in:
Paul Gauthier 2023-07-20 09:39:27 -03:00
parent ea15aa72fa
commit 993bfa6e4c
2 changed files with 10 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/coders/base_coder.py
View file

@ -553,7 +553,16 @@ class Coder:
return add_rel_files_message return add_rel_files_message
def update_cur_messages(self, content, edited): def update_cur_messages(self, content, edited):
self.cur_messages += [dict(role="assistant", content=content)] if self.partial_response_content:
self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
if self.partial_response_function_call:
self.cur_messages += [
dict(
role="assistant",
content=None,
function_call=self.partial_response_function_call,
)
]
def auto_commit(self): def auto_commit(self):
res = self.commit(history=self.cur_messages, prefix="aider: ") res = self.commit(history=self.cur_messages, prefix="aider: ")

12
aider/coders/editblock_func_coder.py
View file

@ -83,18 +83,6 @@ class EditBlockFunctionCoder(Coder):
self.gpt_prompts = EditBlockFunctionPrompts() self.gpt_prompts = EditBlockFunctionPrompts()
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
def update_cur_messages(self, content, edited):
if self.partial_response_content:
self.cur_messages += [dict(role="assistant", content=self.partial_response_content)]
if self.partial_response_function_call:
self.cur_messages += [
dict(
role="assistant",
content=None,
function_call=self.partial_response_function_call,
)
]
def render_incremental_response(self, final=False): def render_incremental_response(self, final=False):
if self.partial_response_content: if self.partial_response_content:
return self.partial_response_content return self.partial_response_content