"""FYN-LLM fitness test suite.

Run with: python -m pipeline.quality fitness
"""
from __future__ import annotations

import argparse
import sys

from config import get_settings
from pipeline.llm_client import LLMClient

from .fitness import FitnessRunner


def main() -> int:
    parser = argparse.ArgumentParser(
        prog="pipeline.quality",
        description="FYN-LLM quality assurance toolkit",
    )
    sub = parser.add_subparsers(dest="command")

    # -- fitness subcommand --
    sub.add_parser("fitness", help="Run LLM fitness tests across four categories")

    args = parser.parse_args()

    if args.command is None:
        parser.print_help()
        return 1

    if args.command == "fitness":
        settings = get_settings()
        client = LLMClient(settings)
        runner = FitnessRunner(client)
        return runner.run_all()

    return 0


if __name__ == "__main__":
    sys.exit(main())