From 93db430178b761e69d1a91e38a5833e0209e92d7 Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Tue, 29 Oct 2024 21:00:12 +0100 Subject: [PATCH] added text-unicon to PALM family --- libs/vertexai/langchain_google_vertexai/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/vertexai/langchain_google_vertexai/_utils.py b/libs/vertexai/langchain_google_vertexai/_utils.py index 4799a4ab..8562ca14 100644 --- 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