Cleanup, use ModelCapabilities only

This commit is contained in:
Fahad
2025-06-23 17:39:47 +04:00
parent 498ea88293
commit 14eaf930ed
5 changed files with 47 additions and 84 deletions

View File

@@ -57,7 +57,7 @@ class TestOpenRouterProvider:
caps = provider.get_capabilities("o3")
assert caps.provider == ProviderType.OPENROUTER
assert caps.model_name == "openai/o3" # Resolved name
assert caps.friendly_name == "OpenRouter"
assert caps.friendly_name == "OpenRouter (openai/o3)"
# Test with a model not in registry - should get generic capabilities
caps = provider.get_capabilities("unknown-model")