bugfix to func coder

This commit is contained in:
Paul Gauthier 2024-08-15 12:35:53 -07:00
parent 3a2ac02024
commit 1a98c2835b

View file

@ -53,6 +53,9 @@ class SingleWholeFileFunctionCoder(Coder):
args = self.parse_partial_args()
if not args:
return ""
for k, v in args.items():
res += "\n"
res += f"{k}:\n"
@ -60,29 +63,6 @@ class SingleWholeFileFunctionCoder(Coder):
return res
if not args:
return
explanation = args.get("explanation")
files = args.get("files", [])
res = ""
if explanation:
res += f"{explanation}\n\n"
for i, file_upd in enumerate(files):
path = file_upd.get("path")
if not path:
continue
content = file_upd.get("content")
if not content:
continue
this_final = (i < len(files) - 1) or final
res += self.live_diffs(path, content, this_final)
return res
def live_diffs(self, fname, content, final):
lines = content.splitlines(keepends=True)
@ -109,6 +89,8 @@ class SingleWholeFileFunctionCoder(Coder):
assert len(chat_files) == 1, chat_files
args = self.parse_partial_args()
if not args:
return []
res = chat_files[0], args["content"]
dump(res)