feat(models): add new models for SiliconFlow and OpenRouter providers

test(models): update tests to validate new providers and registered models
This commit is contained in:
X-iZhang
2026-02-27 14:38:22 +00:00
parent c9e5d925b8
commit d81d41f756
3 changed files with 43 additions and 58 deletions
+5 -3
View File
@@ -33,7 +33,7 @@ class TestModelsRegistry:
def test_entries_are_valid_tuples(self):
"""Test that _MODEL_ENTRIES contains valid (name, model_id, provider) tuples."""
valid_providers = {"anthropic", "openai", "google-genai", "nvidia"}
valid_providers = {"anthropic", "openai", "google-genai", "nvidia", "siliconflow", "openrouter"}
for entry in _MODEL_ENTRIES:
assert len(entry) == 3, f"Entry {entry} doesn't have 3 elements"
name, model_id, provider = entry
@@ -49,9 +49,11 @@ class TestModelsRegistry:
assert isinstance(name, str)
assert isinstance(model_id, str)
# Third-party providers have no registered models (user types model name)
# Third-party providers now have registered models
openrouter_models = get_models_for_provider("openrouter")
assert len(openrouter_models) == 0
assert len(openrouter_models) > 0
siliconflow_models = get_models_for_provider("siliconflow")
assert len(siliconflow_models) > 0
# =============================================================================