#!/usr/bin/env python3
"""Idempotently add Gemma 4 12B Unified to ~/.hermes/config.yaml custom_providers."""

from __future__ import annotations

import argparse
from pathlib import Path

SNIPPET = """
  - name: gemma4-12b-it
    base_url: {base_url}
    api_key: EMPTY
    models:
      - google/gemma-4-12B-it
"""


def main() -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument("--config", default=str(Path.home() / ".hermes" / "config.yaml"))
    parser.add_argument("--base-url", default="http://127.0.0.1:8080/v1")
    parser.add_argument("--dry-run", action="store_true")
    args = parser.parse_args()
    path = Path(args.config)
    text = path.read_text(encoding="utf-8") if path.is_file() else "custom_providers: []\n"
    if "gemma4-12b-it" in text or "google/gemma-4-12B-it" in text:
        print(f"already registered in {path}")
        return 0
    block = SNIPPET.format(base_url=args.base_url.rstrip("/"))
    if "custom_providers:" in text:
        text = text.replace("custom_providers:\n", "custom_providers:\n" + block, 1)
    else:
        text = text.rstrip() + "\ncustom_providers:\n" + block
    if args.dry_run:
        print(block)
        return 0
    path.parent.mkdir(parents=True, exist_ok=True)
    path.write_text(text, encoding="utf-8")
    print(f"added gemma4-12b-it -> {args.base_url} in {path}")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())