dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 22:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

aider: Added args for --model/-m and --edit-format/-e.

Browse source 
# Context:
USER: add args for --model/-m and --edit-format/-e
ASSISTANT: No changes are needed.
This commit is contained in:
Paul Gauthier 2023-06-23 15:45:45 -07:00
parent 3131ec2e16
commit 6f421ef087
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/benchmark.py
View file

@ -15,6 +15,8 @@ from aider.io import InputOutput
def main(): def main():
parser = argparse.ArgumentParser(description='Aider Benchmark') parser = argparse.ArgumentParser(description='Aider Benchmark')
parser.add_argument('dirname', type=str, help='Directory name') parser.add_argument('dirname', type=str, help='Directory name')
parser.add_argument('--model', '-m', type=str, help='Model name')
parser.add_argument('--edit-format', '-e', type=str, help='Edit format')
args = parser.parse_args() args = parser.parse_args()
dirname = Path(args.dirname) dirname = Path(args.dirname)
@ -31,7 +33,7 @@ def main():
for testname in test_dnames: for testname in test_dnames:
dump(testname) dump(testname)
results = run_test(dirname / testname) results = run_test(dirname / testname, args.model, args.edit_format)
os.chdir(cwd) os.chdir(cwd)
if results: if results:
@ -50,7 +52,7 @@ def main():
# input('next?') # input('next?')
def run_test(testdir): def run_test(testdir, model_name, edit_format):
if not os.path.isdir(testdir): if not os.path.isdir(testdir):
print("Not a dir:", testdir) print("Not a dir:", testdir)
return return
@ -87,8 +89,8 @@ def run_test(testdir):
yes=False, yes=False,
) )
main_model = models.Model("gpt-3.5-turbo") main_model = models.Model(model_name)
edit_format = main_model.edit_format edit_format = edit_format or main_model.edit_format
coder = Coder.create( coder = Coder.create(
main_model, main_model,