test: update tests to match new Claude Sonnet 4.5 alias configuration

- Updated sonnet alias to point to claude-sonnet-4.5 instead of 4.1
- Removed references to deprecated 'claude' alias
- Added sonnet4.1 alias for claude-sonnet-4.1 backwards compatibility
- All 809 tests passing
This commit is contained in:
Fahad
2025-10-01 19:57:43 +04:00
parent c0f822ffa2
commit 7efb4094d4
2 changed files with 15 additions and 9 deletions

View File

@@ -72,22 +72,23 @@ class TestOpenRouterProvider:
# Test alias resolution # Test alias resolution
assert provider._resolve_model_name("opus") == "anthropic/claude-opus-4.1" assert provider._resolve_model_name("opus") == "anthropic/claude-opus-4.1"
assert provider._resolve_model_name("sonnet") == "anthropic/claude-sonnet-4.1" assert provider._resolve_model_name("sonnet") == "anthropic/claude-sonnet-4.5"
assert provider._resolve_model_name("sonnet4.1") == "anthropic/claude-sonnet-4.1"
assert provider._resolve_model_name("o3") == "openai/o3" assert provider._resolve_model_name("o3") == "openai/o3"
assert provider._resolve_model_name("o3-mini") == "openai/o3-mini" assert provider._resolve_model_name("o3-mini") == "openai/o3-mini"
assert provider._resolve_model_name("o3mini") == "openai/o3-mini" assert provider._resolve_model_name("o3mini") == "openai/o3-mini"
assert provider._resolve_model_name("o4-mini") == "openai/o4-mini" assert provider._resolve_model_name("o4-mini") == "openai/o4-mini"
assert provider._resolve_model_name("o4-mini") == "openai/o4-mini" assert provider._resolve_model_name("o4-mini") == "openai/o4-mini"
assert provider._resolve_model_name("claude") == "anthropic/claude-sonnet-4.1" assert provider._resolve_model_name("haiku") == "anthropic/claude-3.5-haiku"
assert provider._resolve_model_name("mistral") == "mistralai/mistral-large-2411" assert provider._resolve_model_name("mistral") == "mistralai/mistral-large-2411"
assert provider._resolve_model_name("deepseek") == "deepseek/deepseek-r1-0528" assert provider._resolve_model_name("deepseek") == "deepseek/deepseek-r1-0528"
assert provider._resolve_model_name("r1") == "deepseek/deepseek-r1-0528" assert provider._resolve_model_name("r1") == "deepseek/deepseek-r1-0528"
# Test case-insensitive # Test case-insensitive
assert provider._resolve_model_name("OPUS") == "anthropic/claude-opus-4.1" assert provider._resolve_model_name("OPUS") == "anthropic/claude-opus-4.1"
assert provider._resolve_model_name("SONNET") == "anthropic/claude-sonnet-4.5"
assert provider._resolve_model_name("O3") == "openai/o3" assert provider._resolve_model_name("O3") == "openai/o3"
assert provider._resolve_model_name("Mistral") == "mistralai/mistral-large-2411" assert provider._resolve_model_name("Mistral") == "mistralai/mistral-large-2411"
assert provider._resolve_model_name("CLAUDE") == "anthropic/claude-sonnet-4.1"
# Test direct model names (should pass through unchanged) # Test direct model names (should pass through unchanged)
assert provider._resolve_model_name("anthropic/claude-opus-4.1") == "anthropic/claude-opus-4.1" assert provider._resolve_model_name("anthropic/claude-opus-4.1") == "anthropic/claude-opus-4.1"
@@ -271,7 +272,7 @@ class TestOpenRouterRegistry:
assert len(aliases) > 0 assert len(aliases) > 0
assert "opus" in aliases assert "opus" in aliases
assert "o3" in aliases assert "o3" in aliases
assert "claude" in aliases assert "sonnet" in aliases
def test_registry_capabilities(self): def test_registry_capabilities(self):
"""Test registry provides correct capabilities.""" """Test registry provides correct capabilities."""
@@ -300,12 +301,17 @@ class TestOpenRouterRegistry:
registry = OpenRouterModelRegistry() registry = OpenRouterModelRegistry()
# All these should resolve to Claude Sonnet # All these should resolve to Claude Sonnet 4.5
sonnet_aliases = ["sonnet", "claude", "claude-sonnet", "claude-4.1-sonnet"] sonnet_45_aliases = ["sonnet", "sonnet4.5"]
for alias in sonnet_aliases: for alias in sonnet_45_aliases:
config = registry.resolve(alias) config = registry.resolve(alias)
assert config is not None assert config is not None
assert config.model_name == "anthropic/claude-sonnet-4.1" assert config.model_name == "anthropic/claude-sonnet-4.5"
# Test Sonnet 4.1 alias
config = registry.resolve("sonnet4.1")
assert config is not None
assert config.model_name == "anthropic/claude-sonnet-4.1"
class TestOpenRouterFunctionality: class TestOpenRouterFunctionality:

View File

@@ -89,7 +89,7 @@ class TestOpenRouterModelRegistry:
test_cases = [ test_cases = [
("opus", "anthropic/claude-opus-4.1"), ("opus", "anthropic/claude-opus-4.1"),
("OPUS", "anthropic/claude-opus-4.1"), # Case insensitive ("OPUS", "anthropic/claude-opus-4.1"), # Case insensitive
("claude", "anthropic/claude-sonnet-4.1"), ("sonnet", "anthropic/claude-sonnet-4.5"),
("o3", "openai/o3"), ("o3", "openai/o3"),
("deepseek", "deepseek/deepseek-r1-0528"), ("deepseek", "deepseek/deepseek-r1-0528"),
("mistral", "mistralai/mistral-large-2411"), ("mistral", "mistralai/mistral-large-2411"),