mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 12:45:26 +01:00
107 lines
3.0 KiB
Python
107 lines
3.0 KiB
Python
import json
|
|
|
|
from gpt_engineer.ai import AI
|
|
from gpt_engineer.chat_to_files import to_files
|
|
from gpt_engineer.db import DBs
|
|
|
|
|
|
def setup_sys_prompt(dbs):
|
|
return dbs.identity['setup'] + '\nUseful to know:\n' + dbs.identity['philosophy']
|
|
|
|
|
|
def run(ai: AI, dbs: DBs):
|
|
'''Run the AI on the main prompt and save the results'''
|
|
messages = ai.start(
|
|
setup_sys_prompt(dbs),
|
|
dbs.input['main_prompt'],
|
|
)
|
|
to_files(messages[-1]['content'], dbs.workspace)
|
|
return messages
|
|
|
|
|
|
def clarify(ai: AI, dbs: DBs):
|
|
'''
|
|
Ask the user if they want to clarify anything and save the results to the workspace
|
|
'''
|
|
messages = [ai.fsystem(dbs.identity['qa'])]
|
|
user = dbs.input['main_prompt']
|
|
while True:
|
|
messages = ai.next(messages, user)
|
|
|
|
if messages[-1]['content'].strip().lower() == 'no':
|
|
break
|
|
|
|
print()
|
|
user = input('(answer in text, or "q" to move on)\n')
|
|
print()
|
|
|
|
if not user or user == 'q':
|
|
break
|
|
|
|
user += (
|
|
'\n\n'
|
|
'Is anything else unclear? If yes, only answer in the form:\n'
|
|
'{remaining unclear areas} remaining questions.\n'
|
|
'{Next question}\n'
|
|
'If everything is sufficiently clear, only answer "no".'
|
|
)
|
|
|
|
print()
|
|
return messages
|
|
|
|
|
|
def gen_spec(ai: AI, dbs: DBs):
|
|
'''
|
|
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.next(messages, dbs.identity['spec'])
|
|
messages = ai.next(messages, dbs.identity['respec'])
|
|
messages = ai.next(messages, dbs.identity['spec'])
|
|
|
|
dbs.memory['specification'] = messages[-1]['content']
|
|
|
|
return messages
|
|
|
|
def pre_unit_tests(ai: AI, dbs: DBs):
|
|
'''
|
|
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.next(messages, dbs.identity['unit_tests'])
|
|
|
|
dbs.memory['unit_tests'] = messages[-1]['content']
|
|
to_files(dbs.memory['unit_tests'], dbs.workspace)
|
|
|
|
return messages
|
|
|
|
|
|
def run_clarified(ai: AI, dbs: DBs):
|
|
# get the messages from previous step
|
|
|
|
messages = [
|
|
ai.fsystem(setup_sys_prompt(dbs)),
|
|
ai.fsystem(f"Main prompt: {dbs.input['main_prompt']}"),
|
|
ai.fsystem(f"Specification:\n\n{dbs.memory['specification']}"),
|
|
ai.fsystem(f"Unit tests:\n\n{dbs.memory['unit_tests']}"),
|
|
]
|
|
messages = ai.next(messages, dbs.identity['use_qa'])
|
|
to_files(messages[-1]['content'], dbs.workspace)
|
|
return messages
|
|
|
|
|
|
# Different configs of what steps to run
|
|
STEPS = {
|
|
'default': [run],
|
|
'unit_tests': [gen_spec, pre_unit_tests, run_clarified],
|
|
'clarify': [clarify, run_clarified],
|
|
}
|
|
|
|
# Future steps that can be added:
|
|
# improve_files,
|
|
# add_tests
|
|
# run_tests_and_fix_files,
|
|
# improve_based_on_in_file_feedback_comments
|