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

Try and avoid seaming errors when asking Sonnet to continue

Browse source
This commit is contained in:
Paul Gauthier 2024-06-29 17:01:31 -07:00
parent fe4d75b123
commit 85ef60f083
1 changed files with 18 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

30
aider/coders/base_coder.py
View file

@ -807,8 +807,6 @@ class Coder:
messages = self.format_messages()
self.io.log_llm_history("TO LLM", format_messages(messages))
if self.verbose:
utils.show_messages(messages, functions=self.functions)
@ -842,8 +840,8 @@ class Coder:
exhausted = True
break
# Use prefill to continue the response
self.multi_response_content += self.partial_response_content
self.multi_response_content = self.get_multi_response_content()
if messages[-1]["role"] == "assistant":
messages[-1]["content"] = self.multi_response_content
else:
@ -857,10 +855,8 @@ class Coder:
self.live_incremental_response(True)
self.mdstream = None
if self.multi_response_content:
self.multi_response_content += self.partial_response_content
self.partial_response_content = self.multi_response_content
self.multi_response_content = ""
self.partial_response_content = self.get_multi_response_content(True)
self.multi_response_content = ""
if exhausted:
self.show_exhausted_error()
@ -880,8 +876,6 @@ class Coder:
self.io.tool_output()
self.io.log_llm_history("LLM RESPONSE", format_content("ASSISTANT", content))
if interrupted:
content += "\n^C KeyboardInterrupt"
self.cur_messages += [dict(role="assistant", content=content)]
@ -1074,6 +1068,8 @@ class Coder:
self.partial_response_content = ""
self.partial_response_function_call = dict()
self.io.log_llm_history("TO LLM", format_messages(messages))
interrupted = False
try:
hash_object, completion = send_with_retries(
@ -1089,6 +1085,11 @@ class Coder:
self.keyboard_interrupt()
interrupted = True
finally:
self.io.log_llm_history(
"LLM RESPONSE",
format_content("ASSISTANT", self.partial_response_content),
)
if self.partial_response_content:
self.io.ai_output(self.partial_response_content)
elif self.partial_response_function_call:
@ -1207,8 +1208,13 @@ class Coder:
def render_incremental_response(self, final):
return self.get_multi_response_content()
def get_multi_response_content(self):
return self.multi_response_content + self.partial_response_content
def get_multi_response_content(self, final=False):
cur = self.multi_response_content
new = self.partial_response_content
if new.rstrip() != new and not final:
new = new.rstrip()
return cur + new
def get_rel_fname(self, fname):
return os.path.relpath(fname, self.root)