From 455f529a2d43ae55e628cde08b51e90481721ad2 Mon Sep 17 00:00:00 2001
From: akumar <105435668+akumar-b6i@users.noreply.github.com>
Date: Thu, 14 Mar 2024 04:56:13 +0530
Subject: [PATCH] Adding Support for MedLM Models (#11911)

* Adding Support for MedLM Models

* cr

---------

Co-authored-by: Haotian Zhang <socool.king@gmail.com>
---
 .../llama_index/llms/vertex/utils.py                      | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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 cafdd2977b..e49f1aef02 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"]
 
-- 
GitLab