apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-29 02:46:24 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update increment development, add write_code_guide.py file

Browse source
This commit is contained in:
mannaandpoem 2023-11-29 17:31:15 +08:00
parent 8ff833f1e3
commit a50a1f738f
12 changed files with 290 additions and 449 deletions
Download patch file Download diff file Expand all files Collapse all files

66
startup.py
View file

@ -28,12 +28,11 @@ async def startup(
run_tests: bool = False,
implement: bool = True,
increment: bool = False,
bug_fix: bool = False,
):
"""Run a startup. Be a boss."""
company = Team()
if increment or bug_fix:
if increment:
# 读取文件
prd_path = os.path.join(project_path, 'docs/prd.md')
design_path = os.path.join(project_path, 'docs/system_design.md')
@ -47,64 +46,46 @@ async def startup(
legacy_design = f.read()
with open(api_spec_and_tasks_path, 'r', encoding='utf-8') as f:
legacy_api_spec_and_tasks = f.read()
legacy_project_management = f.read()
# 遍历文件夹，获取所有代码文件
legacy_code = ''
for root, dirs, files in os.walk(code_path):
filenames = [filename for filename in files if filename.endswith('.py')]
legacy_code += f'There are {len(files)} scripts in the current folder: {", ".join(filenames)}\n---\n'
for file in files:
if file.endswith('.py'):
with open(os.path.join(root, file), 'r', encoding='utf-8') as f:
legacy_code += f.read() + '\n---\n'
if bug_fix:
boss_msg = Message(
content=f"Boss's requirement\n:{idea}\n---\nBoss's difference description:{difference_description}\n---\n",
sent_from="Boss",
)
product_manager_msg = Message(
content=f"Product Manager's prd legacy:\n{legacy_prd}\n---\n",
sent_from="ProductManager"
)
architect_msg = Message(
content=f"Architect's design legacy:\n{legacy_design}\n---\n",
sent_from="Architect"
)
project_manager_msg = Message(
content=f"Project Manager's api spec and tasks legacy:\n{legacy_api_spec_and_tasks}\n---\n",
sent_from="ProjectManager"
)
engineer_msg = Message(
content=f"Engineer's code legacy:\n{legacy_code}\n---\n",
sent_from="Engineer"
)
bug_msgs = [boss_msg, product_manager_msg, architect_msg, project_manager_msg, engineer_msg]
else:
company.hire(
[ProductManager(difference_description=difference_description, legacy=legacy_prd, increment=increment),
Architect(legacy=legacy_design, increment=increment),
ProjectManager(legacy=legacy_api_spec_and_tasks, increment=increment)])
legacy_dict = {
'legacy_prd': legacy_prd,
'legacy_design': legacy_design,
'legacy_project_management': legacy_project_management,
'legacy_code': legacy_code
}
company.set_legacy(legacy_dict)
company.hire(
[
ProductManager(increment=increment),
Architect(increment=increment),
ProjectManager(increment=increment)
]
)
else:
company.hire([ProductManager(), Architect(), ProjectManager()])
# if implement or code_review
if bug_fix:
company.hire([Engineer(n_borg=5, bug_msgs=bug_msgs, bug_fix=bug_fix)])
elif implement or code_review:
if (implement or code_review) and increment:
company.hire([Engineer(n_borg=5, use_code_review=code_review, increment=increment)])
elif implement:
# developing features: implement the idea
company.hire([Engineer(n_borg=5, use_code_review=code_review)])
elif (implement or code_review) and increment:
company.hire([Engineer(n_borg=5, use_code_review=code_review, legacy=legacy_code, increment=increment)])
if run_tests or bug_fix:
if run_tests:
# developing features: run tests on the spot and identify bugs
# (bug fixing capability comes soon!)
company.hire([QaEngineer()])
company.invest(investment)
company.start_project(idea)
company.start_project(idea+"\n"+difference_description)
await company.run(n_round=n_round)
@ -118,7 +99,6 @@ def main(
run_tests: bool = False,
implement: bool = True,
increment: bool = False,
bug_fix: bool = False,
):
"""
We are a software startup comprised of AI. By investing in us,
@ -130,10 +110,8 @@ def main(
:param code_review: Whether to use code review.
:return:
"""
asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
asyncio.run(
startup(idea, difference_description, project_path, investment, n_round, code_review, run_tests, implement, increment, bug_fix))
startup(idea, difference_description, project_path, investment, n_round, code_review, run_tests, implement, increment))
if __name__ == "__main__":