refac and func update_files

This commit is contained in:
Paul Gauthier 2023-06-21 21:17:12 -07:00
parent 8c1e1c7267
commit eb062cc504
4 changed files with 35 additions and 10 deletions

View file

@ -21,10 +21,11 @@ class WholeFileCoder(Coder):
self.cur_messages += [dict(role="assistant", content=content)]
def modify_incremental_response(self, final):
resp = self.partial_response_content
return self.update_files(resp, mode="diff")
return self.update_files(mode="diff")
def update_files(self, mode="update"):
content = self.partial_response_content
def update_files(self, content, mode="update"):
edited = set()
chat_files = self.get_inchat_relative_files()
if not chat_files: