Function main

internbootcamp/utils/tool_server/cli.py:320–754

主函数

()

def main():
    """主函数"""
    parser = argparse.ArgumentParser(
        description="分布式Master-Worker服务器架构",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
示例用法:

1. 启动Master服务器 (动态工具发现):
   python cli.py --mode master --port 8000

2. 启动Master服务器 (预配置工具):
   python cli.py --mode master --tools_yaml_path config.yaml --port 8000

3. 在其他机器上启动Worker:
   python cli.py --mode worker --tools_yaml_path config.yaml --master_url http://master_ip:8000 --port 8001 --num_workers 3

4. 启动统一服务器（单机Master+多Worker模式）:
   python cli.py --mode unified --tools_yaml_path config.yaml --port 8000 --num_workers 8
   python cli.py --mode unified --tools_yaml_path config.yaml --port 8000 --keep_running --num_workers 8
   python cli.py --mode unified --tools_yaml_path config.yaml --port 8000 --test_servers --keep_running --timeout_per_query 600 --num_workers 5
        """
    )
    
    parser.add_argument(
        "--mode", 
        choices=["master", "worker", "unified"],
        required=True,
        help="运行模式: master(Master服务器), worker(Worker服务器), unified(统一服务器)"
    )
    parser.add_argument(
        "--tools_yaml_path", 
        required=False,
        help="工具配置YAML文件路径"
    )
    parser.add_argument(
        "--bootcamp_registry",
        required=False,
        help="bootcamp注册表，提供后可将注册表里所有bootcamp的tool挂载到woker server"
    )
    parser.add_argument(
        "--port", 
        type=int, 
        default=None,
        help="服务器端口号 (默认: 随机分配可用端口)"
    )
    parser.add_argument(
        "--host", 
        default="0.0.0.0",
        help="服务器主机地址 (默认: 0.0.0.0)"
    )
    parser.add_argument(
        "--master_url", 
        help="Master服务器URL (Worker模式必需)"
    )
    parser.add_argument(
        "--worker_id", 
        help="Worker ID (Worker模式使用，默认自动生成)"