test: cover provider label formatting
This commit is contained in:
@@ -7,6 +7,7 @@ from hermes_cli.models import (
|
|||||||
fetch_api_models,
|
fetch_api_models,
|
||||||
normalize_provider,
|
normalize_provider,
|
||||||
parse_model_input,
|
parse_model_input,
|
||||||
|
provider_label,
|
||||||
provider_model_ids,
|
provider_model_ids,
|
||||||
validate_requested_model,
|
validate_requested_model,
|
||||||
)
|
)
|
||||||
@@ -111,6 +112,16 @@ class TestNormalizeProvider:
|
|||||||
assert normalize_provider("OpenRouter") == "openrouter"
|
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 ------------------------------------------------------
|
# -- provider_model_ids ------------------------------------------------------
|
||||||
|
|
||||||
class TestProviderModelIds:
|
class TestProviderModelIds:
|
||||||
|
|||||||
Reference in New Issue
Block a user