"""Parser setup for the 'evaluate-refusal' command."""


def setup_evaluate_refusal_parser(parser):
    """Set up the evaluate-refusal command parser."""
    parser.add_argument("--model", type=str, required=True, help="Model to evaluate (path or HuggingFace model name)")
    parser.add_argument("--prompts", type=str, help="JSON file with prompts to test (list of strings)")
    parser.add_argument("--output", type=str, help="Output JSON file for evaluation results")
    parser.add_argument("--threshold", type=float, default=0.35, help="Semantic similarity threshold for refusal detection (default: 0.35)")
    parser.add_argument("--max-new-tokens", type=int, default=150, help="Maximum tokens to generate per response (default: 150)")
    parser.add_argument("--num-prompts", type=int, default=20, help="Number of built-in prompts to use if --prompts not provided (default: 20)")
    parser.add_argument("--verbose", action="store_true", help="Verbose output showing each response")
