mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-25 08:46:48 +02:00
138 lines
4.4 KiB
Python
138 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import asyncio
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
import typer
|
|
|
|
from metagpt.config2 import config
|
|
from metagpt.const import METAGPT_ROOT
|
|
|
|
app = typer.Typer(add_completion=False, pretty_exceptions_show_locals=False)
|
|
|
|
|
|
def generate_repo(
|
|
idea,
|
|
investment,
|
|
n_round,
|
|
code_review,
|
|
run_tests,
|
|
implement,
|
|
project_name,
|
|
inc,
|
|
project_path,
|
|
reqa_file,
|
|
max_auto_summarize_code,
|
|
recover_path,
|
|
):
|
|
"""Run the startup logic. Can be called from CLI or other Python scripts."""
|
|
from metagpt.roles import (
|
|
Architect,
|
|
Engineer,
|
|
ProductManager,
|
|
ProjectManager,
|
|
QaEngineer,
|
|
)
|
|
from metagpt.team import Team
|
|
|
|
config.update_via_cli(project_path, project_name, inc, reqa_file, max_auto_summarize_code)
|
|
|
|
if not recover_path:
|
|
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()])
|
|
else:
|
|
stg_path = Path(recover_path)
|
|
if not stg_path.exists() or not str(stg_path).endswith("team"):
|
|
raise FileNotFoundError(f"{recover_path} not exists or not endswith `team`")
|
|
|
|
company = Team.deserialize(stg_path=stg_path)
|
|
idea = company.idea
|
|
|
|
company.invest(investment)
|
|
company.run_project(idea)
|
|
asyncio.run(company.run(n_round=n_round))
|
|
|
|
|
|
@app.command("", help="Start a new project.")
|
|
def startup(
|
|
idea: str = typer.Argument(None, help="Your innovative idea, such as 'Create a 2048 game.'"),
|
|
investment: float = typer.Option(default=3.0, help="Dollar amount to invest in the AI company."),
|
|
n_round: int = typer.Option(default=5, help="Number of rounds for the simulation."),
|
|
code_review: bool = typer.Option(default=True, help="Whether to use code review."),
|
|
run_tests: bool = typer.Option(default=False, help="Whether to enable QA for adding & running tests."),
|
|
implement: bool = typer.Option(default=True, help="Enable or disable code implementation."),
|
|
project_name: str = typer.Option(default="", help="Unique project name, such as 'game_2048'."),
|
|
inc: bool = typer.Option(default=False, help="Incremental mode. Use it to coop with existing repo."),
|
|
project_path: str = typer.Option(
|
|
default="",
|
|
help="Specify the directory path of the old version project to fulfill the incremental requirements.",
|
|
),
|
|
reqa_file: str = typer.Option(
|
|
default="", help="Specify the source file name for rewriting the quality assurance code."
|
|
),
|
|
max_auto_summarize_code: int = typer.Option(
|
|
default=0,
|
|
help="The maximum number of times the 'SummarizeCode' action is automatically invoked, with -1 indicating "
|
|
"unlimited. This parameter is used for debugging the workflow.",
|
|
),
|
|
recover_path: str = typer.Option(default=None, help="recover the project from existing serialized storage"),
|
|
init_config: bool = typer.Option(default=False, help="Initialize the configuration file for MetaGPT."),
|
|
):
|
|
"""Run a startup. Be a boss."""
|
|
if init_config:
|
|
copy_config_to()
|
|
return
|
|
|
|
if idea is None:
|
|
typer.echo("Missing argument 'IDEA'. Run 'metagpt --help' for more information.")
|
|
raise typer.Exit()
|
|
|
|
return generate_repo(
|
|
idea,
|
|
investment,
|
|
n_round,
|
|
code_review,
|
|
run_tests,
|
|
implement,
|
|
project_name,
|
|
inc,
|
|
project_path,
|
|
reqa_file,
|
|
max_auto_summarize_code,
|
|
recover_path,
|
|
)
|
|
|
|
|
|
def copy_config_to(config_path=METAGPT_ROOT / "config" / "config2.yaml"):
|
|
"""Initialize the configuration file for MetaGPT."""
|
|
target_path = Path.home() / ".metagpt" / "config2.yaml"
|
|
|
|
# 创建目标目录（如果不存在）
|
|
target_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
# 如果目标文件已经存在，则重命名为 .bak
|
|
if target_path.exists():
|
|
backup_path = target_path.with_suffix(".bak")
|
|
target_path.rename(backup_path)
|
|
print(f"Existing configuration file backed up at {backup_path}")
|
|
|
|
# 复制文件
|
|
shutil.copy(str(config_path), target_path)
|
|
print(f"Configuration file initialized at {target_path}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|