test: cover provider label formatting
This commit is contained in:
parent
c2c37ef158
commit
b5a7e807d0
1 changed files with 11 additions and 0 deletions
|
|
@ -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:
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue