Function main

cli/main.py:469–505

(argv: Sequence[str] | None = None)

def main(argv: Sequence[str] | None = None) -> int:
    args = build_parser().parse_args(argv)

    if args.command == "benchmark":
        return handle_benchmark_command(args)
    if args.command == "inspect":
        return handle_inspect(args.benchmark_file, args.benchmark_name, args.model, args.config_file)
    if args.command == "run":
        return handle_run(
            args.benchmark_file,
            args.benchmark_name,
            args.model,
            args.ip_addr,
            args.prompt_bundle,
            args.config_file,
            args.output_file,
        )
    if args.command == "experiment":
        return handle_experiment(
            args.benchmark_file,
            args.benchmark_name,
            args.models,
            args.ip_addr,
            args.repeat,
            args.prompt_bundle,
            args.config_file,
            args.output_file,
        )
    if args.command == "analyze":
        return handle_analysis_command(args)
    if args.command == "doctor":
        return handle_doctor(
            config_file=args.config_file,
            benchmark_file=args.benchmark_file,
            model_identifier=args.model,
        )
    raise ValueError(f"Unsupported command: {args.command}")


if __name__ == "__main__":