test: cover provider label formatting

This commit is contained in:
teknium1
2026-03-14 03:39:12 -07:00
parent c2c37ef158
commit b5a7e807d0

View File

@@ -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: