diff --git a/tests/hermes_cli/test_model_validation.py b/tests/hermes_cli/test_model_validation.py index 8b8f34444..f8ce868e2 100644 --- a/tests/hermes_cli/test_model_validation.py +++ b/tests/hermes_cli/test_model_validation.py @@ -7,6 +7,7 @@ from hermes_cli.models import ( fetch_api_models, normalize_provider, parse_model_input, + provider_label, provider_model_ids, validate_requested_model, ) @@ -111,6 +112,16 @@ class TestNormalizeProvider: assert normalize_provider("OpenRouter") == "openrouter" +class TestProviderLabel: + def test_known_labels_and_auto(self): + assert provider_label("anthropic") == "Anthropic" + assert provider_label("kimi") == "Kimi / Moonshot" + assert provider_label("auto") == "Auto" + + def test_unknown_provider_preserves_original_name(self): + assert provider_label("my-custom-provider") == "my-custom-provider" + + # -- provider_model_ids ------------------------------------------------------ class TestProviderModelIds: