diff --git a/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/utils.py b/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/utils.py index cafdd2977bc0e86ac1bd96104f97d0783a70d36e..e49f1aef029c3725fe3cab84d99d19a8c250fcb5 100644 --- a/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/utils.py +++ b/llama-index-integrations/llms/llama-index-llms-vertex/llama_index/llms/vertex/utils.py @@ -19,7 +19,13 @@ from vertexai.language_models import InputOutputTextPair from llama_index.core.base.llms.types import ChatMessage, MessageRole CHAT_MODELS = ["chat-bison", "chat-bison-32k", "chat-bison@001"] -TEXT_MODELS = ["text-bison", "text-bison-32k", "text-bison@001"] +TEXT_MODELS = [ + "text-bison", + "text-bison-32k", + "text-bison@001", + "medlm-medium", + "medlm-large", +] CODE_MODELS = ["code-bison", "code-bison-32k", "code-bison@001"] CODE_CHAT_MODELS = ["codechat-bison", "codechat-bison-32k", "codechat-bison@001"]