feat(genai): Add samples for Embeddings, Provisioned Throughput, Safety

genai/embeddings/embeddings-docretrieval-with-txt.js

@@ -0,0 +1,59 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+// [START googlegenaisdk_embeddings_docretrieval_with_txt]
+const {GoogleGenAI} = require('@google/genai');
+
+const GOOGLE_CLOUD_PROJECT = process.env.GOOGLE_CLOUD_PROJECT;
+
+async function generateEmbeddingsForRetrieval(
+  projectId = GOOGLE_CLOUD_PROJECT
+) {
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+  });
+
+  const prompt = [
+    "How do I get a driver's license/learner's permit?",
+    "How long is my driver's license valid for?",
+    "Driver's knowledge test study guide",
+  ];
+
+  const response = await client.models.embedContent({
+    model: 'gemini-embedding-001',
+    contents: prompt,
+    config: {
+      taskType: 'RETRIEVAL_DOCUMENT', // Optional
+      outputDimensionality: 3072, // Optional
+      title: "Driver's License", // Optional
+    },
+  });
+
+  console.log(response);
+
+  // Example response:
+  //  embeddings=[ContentEmbedding(values=[-0.06302902102470398, 0.00928034819662571, 0.014716853387653828, -0.028747491538524628, ... ],
+  //  statistics=ContentEmbeddingStatistics(truncated=False, token_count=13.0))]
+  //  metadata=EmbedContentMetadata(billable_character_count=112)
+
+  return response;
+}
+// [END googlegenaisdk_embeddings_docretrieval_with_txt]
+
+module.exports = {
+  generateEmbeddingsForRetrieval,
+};

genai/express-mode/api-key-example.js

@@ -0,0 +1,43 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+// [START googlegenaisdk_vertexai_express_mode]
+const {GoogleGenAI} = require('@google/genai');
+const API_KEY = 'PUT HERE YOUR API KEY';
+
+async function generateWithApiKey(apiKey = API_KEY) {
+  const client = new GoogleGenAI({
+    vertexai: true,
+    apiKey: apiKey,
+  });
+
+  const response = await client.models.generateContentStream({
+    model: 'gemini-2.5-flash',
+    contents: 'Explain bubble sort to me.',
+  });
+
+  console.log(response.text);
+
+  // Example response:
+  //  Bubble Sort is a simple sorting algorithm that repeatedly steps through the list
+
+  return response;
+}
+// [END googlegenaisdk_vertexai_express_mode]
+
+module.exports = {
+  generateWithApiKey,
+};

genai/package.json

@@ -15,10 +15,11 @@
   "dependencies": {
     "@google/genai": "1.30.0",
     "axios": "^1.6.2",
+    "luxon": "^3.7.2",
+    "node-fetch": "^2.7.0",
+    "openai": "^6.9.1",
     "canvas": "^3.2.0",
     "google-auth-library": "^10.3.0",
-    "luxon": "^3.7.1",
-    "openai": "^5.19.1",
     "proxyquire": "^2.1.3",
     "supertest": "^7.0.0"
   },

genai/provisioned-throughput/provisionedthroughput-with-txt.js

@@ -0,0 +1,62 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+// [START googlegenaisdk_provisionedthroughput_with_txt]
+const {GoogleGenAI} = require('@google/genai');
+
+const GOOGLE_CLOUD_PROJECT = process.env.GOOGLE_CLOUD_PROJECT;
+const GOOGLE_CLOUD_LOCATION = process.env.GOOGLE_CLOUD_LOCATION || 'global';
+
+async function generateWithProvisionedThroughput(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION
+) {
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+    httpOptions: {
+      apiVersion: 'v1',
+      headers: {
+        // Options:
+        // - "dedicated": Use Provisioned Throughput
+        // - "shared": Use pay-as-you-go
+        // https://cloud.google.com/vertex-ai/generative-ai/docs/use-provisioned-throughput
+        'X-Vertex-AI-LLM-Request-Type': 'shared',
+      },
+    },
+  });
+
+  const response = await client.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents: 'How does AI work?',
+  });
+
+  console.log(response.text);
+
+  // Example response:
+  //  Okay, let's break down how AI works. It's a broad field, so I'll focus on the ...
+  //  Here's a simplified overview:
+  //  ...
+
+  return response.text;
+}
+
+// [END googlegenaisdk_provisionedthroughput_with_txt]
+
+module.exports = {
+  generateWithProvisionedThroughput,
+};

genai/safety/safety-with-txt.js

