Added support for medlm-large-1.5@001 in vertexai GoogleModelFamily (#…

…600)
langchain-ai · Nov 14, 2024 · 8bc67d8 · 8bc67d8
1 parent f199127
commit 8bc67d8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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