mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-26 01:06:27 +02:00
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
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:
|
|
# 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)
|
|
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)
|