diff --git a/studio/backend/core/inference/providers.py b/studio/backend/core/inference/providers.py index 8188d8d603..98a884df23 100644 --- a/studio/backend/core/inference/providers.py +++ b/studio/backend/core/inference/providers.py @@ -33,7 +33,7 @@ PROVIDER_REGISTRY: dict[str, dict[str, Any]] = { "base_url": "https://api.mistral.ai/v1", "default_models": [ "mistral-large-latest", - "mistral-small-latest", + "mistral-small-2506", "codestral-latest", ], "supports_streaming": True, diff --git a/studio/backend/tests/test_providers_api.py b/studio/backend/tests/test_providers_api.py index 5790187a0e..30a5362217 100644 --- a/studio/backend/tests/test_providers_api.py +++ b/studio/backend/tests/test_providers_api.py @@ -42,7 +42,7 @@ PASSWORD = os.getenv("STUDIO_TEST_PASSWORD", "") # Map provider_type → (env var name, model to use for inference test) _PROVIDER_CONFIGS: dict[str, tuple[str, str]] = { "openai": ("OPENAI_API_KEY", "gpt-4o-mini"), - "mistral": ("MISTRAL_API_KEY", "mistral-small-latest"), + "mistral": ("MISTRAL_API_KEY", "mistral-small-2506"), "gemini": ("GEMINI_API_KEY", "gemini-2.5-flash"), "cohere": ("COHERE_API_KEY", "command-a-03-2025"), "openrouter": ("OPENROUTER_API_KEY", "openai/gpt-4o-mini"),