cleanup Coder params

This commit is contained in:
Paul Gauthier 2023-05-18 18:21:44 -07:00
parent a59575fa4b
commit ba9307374e
3 changed files with 19 additions and 22 deletions

View file

@ -28,14 +28,21 @@ class Coder:
last_aider_commit_hash = None
last_asked_for_commit_time = 0
def __init__(self, main_model, fnames, pretty, show_diffs, auto_commits, io, dry_run):
def __init__(
self,
io,
main_model="gpt-4",
fnames=None,
pretty=True,
show_diffs=False,
auto_commits=True,
dry_run=False,
):
self.abs_fnames = set()
self.io = io
self.cur_messages = []
self.done_messages = []
self.io = io
self.auto_commits = auto_commits
self.dry_run = dry_run

View file

@ -88,8 +88,6 @@ def main(args=None, input=None, output=None):
default=False,
)
args = parser.parse_args(args)
fnames = args.files
pretty = args.pretty
io = InputOutput(
args.pretty,
@ -103,13 +101,13 @@ def main(args=None, input=None, output=None):
io.tool(*sys.argv, log_only=True)
coder = Coder(
args.model,
fnames,
pretty,
args.show_diffs,
args.auto_commits,
io,
args.dry_run,
main_model=args.model,
fnames=args.files,
pretty=args.pretty,
show_diffs=args.show_diffs,
auto_commits=args.auto_commits,
dry_run=args.dry_run,
)
if args.auto_commits:
coder.commit(ask=True, prefix="wip: ", which="repo_files")