From f3b28f291629665950dd4fa6c376a2b8de4ab214 Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Mon, 30 Mar 2026 05:48:11 +0000 Subject: [PATCH] fix: update Mistral model to mistral-small-2506 --- studio/backend/core/inference/providers.py | 2 +- studio/backend/tests/test_providers_api.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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"),