stand alone junior message

This commit is contained in:
Paul Gauthier 2024-09-25 08:41:49 -07:00
parent c912982747
commit 075bc828f6
2 changed files with 10 additions and 6 deletions

View file

@ -4,10 +4,13 @@ from .base_prompts import CoderPrompts
class AskPrompts(CoderPrompts):
main_system = """Act as an expert code analyst.
main_system = """Act as an expert senior engineer and provide direction to your junior engineer.
Study the change request and the current code.
Describe how to modify the code to complete the request.
Explain all needed code changes clearly and completely, but concisely.
The junior engineer will rely solely on your instructions, so make them unambiguous and complete.
Always reply to the user in the same language they are using.
Always reply in the same language as the change request.
"""
example_messages = []

View file

@ -550,9 +550,9 @@ def run_test_real(
)
# senior_model = models.Model("o1-mini")
# senior_model = models.Model("o1-preview")
senior_model = models.Model("o1-preview")
# senior_model = models.Model("gpt-4o")
senior_model = models.Model("openrouter/anthropic/claude-3.5-sonnet")
# senior_model = models.Model("openrouter/anthropic/claude-3.5-sonnet")
# senior_model = models.Model("openrouter/deepseek/deepseek-chat")
# junior_model = models.Model("gpt-4o")
@ -622,12 +622,13 @@ def run_test_real(
"""
response = coder.run(with_message=instructions, preproc=False)
junior_coder = Coder.create(
from_coder=coder,
# from_coder=coder,
main_model=junior_model,
edit_format=junior_edit_format,
**coder_kwargs,
)
response = junior_coder.run(with_message="make those changes", preproc=False)
response = junior_coder.run(with_message=response, preproc=False)
coder.move_back_cur_messages("I made those changes to the files.")
dur += time.time() - start