From f2534fe9a20ec6279d27e349a416a73552d34e53 Mon Sep 17 00:00:00 2001 From: Fahad Date: Sat, 21 Jun 2025 15:11:37 +0400 Subject: [PATCH] Fixed mistral model name --- tests/test_openrouter_provider.py | 4 ++-- tests/test_openrouter_registry.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_openrouter_provider.py b/tests/test_openrouter_provider.py index 06d6d95..0dd2b78 100644 --- a/tests/test_openrouter_provider.py +++ b/tests/test_openrouter_provider.py @@ -79,14 +79,14 @@ class TestOpenRouterProvider: 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("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("r1") == "deepseek/deepseek-r1-0528" # Test case-insensitive assert provider._resolve_model_name("OPUS") == "anthropic/claude-3-opus" 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" # Test direct model names (should pass through unchanged) diff --git a/tests/test_openrouter_registry.py b/tests/test_openrouter_registry.py index 44b053c..1a5a2f1 100644 --- a/tests/test_openrouter_registry.py +++ b/tests/test_openrouter_registry.py @@ -79,7 +79,7 @@ class TestOpenRouterModelRegistry: ("claude", "anthropic/claude-3-sonnet"), ("o3", "openai/o3"), ("deepseek", "deepseek/deepseek-r1-0528"), - ("mistral", "mistral/mistral-large"), + ("mistral", "mistralai/mistral-large-2411"), ] for alias, expected_model in test_cases: