aljaz/gpt-engineer
1
0
Fork 0
mirror of https://github.com/aljazceru/gpt-engineer.git synced 2025-12-17 12:45:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Black

Browse Source
This commit is contained in:
Anton Osika
2023-06-17 21:32:46 +02:00
parent 8ac2cd81b9
commit a4019fb82f
1 changed files with 36 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
gpt_engineer/steps.py
View File

@@ -30,7 +30,7 @@ def clarify(ai: AI, dbs: DBs):
while True: while True:
messages = ai.next(messages, user) messages = ai.next(messages, user)
if messages[-1]['content'].strip().lower().startswith("no"): if messages[-1]["content"].strip().lower().startswith("no"):
break break
print() print()
@@ -53,29 +53,37 @@ def clarify(ai: AI, dbs: DBs):
def gen_spec(ai: AI, dbs: DBs): def gen_spec(ai: AI, dbs: DBs):
''' """
Generate a spec from the main prompt + clarifications and save the results to the workspace Generate a spec from the main prompt + clarifications and save the results to the workspace
''' """
messages = [ai.fsystem(setup_sys_prompt(dbs)), ai.fsystem(f"Instructions: {dbs.input['main_prompt']}")] messages = [
ai.fsystem(setup_sys_prompt(dbs)),
ai.fsystem(f"Instructions: {dbs.input['main_prompt']}"),
]
messages = ai.next(messages, dbs.identity['spec']) messages = ai.next(messages, dbs.identity["spec"])
messages = ai.next(messages, dbs.identity['respec']) messages = ai.next(messages, dbs.identity["respec"])
messages = ai.next(messages, dbs.identity['spec']) messages = ai.next(messages, dbs.identity["spec"])
dbs.memory['specification'] = messages[-1]['content'] dbs.memory["specification"] = messages[-1]["content"]
return messages return messages
def pre_unit_tests(ai: AI, dbs: DBs): def pre_unit_tests(ai: AI, dbs: DBs):
''' """
Generate unit tests based on the specification, that should work. Generate unit tests based on the specification, that should work.
''' """
messages = [ai.fsystem(setup_sys_prompt(dbs)), ai.fuser(f"Instructions: {dbs.input['main_prompt']}"), ai.fuser(f"Specification:\n\n{dbs.memory['specification']}")] messages = [
ai.fsystem(setup_sys_prompt(dbs)),
ai.fuser(f"Instructions: {dbs.input['main_prompt']}"),
ai.fuser(f"Specification:\n\n{dbs.memory['specification']}"),
]
messages = ai.next(messages, dbs.identity['unit_tests']) messages = ai.next(messages, dbs.identity["unit_tests"])
dbs.memory['unit_tests'] = messages[-1]['content'] dbs.memory["unit_tests"] = messages[-1]["content"]
to_files(dbs.memory['unit_tests'], dbs.workspace) to_files(dbs.memory["unit_tests"], dbs.workspace)
return messages return messages
@@ -101,19 +109,19 @@ def execute_workspace(ai: AI, dbs: DBs):
def execute_entrypoint(ai, dbs): def execute_entrypoint(ai, dbs):
command = dbs.workspace['run.sh'] command = dbs.workspace["run.sh"]
print('Do you want to execute this code?') print("Do you want to execute this code?")
print() print()
print(command) print(command)
print() print()
print('If yes, press enter. If no, type "no"') print('If yes, press enter. If no, type "no"')
print() print()
if input() == 'no': if input() == "no":
print('Ok, not executing the code.') print("Ok, not executing the code.")
print('Executing the code...') print("Executing the code...")
print() print()
subprocess.run('bash run.sh', shell=True, cwd=dbs.workspace.path) subprocess.run("bash run.sh", shell=True, cwd=dbs.workspace.path)
return [] return []
@@ -130,19 +138,19 @@ def gen_entrypoint(ai, dbs):
user="Information about the codebase:\n\n" + dbs.workspace["all_output.txt"], user="Information about the codebase:\n\n" + dbs.workspace["all_output.txt"],
) )
print() print()
[[lang, command]] = parse_chat(messages[-1]['content']) [[lang, command]] = parse_chat(messages[-1]["content"])
assert lang in ['', 'bash', 'sh'] assert lang in ["", "bash", "sh"]
dbs.workspace['run.sh'] = command dbs.workspace["run.sh"] = command
return messages return messages
# Different configs of what steps to run # Different configs of what steps to run
STEPS = { STEPS = {
'default': [gen_spec, pre_unit_tests, run_clarified, execute_workspace], "default": [gen_spec, pre_unit_tests, run_clarified, execute_workspace],
'benchmark': [gen_spec, pre_unit_tests, run_clarified, gen_entrypoint], "benchmark": [gen_spec, pre_unit_tests, run_clarified, gen_entrypoint],
'simple': [run, execute_workspace], "simple": [run, execute_workspace],
'clarify': [clarify, run_clarified, gen_entrypoint], "clarify": [clarify, run_clarified, gen_entrypoint],
'execute_only': [execute_entrypoint], "execute_only": [execute_entrypoint],
} }
# Future steps that can be added: # Future steps that can be added: