mirror of
https://github.com/nesquena/hermes-webui.git
synced 2026-07-18 21:50:55 +00:00
sbe27
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
"""#5311 / #5611: OpenCode Go's generic live probe used to return public-catalog
|
|
models not enabled on the Go tier, so the picker intentionally falls back to a curated
|
|
static _PROVIDER_MODELS list. Keep that curated list in sync with the public Go docs
|
|
and documented Go models endpoint while excluding preview/free-only Zen models.
|
|
"""
|
|
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
CONFIG_PATH = ROOT / "api" / "config.py"
|
|
CONFIG = CONFIG_PATH.read_text(encoding="utf-8")
|
|
|
|
EXPECTED_OPENCODE_GO_MODEL_IDS = [
|
|
"minimax-m3",
|
|
"minimax-m2.7",
|
|
"minimax-m2.5",
|
|
"kimi-k2.7-code",
|
|
"kimi-k2.6",
|
|
"kimi-k2.5",
|
|
"glm-5.2",
|
|
"glm-5.1",
|
|
"glm-5",
|
|
"deepseek-v4-pro",
|
|
"deepseek-v4-flash",
|
|
"qwen3.7-max",
|
|
"qwen3.7-plus",
|
|
"qwen3.6-plus",
|
|
"qwen3.5-plus",
|
|
"mimo-v2-pro",
|
|
"mimo-v2-omni",
|
|
"mimo-v2.5-pro",
|
|
"mimo-v2.5",
|
|
]
|
|
|
|
|
|
def _opencode_go_static_models():
|
|
tree = ast.parse(CONFIG, filename=str(CONFIG_PATH))
|
|
for node in tree.body:
|
|
if not isinstance(node, ast.Assign):
|
|
continue
|
|
if not any(isinstance(target, ast.Name) and target.id == "_PROVIDER_MODELS" for target in node.targets):
|
|
continue
|
|
provider_models = ast.literal_eval(node.value)
|
|
return provider_models["opencode-go"]
|
|
raise AssertionError("_PROVIDER_MODELS assignment not found")
|
|
|
|
|
|
def test_opencode_go_skips_live_models_probe():
|
|
# The provider-loop must special-case opencode-go to skip the old generic
|
|
# live probe and fall through to this curated static list.
|
|
body = CONFIG[CONFIG.index("def get_available_models"):]
|
|
body = body[: body.index("\ndef ", 1)]
|
|
assert 'elif pid == "opencode-go":' in body
|
|
idx = body.index('elif pid == "opencode-go":')
|
|
branch = body[idx: idx + 400]
|
|
assert "_models_from_live_provider_ids" not in branch.split("else:")[0]
|
|
|
|
|
|
def test_opencode_go_static_models_match_documented_endpoint_snapshot():
|
|
# Snapshot from https://opencode.ai/zen/go/v1/models on 2026-07-07.
|
|
# This catches stale-list regressions such as missing GLM-5.2 / MiniMax M3.
|
|
models = _opencode_go_static_models()
|
|
assert [model["id"] for model in models] == EXPECTED_OPENCODE_GO_MODEL_IDS
|
|
|
|
|
|
def test_opencode_go_recent_additions_have_human_labels():
|
|
labels = {model["id"]: model["label"] for model in _opencode_go_static_models()}
|
|
assert labels["glm-5.2"] == "GLM-5.2"
|
|
assert labels["minimax-m3"] == "MiniMax M3"
|
|
assert labels["kimi-k2.7-code"] == "Kimi K2.7 Code"
|
|
assert labels["qwen3.7-max"] == "Qwen3.7 Max"
|