From 6a8bccbdb77b0536ecb6abb974fc7d6f969f97f9 Mon Sep 17 00:00:00 2001
From: Tanya Sharma <tanyaasharma@google.com>
Date: Mon, 17 Jun 2024 19:07:56 +0530
Subject: [PATCH] changing model names

---
 .../src/models/chat/integration_googlevertexai-streaming.ts     | 2 +-
 examples/src/models/chat/integration_googlevertexai-tools.ts    | 2 +-
 examples/src/models/chat/integration_googlevertexai-wsa.ts      | 2 +-
 examples/src/models/chat/integration_googlevertexai.ts          | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/src/models/chat/integration_googlevertexai-streaming.ts b/examples/src/models/chat/integration_googlevertexai-streaming.ts
index 37e6dd9e794b..d8e5d094a75c 100644
--- a/examples/src/models/chat/integration_googlevertexai-streaming.ts
+++ b/examples/src/models/chat/integration_googlevertexai-streaming.ts
@@ -4,7 +4,7 @@ import { ChatVertexAI } from "@langchain/google-vertexai";
 
 const model = new ChatVertexAI({
   temperature: 0.7,
-  model: "gemini-1.5-pro-001",
+  model: "gemini-1.5-flash-001",
 });
 const stream = await model.stream([
   ["system", "You are a funny assistant that answers in pirate language."],
diff --git a/examples/src/models/chat/integration_googlevertexai-tools.ts b/examples/src/models/chat/integration_googlevertexai-tools.ts
index f0f595848307..e36dd82c984e 100644
--- a/examples/src/models/chat/integration_googlevertexai-tools.ts
+++ b/examples/src/models/chat/integration_googlevertexai-tools.ts
@@ -25,7 +25,7 @@ const geminiCalculatorTool: GeminiTool = {
 
 const model = new ChatVertexAI({
   temperature: 0.7,
-  model: "gemini-pro-vision",
+  model: "gemini-1.5-flash-001",
 }).bind({
   tools: [geminiCalculatorTool],
 });
diff --git a/examples/src/models/chat/integration_googlevertexai-wsa.ts b/examples/src/models/chat/integration_googlevertexai-wsa.ts
index a59e036faeb0..61fbe4e56d5e 100644
--- a/examples/src/models/chat/integration_googlevertexai-wsa.ts
+++ b/examples/src/models/chat/integration_googlevertexai-wsa.ts
@@ -13,7 +13,7 @@ const calculatorSchema = z.object({
 
 const model = new ChatVertexAI({
   temperature: 0.7,
-  model: "gemini-pro-vision",
+  model: "gemini-1.5-flash-001",
 }).withStructuredOutput(calculatorSchema);
 
 const response = await model.invoke("What is 1628253239 times 81623836?");
diff --git a/examples/src/models/chat/integration_googlevertexai.ts b/examples/src/models/chat/integration_googlevertexai.ts
index 06aa1e854b76..3fa1f2c1caae 100644
--- a/examples/src/models/chat/integration_googlevertexai.ts
+++ b/examples/src/models/chat/integration_googlevertexai.ts
@@ -4,7 +4,7 @@ import { ChatVertexAI } from "@langchain/google-vertexai";
 
 const model = new ChatVertexAI({
   temperature: 0.7,
-  model: "gemini-pro-vision",
+  model: "gemini-1.5-flash-001",
 });
 
 const response = await model.invoke("Why is the ocean blue?");