diff --git a/libs/vertexai/langchain_google_vertexai/_utils.py b/libs/vertexai/langchain_google_vertexai/_utils.py index d6f25b42..f0e4d39d 100644 --- a/libs/vertexai/langchain_google_vertexai/_utils.py +++ b/libs/vertexai/langchain_google_vertexai/_utils.py @@ -129,6 +129,7 @@ def _missing_(cls, value: Any) -> "GoogleModelFamily": gemini_advanced_models = { "medlm-large-1.5-preview", "medlm-large-1.5-001", + "medlm-large-1.5@001", } if "gemini-1.5" in model_name or model_name in gemini_advanced_models: return GoogleModelFamily.GEMINI_ADVANCED