fix mistral tests

Author: lkuligin

libs/vertexai/langchain_google_vertexai/model_garden_maas/_base.py

@@ -25,12 +25,7 @@
 
 from langchain_google_vertexai._base import _VertexAIBase
 
-_MISTRAL_MODELS: list[str] = [
-    "mistral-nemo@2407",
-    "mistral-large-2411@001",
-    "mistral-small-2503@001",
-    "codestral-2501@001",
-]
+_MISTRAL_MODELS: list[str] = ["mistral-medium-3", "mistral-small-2503", "codestral-2"]
 _LLAMA_MODELS: list[str] = [
     "meta/llama-3.2-90b-vision-instruct-maas",
     "meta/llama-3.3-70b-instruct-maas",

libs/vertexai/tests/integration_tests/test_maas.py

@@ -19,17 +19,13 @@
 model_names = _LLAMA_MODELS + _MISTRAL_MODELS
 # Fix tool support for new Mistral and Llama models
 model_names_with_tools_support = [
-    "mistral-nemo@2407",
+    "mistral-medium-3",
 ]
 model_locations = {
     "meta/llama-4-maverick-17b-128e-instruct-maas": "us-east5",
     "meta/llama-4-scout-17b-16e-instruct-maas": "us-east5",
 }
 
-# TODO: mistral-nemo@2407 missing from Cloud project
-model_names.remove("mistral-nemo@2407")
-model_names_with_tools_support.remove("mistral-nemo@2407")
-
 
 @pytest.mark.extended
 @pytest.mark.parametrize("model_name", model_names)