import argparse

from benchmark.tau_bench.model_utils.model.model import Platform


def api_parser() -> argparse.ArgumentParser:
    parser: argparse.ArgumentParser = argparse.ArgumentParser()
    parser.add_argument("--model", type=str)
    parser.add_argument("--base-url", type=str)
    parser.add_argument(
        "--platform", type=str, required=True, choices=[e.value for e in Platform]
    )
    return parser
