mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-28 18:36:22 +02:00
118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import asyncio
|
|
import os
|
|
import platform
|
|
|
|
import fire
|
|
|
|
from metagpt.roles import (
|
|
Architect,
|
|
Engineer,
|
|
ProductManager,
|
|
ProjectManager,
|
|
QaEngineer,
|
|
)
|
|
from metagpt.schema import Message
|
|
from metagpt.team import Team
|
|
from metagpt.utils.special_tokens import MSG_SEP
|
|
|
|
|
|
async def startup(
|
|
idea: str,
|
|
difference_description: str = "",
|
|
project_path: str = "",
|
|
investment: float = 3.0,
|
|
n_round: int = 5,
|
|
code_review: bool = False,
|
|
run_tests: bool = False,
|
|
implement: bool = True,
|
|
increment: bool = False,
|
|
):
|
|
"""Run a startup. Be a boss."""
|
|
company = Team()
|
|
|
|
if increment:
|
|
# 读取文件
|
|
prd_path = os.path.join(project_path, 'docs/prd.md')
|
|
design_path = os.path.join(project_path, 'docs/system_design.md')
|
|
api_spec_and_tasks_path = os.path.join(project_path, 'docs/api_spec_and_tasks.md')
|
|
code_path = os.path.join(project_path, os.path.basename(project_path))
|
|
|
|
with open(prd_path, 'r', encoding='utf-8') as f:
|
|
legacy_prd = f.read()
|
|
|
|
with open(design_path, 'r', encoding='utf-8') as f:
|
|
legacy_design = f.read()
|
|
|
|
with open(api_spec_and_tasks_path, 'r', encoding='utf-8') as f:
|
|
legacy_project_management = f.read()
|
|
|
|
legacy_code = ''
|
|
for root, dirs, files in os.walk(code_path):
|
|
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'
|
|
|
|
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 (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)])
|
|
|
|
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+"\n"+difference_description)
|
|
await company.run(n_round=n_round)
|
|
|
|
|
|
def main(
|
|
idea: str,
|
|
difference_description: str = "",
|
|
project_path: str = "",
|
|
investment: float = 3.0,
|
|
n_round: int = 5,
|
|
code_review: bool = True,
|
|
run_tests: bool = False,
|
|
implement: bool = True,
|
|
increment: bool = False,
|
|
):
|
|
"""
|
|
We are a software startup comprised of AI. By investing in us,
|
|
you are empowering a future filled with limitless possibilities.
|
|
:param idea: Your innovative idea, such as "Creating a snake game."
|
|
:param investment: As an investor, you have the opportunity to contribute
|
|
a certain dollar amount to this AI company.
|
|
:param n_round:
|
|
:param code_review: Whether to use code review.
|
|
:return:
|
|
"""
|
|
asyncio.run(
|
|
startup(idea, difference_description, project_path, investment, n_round, code_review, run_tests, implement, increment))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fire.Fire(main)
|