feat(vertexai): Update sample models to gemini-1.5-flash-001 (#3705)

* feat(vertexai): Update sample models to gemini-1.5-flash-001 * fix: Revert model changes for grounding until feature is ready for release
GoogleCloudPlatform · May 24, 2024 · c401daf · c401daf
1 parent d271b5d
commit c401daf
Show file tree

Hide file tree

Showing 34 changed files with 46 additions and 46 deletions.
diff --git a/generative-ai/snippets/count-tokens/countTokens.js b/generative-ai/snippets/count-tokens/countTokens.js
@@ -22,7 +22,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function countTokens(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/count-tokens/countTokensAdvanced.js b/generative-ai/snippets/count-tokens/countTokensAdvanced.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function countTokens(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.5-pro-preview-0409'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/function-calling/functionCallingAdvanced.js b/generative-ai/snippets/function-calling/functionCallingAdvanced.js
@@ -65,7 +65,7 @@ const generationConfig = {
 async function functionCallingAdvanced(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-001'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/function-calling/functionCallingBasic.js b/generative-ai/snippets/function-calling/functionCallingBasic.js
@@ -46,7 +46,7 @@ const functionDeclarations = [
 async function functionCallingBasic(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-001'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/function-calling/functionCallingStreamChat.js b/generative-ai/snippets/function-calling/functionCallingStreamChat.js
@@ -55,7 +55,7 @@ const functionResponseParts = [
 async function functionCallingStreamChat(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/function-calling/functionCallingStreamContent.js b/generative-ai/snippets/function-calling/functionCallingStreamContent.js
@@ -55,7 +55,7 @@ const functionResponseParts = [
 async function functionCallingStreamContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/gemini-text-input.js b/generative-ai/snippets/gemini-text-input.js
@@ -22,7 +22,7 @@ async function generate_from_text_input(projectId = 'PROJECT_ID') {
   const vertexAI = new VertexAI({project: projectId, location: 'us-central1'});
 
   const generativeModel = vertexAI.getGenerativeModel({
-    model: 'gemini-1.0-pro-002',
+    model: 'gemini-1.5-flash-001',
   });
 
   const prompt =

diff --git a/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js b/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function generateContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.5-pro-preview-0409'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex AI
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/inference/nonStreamTextBasic.js b/generative-ai/snippets/inference/nonStreamTextBasic.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function generateContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/inference/streamMultiModalityBasic.js b/generative-ai/snippets/inference/streamMultiModalityBasic.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function generateContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.5-pro-preview-0409'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex AI
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/inference/streamTextBasic.js b/generative-ai/snippets/inference/streamTextBasic.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function generateContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/nonStreamingChat.js b/generative-ai/snippets/nonStreamingChat.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function createNonStreamingChat(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/nonStreamingContent.js b/generative-ai/snippets/nonStreamingContent.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function createNonStreamingContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/safetySettings.js b/generative-ai/snippets/safetySettings.js
@@ -25,7 +25,7 @@ const {
 async function setSafetySettings(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-001'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/streamChat.js b/generative-ai/snippets/streamChat.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function createStreamChat(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/streamContent.js b/generative-ai/snippets/streamContent.js
@@ -21,7 +21,7 @@ const {VertexAI} = require('@google-cloud/vertexai');
 async function createStreamContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-1.0-pro-002'
+  model = 'gemini-1.5-flash-001'
 ) {
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});

diff --git a/generative-ai/snippets/test/count-tokens/countTokens.test.js b/generative-ai/snippets/test/count-tokens/countTokens.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Count tokens', async () => {
   /**
@@ -30,7 +30,7 @@ describe('Count tokens', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should count tokens', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/count-tokens/countTokensAdvanced.test.js b/generative-ai/snippets/test/count-tokens/countTokensAdvanced.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.5-pro-preview-0409';
+const model = 'gemini-1.5-flash-001';
 
 describe('Count tokens advanced', async () => {
   /**
@@ -30,7 +30,7 @@ describe('Count tokens advanced', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should count tokens in a multimodal prompt', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/function-calling/functionCallingAdvanced.test.js b/generative-ai/snippets/test/function-calling/functionCallingAdvanced.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-001';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Function Calling Advanced', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Function Calling Advanced', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should define multiple functions and have the model invoke the specified one', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/function-calling/functionCallingBasic.test.js b/generative-ai/snippets/test/function-calling/functionCallingBasic.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-001';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Function Calling', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Function Calling', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should define a function and have the model invoke it', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/function-calling/functionCallingStreamChat.test.js b/generative-ai/snippets/test/function-calling/functionCallingStreamChat.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.5-pro-preview-0409';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Function Calling Stream Chat', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Function Calling Stream Chat', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.5-pro-preview-0409';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should create stream chat and begin the conversation the same in each instance', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/function-calling/functionCallingStreamContent.test.js b/generative-ai/snippets/test/function-calling/functionCallingStreamContent.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Function Calling Stream Content', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Function Calling Stream Content', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should create stream chat and begin the conversation the same in each instance', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/gemini-system-instruction.test.js b/generative-ai/snippets/test/gemini-system-instruction.test.js
@@ -21,8 +21,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Set sytem instruction', async () => {
-  it('should set sytem instruction', async () => {
+describe('Set system instruction', async () => {
+  it('should set system instruction', async () => {
     const output = execSync(`node ./gemini-system-instruction.js ${projectId}`);
 
     assert(output.length > 0);

diff --git a/generative-ai/snippets/test/grounding/groundingPrivateDataBasic.test.js b/generative-ai/snippets/test/grounding/groundingPrivateDataBasic.test.js
@@ -31,7 +31,7 @@ describe('Private data grounding', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.0-pro-002';
 
   it('should ground results in private VertexAI search data', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/grounding/groundingPublicDataBasic.test.js b/generative-ai/snippets/test/grounding/groundingPublicDataBasic.test.js
@@ -30,12 +30,12 @@ describe('Google search grounding', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.0-pro-002';
 
   it('should ground results in public search data', async () => {
     const output = execSync(
       `node ./grounding/groundingPublicDataBasic.js ${projectId} ${location} ${model}`
     );
-    assert(output.match(/webSearchQueries.*Why is the sky blue?/));
+    assert(output.match(/webSearchQueries.*why is the sky blue?/));
   });
 });
diff --git a/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js b/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
@@ -30,7 +30,7 @@ describe('Generative AI Multimodal Text Inference', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.5-pro-preview-0409';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should generate text based on a prompt containing text, a video, and an image', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js b/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Basic Text Inference', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Basic Text Inference', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should create a generative text model and infer text from a prompt', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js b/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.5-pro-preview-0409';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Basic Multimodal Text Inference Streaming', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Basic Multimodal Text Inference Streaming', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.5-pro-preview-0409';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should create a generative text model and infer text from a prompt, streaming the results', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/inference/streamTextBasic.test.js b/generative-ai/snippets/test/inference/streamTextBasic.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Basic Text Inference Streaming', () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI Basic Text Inference Streaming', () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should create a generative text model and infer text from a prompt, streaming the results', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/nonStreamingChat.test.js b/generative-ai/snippets/test/nonStreamingChat.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI NonStreaming Chat', async () => {
   /**
@@ -30,7 +30,7 @@ describe('Generative AI NonStreaming Chat', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-1.0-pro';
+  // const model = 'gemini-1.5-flash-001';
 
   it('should create nonstreaming chat and begin the conversation the same in each instance', async () => {
     const output = execSync(

diff --git a/generative-ai/snippets/test/nonStreamingContent.test.js b/generative-ai/snippets/test/nonStreamingContent.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI NonStreaming Content', () => {
   /**

diff --git a/generative-ai/snippets/test/safetySettings.test.js b/generative-ai/snippets/test/safetySettings.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-001';
+const model = 'gemini-1.5-flash-001';
 
 describe('Safety settings', async () => {
   /**

diff --git a/generative-ai/snippets/test/streamChat.test.js b/generative-ai/snippets/test/streamChat.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
-const model = 'gemini-1.0-pro-002';
+const model = 'gemini-1.5-flash-001';
 
 describe('Generative AI Stream Chat', () => {
   /**