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

Co-authored-by: Guiners <rkoza@softserveinc.com>
Co-authored-by: Holt Skinner <13262395+holtskinner@users.noreply.github.com>

Author: 3 people

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/embeddings/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "nodejs-genai-embeddings-samples",
+  "version": "0.0.1",
+  "private": true,
+  "license": "Apache-2.0",
+  "author": "Google LLC",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/GoogleCloudPlatform/nodejs-docs-samples.git"
+  },
+  "engines": {
+    "node": ">=16.0.0"
+  },
+  "files": [
+    "*.js"
+  ],
+  "scripts": {
+    "test": "c8 mocha -p -j 2 --timeout 2400000 test/*.test.js test/**/*.test.js"
+  },
+  "dependencies": {
+    "@google/genai": "1.30.0"
+  },
+  "devDependencies": {
+    "c8": "^10.0.0",
+    "chai": "^4.5.0",
+    "mocha": "^10.0.0"
+  }
+}

genai/embeddings/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-docretrieval-with-txt.js');
+const {delay} = require('../../test/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/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 = 'YOUR_EXPRESS_MODE_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/express-mode/package.json

@@ -0,0 +1,29 @@
+{
+  "name": "nodejs-genai-express-mode-samples",
+  "version": "0.0.1",
+  "private": true,
+  "license": "Apache-2.0",
+  "author": "Google LLC",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/GoogleCloudPlatform/nodejs-docs-samples.git"
+  },
+  "engines": {
+    "node": ">=16.0.0"
+  },
+  "files": [
+    "*.js"
+  ],
+  "scripts": {
+    "test": "c8 mocha -p -j 2 --timeout 2400000 test/*.test.js test/**/*.test.js"
+  },
+  "dependencies": {
+    "@google/genai": "1.30.0"
+  },
+  "devDependencies": {
+    "c8": "^10.0.0",
+    "chai": "^4.5.0",
+    "mocha": "^10.0.0",
+    "proxyquire": "^2.1.3"
+  }
+}

genai/express-mode/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('../../test/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('../api-key-example.js', {
+      '@google/genai': {GoogleGenAI: MockGoogleGenAI},
+    });
+
+    const response = await sample.generateWithApiKey('FAKE_API_KEY');
+
+    assert.strictEqual(response.text, mockGenerateContentStreamResult.text);
+  });
+});

genai/image-generation/package.json

@@ -11,7 +11,8 @@
   },
   "dependencies": {
     "@google/genai": "1.34.0",
-    "@google-cloud/storage": "^7.17.3"
+    "@google-cloud/storage": "^7.17.3",
+    "date-fns": "^4.1.0"
   },
   "devDependencies": {
     "c8": "^10.0.0",

genai/image-generation/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);
   });
 });