Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/hermes_cli/test_local_runtime_picker_row.py

@@ -0,0 +1,90 @@
+"""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"