| | |
| | |
| | import asyncio |
| | import platform |
| | import fire |
| |
|
| | from metagpt.roles import Architect, Engineer, ProductManager, ProjectManager, QaEngineer |
| | from metagpt.software_company import SoftwareCompany |
| |
|
| |
|
| | async def startup(idea: str, investment: float = 3.0, n_round: int = 5, |
| | code_review: bool = False, run_tests: bool = False): |
| | """Run a startup. Be a boss.""" |
| | company = SoftwareCompany() |
| | company.hire([ProductManager(), |
| | Architect(), |
| | ProjectManager(), |
| | Engineer(n_borg=5, use_code_review=code_review)]) |
| | if run_tests: |
| | |
| | company.hire([QaEngineer()]) |
| | company.invest(investment) |
| | company.start_project(idea) |
| | await company.run(n_round=n_round) |
| |
|
| |
|
| | def main(idea: str, investment: float = 3.0, n_round: int = 5, code_review: bool = False, run_tests: 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: |
| | """ |
| | if platform.system() == "Windows": |
| | asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) |
| | asyncio.run(startup(idea, investment, n_round, code_review, run_tests)) |
| |
|
| |
|
| | if __name__ == '__main__': |
| | fire.Fire(main) |
| |
|
