torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed mistral model name

Browse Source
This commit is contained in:
Fahad
2025-06-21 15:11:37 +04:00
parent 9079d06941
commit f2534fe9a2
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_openrouter_provider.py
View File

@@ -79,14 +79,14 @@ class TestOpenRouterProvider:
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-high") == "openai/o4-mini-high" assert provider._resolve_model_name("o4-mini-high") == "openai/o4-mini-high"
assert provider._resolve_model_name("claude") == "anthropic/claude-3-sonnet" assert provider._resolve_model_name("claude") == "anthropic/claude-3-sonnet"
assert provider._resolve_model_name("mistral") == "mistral/mistral-large" 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-3-opus" assert provider._resolve_model_name("OPUS") == "anthropic/claude-3-opus"
assert provider._resolve_model_name("O3") == "openai/o3" assert provider._resolve_model_name("O3") == "openai/o3"
assert provider._resolve_model_name("Mistral") == "mistral/mistral-large" assert provider._resolve_model_name("Mistral") == "mistralai/mistral-large-2411"
assert provider._resolve_model_name("CLAUDE") == "anthropic/claude-3-sonnet" assert provider._resolve_model_name("CLAUDE") == "anthropic/claude-3-sonnet"
# Test direct model names (should pass through unchanged) # Test direct model names (should pass through unchanged)

2
tests/test_openrouter_registry.py
View File

@@ -79,7 +79,7 @@ class TestOpenRouterModelRegistry:
("claude", "anthropic/claude-3-sonnet"), ("claude", "anthropic/claude-3-sonnet"),
("o3", "openai/o3"), ("o3", "openai/o3"),
("deepseek", "deepseek/deepseek-r1-0528"), ("deepseek", "deepseek/deepseek-r1-0528"),
("mistral", "mistral/mistral-large"), ("mistral", "mistralai/mistral-large-2411"),
] ]
for alias, expected_model in test_cases: for alias, expected_model in test_cases: