Skip to content

Commit

Permalink
changing model names
Browse files Browse the repository at this point in the history
  • Loading branch information
tanyaasharma committed Jun 17, 2024
1 parent 1f3209a commit 6a8bccb
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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."],
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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],
});
Expand Down
2 changes: 1 addition & 1 deletion examples/src/models/chat/integration_googlevertexai-wsa.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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?");
Expand Down
2 changes: 1 addition & 1 deletion examples/src/models/chat/integration_googlevertexai.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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?");
Expand Down

0 comments on commit 6a8bccb

Please sign in to comment.