File size: 1,426 Bytes
676f5d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | #!/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())
|