91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
"""The llamacpp provider row in the model picker payload.
|
|
|
|
Contract: staged local GGUFs appear as a selectable provider row in
|
|
build_models_payload — the same payload /api/model/options and the desktop
|
|
picker consume — whenever models are staged, without any credential."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def hermes_home(tmp_path, monkeypatch):
|
|
home = tmp_path / ".hermes"
|
|
home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(home))
|
|
return home
|
|
|
|
|
|
def _stage(home, *names):
|
|
mdir = home / "models"
|
|
mdir.mkdir(exist_ok=True)
|
|
for name in names:
|
|
(mdir / f"{name}.gguf").write_bytes(b"GGUF" + b"\x00" * 64)
|
|
|
|
|
|
def test_no_staged_models_no_row(hermes_home):
|
|
from hermes_cli.inventory import _local_runtime_row, load_picker_context
|
|
|
|
assert _local_runtime_row(load_picker_context()) is None
|
|
|
|
|
|
def test_staged_models_make_a_selectable_row(hermes_home):
|
|
from hermes_cli.inventory import _local_runtime_row, load_picker_context
|
|
|
|
_stage(hermes_home, "Qwen3-4B-Instruct-2507-UD-Q8_K_XL", "Some-Other-Model")
|
|
row = _local_runtime_row(load_picker_context())
|
|
assert row is not None
|
|
assert row["slug"] == "llamacpp"
|
|
assert row["authenticated"] is True
|
|
assert "Qwen3-4B-Instruct-2507-UD-Q8_K_XL" in row["models"]
|
|
assert row["total_models"] == 2
|
|
|
|
|
|
def test_row_marks_current_when_config_points_at_llamacpp(hermes_home):
|
|
from hermes_cli.inventory import _local_runtime_row, load_picker_context
|
|
|
|
_stage(hermes_home, "M")
|
|
ctx = load_picker_context().with_overrides(current_provider="llamacpp")
|
|
row = _local_runtime_row(ctx)
|
|
assert row is not None and row["is_current"] is True
|
|
|
|
|
|
def test_full_payload_includes_local_row(hermes_home):
|
|
"""Through the REAL payload builder — the shape the desktop picker eats."""
|
|
from hermes_cli.inventory import build_models_payload, load_picker_context
|
|
|
|
_stage(hermes_home, "Local-Model-X")
|
|
payload = build_models_payload(
|
|
load_picker_context(),
|
|
probe_custom_providers=False,
|
|
probe_current_custom_provider=False,
|
|
)
|
|
slugs = [p["slug"] for p in payload["providers"]]
|
|
assert "llamacpp" in slugs
|
|
row = payload["providers"][slugs.index("llamacpp")]
|
|
assert row["models"] == ["Local-Model-X"]
|
|
|
|
|
|
def test_explicit_only_filter_keeps_local_row_on_any_profile(hermes_home):
|
|
"""The desktop dropdown requests explicit_only=True, and the local row
|
|
has no config credential by design (credential is reachability). The
|
|
filter must treat staged models as explicit configuration — otherwise
|
|
the row only survives on the profile whose config points at llamacpp,
|
|
and every other profile's dropdown silently loses local models."""
|
|
from hermes_cli.inventory import _filter_explicit_provider_rows, _local_runtime_row, load_picker_context
|
|
|
|
_stage(hermes_home, "Qwen3.8-27B-UD-Q5_K_XL")
|
|
ctx = load_picker_context()
|
|
row = _local_runtime_row(ctx)
|
|
assert row is not None
|
|
|
|
# Simulate a profile whose current provider is a cloud one (the normal
|
|
# profile's shape): explicit-only filtering must keep the local row.
|
|
import dataclasses
|
|
|
|
ctx = dataclasses.replace(ctx, current_provider="anthropic")
|
|
kept = _filter_explicit_provider_rows([row], ctx)
|
|
assert kept, "explicit-only filter dropped the local-runtime row"
|
|
assert kept[0]["slug"] == "llamacpp"
|