@@ -0,0 +1,115 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+// [START googlegenaisdk_safety_with_txt]
+const {GoogleGenAI} = require('@google/genai');
+
+const GOOGLE_CLOUD_PROJECT = process.env.GOOGLE_CLOUD_PROJECT;
+const GOOGLE_CLOUD_LOCATION = process.env.GOOGLE_CLOUD_LOCATION || 'global';
+
+async function generateWithSafetySettings(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION
+) {
+  const client = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+  });
+
+  const systemInstruction = 'Be as mean as possible.';
+
+  const prompt =
+    'Write a list of 5 disrespectful things that I might say to the universe after stubbing my toe in the dark.';
+
+  const safetySettings = [
+    {
+      category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+      threshold: 'BLOCK_LOW_AND_ABOVE',
+    },
+    {
+      category: 'HARM_CATEGORY_HARASSMENT',
+      threshold: 'BLOCK_LOW_AND_ABOVE',
+    },
+    {
+      category: 'HARM_CATEGORY_HATE_SPEECH',
+      threshold: 'BLOCK_LOW_AND_ABOVE',
+    },
+    {
+      category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
+      threshold: 'BLOCK_LOW_AND_ABOVE',
+    },
+  ];
+
+  const response = await client.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents: prompt,
+    config: {
+      systemInstruction: systemInstruction,
+      safetySettings: safetySettings,
+    },
+  });
+
+  // console.log(response.text);
+  // console.log(response.candidates[0].finishMessage);
+  //
+  // for (const each of response.candidates[0].safetyRatings) {
+  //   console.log('\nCategory:', String(each.category));
+  //   console.log('Is Blocked:', each.blocked);
+  //   console.log('Probability:', each.probability);
+  //   console.log('Probability Score:', each.probabilityScore);
+  //   console.log('Severity:', each.severity);
+  //   console.log('Severity Score:', each.severityScore);
+  // }
+
+  // Example response:
+  //
+  //     Category:  HarmCategory.HARM_CATEGORY_HATE_SPEECH
+  //     Is Blocked: False
+  //     Probability:  HarmProbability.NEGLIGIBLE
+  //     Probability Score:  2.547714e-05
+  //     Severity: HarmSeverity.HARM_SEVERITY_NEGLIGIBLE
+  //     Severity Score: None
+  //
+  //     Category:  HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT
+  //     Is Blocked: False
+  //     Probability:  HarmProbability.NEGLIGIBLE
+  //     Probability Score:  3.6103818e-06
+  //     Severity: HarmSeverity.HARM_SEVERITY_NEGLIGIBLE
+  //     Severity Score: None
+  //
+  //     Category:  HarmCategory.HARM_CATEGORY_HARASSMENT
+  //     Is Blocked: True
+  //     Probability:  HarmProbability.MEDIUM
+  //     Probability Score:  0.71599233
+  //     Severity: HarmSeverity.HARM_SEVERITY_MEDIUM
+  //     Severity Score: 0.30782545
+  //
+  //     Category:  HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT
+  //     Is Blocked: False
+  //     Probability:  HarmProbability.NEGLIGIBLE
+  //     Probability Score:  1.5624657e-05
+  //     Severity: HarmSeverity.HARM_SEVERITY_NEGLIGIBLE
+  //     Severity Score: None
+
+  return response;
+}
+
+// [END googlegenaisdk_safety_with_txt]
+
+module.exports = {
+  generateWithSafetySettings,
+};

genai/test/api-key-example.test.js

@@ -0,0 +1,53 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+const {assert} = require('chai');
+const {describe, it} = require('mocha');
+const {delay} = require('./util');
+const proxyquire = require('proxyquire').noCallThru();
+
+describe('vertexai-express-mode', () => {
+  it('should call generateContentStream and return the mocked response', async function () {
+    this.timeout(10000);
+
+    this.retries(4);
+    await delay(this.test);
+
+    const mockGenerateContentStreamResult = {
+      text: 'Bubble sort works by repeatedly swapping adjacent elements until sorted.',
+    };
+
+    class MockModels {
+      async generateContentStream() {
+        return mockGenerateContentStreamResult;
+      }
+    }
+
+    class MockGoogleGenAI {
+      constructor() {
+        this.models = new MockModels();
+      }
+    }
+
+    const sample = proxyquire('../express-mode/api-key-example.js', {
+      '@google/genai': {GoogleGenAI: MockGoogleGenAI},
+    });
+
+    const response = await sample.generateWithApiKey('FAKE_API_KEY');
+
+    assert.strictEqual(response.text, mockGenerateContentStreamResult.text);
+  });
+});

genai/test/embeddings-docretrieval-with-txt.test.js

@@ -0,0 +1,31 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+'use strict';
+
+const {assert} = require('chai');
+const {describe, it} = require('mocha');
+
+const projectId = process.env.CAIP_PROJECT_ID;
+const sample = require('../embeddings/embeddings-docretrieval-with-txt.js');
+const {delay} = require('./util');
+
+describe('embeddings-docretrieval-with-txt', () => {
+  it('should return an object containing embeddings and metadata', async function () {
+    this.retries(4);
+    await delay(this.test);
+    const result = await sample.generateEmbeddingsForRetrieval(projectId);
+    assert.containsAllKeys(result, ['embeddings', 'metadata']);
+  });
+});

genai/test/imggen-mmflash-edit-img-with-txt-img.test.js

@@ -28,6 +28,7 @@ describe('imggen-mmflash-edit-img-with-txt-img', async () => {
     this.retries(4);
     await delay(this.test);
     const response = await sample.generateImage(projectId, location);
+    console.log(response);
     assert(response);
   });
 });