| |
| """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()) |
|
|
