diff --git a/libs/vertexai/langchain_google_vertexai/model_garden_maas/_base.py b/libs/vertexai/langchain_google_vertexai/model_garden_maas/_base.py index 42a19dbd5..9564b42b7 100644 --- a/libs/vertexai/langchain_google_vertexai/model_garden_maas/_base.py +++ b/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", diff --git a/libs/vertexai/tests/integration_tests/test_maas.py b/libs/vertexai/tests/integration_tests/test_maas.py index ab482f15c..53de14932 100644 --- a/libs/vertexai/tests/integration_tests/test_maas.py +++ b/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)