mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-05-09 07:42:38 +02:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
from pathlib import Path
|
||
|
|
import asyncio
|
||
|
|
import typer
|
||
|
|
|
||
|
|
app = typer.Typer()
|
||
|
|
|
||
|
|
|
||
|
|
@app.command()
|
||
|
|
def startup(
|
||
|
|
idea: str = typer.Argument(..., help="Your innovative idea, such as 'Create a 2048 game.'"),
|
||
|
|
investment: float = typer.Option(3.0, help="Dollar amount to invest in the AI company."),
|
||
|
|
n_round: int = typer.Option(5, help="Number of rounds for the simulation."),
|
||
|
|
code_review: bool = typer.Option(True, help="Whether to use code review."),
|
||
|
|
run_tests: bool = typer.Option(False, help="Whether to enable QA for adding & running tests."),
|
||
|
|
implement: bool = typer.Option(True, help="Enable or disable code implementation."),
|
||
|
|
project_name: str = typer.Option("", help="Unique project name, such as 'game_2048'"),
|
||
|
|
):
|
||
|
|
"""Run a startup. Be a boss."""
|
||
|
|
from metagpt.roles import ProductManager, Architect, ProjectManager, Engineer, QaEngineer
|
||
|
|
from metagpt.team import Team
|
||
|
|
|
||
|
|
company = Team()
|
||
|
|
company.hire(
|
||
|
|
[
|
||
|
|
ProductManager(),
|
||
|
|
Architect(),
|
||
|
|
ProjectManager(),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
if implement or code_review:
|
||
|
|
company.hire([Engineer(n_borg=5, use_code_review=code_review)])
|
||
|
|
|
||
|
|
if run_tests:
|
||
|
|
company.hire([QaEngineer()])
|
||
|
|
|
||
|
|
company.invest(investment)
|
||
|
|
company.start_project(project_name, idea)
|
||
|
|
asyncio.run(company.run(n_round=n_round))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
app()
|