singlewholefile-func

This commit is contained in:
Paul Gauthier 2023-06-29 18:13:42 -07:00
parent 271807123a
commit ddac41f106
4 changed files with 158 additions and 1 deletions

View file

@ -59,6 +59,7 @@ class Coder:
from . import (
EditBlockCoder,
EditBlockFunctionCoder,
SingleWholeFileFunctionCoder,
WholeFileCoder,
WholeFileFunctionCoder,
)
@ -87,6 +88,8 @@ class Coder:
return WholeFileCoder(main_model, io, **kwargs)
elif edit_format == "whole-func":
return WholeFileFunctionCoder(main_model, io, **kwargs)
elif edit_format == "single-whole-func":
return SingleWholeFileFunctionCoder(main_model, io, **kwargs)
elif edit_format == "diff-func":
return EditBlockFunctionCoder("list", main_model, io, **kwargs)
elif edit_format == "diff-func-string":