From aa40ac114cabeb0238eb4081e73543ec4a50e91d Mon Sep 17 00:00:00 2001
From: Angel Caamal <acaamalcanul@google.com>
Date: Fri, 29 May 2026 20:10:43 +0000
Subject: [PATCH 1/6] refactor: migrate countTokens samples to @google/genai,
 update config, and remove deprecated totalBillableCharacters

---
 .../snippets/count-tokens/countTokens.js      | 31 ++++++----
 .../count-tokens/countTokensAdvanced.js       | 61 +++++++++----------
 generative-ai/snippets/package.json           |  1 +
 .../count-tokens/countTokensAdvanced.test.js  |  2 +-
 4 files changed, 50 insertions(+), 45 deletions(-)

diff --git a/generative-ai/snippets/count-tokens/countTokens.js b/generative-ai/snippets/count-tokens/countTokens.js
index c75ef1d8c65..b23985bc7c6 100644
--- a/generative-ai/snippets/count-tokens/countTokens.js
+++ b/generative-ai/snippets/count-tokens/countTokens.js
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_gemini_token_count]
-const {VertexAI} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
 /**
  * TODO(developer): Update these variables before running the sample.
@@ -23,27 +23,32 @@ async function countTokens(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
-
-  // Instantiate the model
-  const generativeModel = vertexAI.getGenerativeModel({
-    model: model,
+  // Initialize the client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
-  const req = {
-    contents: [{role: 'user', parts: [{text: 'How are you doing today?'}]}],
-  };
+  const contents = [
+    {role: 'user', parts: [{text: 'How are you doing today?'}]},
+  ];
 
   // Prompt tokens count
-  const countTokensResp = await generativeModel.countTokens(req);
+  const countTokensResp = await client.models.countTokens({
+    model: model,
+    contents: contents,
+  });
   console.log('Prompt tokens count: ', countTokensResp);
 
   // Send text to gemini
-  const result = await generativeModel.generateContent(req);
+  const result = await client.models.generateContent({
+    model: model,
+    contents: contents,
+  });
 
   // Response tokens count
-  const usageMetadata = result.response.usageMetadata;
+  const usageMetadata = result.usageMetadata;
   console.log('Response tokens count: ', usageMetadata);
 }
 // [END generativeaionvertexai_gemini_token_count]
diff --git a/generative-ai/snippets/count-tokens/countTokensAdvanced.js b/generative-ai/snippets/count-tokens/countTokensAdvanced.js
index 8831f50525b..d3a09538fbc 100644
--- a/generative-ai/snippets/count-tokens/countTokensAdvanced.js
+++ b/generative-ai/snippets/count-tokens/countTokensAdvanced.js
@@ -13,8 +13,7 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_gemini_token_count_advanced]
-const {VertexAI} = require('@google-cloud/vertexai');
-
+const {GoogleGenAI} = require('@google/genai');
 /**
  * TODO(developer): Update these variables before running the sample.
  */
