mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-19 15:18:03 +00:00
2b5d4ae916
Preserve the root cause and precedence direction from #43538 while applying the merge before truncation and covering all declared model shapes. Co-authored-by: liuhao1024 <sunsky.lau@gmail.com>
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
"""Configured models extend built-in picker rows."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from hermes_cli.model_switch import list_authenticated_providers
|
|
|
|
|
|
def _provider_row(configured_models, *, max_models=None):
|
|
with (
|
|
patch(
|
|
"agent.models_dev.fetch_models_dev",
|
|
return_value={"deepseek": {"env": ["DEEPSEEK_API_KEY"], "name": "DeepSeek"}},
|
|
),
|
|
patch(
|
|
"agent.models_dev.PROVIDER_TO_MODELS_DEV",
|
|
{"deepseek": "deepseek"},
|
|
),
|
|
patch(
|
|
"hermes_cli.models.cached_provider_model_ids",
|
|
return_value=["live-a", "shared"],
|
|
),
|
|
patch("hermes_cli.providers.HERMES_OVERLAYS", {}),
|
|
patch.dict("os.environ", {"DEEPSEEK_API_KEY": "test-key"}),
|
|
):
|
|
rows = list_authenticated_providers(
|
|
current_provider="deepseek",
|
|
user_providers={"deepseek": {"models": configured_models}},
|
|
max_models=max_models,
|
|
)
|
|
return next(row for row in rows if row["slug"] == "deepseek")
|
|
|
|
|
|
def test_configured_models_precede_and_deduplicate_discovered_models():
|
|
row = _provider_row({"configured-x": {}, "shared": {}})
|
|
|
|
assert row["models"] == ["configured-x", "shared", "live-a"]
|
|
assert row["total_models"] == 3
|
|
|
|
|
|
def test_configured_models_are_merged_before_picker_limit():
|
|
row = _provider_row(["configured-x", "configured-y"], max_models=2)
|
|
|
|
assert row["models"] == ["configured-x", "configured-y"]
|
|
assert row["total_models"] == 4
|