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

improve execution step

Browse Source
This commit is contained in:
Anton Osika
2023-06-17 20:36:51 +02:00
parent a3cb4964e9
commit e8f47c1e70
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
gpt_engineer/steps.py
View File

@@ -56,7 +56,7 @@ 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"Main prompt: {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'])
@@ -97,10 +97,11 @@ def run_clarified(ai: AI, dbs: DBs):
def execute_workspace(ai: AI, dbs: DBs): def execute_workspace(ai: AI, dbs: DBs):
messages = ai.start( messages = ai.start(
system=( system=(
f"You will get infomation about a codebase that is currently on disk in the folder {dbs.workspace.path}.\n" f"You will get information about a codebase that is currently on disk in the current folder.\n"
"From this you will answer with one code block that includes all the necessary macos terminal commands to " "From this you will answer with one code block that includes all the necessary macos terminal commands to "
"a) install dependencies " "a) install dependencies "
"b) run the necessary parts of the codebase to try it.\n" "b) run all necessary parts of the codebase (in parallell if necessary).\n"
"Do not install globally. Do not use sudo.\n"
"Do not explain the code, just give the commands.\n" "Do not explain the code, just give the commands.\n"
), ),
user="Information about the codebase:\n\n" + dbs.workspace["all_output.txt"], user="Information about the codebase:\n\n" + dbs.workspace["all_output.txt"],
@@ -109,6 +110,8 @@ def execute_workspace(ai: AI, dbs: DBs):
[[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
print('Do you want to execute this code?') print('Do you want to execute this code?')
print(command) print(command)
print() print()
@@ -119,7 +122,9 @@ def execute_workspace(ai: AI, dbs: DBs):
return messages return messages
print('Executing the code...') print('Executing the code...')
print() print()
subprocess.run(command, shell=True)
# Run the subprocess in dbs.workspace.path
subprocess.run('bash run.sh', shell=True, cwd=dbs.workspace.path)
return messages return messages
@@ -128,6 +133,7 @@ STEPS = {
'default': [gen_spec, pre_unit_tests, run_clarified], 'default': [gen_spec, pre_unit_tests, run_clarified],
'simple': [run, execute_workspace], 'simple': [run, execute_workspace],
'clarify': [clarify, run_clarified], 'clarify': [clarify, run_clarified],
'execute_only': [execute_workspace],
} }
# Future steps that can be added: # Future steps that can be added: