accept diff-func(-list,-string,)

This commit is contained in:
Paul Gauthier 2023-07-01 06:17:55 -07:00
parent 92e5f43abe
commit 059bfc75ab

View file

@ -90,9 +90,9 @@ class Coder:
return WholeFileFunctionCoder(main_model, io, **kwargs) return WholeFileFunctionCoder(main_model, io, **kwargs)
elif edit_format == "single-whole-func": elif edit_format == "single-whole-func":
return SingleWholeFileFunctionCoder(main_model, io, **kwargs) return SingleWholeFileFunctionCoder(main_model, io, **kwargs)
elif edit_format == "diff-func": elif edit_format == "diff-func-list":
return EditBlockFunctionCoder("list", main_model, io, **kwargs) return EditBlockFunctionCoder("list", main_model, io, **kwargs)
elif edit_format == "diff-func-string": elif edit_format in ("diff-func", "diff-func-string"):
return EditBlockFunctionCoder("string", main_model, io, **kwargs) return EditBlockFunctionCoder("string", main_model, io, **kwargs)
else: else:
raise ValueError(f"Unknown edit format {edit_format}") raise ValueError(f"Unknown edit format {edit_format}")