a6deb961ea
Keep the Use-button case (staged but absent from the spawn-only live listing is accepted) and the control (never staged stays rejected); the other five re-asserted the same branch from different angles.
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
"""Tests for the managed llama.cpp runtime branch in /model validation (#115237)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
from hermes_cli.models_validate import validate_requested_model
|
|
|
|
|
|
def _validate(model, staged=("Qwen3-4B-Q4_K_M",), live=("other-model",), **kw):
|
|
"""validate against a managed-local world: staged files on disk + a live listing that
|
|
(spawn-only as it is) hasn't learned the new file yet."""
|
|
with (
|
|
patch("hermes_cli.models.fetch_api_models", return_value=list(live)),
|
|
patch(
|
|
"hermes_cli.models.probe_api_models",
|
|
return_value={
|
|
"models": list(live),
|
|
"probed_url": "http://127.0.0.1:18434/v1/models",
|
|
"resolved_base_url": "http://127.0.0.1:18434/v1",
|
|
"suggested_base_url": None,
|
|
"used_fallback": False,
|
|
},
|
|
),
|
|
patch(
|
|
"hermes_cli.local_runtime.bootstrap.staged_model_ids",
|
|
return_value=list(staged),
|
|
),
|
|
):
|
|
return validate_requested_model(
|
|
model, "llamacpp", base_url="http://127.0.0.1:18434/v1", **kw
|
|
)
|
|
|
|
|
|
class TestManagedLocalValidation:
|
|
def test_staged_but_not_live_accepted(self):
|
|
"""The Use button's exact case: downloaded, not yet in the spawn-only live listing."""
|
|
result = _validate("Qwen3-4B-Q4_K_M")
|
|
assert (result["accepted"], result["persist"], result["recognized"]) == (
|
|
True,
|
|
True,
|
|
True,
|
|
)
|
|
assert "managed local-models library" in result["message"]
|
|
|
|
def test_not_staged_and_not_live_still_rejected(self):
|
|
result = _validate("Llama-4-90B-Q4_K_M", staged=())
|
|
assert result["accepted"] is False
|
|
assert "was not found in this provider's model listing" in result["message"]
|