This commit is contained in:
Paul Gauthier 2024-09-10 15:21:54 -07:00
parent abd484bfa7
commit 39ae106bb3

View file

@ -549,15 +549,15 @@ def run_test_real(
chat_history_file=history_fname, chat_history_file=history_fname,
) )
# ask_model = models.Model("gpt-4o") ask_model = models.Model("gpt-4o")
ask_model = models.Model("openrouter/anthropic/claude-3.5-sonnet") # ask_model = models.Model("openrouter/anthropic/claude-3.5-sonnet")
# ask_model = models.Model("openrouter/deepseek/deepseek-coder") # ask_model = models.Model("openrouter/deepseek/deepseek-coder")
# whole_model = models.Model("gpt-4o") # whole_model = models.Model("gpt-4o")
# whole_model = models.Model("openrouter/anthropic/claude-3.5-sonnet") # whole_model = models.Model("openrouter/anthropic/claude-3.5-sonnet")
# whole_model = models.Model("openrouter/deepseek/deepseek-coder") # whole_model = models.Model("openrouter/deepseek/deepseek-coder")
whole_model = models.Model("openrouter/anthropic/claude-3-haiku-20240307") # whole_model = models.Model("openrouter/anthropic/claude-3-haiku-20240307")
# whole_model = models.Model("gpt-4o-mini") whole_model = models.Model("gpt-4o-mini")
main_model = ask_model main_model = ask_model
edit_format = "ask-whole" edit_format = "ask-whole"
@ -618,7 +618,7 @@ def run_test_real(
whole_coder = Coder.create( whole_coder = Coder.create(
from_coder=coder, from_coder=coder,
main_model=whole_model, main_model=whole_model,
edit_format="diff", edit_format="whole",
**coder_kwargs, **coder_kwargs,
) )
response = whole_coder.run(with_message="make those changes", preproc=False) response = whole_coder.run(with_message="make those changes", preproc=False)