refactor: migrate countTokens samples to @google/genai, update config, and remove deprecated totalBillableCharacters

Author: angelcaamal

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]

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);

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"
   },

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+/));
   });
 });