From 993bfa6e4cc1b160dfdb5d8a9328a1ec82da6c13 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 20 Jul 2023 09:39:27 -0300 Subject: [PATCH] cleanup --- aider/coders/base_coder.py | 11 ++++++++++- aider/coders/editblock_func_coder.py | 12 ------------ 2 files changed, 10 insertions(+), 13 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 4e9adbf3b..f401e38b4 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -553,7 +553,16 @@ class Coder: return add_rel_files_message 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): res = self.commit(history=self.cur_messages, prefix="aider: ") diff --git a/aider/coders/editblock_func_coder.py b/aider/coders/editblock_func_coder.py index 2c834255e..3f7acbaf5 100644 --- a/aider/coders/editblock_func_coder.py +++ b/aider/coders/editblock_func_coder.py @@ -83,18 +83,6 @@ class EditBlockFunctionCoder(Coder): self.gpt_prompts = EditBlockFunctionPrompts() 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): if self.partial_response_content: return self.partial_response_content