From a889cd14f38bfb0fa8158799defcb6a10ad9a4de Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Thu, 11 Apr 2024 12:34:19 -0700
Subject: [PATCH] docs: use vertexai in chat model tabs (#20352)

---
 docs/src/theme/ChatModelTabs.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/docs/src/theme/ChatModelTabs.js b/docs/src/theme/ChatModelTabs.js
index 09cab1f99f0..073171c99d4 100644
--- a/docs/src/theme/ChatModelTabs.js
+++ b/docs/src/theme/ChatModelTabs.js
@@ -31,7 +31,7 @@ os.environ["${apiKeyName}"] = getpass.getpass()`;
  * @property {boolean} [hideAnthropic] - Whether or not to hide Anthropic chat model.
  * @property {boolean} [hideFireworks] - Whether or not to hide Fireworks chat model.
  * @property {boolean} [hideMistral] - Whether or not to hide Mistral chat model.
- * @property {boolean} [hideGoogle] - Whether or not to hide Google chat model.
+ * @property {boolean} [hideGoogle] - Whether or not to hide Google VertexAI chat model.
  * @property {boolean} [hideTogether] - Whether or not to hide Together chat model.
  * @property {string} [customVarName] - Custom variable name for the model. Defaults to `model`.
  */
@@ -111,9 +111,9 @@ export default function ChatModelTabs(props) {
     {
       value: "Google",
       label: "Google",
-      text: `from langchain_google_genai import ChatGoogleGenerativeAI\n\n${llmVarName} = ChatGoogleGenerativeAI(${googleParamsOrDefault})`,
+      text: `from langchain_google_vertexai import ChatVertexAI\n\n${llmVarName} = ChatVertexAI(${googleParamsOrDefault})`,
       apiKeyName: "GOOGLE_API_KEY",
-      packageName: "langchain-google-genai",
+      packageName: "langchain-google-vertexai",
       default: false,
       shouldHide: hideGoogle,
     },