def parse_args()

in scripts/comment_resolver_runner.py [0:0]

• 67 lines of code
• 1 McCabe index (conditional complexity)

def parse_args(args):
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--llm",
        help="LLM",
        choices=["openai"],
        default="openai",
    )
    parser.add_argument(
        "--create-db",
        action="store_true",
        help="If set, the local Qdrant database will be created and populated.",
    )
    parser.add_argument(
        "--dataset-file",
        type=str,
        default="data/fixed_comments.json",
        help="Dataset file to upload as Qdrant database.",
    )
    parser.add_argument(
        "--output-csv",
        type=str,
        default="metrics_results.csv",
        help="Output CSV file for results.",
    )
    parser.add_argument(
        "--prompt-types",
        nargs="+",
        default=["zero-shot"],
        help="Types of prompts to use.",
    )
    parser.add_argument(
        "--diff-length-limits",
        nargs="+",
        type=int,
        default=[1000],
        help="Diff length limits to enforce when searching for examples.",
    )
    parser.add_argument(
        "--hunk-sizes",
        nargs="+",
        type=int,
        default=[20],
        help="Hunk sizes to enforce when searching for examples.",
    )
    parser.add_argument(
        "--generation-limit",
        type=int,
        default=100,
        help="Maximum number of generations.",
    )
    parser.add_argument(
        "--revision-id",
        type=int,
        help="Revision ID for individual fix generation.",
    )
    parser.add_argument(
        "--diff-id",
        type=int,
        help="Diff ID for individual fix generation.",
    )
    parser.add_argument(
        "--comment-id",
        type=int,
        help="Comment ID for individual fix generation.",
    )

    return parser.parse_args(args)