added text-unicon to PALM family

langchain-ai · Oct 30, 2024 · 93db430 · 93db430
1 parent 4f37fac
commit 93db430
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/vertexai/langchain_google_vertexai/_utils.py b/libs/vertexai/langchain_google_vertexai/_utils.py
@@ -138,7 +138,7 @@ def _missing_(cls, value: Any) -> "GoogleModelFamily":
             return GoogleModelFamily.CODEY
         if "medlm-medium@latest" in model_name:
             return GoogleModelFamily.GEMINI
-        if "bison" in model_name or "medlm" in model_name:
+        if "bison" in model_name or "medlm" in model_name or "unicorn" in model_name:
             return GoogleModelFamily.PALM
         return GoogleModelFamily.GEMINI