diff --git a/metagpt/startup.py b/metagpt/startup.py index cacf68113..cd5b4dac7 100644 --- a/metagpt/startup.py +++ b/metagpt/startup.py @@ -1,11 +1,13 @@ #!/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) @@ -64,9 +66,9 @@ def generate_repo( asyncio.run(company.run(n_round=n_round)) -@app.command() +@app.command("", help="Start a new project.") def startup( - idea: str = typer.Argument(..., help="Your innovative idea, such as 'Create a 2048 game.'"), + 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."), @@ -87,8 +89,17 @@ def startup( "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, @@ -105,5 +116,23 @@ def startup( ) +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()