Orion
516 lines
17 KiB
Python
Executable File
516 lines
17 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
"""Interactive Hermes model/provider switcher.
|
||
|
||
Secrets are entered with hidden prompts and stored only in Hermes' local .env
|
||
as environment variables referenced from config.yaml. Do not commit .env.
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import getpass
|
||
import os
|
||
import re
|
||
import shutil
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
from typing import Any
|
||
|
||
try:
|
||
import yaml
|
||
except ImportError: # pragma: no cover
|
||
print("缺少 PyYAML：请先安装 `python3 -m pip install pyyaml`", file=sys.stderr)
|
||
sys.exit(1)
|
||
|
||
|
||
PROVIDERS: list[dict[str, Any]] = [
|
||
{
|
||
"id": "custom",
|
||
"name": "Custom / OpenAI-compatible endpoint",
|
||
"models": ["gpt-5.5", "gpt-5.1", "claude-sonnet-4.5", "gemini-2.5-pro", "qwen3-coder-plus"],
|
||
"base_url": "",
|
||
"api_mode": "auto",
|
||
"secret_env": "HERMES_MODEL_API_KEY",
|
||
"notes": "任意 OpenAI 兼容中转/代理；最通用。",
|
||
},
|
||
{
|
||
"id": "openrouter",
|
||
"name": "OpenRouter",
|
||
"models": ["openai/gpt-5.1", "openai/gpt-5", "anthropic/claude-sonnet-4.5", "google/gemini-2.5-pro", "deepseek/deepseek-chat-v3.1"],
|
||
"base_url": "https://openrouter.ai/api/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "OPENROUTER_API_KEY",
|
||
},
|
||
{
|
||
"id": "openai-api",
|
||
"name": "OpenAI API",
|
||
"models": ["gpt-5.1", "gpt-5", "gpt-5-mini", "gpt-4.1", "o4-mini"],
|
||
"base_url": "https://api.openai.com/v1",
|
||
"api_mode": "codex_responses",
|
||
"secret_env": "OPENAI_API_KEY",
|
||
},
|
||
{
|
||
"id": "anthropic",
|
||
"name": "Anthropic Claude",
|
||
"models": ["claude-sonnet-4-5", "claude-opus-4-1", "claude-haiku-4-5", "claude-3-7-sonnet-latest"],
|
||
"base_url": "https://api.anthropic.com",
|
||
"api_mode": "anthropic_messages",
|
||
"secret_env": "ANTHROPIC_API_KEY",
|
||
},
|
||
{
|
||
"id": "google",
|
||
"name": "Google Gemini API / OpenAI-compatible",
|
||
"models": ["gemini-2.5-pro", "gemini-2.5-flash", "gemini-2.0-flash"],
|
||
"base_url": "https://generativelanguage.googleapis.com/v1beta/openai/",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "GEMINI_API_KEY",
|
||
},
|
||
{
|
||
"id": "deepseek",
|
||
"name": "DeepSeek",
|
||
"models": ["deepseek-chat", "deepseek-reasoner"],
|
||
"base_url": "https://api.deepseek.com/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "DEEPSEEK_API_KEY",
|
||
},
|
||
{
|
||
"id": "alibaba",
|
||
"name": "Alibaba DashScope / Qwen",
|
||
"models": ["qwen3-coder-plus", "qwen-plus", "qwen-max", "qwen-turbo"],
|
||
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "DASHSCOPE_API_KEY",
|
||
},
|
||
{
|
||
"id": "zai",
|
||
"name": "Z.AI / GLM",
|
||
"models": ["glm-4.6", "glm-4.5", "glm-4-plus"],
|
||
"base_url": "https://open.bigmodel.cn/api/paas/v4",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "GLM_API_KEY",
|
||
},
|
||
{
|
||
"id": "xai",
|
||
"name": "xAI Grok",
|
||
"models": ["grok-4", "grok-code-fast-1", "grok-3"],
|
||
"base_url": "https://api.x.ai/v1",
|
||
"api_mode": "codex_responses",
|
||
"secret_env": "XAI_API_KEY",
|
||
},
|
||
{
|
||
"id": "kimi-for-coding",
|
||
"name": "Kimi / Moonshot Coding",
|
||
"models": ["kimi-k2", "kimi-latest", "kimi-thinking-preview"],
|
||
"base_url": "https://api.kimi.com/coding",
|
||
"api_mode": "anthropic_messages",
|
||
"secret_env": "KIMI_API_KEY",
|
||
},
|
||
{
|
||
"id": "minimax",
|
||
"name": "MiniMax Global",
|
||
"models": ["MiniMax-M2", "MiniMax-Text-01"],
|
||
"base_url": "https://api.minimax.io/anthropic",
|
||
"api_mode": "anthropic_messages",
|
||
"secret_env": "MINIMAX_API_KEY",
|
||
},
|
||
{
|
||
"id": "minimax-cn",
|
||
"name": "MiniMax CN",
|
||
"models": ["MiniMax-M2", "MiniMax-Text-01"],
|
||
"base_url": "https://api.minimax.chat/anthropic",
|
||
"api_mode": "anthropic_messages",
|
||
"secret_env": "MINIMAX_CN_API_KEY",
|
||
},
|
||
{
|
||
"id": "vercel",
|
||
"name": "Vercel AI Gateway",
|
||
"models": ["openai/gpt-5.1", "anthropic/claude-sonnet-4.5", "google/gemini-2.5-pro"],
|
||
"base_url": "https://ai-gateway.vercel.sh/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "AI_GATEWAY_API_KEY",
|
||
},
|
||
{
|
||
"id": "opencode",
|
||
"name": "OpenCode Zen",
|
||
"models": ["anthropic/claude-sonnet-4.5", "openai/gpt-5.1"],
|
||
"base_url": "https://opencode.ai/zen/v1",
|
||
"api_mode": "auto",
|
||
"secret_env": "OPENCODE_ZEN_API_KEY",
|
||
},
|
||
{
|
||
"id": "opencode-go",
|
||
"name": "OpenCode Go",
|
||
"models": ["anthropic/claude-sonnet-4.5", "openai/gpt-5.1"],
|
||
"base_url": "https://opencode.ai/go/v1",
|
||
"api_mode": "auto",
|
||
"secret_env": "OPENCODE_GO_API_KEY",
|
||
},
|
||
{
|
||
"id": "kilo",
|
||
"name": "Kilo Code Gateway",
|
||
"models": ["anthropic/claude-sonnet-4.5", "openai/gpt-5.1", "google/gemini-2.5-pro"],
|
||
"base_url": "https://kilocode.ai/api/openrouter/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "KILOCODE_API_KEY",
|
||
},
|
||
{
|
||
"id": "huggingface",
|
||
"name": "Hugging Face Inference Providers",
|
||
"models": ["openai/gpt-oss-120b", "meta-llama/Llama-3.3-70B-Instruct"],
|
||
"base_url": "https://router.huggingface.co/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "HF_TOKEN",
|
||
},
|
||
{
|
||
"id": "novita",
|
||
"name": "Novita AI",
|
||
"models": ["deepseek/deepseek-v3.1", "qwen/qwen3-coder-480b-a35b-instruct"],
|
||
"base_url": "https://api.novita.ai/v3/openai",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "NOVITA_API_KEY",
|
||
},
|
||
{
|
||
"id": "nvidia",
|
||
"name": "NVIDIA NIM",
|
||
"models": ["openai/gpt-oss-120b", "nvidia/llama-3.1-nemotron-ultra-253b-v1"],
|
||
"base_url": "https://integrate.api.nvidia.com/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "NVIDIA_API_KEY",
|
||
},
|
||
{
|
||
"id": "gmi",
|
||
"name": "GMI Cloud",
|
||
"models": ["deepseek-ai/DeepSeek-V3.1", "Qwen/Qwen3-Coder-480B-A35B-Instruct"],
|
||
"base_url": "https://api.gmi-serving.com/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "GMI_API_KEY",
|
||
},
|
||
{
|
||
"id": "arcee",
|
||
"name": "Arcee AI",
|
||
"models": ["arcee-ai/coder-large", "arcee-ai/spotlight"],
|
||
"base_url": "https://api.arcee.ai/api/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "ARCEE_API_KEY",
|
||
},
|
||
{
|
||
"id": "xiaomi",
|
||
"name": "Xiaomi MiMo",
|
||
"models": ["mimo-vl-7b", "mimo-7b"],
|
||
"base_url": "https://api.mimo.mi.com/openai/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "XIAOMI_API_KEY",
|
||
},
|
||
{
|
||
"id": "lmstudio",
|
||
"name": "LM Studio Local",
|
||
"models": ["local-model"],
|
||
"base_url": "http://127.0.0.1:1234/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "LM_API_KEY",
|
||
"allow_blank_key": True,
|
||
},
|
||
{
|
||
"id": "ollama-cloud",
|
||
"name": "Ollama Cloud / Local OpenAI-compatible",
|
||
"models": ["gpt-oss:120b", "llama3.3:70b", "qwen3-coder"],
|
||
"base_url": "http://127.0.0.1:11434/v1",
|
||
"api_mode": "chat_completions",
|
||
"secret_env": "OLLAMA_API_KEY",
|
||
"allow_blank_key": True,
|
||
},
|
||
]
|
||
|
||
API_MODES = [
|
||
("auto", "自动/不写入 api_mode"),
|
||
("chat_completions", "OpenAI Chat Completions /chat/completions"),
|
||
("codex_responses", "OpenAI Responses /responses，常用于 GPT-5.x、xAI"),
|
||
("anthropic_messages", "Anthropic Messages /v1/messages"),
|
||
("bedrock_converse", "AWS Bedrock Converse"),
|
||
]
|
||
|
||
ENV_NAME_RE = re.compile(r"^[A-Za-z_][A-Za-z0-9_]*$")
|
||
DOTENV_SAFE_VALUE_RE = re.compile(r"^[A-Za-z0-9_./:@%+=,\-]+$")
|
||
|
||
|
||
def hermes_home() -> Path:
|
||
return Path(os.environ.get("HERMES_HOME") or Path.home() / ".hermes").expanduser()
|
||
|
||
|
||
def config_path() -> Path:
|
||
env_path = os.environ.get("HERMES_CONFIG")
|
||
return Path(env_path).expanduser() if env_path else hermes_home() / "config.yaml"
|
||
|
||
|
||
def env_path() -> Path:
|
||
env_path_override = os.environ.get("HERMES_ENV_PATH")
|
||
return Path(env_path_override).expanduser() if env_path_override else hermes_home() / ".env"
|
||
|
||
|
||
def prompt(text: str, default: str = "") -> str:
|
||
suffix = f" [{default}]" if default else ""
|
||
try:
|
||
value = input(f"{text}{suffix}: ").strip()
|
||
except EOFError:
|
||
print()
|
||
return default
|
||
return value if value else default
|
||
|
||
|
||
def choose(title: str, options: list[str], default_index: int = 0) -> int:
|
||
print(f"\n{title}")
|
||
for index, option in enumerate(options, 1):
|
||
marker = "*" if index - 1 == default_index else " "
|
||
print(f" {marker} {index:2d}) {option}")
|
||
while True:
|
||
try:
|
||
raw = input(f"请选择 1-{len(options)} [{default_index + 1}]: ").strip()
|
||
except EOFError:
|
||
print()
|
||
return default_index
|
||
if not raw:
|
||
return default_index
|
||
if raw.isdigit() and 1 <= int(raw) <= len(options):
|
||
return int(raw) - 1
|
||
print("输入无效，请重新选择。")
|
||
|
||
|
||
def confirm(text: str, default: bool = True) -> bool:
|
||
suffix = "Y/n" if default else "y/N"
|
||
while True:
|
||
try:
|
||
raw = input(f"{text} [{suffix}]: ").strip().lower()
|
||
except EOFError:
|
||
print()
|
||
return default
|
||
if not raw:
|
||
return default
|
||
if raw in {"y", "yes", "1", "true", "是"}:
|
||
return True
|
||
if raw in {"n", "no", "0", "false", "否"}:
|
||
return False
|
||
print("请输入 y 或 n。")
|
||
|
||
|
||
def safe_env_name(value: str, fallback: str) -> str:
|
||
name = re.sub(r"[^A-Za-z0-9_]+", "_", value).upper().strip("_")
|
||
if not name or not name[0].isalpha():
|
||
name = fallback
|
||
return name
|
||
|
||
|
||
def validate_env_name(value: str) -> str:
|
||
if not ENV_NAME_RE.fullmatch(value):
|
||
raise SystemExit(f"环境变量名无效：{value!r}。请使用类似 OPENAI_API_KEY 的格式。")
|
||
return value
|
||
|
||
|
||
def ensure_mapping(parent: dict[str, Any], key: str) -> dict[str, Any]:
|
||
value = parent.get(key)
|
||
if isinstance(value, dict):
|
||
return value
|
||
if value is not None:
|
||
print(f"配置项 {key!r} 不是 YAML 字典，已重置。", file=sys.stderr)
|
||
value = {}
|
||
parent[key] = value
|
||
return value
|
||
|
||
|
||
def load_yaml(path: Path) -> dict[str, Any]:
|
||
if not path.exists():
|
||
return {}
|
||
with path.open("r", encoding="utf-8-sig") as handle:
|
||
data = yaml.safe_load(handle) or {}
|
||
if not isinstance(data, dict):
|
||
raise SystemExit(f"配置文件不是 YAML 字典：{path}")
|
||
return data
|
||
|
||
|
||
def save_yaml(path: Path, data: dict[str, Any]) -> None:
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
tmp = path.with_suffix(path.suffix + ".tmp")
|
||
with tmp.open("w", encoding="utf-8") as handle:
|
||
yaml.safe_dump(data, handle, allow_unicode=True, sort_keys=False)
|
||
os.replace(tmp, path)
|
||
try:
|
||
path.chmod(0o600)
|
||
except OSError:
|
||
pass
|
||
|
||
|
||
def upsert_env(path: Path, key: str, value: str) -> None:
|
||
validate_env_name(key)
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
lines: list[str] = []
|
||
if path.exists():
|
||
lines = path.read_text(encoding="utf-8-sig", errors="replace").splitlines()
|
||
rendered = f"{key}={render_dotenv_value(value)}"
|
||
pattern = re.compile(rf"^\s*(?:export\s+)?{re.escape(key)}\s*=")
|
||
replaced = False
|
||
out: list[str] = []
|
||
for line in lines:
|
||
if pattern.match(line):
|
||
out.append(rendered)
|
||
replaced = True
|
||
else:
|
||
out.append(line)
|
||
if not replaced:
|
||
out.append(rendered)
|
||
path.write_text("\n".join(out).rstrip() + "\n", encoding="utf-8")
|
||
try:
|
||
path.chmod(0o600)
|
||
except OSError:
|
||
pass
|
||
|
||
|
||
def render_dotenv_value(value: str) -> str:
|
||
if DOTENV_SAFE_VALUE_RE.fullmatch(value):
|
||
return value
|
||
escaped = (
|
||
value.replace("\\", "\\\\")
|
||
.replace('"', '\\"')
|
||
.replace("$", "\\$")
|
||
.replace("\n", "\\n")
|
||
.replace("\r", "\\r")
|
||
)
|
||
return f'"{escaped}"'
|
||
|
||
|
||
def masked_env_ref(env_name: str) -> str:
|
||
return "${" + env_name + "}"
|
||
|
||
|
||
def maybe_restart_gateway() -> None:
|
||
if not shutil.which("hermes"):
|
||
print("未找到 hermes 命令，跳过 gateway restart。")
|
||
return
|
||
if not confirm("是否重启 Hermes gateway 让 Telegram/微信等立即生效？", True):
|
||
return
|
||
result = subprocess.run(["hermes", "gateway", "restart"], text=True)
|
||
if result.returncode != 0:
|
||
print("gateway restart 失败；你可以稍后手动运行 `hermes gateway restart`。", file=sys.stderr)
|
||
|
||
|
||
def main() -> int:
|
||
print("Hermes 模型/Provider 一键交互切换")
|
||
print("说明：API Key 隐藏输入，写入本机 ~/.hermes/.env；config.yaml 只保存 ${ENV_VAR} 引用。")
|
||
|
||
cfg_file = Path(prompt("config.yaml 路径", str(config_path()))).expanduser()
|
||
dotenv_file = Path(prompt(".env 路径", str(env_path()))).expanduser()
|
||
|
||
provider_labels = [f"{item['id']} — {item['name']}" for item in PROVIDERS]
|
||
provider = PROVIDERS[choose("选择提供商", provider_labels, 0)]
|
||
|
||
model_options = list(provider["models"]) + ["手动输入其它模型名"]
|
||
model_index = choose("选择模型", model_options, 0)
|
||
model = prompt("模型名") if model_index == len(model_options) - 1 else model_options[model_index]
|
||
if not model:
|
||
raise SystemExit("模型名不能为空。")
|
||
|
||
default_base = str(provider.get("base_url") or "")
|
||
base_url = prompt("Base URL（回车使用默认；输入 - 清空）", default_base)
|
||
if base_url.lower() in {"-", "none", "null"}:
|
||
base_url = ""
|
||
|
||
default_mode = str(provider.get("api_mode") or "auto")
|
||
mode_index = next((i for i, (mode, _) in enumerate(API_MODES) if mode == default_mode), 0)
|
||
api_mode = API_MODES[choose("选择 API 协议模式", [f"{m} — {label}" for m, label in API_MODES], mode_index)][0]
|
||
|
||
default_secret_env = str(provider.get("secret_env") or safe_env_name(provider["id"], "HERMES_MODEL") + "_API_KEY")
|
||
secret_env = validate_env_name(prompt("API Key 保存到哪个环境变量名", default_secret_env))
|
||
|
||
allow_blank_key = bool(provider.get("allow_blank_key"))
|
||
api_key = getpass.getpass("API Key（隐藏输入；留空=不修改现有值）: ").strip()
|
||
if api_key:
|
||
api_key_ref = masked_env_ref(secret_env)
|
||
elif allow_blank_key:
|
||
api_key_ref = ""
|
||
else:
|
||
api_key_ref = masked_env_ref(secret_env)
|
||
print(f"未输入新 key，将保留/引用现有 {secret_env}。")
|
||
|
||
context_length_raw = prompt("context_length（可选，留空则清除现有值）")
|
||
service_tier = prompt("service_tier（可选，例如 flex/default，留空则清除现有值）")
|
||
|
||
cfg = load_yaml(cfg_file)
|
||
model_cfg = ensure_mapping(cfg, "model")
|
||
|
||
model_cfg["default"] = model
|
||
model_cfg["provider"] = provider["id"]
|
||
if base_url:
|
||
model_cfg["base_url"] = base_url.rstrip("/")
|
||
else:
|
||
model_cfg.pop("base_url", None)
|
||
if api_key_ref:
|
||
model_cfg["api_key"] = api_key_ref
|
||
elif allow_blank_key:
|
||
model_cfg.pop("api_key", None)
|
||
if api_mode and api_mode != "auto":
|
||
model_cfg["api_mode"] = api_mode
|
||
else:
|
||
model_cfg.pop("api_mode", None)
|
||
if context_length_raw:
|
||
try:
|
||
model_cfg["context_length"] = int(context_length_raw.replace(",", ""))
|
||
except ValueError:
|
||
print("context_length 不是数字，已保留原值。")
|
||
else:
|
||
model_cfg.pop("context_length", None)
|
||
if service_tier:
|
||
ensure_mapping(cfg, "agent")["service_tier"] = service_tier
|
||
else:
|
||
agent_cfg = cfg.get("agent")
|
||
if isinstance(agent_cfg, dict):
|
||
agent_cfg.pop("service_tier", None)
|
||
|
||
if confirm("是否同时把 delegation 子代理模型配置成一样？", True):
|
||
delegation = ensure_mapping(cfg, "delegation")
|
||
delegation["model"] = model
|
||
delegation["provider"] = provider["id"]
|
||
if base_url:
|
||
delegation["base_url"] = base_url.rstrip("/")
|
||
else:
|
||
delegation.pop("base_url", None)
|
||
if api_key_ref:
|
||
delegation["api_key"] = api_key_ref
|
||
elif allow_blank_key:
|
||
delegation.pop("api_key", None)
|
||
if api_mode and api_mode != "auto":
|
||
delegation["api_mode"] = api_mode
|
||
else:
|
||
delegation.pop("api_mode", None)
|
||
|
||
print("\n即将写入：")
|
||
print(f" provider: {provider['id']}")
|
||
print(f" model: {model}")
|
||
print(f" base_url: {base_url or '(empty)'}")
|
||
print(f" api_key: {api_key_ref or '(empty/no key)'}")
|
||
print(f" api_mode: {api_mode}")
|
||
print(f" config: {cfg_file}")
|
||
print(f" env: {dotenv_file}")
|
||
|
||
if not confirm("确认写入？", True):
|
||
print("已取消。")
|
||
return 1
|
||
|
||
try:
|
||
if api_key:
|
||
upsert_env(dotenv_file, secret_env, api_key)
|
||
print(f"API Key 已写入 {dotenv_file}。")
|
||
save_yaml(cfg_file, cfg)
|
||
except OSError as exc:
|
||
print(f"写入失败：{exc}", file=sys.stderr)
|
||
return 1
|
||
print("配置已写入。")
|
||
maybe_restart_gateway()
|
||
print("完成。")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
raise SystemExit(main())
|
||
except KeyboardInterrupt:
|
||
print("\n已取消。", file=sys.stderr)
|
||
raise SystemExit(130)
|