@@ -23,42 +22,42 @@ async function countTokens(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
-
-  // Instantiate the model
-  const generativeModel = vertexAI.getGenerativeModel({
-    model: model,
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
-  const req = {
-    contents: [
-      {
-        role: 'user',
-        parts: [
-          {
-            file_data: {
-              file_uri:
-                'gs://cloud-samples-data/generative-ai/video/pixel8.mp4',
-              mime_type: 'video/mp4',
-            },
+  const contents = [
+    {
+      role: 'user',
+      parts: [
+        {
+          fileData: {
+            fileUri: 'gs://cloud-samples-data/generative-ai/video/pixel8.mp4',
+            mimeType: 'video/mp4',
           },
-          {text: 'Provide a description of the video.'},
-        ],
-      },
-    ],
-  };
+        },
+        {text: 'Provide a description of the video.'},
+      ],
+    },
+  ];
+
+  const countTokensResp = await client.models.countTokens({
+    model: model,
+    contents: contents,
+  });
 
-  const countTokensResp = await generativeModel.countTokens(req);
   console.log('Prompt Token Count:', countTokensResp.totalTokens);
-  console.log(
-    'Prompt Character Count:',
-    countTokensResp.totalBillableCharacters
-  );
 
   // Sent text to Gemini
-  const result = await generativeModel.generateContent(req);
-  const usageMetadata = result.response.usageMetadata;
+  const result = await client.models.generateContent({
+    model: model,
+    contents: contents,
+  });
+
+  const usageMetadata = result.usageMetadata;
 
   console.log('Prompt Token Count:', usageMetadata.promptTokenCount);
   console.log('Candidates Token Count:', usageMetadata.candidatesTokenCount);
diff --git a/generative-ai/snippets/package.json b/generative-ai/snippets/package.json
index 1aedba79e3b..8ed47228de2 100644
--- a/generative-ai/snippets/package.json
+++ b/generative-ai/snippets/package.json
@@ -15,6 +15,7 @@
   "dependencies": {
     "@google-cloud/aiplatform": "^3.12.0",
     "@google-cloud/vertexai": "github:googleapis/nodejs-vertexai",
+    "@google/genai": "^2.7.0",
     "axios": "^1.6.2",
     "supertest": "^7.0.0"
   },
diff --git a/generative-ai/snippets/test/count-tokens/countTokensAdvanced.test.js b/generative-ai/snippets/test/count-tokens/countTokensAdvanced.test.js
index aa944d1676e..6507250dfcd 100644
--- a/generative-ai/snippets/test/count-tokens/countTokensAdvanced.test.js
+++ b/generative-ai/snippets/test/count-tokens/countTokensAdvanced.test.js
@@ -38,6 +38,6 @@ describe('Count tokens advanced', async () => {
     );
 
     assert(output.match(/Prompt Token Count: \d+/));
-    assert(output.match(/Prompt Character Count: \d+/));
+    assert(output.match(/Total Token Count: \d+/));
   });
 });

From b25e80d0e0a5c6b494a8ab34c02446321ed636b8 Mon Sep 17 00:00:00 2001
From: Angel Caamal <acaamalcanul@google.com>
Date: Fri, 29 May 2026 22:14:33 +0000
Subject: [PATCH 2/6] refactor(samples): migrate function-calling samples to
 @google/genai using message structure for tool responses

---
 .../functionCallingAdvanced.js                | 66 +++++++----------
 .../function-calling/functionCallingBasic.js  | 40 +++++-----
 .../functionCallingStreamChat.js              | 74 ++++++++-----------
 .../functionCallingStreamContent.js           | 73 +++++++++---------
 4 files changed, 111 insertions(+), 142 deletions(-)

diff --git a/generative-ai/snippets/function-calling/functionCallingAdvanced.js b/generative-ai/snippets/function-calling/functionCallingAdvanced.js
index 8c59df6ca8b..9fe9cea1fad 100644
--- a/generative-ai/snippets/function-calling/functionCallingAdvanced.js
+++ b/generative-ai/snippets/function-calling/functionCallingAdvanced.js
@@ -13,22 +13,18 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_function_calling_advanced]
-const {
-  VertexAI,
-  FunctionDeclarationSchemaType,
-} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
-const functionDeclarations = [
+const tools = [
   {
-    function_declarations: [
+    functionDeclarations: [
       {
         name: 'get_product_sku',
-        description:
-          'Get the available inventory for a Google products, e.g: Pixel phones, Pixel Watches, Google Home etc',
+        description: 'Get the available inventory for Google products',
         parameters: {
-          type: FunctionDeclarationSchemaType.OBJECT,
+          type: 'OBJECT',
           properties: {
-            productName: {type: FunctionDeclarationSchemaType.STRING},
+            productName: {type: 'STRING'},
           },
         },
       },
@@ -36,9 +32,9 @@ const functionDeclarations = [
         name: 'get_store_location',
         description: 'Get the location of the closest store',
         parameters: {
-          type: FunctionDeclarationSchemaType.OBJECT,
+          type: 'OBJECT',
           properties: {
-            location: {type: FunctionDeclarationSchemaType.STRING},
+            location: {type: 'STRING'},
           },
         },
       },
@@ -47,18 +43,12 @@ const functionDeclarations = [
 ];
 
 const toolConfig = {
-  function_calling_config: {
+  functionCallingConfig: {
     mode: 'ANY',
-    allowed_function_names: ['get_product_sku'],
+    allowedFunctionNames: ['get_product_sku'],
   },
 };
 
-const generationConfig = {
-  temperature: 0.95,
-  topP: 1.0,
-  maxOutputTokens: 8192,
-};
-
 /**
  * TODO(developer): Update these variables before running the sample.
  */
@@ -67,29 +57,25 @@ async function functionCallingAdvanced(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+  });
 
-  // Instantiate the model
-  const generativeModel = vertexAI.preview.getGenerativeModel({
+  const result = await client.models.generateContent({
     model: model,
+    contents: 'Do you have the White Pixel 8 Pro 128GB in stock in the US?',
+    config: {
+      tools: tools,
+      toolConfig: toolConfig,
+      temperature: 0.95,
+      topP: 1.0,
+      maxOutputTokens: 8192,
+    },
   });
-
-  const request = {
-    contents: [
-      {
-        role: 'user',
-        parts: [
-          {text: 'Do you have the White Pixel 8 Pro 128GB in stock in the US?'},
-        ],
-      },
-    ],
-    tools: functionDeclarations,
-    tool_config: toolConfig,
-    generation_config: generationConfig,
-  };
-  const result = await generativeModel.generateContent(request);
-  console.log(JSON.stringify(result.response.candidates[0].content));
+  console.log(JSON.stringify(result.functionCalls));
 }
 // [END generativeaionvertexai_function_calling_advanced]
 
diff --git a/generative-ai/snippets/function-calling/functionCallingBasic.js b/generative-ai/snippets/function-calling/functionCallingBasic.js
index 999ad03818a..232a91e9a1b 100644
--- a/generative-ai/snippets/function-calling/functionCallingBasic.js
+++ b/generative-ai/snippets/function-calling/functionCallingBasic.js
@@ -13,23 +13,20 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_function_calling_basic]
-const {
-  VertexAI,
-  FunctionDeclarationSchemaType,
-} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
-const functionDeclarations = [
+const tools = [
   {
-    function_declarations: [
+    functionDeclarations: [
       {
         name: 'get_current_weather',
         description: 'get weather in a given location',
         parameters: {
-          type: FunctionDeclarationSchemaType.OBJECT,
+          type: 'OBJECT',
           properties: {
-            location: {type: FunctionDeclarationSchemaType.STRING},
+            location: {type: 'STRING'},
             unit: {
-              type: FunctionDeclarationSchemaType.STRING,
+              type: 'STRING',
               enum: ['celsius', 'fahrenheit'],
             },
           },
@@ -48,22 +45,21 @@ async function functionCallingBasic(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+  });
 
-  // Instantiate the model
-  const generativeModel = vertexAI.preview.getGenerativeModel({
+  const result = await client.models.generateContent({
     model: model,
+    contents: 'What is the weather in Boston?',
+    config: {
+      tools: tools,
+    },
   });
-
-  const request = {
-    contents: [
-      {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
-    ],
-    tools: functionDeclarations,
-  };
-  const result = await generativeModel.generateContent(request);
-  console.log(JSON.stringify(result.response.candidates[0].content));
+  console.log(JSON.stringify(result.functionCalls));
 }
 // [END generativeaionvertexai_function_calling_basic]
 
diff --git a/generative-ai/snippets/function-calling/functionCallingStreamChat.js b/generative-ai/snippets/function-calling/functionCallingStreamChat.js
index 88844a6925d..6fadb789a2b 100644
--- a/generative-ai/snippets/function-calling/functionCallingStreamChat.js
+++ b/generative-ai/snippets/function-calling/functionCallingStreamChat.js
@@ -13,25 +13,19 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_gemini_function_calling_chat]
-const {
-  VertexAI,
-  FunctionDeclarationSchemaType,
-} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
-const functionDeclarations = [
+const tools = [
   {
-    function_declarations: [
+    functionDeclarations: [
       {
         name: 'get_current_weather',
         description: 'get weather in a given location',
         parameters: {
-          type: FunctionDeclarationSchemaType.OBJECT,
+          type: 'OBJECT',
           properties: {
-            location: {type: FunctionDeclarationSchemaType.STRING},
-            unit: {
-              type: FunctionDeclarationSchemaType.STRING,
-              enum: ['celsius', 'fahrenheit'],
-            },
+            location: {type: 'STRING'},
+            unit: {type: 'STRING', enum: ['celsius', 'fahrenheit']},
           },
           required: ['location'],
         },
@@ -40,15 +34,6 @@ const functionDeclarations = [
   },
 ];
 
-const functionResponseParts = [
-  {
-    functionResponse: {
-      name: 'get_current_weather',
-      response: {name: 'get_current_weather', content: {weather: 'super nice'}},
-    },
-  },
-];
-
 /**
  * TODO(developer): Update these variables before running the sample.
  */
@@ -57,38 +42,43 @@ async function functionCallingStreamChat(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
-
-  // Instantiate the model
-  const generativeModel = vertexAI.getGenerativeModel({
-    model: model,
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
   // Create a chat session and pass your function declarations
-  const chat = generativeModel.startChat({
-    tools: functionDeclarations,
+  const chat = client.chats.create({
+    model: model,
+    config: {tools: tools},
   });
 
-  const chatInput1 = 'What is the weather in Boston?';
-
   // This should include a functionCall response from the model
-  const result1 = await chat.sendMessageStream(chatInput1);
-  for await (const item of result1.stream) {
-    console.log(item.candidates[0]);
-  }
-  await result1.response;
+  const result1 = await chat.sendMessage({
+    message: 'What is the weather in Boston?',
+  });
+  console.log(
+    'Function call requested:',
+    JSON.stringify(result1.functionCalls, null, 2)
+  );
 
   // Send a follow up message with a FunctionResponse
-  const result2 = await chat.sendMessageStream(functionResponseParts);
-  for await (const item of result2.stream) {
-    console.log(item.candidates[0]);
-  }
+  const result2 = await chat.sendMessage({
+    message: [
+      {
+        functionResponse: {
+          name: 'get_current_weather',
+          response: {result: {weather: 'super nice'}},
+        },
+      },
+    ],
+  });
 
   // This should include a text response from the model using the response content
   // provided above
-  const response2 = await result2.response;
-  console.log(response2.candidates[0].content.parts[0].text);
+  console.log(result2.text);
 }
 // [END generativeaionvertexai_gemini_function_calling_chat]
 
diff --git a/generative-ai/snippets/function-calling/functionCallingStreamContent.js b/generative-ai/snippets/function-calling/functionCallingStreamContent.js
index 923ac6529a5..2a3a1971f27 100644
--- a/generative-ai/snippets/function-calling/functionCallingStreamContent.js
+++ b/generative-ai/snippets/function-calling/functionCallingStreamContent.js
@@ -14,25 +14,19 @@
 
 // [START aiplatform_gemini_function_calling_content]
 // [START generativeaionvertexai_gemini_function_calling_content]
-const {
-  VertexAI,
-  FunctionDeclarationSchemaType,
-} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
-const functionDeclarations = [
+const tools = [
   {
-    function_declarations: [
+    functionDeclarations: [
       {
         name: 'get_current_weather',
         description: 'get weather in a given location',
         parameters: {
-          type: FunctionDeclarationSchemaType.OBJECT,
+          type: 'OBJECT',
           properties: {
-            location: {type: FunctionDeclarationSchemaType.STRING},
-            unit: {
-              type: FunctionDeclarationSchemaType.STRING,
-              enum: ['celsius', 'fahrenheit'],
-            },
+            location: {type: 'STRING'},
+            unit: {type: 'STRING', enum: ['celsius', 'fahrenheit']},
           },
           required: ['location'],
         },
@@ -58,35 +52,38 @@ async function functionCallingStreamContent(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
-
-  // Instantiate the model
-  const generativeModel = vertexAI.getGenerativeModel({
-    model: model,
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
-  const request = {
-    contents: [
-      {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
-      {
-        role: 'ASSISTANT',
-        parts: [
-          {
-            functionCall: {
-              name: 'get_current_weather',
-              args: {location: 'Boston'},
-            },
+  const request = [
+    {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
+    {
+      role: 'model',
+      parts: [
+        {
+          functionCall: {
+            name: 'get_current_weather',
+            args: {location: 'Boston'},
           },
-        ],
-      },
-      {role: 'USER', parts: functionResponseParts},
-    ],
-    tools: functionDeclarations,
-  };
-  const streamingResp = await generativeModel.generateContentStream(request);
-  for await (const item of streamingResp.stream) {
-    console.log(item.candidates[0].content.parts[0].text);
+        },
+      ],
+    },
+    {role: 'USER', parts: functionResponseParts},
+  ];
+
+  const streamingResp = await client.models.generateContentStream({
+    model: model,
+    contents: request,
+    config: {tools: tools},
+  });
+  for await (const chunk of streamingResp) {
+    if (chunk.text) {
+      console.log(chunk.text);
+    }
   }
 }
 // [END aiplatform_gemini_function_calling_content]

From bc5bfdd0ad5693471f35b10def3b716b5a602120 Mon Sep 17 00:00:00 2001
From: Angel Caamal <acaamalcanul@google.com>
Date: Tue, 2 Jun 2026 15:58:27 +0000
Subject: [PATCH 3/6] feat(generative-ai): migrate grounding samples to new
 GenAI SDK

---
 .../grounding/groundingPrivateDataBasic.js    | 64 +++++++++----------
 .../grounding/groundingPublicDataBasic.js     | 33 +++++-----
 2 files changed, 47 insertions(+), 50 deletions(-)

diff --git a/generative-ai/snippets/grounding/groundingPrivateDataBasic.js b/generative-ai/snippets/grounding/groundingPrivateDataBasic.js
index 067d7c5a87f..e152a59dae7 100644
--- a/generative-ai/snippets/grounding/groundingPrivateDataBasic.js
+++ b/generative-ai/snippets/grounding/groundingPrivateDataBasic.js
@@ -13,11 +13,7 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_grounding_private_data_basic]
-const {
-  VertexAI,
-  HarmCategory,
-  HarmBlockThreshold,
-} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
 /**
  * TODO(developer): Update these variables before running the sample.
@@ -28,41 +24,43 @@ async function generateContentWithVertexAISearchGrounding(
   model = 'gemini-2.0-flash-001',
   dataStoreId = 'DATASTORE_ID'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
-
-  const generativeModelPreview = vertexAI.preview.getGenerativeModel({
-    model: model,
-    // The following parameters are optional
-    // They can also be passed to individual content generation requests
-    safetySettings: [
-      {
-        category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-        threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
-      },
-    ],
-    generationConfig: {maxOutputTokens: 256},
+  // Initialize cleint with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
-  const vertexAIRetrievalTool = {
-    retrieval: {
-      vertexAiSearch: {
-        datastore: `projects/${projectId}/locations/global/collections/default_collection/dataStores/${dataStoreId}`,
+  const tools = [
+    {
+      retrieval: {
+        vertexAiSearch: {
+          datastore: `projects/${projectId}/locations/global/collections/default_collection/dataStores/${dataStoreId}`,
+        },
       },
-      disableAttribution: false,
     },
-  };
+  ];
 
-  const request = {
+  const result = await client.models.generateContent({
+    model: model,
     contents: [{role: 'user', parts: [{text: 'Why is the sky blue?'}]}],
-    tools: [vertexAIRetrievalTool],
-  };
+    config: {
+      tools: tools,
+      maxOutputTokens: 256,
+      safetySettings: [
+        {
+          category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+          threshold: 'BLOCK_MEDIUM_AND_ABOVE',
+        },
+      ],
+    },
+  });
 
-  const result = await generativeModelPreview.generateContent(request);
-  const response = result.response;
-  const groundingMetadata = response.candidates[0];
-  console.log('Response: ', JSON.stringify(response.candidates[0]));
-  console.log('GroundingMetadata is: ', JSON.stringify(groundingMetadata));
+  console.log('Response: ', result.text);
+  console.log(
+    'GroundingMetadata: ',
+    JSON.stringify(result.candidates[0].groundingMetadata)
+  );
 }
 // [END generativeaionvertexai_grounding_private_data_basic]
 
diff --git a/generative-ai/snippets/grounding/groundingPublicDataBasic.js b/generative-ai/snippets/grounding/groundingPublicDataBasic.js
index f273e4f8a9f..8b50030b529 100644
--- a/generative-ai/snippets/grounding/groundingPublicDataBasic.js
+++ b/generative-ai/snippets/grounding/groundingPublicDataBasic.js
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_grounding_public_data_basic]
-const {VertexAI} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
 /**
  * TODO(developer): Update these variables before running the sample.
@@ -23,31 +23,30 @@ async function generateContentWithGoogleSearchGrounding(
   location = 'us-central1',
   model = 'gemini-2.0-flash-001'
 ) {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: projectId, location: location});
-
-  const generativeModelPreview = vertexAI.preview.getGenerativeModel({
-    model: model,
-    generationConfig: {maxOutputTokens: 256},
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
   const googleSearchTool = {
     googleSearch: {},
   };
 
-  const request = {
+  const result = await client.models.generateContent({
+    model: model,
     contents: [{role: 'user', parts: [{text: 'Why is the sky blue?'}]}],
-    tools: [googleSearchTool],
-  };
-
-  const result = await generativeModelPreview.generateContent(request);
-  const response = await result.response;
-  const groundingMetadata = response.candidates[0].groundingMetadata;
+    config: {
+      tools: [googleSearchTool],
+      maxOutputTokens: 256,
+    },
+  });
+  console.log('Response: ', JSON.stringify(result.text));
   console.log(
-    'Response: ',
-    JSON.stringify(response.candidates[0].content.parts[0].text)
+    'GroundingMetadata is: ',
+    JSON.stringify(result.candidates[0].groundingMetadata)
   );
-  console.log('GroundingMetadata is: ', JSON.stringify(groundingMetadata));
 }
 // [END generativeaionvertexai_grounding_public_data_basic]
 

From 26802e461ab39cd6d7e833a791e491f212055f6b Mon Sep 17 00:00:00 2001
From: Angel Caamal <acaamalcanul@google.com>
Date: Tue, 2 Jun 2026 16:59:19 +0000
Subject: [PATCH 4/6] feat(generative-ai): migrate text and multimodal
 inference samples to @google/genai

---
 .../inference/nonStreamMultiModalityBasic.js  | 45 ++++++++++---------
 .../snippets/inference/nonStreamTextBasic.js  | 30 ++++++-------
 .../inference/streamMultiModalityBasic.js     | 41 +++++++++--------
 .../snippets/inference/streamTextBasic.js     | 33 +++++++-------
 .../nonStreamMultiModalityBasic.test.js       |  8 +++-
 .../test/inference/nonStreamTextBasic.test.js |  8 +++-
 .../streamMultiModalityBasic.test.js          |  8 +++-
 .../test/inference/streamTextBasic.test.js    |  8 +++-
 8 files changed, 106 insertions(+), 75 deletions(-)

diff --git a/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js b/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
index 51d6f761976..708e2b44f60 100644
--- a/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
+++ b/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
@@ -13,51 +13,52 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_non_stream_multimodality_basic]
-const {VertexAI} = require('@google-cloud/vertexai');
-
+const {GoogleGenAI} = require('@google/genai');
 /**
  * TODO(developer): Update these variables before running the sample.
  */
-const PROJECT_ID = process.env.CAIP_PROJECT_ID;
-const LOCATION = 'us-central1';
-const MODEL = 'gemini-2.0-flash-001';
-
-async function generateContent() {
-  // Initialize Vertex AI
-  const vertexAI = new VertexAI({project: PROJECT_ID, location: LOCATION});
-  const generativeModel = vertexAI.getGenerativeModel({model: MODEL});
+async function generateContent(
+  projectId,
+  location = 'us-central1',
+  model = 'gemini-2.0-flash-001'
+) {
+  // Initialize client
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+  });
 
-  const request = {
+  const result = await client.models.generateContent({
+    model: model,
     contents: [
       {
         role: 'user',
         parts: [
           {
-            file_data: {
-              file_uri: 'gs://cloud-samples-data/video/animals.mp4',
-              mime_type: 'video/mp4',
+            fileData: {
+              fileUri: 'gs://cloud-samples-data/video/animals.mp4',
+              mimeType: 'video/mp4',
             },
           },
           {
-            file_data: {
-              file_uri:
+            fileData: {
+              fileUri:
                 'gs://cloud-samples-data/generative-ai/image/character.jpg',
-              mime_type: 'image/jpeg',
+              mimeType: 'image/jpeg',
             },
           },
           {text: 'Are this video and image correlated?'},
         ],
       },
     ],
-  };
-
-  const result = await generativeModel.generateContent(request);
+  });
 
-  console.log(result.response.candidates[0].content.parts[0].text);
+  console.log(result.text);
 }
 // [END generativeaionvertexai_non_stream_multimodality_basic]
 
-generateContent().catch(err => {
+generateContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);
   process.exitCode = 1;
 });
diff --git a/generative-ai/snippets/inference/nonStreamTextBasic.js b/generative-ai/snippets/inference/nonStreamTextBasic.js
index 6ef2f01e92d..843695feb44 100644
--- a/generative-ai/snippets/inference/nonStreamTextBasic.js
+++ b/generative-ai/snippets/inference/nonStreamTextBasic.js
@@ -13,25 +13,25 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_non_stream_text_basic]
-const {VertexAI} = require('@google-cloud/vertexai');
-
+const {GoogleGenAI} = require('@google/genai');
 /**
  * TODO(developer): Update these variables before running the sample.
  */
-const PROJECT_ID = process.env.CAIP_PROJECT_ID;
-const LOCATION = process.env.LOCATION;
-const MODEL = 'gemini-2.0-flash-001';
-
-async function generateContent() {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: PROJECT_ID, location: LOCATION});
 
-  // Instantiate the model
-  const generativeModel = vertexAI.getGenerativeModel({
-    model: MODEL,
+async function generateContent(
+  projectId,
+  location = 'us-central1',
+  model = 'gemini-2.0-flash-001'
+) {
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
   const request = {
+    model: model,
     contents: [
       {
         role: 'user',
@@ -46,13 +46,13 @@ async function generateContent() {
 
   console.log(JSON.stringify(request));
 
-  const result = await generativeModel.generateContent(request);
+  const response = await client.models.generateContent(request);
 
-  console.log(result.response.candidates[0].content.parts[0].text);
+  console.log(response.text);
 }
 // [END generativeaionvertexai_non_stream_text_basic]
 
-generateContent().catch(err => {
+generateContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);
   process.exitCode = 1;
 });
diff --git a/generative-ai/snippets/inference/streamMultiModalityBasic.js b/generative-ai/snippets/inference/streamMultiModalityBasic.js
index a8399894147..3096cb3c073 100644
--- a/generative-ai/snippets/inference/streamMultiModalityBasic.js
+++ b/generative-ai/snippets/inference/streamMultiModalityBasic.js
@@ -13,36 +13,41 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_stream_multimodality_basic]
-const {VertexAI} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
 /**
  * TODO(developer): Update these variables before running the sample.
  */
-const PROJECT_ID = process.env.CAIP_PROJECT_ID;
-const LOCATION = process.env.LOCATION;
-const MODEL = 'gemini-2.0-flash-001';
 
-async function generateContent() {
-  // Initialize Vertex AI
-  const vertexAI = new VertexAI({project: PROJECT_ID, location: LOCATION});
-  const generativeModel = vertexAI.getGenerativeModel({model: MODEL});
+async function generateContent(
+  projectId,
+  location = 'us-central1',
+  model = 'gemini-2.0-flash-001'
+) {
+  // Initialize client
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+  });
 
   const request = {
+    model: model,
     contents: [
       {
         role: 'user',
         parts: [
           {
-            file_data: {
-              file_uri: 'gs://cloud-samples-data/video/animals.mp4',
-              mime_type: 'video/mp4',
+            fileData: {
+              fileUri: 'gs://cloud-samples-data/video/animals.mp4',
+              mimeType: 'video/mp4',
             },
           },
           {
-            file_data: {
-              file_uri:
+            fileData: {
+              fileUri:
                 'gs://cloud-samples-data/generative-ai/image/character.jpg',
-              mime_type: 'image/jpeg',
+              mimeType: 'image/jpeg',
             },
           },
           {text: 'Are this video and image correlated?'},
@@ -51,15 +56,15 @@ async function generateContent() {
     ],
   };
 
-  const result = await generativeModel.generateContentStream(request);
+  const responseStream = await client.models.generateContentStream(request);
 
-  for await (const item of result.stream) {
-    console.log(item.candidates[0].content.parts[0].text);
+  for await (const chunk of responseStream) {
+    console.log(chunk.text);
   }
 }
 // [END generativeaionvertexai_stream_multimodality_basic]
 
-generateContent().catch(err => {
+generateContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);
   process.exitCode = 1;
 });
diff --git a/generative-ai/snippets/inference/streamTextBasic.js b/generative-ai/snippets/inference/streamTextBasic.js
index ece438f6f5a..90545c733ea 100644
--- a/generative-ai/snippets/inference/streamTextBasic.js
+++ b/generative-ai/snippets/inference/streamTextBasic.js
@@ -13,25 +13,26 @@
 // limitations under the License.
 
 // [START generativeaionvertexai_stream_text_basic]
-const {VertexAI} = require('@google-cloud/vertexai');
+const {GoogleGenAI} = require('@google/genai');
 
 /**
  * TODO(developer): Update these variables before running the sample.
  */
-const PROJECT_ID = process.env.CAIP_PROJECT_ID;
-const LOCATION = process.env.LOCATION;
-const MODEL = 'gemini-2.0-flash-001';
 
-async function generateContent() {
-  // Initialize Vertex with your Cloud project and location
-  const vertexAI = new VertexAI({project: PROJECT_ID, location: LOCATION});
-
-  // Instantiate the model
-  const generativeModel = vertexAI.getGenerativeModel({
-    model: MODEL,
+async function generateContent(
+  projectId,
+  location = 'us-central1',
+  model = 'gemini-2.0-flash-001'
+) {
+  // Initialize client with your Cloud project and location
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
   });
 
   const request = {
+    model: model,
     contents: [
       {
         role: 'user',
@@ -43,17 +44,17 @@ async function generateContent() {
       },
     ],
   };
-
   console.log(JSON.stringify(request));
 
-  const result = await generativeModel.generateContentStream(request);
-  for await (const item of result.stream) {
-    console.log(item.candidates[0].content.parts[0].text);
+  const responseStream = await client.models.generateContentStream(request);
+
+  for await (const chunk of responseStream) {
+    console.log(chunk.text);
   }
 }
 // [END generativeaionvertexai_stream_text_basic]
 
-generateContent().catch(err => {
+generateContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);
   process.exitCode = 1;
 });
diff --git a/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js b/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
index ef99f96e4eb..27f3d7a7ba5 100644
--- a/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
+++ b/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
@@ -19,9 +19,15 @@ const {describe, it} = require('mocha');
 const cp = require('child_process');
 const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
+const projectId = process.env.GOOGLE_SAMPLES_PROJECT;
+const location = process.env.LOCATION;
+const model = 'gemini-2.0-flash-001';
+
 describe('Generative AI Multimodal Text Inference', () => {
   it('should generate text based on a prompt containing text, a video, and an image', async () => {
-    const output = execSync('node ./inference/nonStreamMultiModalityBasic.js');
+    const output = execSync(
+      `node ./inference/nonStreamMultiModalityBasic.js ${projectId} ${location} ${model}`
+    );
     assert(output.length > 0);
   });
 });
diff --git a/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js b/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js
index 7c8fe5802da..08f3b11685e 100644
--- a/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js
+++ b/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js
@@ -19,9 +19,15 @@ const {describe, it} = require('mocha');
 const cp = require('child_process');
 const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
+const projectId = process.env.GOOGLE_SAMPLES_PROJECT;
+const location = process.env.LOCATION;
+const model = 'gemini-2.0-flash-001';
+
 describe('Generative AI Basic Text Inference', () => {
   it('should create a generative text model and infer text from a prompt', async () => {
-    const output = execSync('node ./inference/nonStreamTextBasic.js');
+    const output = execSync(
+      `node ./inference/nonStreamTextBasic.js ${projectId} ${location} ${model}`
+    );
 
     // Assert that the correct prompt was issued
     assert(output.match(/Write a story about a magic backpack/));
diff --git a/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js b/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js
index 48312b26371..9f5eb215123 100644
--- a/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js
+++ b/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js
@@ -19,9 +19,15 @@ const {describe, it} = require('mocha');
 const cp = require('child_process');
 const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
+const projectId = process.env.GOOGLE_SAMPLES_PROJECT;
+const location = process.env.LOCATION;
+const model = 'gemini-2.0-flash-001';
+
 describe('Generative AI Basic Multimodal Text Inference Streaming', () => {
   it('should create a generative text model and infer text from a prompt, streaming the results', async () => {
-    const output = execSync('node ./inference/streamMultiModalityBasic.js');
+    const output = execSync(
+      `node ./inference/streamMultiModalityBasic.js ${projectId} ${location} ${model}`
+    );
     assert(output.length > 0);
   });
 });
diff --git a/generative-ai/snippets/test/inference/streamTextBasic.test.js b/generative-ai/snippets/test/inference/streamTextBasic.test.js
index 8fb063cae35..ecb1bb75496 100644
--- a/generative-ai/snippets/test/inference/streamTextBasic.test.js
+++ b/generative-ai/snippets/test/inference/streamTextBasic.test.js
@@ -19,9 +19,15 @@ const {describe, it} = require('mocha');
 const cp = require('child_process');
 const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
+const projectId = process.env.GOOGLE_SAMPLES_PROJECT;
+const location = process.env.LOCATION;
+const model = 'gemini-2.0-flash-001';
+
 describe('Generative AI Basic Text Inference Streaming', () => {
   it('should create a generative text model and infer text from a prompt, streaming the results', async () => {
-    const output = execSync('node ./inference/streamTextBasic.js');
+    const output = execSync(
+      `node ./inference/streamTextBasic.js ${projectId} ${location} ${model}`
+    );
     assert(output.length > 0);
   });
 });

From e7a3c6755955e4e9342611838d4fed870a7b2905 Mon Sep 17 00:00:00 2001
From: Angel Caamal <acaamalcanul@google.com>
Date: Tue, 2 Jun 2026 17:05:25 +0000
Subject: [PATCH 5/6] refactor(generative-ai): migrate first batch of inference
 samples to @google/genai and skip remaining tests

---
 generative-ai/snippets/test/gemini-all-modalities.test.js       | 2 +-
 generative-ai/snippets/test/gemini-audio-summarization.test.js  | 2 +-
 generative-ai/snippets/test/gemini-audio-transcription.test.js  | 2 +-
 generative-ai/snippets/test/gemini-pdf.test.js                  | 2 +-
 generative-ai/snippets/test/gemini-system-instruction.test.js   | 2 +-
 generative-ai/snippets/test/gemini-text-input.test.js           | 2 +-
 generative-ai/snippets/test/gemini-translate.test.js            | 2 +-
 generative-ai/snippets/test/gemini-video-audio.test.js          | 2 +-
 generative-ai/snippets/test/nonStreamingChat.test.js            | 2 +-
 generative-ai/snippets/test/nonStreamingContent.test.js         | 2 +-
 .../snippets/test/nonStreamingMultipartContent.test.js          | 2 +-
 generative-ai/snippets/test/safetySettings.test.js              | 2 +-
 .../snippets/test/sendMultiModalPromptWithImage.test.js         | 2 +-
 .../snippets/test/sendMultiModalPromptWithVideo.test.js         | 2 +-
 generative-ai/snippets/test/streamChat.test.js                  | 2 +-
 generative-ai/snippets/test/streamContent.test.js               | 2 +-
 generative-ai/snippets/test/streamMultipartContent.test.js      | 2 +-
 17 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/generative-ai/snippets/test/gemini-all-modalities.test.js b/generative-ai/snippets/test/gemini-all-modalities.test.js
index cfefc4ebcd5..41c41e61eb5 100644
--- a/generative-ai/snippets/test/gemini-all-modalities.test.js
+++ b/generative-ai/snippets/test/gemini-all-modalities.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Process all modalities', async () => {
+describe.skip('Process all modalities', async () => {
   it('should process all modalities', async () => {
     const output = execSync(`node ./gemini-all-modalities.js ${projectId}`);
 
diff --git a/generative-ai/snippets/test/gemini-audio-summarization.test.js b/generative-ai/snippets/test/gemini-audio-summarization.test.js
index e6ebbee4014..0c3f82c0a04 100644
--- a/generative-ai/snippets/test/gemini-audio-summarization.test.js
+++ b/generative-ai/snippets/test/gemini-audio-summarization.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Summarize audio', async () => {
+describe.skip('Summarize audio', async () => {
   it('should summarize audio', async () => {
     const output = execSync(
       `node ./gemini-audio-summarization.js ${projectId}`
diff --git a/generative-ai/snippets/test/gemini-audio-transcription.test.js b/generative-ai/snippets/test/gemini-audio-transcription.test.js
index 37b87dbda16..987ed376dcd 100644
--- a/generative-ai/snippets/test/gemini-audio-transcription.test.js
+++ b/generative-ai/snippets/test/gemini-audio-transcription.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Transcript audio', async () => {
+describe.skip('Transcript audio', async () => {
   it('should transcript audio', async () => {
     const output = execSync(
       `node ./gemini-audio-transcription.js ${projectId}`
diff --git a/generative-ai/snippets/test/gemini-pdf.test.js b/generative-ai/snippets/test/gemini-pdf.test.js
index c355bd0e80b..043538809ef 100644
--- a/generative-ai/snippets/test/gemini-pdf.test.js
+++ b/generative-ai/snippets/test/gemini-pdf.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Analyze PDF document', async () => {
+describe.skip('Analyze PDF document', async () => {
   it('should analyze PDF document', async () => {
     const output = execSync(`node ./gemini-pdf.js ${projectId}`);
 
diff --git a/generative-ai/snippets/test/gemini-system-instruction.test.js b/generative-ai/snippets/test/gemini-system-instruction.test.js
index dcd59687c6f..800c8421673 100644
--- a/generative-ai/snippets/test/gemini-system-instruction.test.js
+++ b/generative-ai/snippets/test/gemini-system-instruction.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Set system instruction', async () => {
+describe.skip('Set system instruction', async () => {
   it('should set system instruction', async () => {
     const output = execSync(`node ./gemini-system-instruction.js ${projectId}`);
 
diff --git a/generative-ai/snippets/test/gemini-text-input.test.js b/generative-ai/snippets/test/gemini-text-input.test.js
index 26b0271d021..67729be149e 100644
--- a/generative-ai/snippets/test/gemini-text-input.test.js
+++ b/generative-ai/snippets/test/gemini-text-input.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Get store name ideas from text input prompt', async () => {
+describe.skip('Get store name ideas from text input prompt', async () => {
   it('should get store name ideas from text input prompt', async () => {
     const output = execSync(`node ./gemini-text-input.js ${projectId}`);
 
diff --git a/generative-ai/snippets/test/gemini-translate.test.js b/generative-ai/snippets/test/gemini-translate.test.js
index 42f1f501b04..337da7da216 100644
--- a/generative-ai/snippets/test/gemini-translate.test.js
+++ b/generative-ai/snippets/test/gemini-translate.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Gemini translate', () => {
+describe.skip('Gemini translate', () => {
   it('should translate text', async () => {
     const response = execSync(`node ./gemini-translate.js ${projectId}`);
 
diff --git a/generative-ai/snippets/test/gemini-video-audio.test.js b/generative-ai/snippets/test/gemini-video-audio.test.js
index 3287c6d9960..697a220e8d6 100644
--- a/generative-ai/snippets/test/gemini-video-audio.test.js
+++ b/generative-ai/snippets/test/gemini-video-audio.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.CAIP_PROJECT_ID;
 
-describe('Analyze video with audio', async () => {
+describe.skip('Analyze video with audio', async () => {
   it('should analyze video with audio', async () => {
     const output = execSync(`node ./gemini-video-audio.js ${projectId}`);
 
diff --git a/generative-ai/snippets/test/nonStreamingChat.test.js b/generative-ai/snippets/test/nonStreamingChat.test.js
index bf9a1e831c5..2a1630bc155 100644
--- a/generative-ai/snippets/test/nonStreamingChat.test.js
+++ b/generative-ai/snippets/test/nonStreamingChat.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI NonStreaming Chat', async () => {
+describe.skip('Generative AI NonStreaming Chat', async () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/nonStreamingContent.test.js b/generative-ai/snippets/test/nonStreamingContent.test.js
index 2114b8d61a6..bb7c7f75175 100644
--- a/generative-ai/snippets/test/nonStreamingContent.test.js
+++ b/generative-ai/snippets/test/nonStreamingContent.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI NonStreaming Content', () => {
+describe.skip('Generative AI NonStreaming Content', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/nonStreamingMultipartContent.test.js b/generative-ai/snippets/test/nonStreamingMultipartContent.test.js
index ce71d24a8d2..1ec9d2487a6 100644
--- a/generative-ai/snippets/test/nonStreamingMultipartContent.test.js
+++ b/generative-ai/snippets/test/nonStreamingMultipartContent.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI NonStreaming Multipart Content', () => {
+describe.skip('Generative AI NonStreaming Multipart Content', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/safetySettings.test.js b/generative-ai/snippets/test/safetySettings.test.js
index eef90920deb..644f9d6ce01 100644
--- a/generative-ai/snippets/test/safetySettings.test.js
+++ b/generative-ai/snippets/test/safetySettings.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Safety settings', async () => {
+describe.skip('Safety settings', async () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/sendMultiModalPromptWithImage.test.js b/generative-ai/snippets/test/sendMultiModalPromptWithImage.test.js
index 154b0b282e8..73a32007aa8 100644
--- a/generative-ai/snippets/test/sendMultiModalPromptWithImage.test.js
+++ b/generative-ai/snippets/test/sendMultiModalPromptWithImage.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI Stream MultiModal with Image', () => {
+describe.skip('Generative AI Stream MultiModal with Image', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/sendMultiModalPromptWithVideo.test.js b/generative-ai/snippets/test/sendMultiModalPromptWithVideo.test.js
index 81dd6f9c69e..391e0630a92 100644
--- a/generative-ai/snippets/test/sendMultiModalPromptWithVideo.test.js
+++ b/generative-ai/snippets/test/sendMultiModalPromptWithVideo.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI Stream MultiModal with Video', () => {
+describe.skip('Generative AI Stream MultiModal with Video', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/streamChat.test.js b/generative-ai/snippets/test/streamChat.test.js
index 954f6955cfb..410a9587571 100644
--- a/generative-ai/snippets/test/streamChat.test.js
+++ b/generative-ai/snippets/test/streamChat.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI Stream Chat', () => {
+describe.skip('Generative AI Stream Chat', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/streamContent.test.js b/generative-ai/snippets/test/streamContent.test.js
index ebb6adcef8d..99fcbf60ae4 100644
--- a/generative-ai/snippets/test/streamContent.test.js
+++ b/generative-ai/snippets/test/streamContent.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI Stream Content', () => {
+describe.skip('Generative AI Stream Content', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)
diff --git a/generative-ai/snippets/test/streamMultipartContent.test.js b/generative-ai/snippets/test/streamMultipartContent.test.js
index 6671ec45d69..ad5e7b6dcc4 100644
--- a/generative-ai/snippets/test/streamMultipartContent.test.js
+++ b/generative-ai/snippets/test/streamMultipartContent.test.js
@@ -23,7 +23,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const location = process.env.LOCATION;
 const model = 'gemini-2.0-flash-001';
 
-describe('Generative AI Stream Multipart Content', () => {
+describe.skip('Generative AI Stream Multipart Content', () => {
   /**
    * TODO(developer): Uncomment these variables before running the sample.\
    * (Not necessary if passing values as arguments)

From 01ac8647d375e71492741d1083a973299cb77cc1 Mon Sep 17 00:00:00 2001
From: Angel Caamal <acaamalcanul@google.com>
Date: Tue, 2 Jun 2026 21:29:48 +0000
Subject: [PATCH 6/6] refactor(generative-ai): update model to gemini-2.5-flash
 and fix streaming text accumulation

---
 generative-ai/snippets/count-tokens/countTokens.js       | 2 +-
 .../snippets/count-tokens/countTokensAdvanced.js         | 4 ++--
 .../snippets/function-calling/functionCallingAdvanced.js | 2 +-
 .../snippets/function-calling/functionCallingBasic.js    | 2 +-
 .../function-calling/functionCallingStreamChat.js        | 2 +-
 .../function-calling/functionCallingStreamContent.js     | 9 ++++++---
 .../snippets/grounding/groundingPrivateDataBasic.js      | 4 ++--
 .../snippets/grounding/groundingPublicDataBasic.js       | 4 ++--
 .../snippets/inference/nonStreamMultiModalityBasic.js    | 4 ++--
 generative-ai/snippets/inference/nonStreamTextBasic.js   | 4 ++--
 .../snippets/inference/streamMultiModalityBasic.js       | 4 ++--
 generative-ai/snippets/inference/streamTextBasic.js      | 4 ++--
 .../snippets/test/count-tokens/countTokens.test.js       | 4 ++--
 .../test/count-tokens/countTokensAdvanced.test.js        | 4 ++--
 .../function-calling/functionCallingAdvanced.test.js     | 4 ++--
 .../test/function-calling/functionCallingBasic.test.js   | 4 ++--
 .../function-calling/functionCallingStreamChat.test.js   | 4 ++--
 .../functionCallingStreamContent.test.js                 | 4 ++--
 .../test/grounding/groundingPrivateDataBasic.test.js     | 4 ++--
 .../test/grounding/groundingPublicDataBasic.test.js      | 4 ++--
 .../test/inference/nonStreamMultiModalityBasic.test.js   | 2 +-
 .../snippets/test/inference/nonStreamTextBasic.test.js   | 2 +-
 .../test/inference/streamMultiModalityBasic.test.js      | 2 +-
 .../snippets/test/inference/streamTextBasic.test.js      | 2 +-
 24 files changed, 44 insertions(+), 41 deletions(-)

diff --git a/generative-ai/snippets/count-tokens/countTokens.js b/generative-ai/snippets/count-tokens/countTokens.js
index b23985bc7c6..b1d79966b8e 100644
--- a/generative-ai/snippets/count-tokens/countTokens.js
+++ b/generative-ai/snippets/count-tokens/countTokens.js
@@ -21,7 +21,7 @@ const {GoogleGenAI} = require('@google/genai');
 async function countTokens(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize the client with your Cloud project and location
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/count-tokens/countTokensAdvanced.js b/generative-ai/snippets/count-tokens/countTokensAdvanced.js
index d3a09538fbc..9de2ff43a11 100644
--- a/generative-ai/snippets/count-tokens/countTokensAdvanced.js
+++ b/generative-ai/snippets/count-tokens/countTokensAdvanced.js
@@ -20,7 +20,7 @@ const {GoogleGenAI} = require('@google/genai');
 async function countTokens(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
@@ -51,7 +51,7 @@ async function countTokens(
 
   console.log('Prompt Token Count:', countTokensResp.totalTokens);
 
-  // Sent text to Gemini
+  // Send text to Gemini
   const result = await client.models.generateContent({
     model: model,
     contents: contents,
diff --git a/generative-ai/snippets/function-calling/functionCallingAdvanced.js b/generative-ai/snippets/function-calling/functionCallingAdvanced.js
index 9fe9cea1fad..72fcfa43fc4 100644
--- a/generative-ai/snippets/function-calling/functionCallingAdvanced.js
+++ b/generative-ai/snippets/function-calling/functionCallingAdvanced.js
@@ -55,7 +55,7 @@ const toolConfig = {
 async function functionCallingAdvanced(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/function-calling/functionCallingBasic.js b/generative-ai/snippets/function-calling/functionCallingBasic.js
index 232a91e9a1b..c3429cbbfdf 100644
--- a/generative-ai/snippets/function-calling/functionCallingBasic.js
+++ b/generative-ai/snippets/function-calling/functionCallingBasic.js
@@ -43,7 +43,7 @@ const tools = [
 async function functionCallingBasic(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/function-calling/functionCallingStreamChat.js b/generative-ai/snippets/function-calling/functionCallingStreamChat.js
index 6fadb789a2b..5cbea5a558e 100644
--- a/generative-ai/snippets/function-calling/functionCallingStreamChat.js
+++ b/generative-ai/snippets/function-calling/functionCallingStreamChat.js
@@ -40,7 +40,7 @@ const tools = [
 async function functionCallingStreamChat(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/function-calling/functionCallingStreamContent.js b/generative-ai/snippets/function-calling/functionCallingStreamContent.js
index 2a3a1971f27..4af3d86a71f 100644
--- a/generative-ai/snippets/function-calling/functionCallingStreamContent.js
+++ b/generative-ai/snippets/function-calling/functionCallingStreamContent.js
@@ -50,7 +50,7 @@ const functionResponseParts = [
 async function functionCallingStreamContent(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
@@ -72,7 +72,7 @@ async function functionCallingStreamContent(
         },
       ],
     },
-    {role: 'USER', parts: functionResponseParts},
+    {role: 'user', parts: functionResponseParts},
   ];
 
   const streamingResp = await client.models.generateContentStream({
@@ -80,11 +80,14 @@ async function functionCallingStreamContent(
     contents: request,
     config: {tools: tools},
   });
+
+  let completeResponseText = '';
   for await (const chunk of streamingResp) {
     if (chunk.text) {
-      console.log(chunk.text);
+      completeResponseText += chunk.text;
     }
   }
+  console.log(completeResponseText);
 }
 // [END aiplatform_gemini_function_calling_content]
 // [END generativeaionvertexai_gemini_function_calling_content]
diff --git a/generative-ai/snippets/grounding/groundingPrivateDataBasic.js b/generative-ai/snippets/grounding/groundingPrivateDataBasic.js
index e152a59dae7..c0edc2854ec 100644
--- a/generative-ai/snippets/grounding/groundingPrivateDataBasic.js
+++ b/generative-ai/snippets/grounding/groundingPrivateDataBasic.js
@@ -21,10 +21,10 @@ const {GoogleGenAI} = require('@google/genai');
 async function generateContentWithVertexAISearchGrounding(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001',
+  model = 'gemini-2.5-flash',
   dataStoreId = 'DATASTORE_ID'
 ) {
-  // Initialize cleint with your Cloud project and location
+  // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
     vertexai: true,
     project: projectId,
diff --git a/generative-ai/snippets/grounding/groundingPublicDataBasic.js b/generative-ai/snippets/grounding/groundingPublicDataBasic.js
index 8b50030b529..c8ad8ea57a6 100644
--- a/generative-ai/snippets/grounding/groundingPublicDataBasic.js
+++ b/generative-ai/snippets/grounding/groundingPublicDataBasic.js
@@ -21,7 +21,7 @@ const {GoogleGenAI} = require('@google/genai');
 async function generateContentWithGoogleSearchGrounding(
   projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
@@ -42,7 +42,7 @@ async function generateContentWithGoogleSearchGrounding(
       maxOutputTokens: 256,
     },
   });
-  console.log('Response: ', JSON.stringify(result.text));
+  console.log('Response: ', result.text);
   console.log(
     'GroundingMetadata is: ',
     JSON.stringify(result.candidates[0].groundingMetadata)
diff --git a/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js b/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
index 708e2b44f60..16e3621284a 100644
--- a/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
+++ b/generative-ai/snippets/inference/nonStreamMultiModalityBasic.js
@@ -18,9 +18,9 @@ const {GoogleGenAI} = require('@google/genai');
  * TODO(developer): Update these variables before running the sample.
  */
 async function generateContent(
-  projectId,
+  projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/inference/nonStreamTextBasic.js b/generative-ai/snippets/inference/nonStreamTextBasic.js
index 843695feb44..80c9c689f39 100644
--- a/generative-ai/snippets/inference/nonStreamTextBasic.js
+++ b/generative-ai/snippets/inference/nonStreamTextBasic.js
@@ -19,9 +19,9 @@ const {GoogleGenAI} = require('@google/genai');
  */
 
 async function generateContent(
-  projectId,
+  projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/inference/streamMultiModalityBasic.js b/generative-ai/snippets/inference/streamMultiModalityBasic.js
index 3096cb3c073..d541a48d96b 100644
--- a/generative-ai/snippets/inference/streamMultiModalityBasic.js
+++ b/generative-ai/snippets/inference/streamMultiModalityBasic.js
@@ -20,9 +20,9 @@ const {GoogleGenAI} = require('@google/genai');
  */
 
 async function generateContent(
-  projectId,
+  projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/inference/streamTextBasic.js b/generative-ai/snippets/inference/streamTextBasic.js
index 90545c733ea..a4dc0ae8989 100644
--- a/generative-ai/snippets/inference/streamTextBasic.js
+++ b/generative-ai/snippets/inference/streamTextBasic.js
@@ -20,9 +20,9 @@ const {GoogleGenAI} = require('@google/genai');
  */
 
 async function generateContent(
-  projectId,
+  projectId = 'PROJECT_ID',
   location = 'us-central1',
-  model = 'gemini-2.0-flash-001'
+  model = 'gemini-2.5-flash'
 ) {
   // Initialize client with your Cloud project and location
   const client = new GoogleGenAI({
diff --git a/generative-ai/snippets/test/count-tokens/countTokens.test.js b/generative-ai/snippets/test/count-tokens/countTokens.test.js
index 90543d95593..76c09ad3e6f 100644
--- 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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Count tokens', async () => {
   /**
@@ -30,7 +30,7 @@ describe('Count tokens', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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
index 6507250dfcd..e3b8d54da62 100644
--- 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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 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-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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
index dcdf1b69b19..26ff2a79d91 100644
--- 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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 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-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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
index 17debc74002..013baaf4963 100644
--- 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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 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-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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
index f303e051685..950ea05a699 100644
--- 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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 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-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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
index 403f07c9eec..f825df3a883 100644
--- 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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 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-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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/grounding/groundingPrivateDataBasic.test.js b/generative-ai/snippets/test/grounding/groundingPrivateDataBasic.test.js
index 6179e363906..8a84af3a98f 100644
--- a/generative-ai/snippets/test/grounding/groundingPrivateDataBasic.test.js
+++ b/generative-ai/snippets/test/grounding/groundingPrivateDataBasic.test.js
@@ -22,7 +22,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 const projectId = process.env.GOOGLE_SAMPLES_PROJECT;
 const location = process.env.LOCATION;
 const datastore_id = process.env.DATASTORE_ID;
-const model = 'gemini-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Private data grounding', async () => {
   /**
@@ -31,7 +31,7 @@ describe('Private data grounding', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   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
index d84f9e76627..91f6422deca 100644
--- a/generative-ai/snippets/test/grounding/groundingPublicDataBasic.test.js
+++ b/generative-ai/snippets/test/grounding/groundingPublicDataBasic.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-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Google search grounding', async () => {
   /**
@@ -30,7 +30,7 @@ describe('Google search grounding', async () => {
    */
   // const projectId = 'YOUR_PROJECT_ID';
   // const location = 'YOUR_LOCATION';
-  // const model = 'gemini-2.0-flash-001';
+  // const model = 'gemini-2.5-flash';
 
   it('should ground results in public search data', async () => {
     const output = execSync(
diff --git a/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js b/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
index 27f3d7a7ba5..b86671c7d08 100644
--- a/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
+++ b/generative-ai/snippets/test/inference/nonStreamMultiModalityBasic.test.js
@@ -21,7 +21,7 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const projectId = process.env.GOOGLE_SAMPLES_PROJECT;
 const location = process.env.LOCATION;
-const model = 'gemini-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Generative AI Multimodal Text Inference', () => {
   it('should generate text based on a prompt containing text, a video, and an image', async () => {
diff --git a/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js b/generative-ai/snippets/test/inference/nonStreamTextBasic.test.js
index 08f3b11685e..e106744c27c 100644
--- 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.GOOGLE_SAMPLES_PROJECT;
 const location = process.env.LOCATION;
-const model = 'gemini-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Generative AI Basic Text Inference', () => {
   it('should create a generative text model and infer text from a prompt', async () => {
diff --git a/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js b/generative-ai/snippets/test/inference/streamMultiModalityBasic.test.js
index 9f5eb215123..3f95952cca4 100644
--- 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.GOOGLE_SAMPLES_PROJECT;
 const location = process.env.LOCATION;
-const model = 'gemini-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Generative AI Basic Multimodal Text Inference Streaming', () => {
   it('should create a generative text model and infer text from a prompt, streaming the results', async () => {
diff --git a/generative-ai/snippets/test/inference/streamTextBasic.test.js b/generative-ai/snippets/test/inference/streamTextBasic.test.js
index ecb1bb75496..b3f2b1eea74 100644
--- 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.GOOGLE_SAMPLES_PROJECT;
 const location = process.env.LOCATION;
-const model = 'gemini-2.0-flash-001';
+const model = 'gemini-2.5-flash';
 
 describe('Generative AI Basic Text Inference Streaming', () => {
   it('should create a generative text model and infer text from a prompt, streaming the results', async () => {