| |
| |
| 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) |
|
|
