feat(genai): Adding content cache samples

genai/content-cache/content-cache-create-with-txt-gcs-pdf.js

@@ -0,0 +1,82 @@
+// 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_contentcache_create_with_txt_gcs_pdf]
+
+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 generateContent(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION
+) {
+  const ai = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+    httpOptions: {
+      apiVersion: 'v1',
+    },
+  });
+
+  const systemInstruction = `
+  You are an expert researcher. You always stick to the facts in the sources provided, and never make up new facts.
+  Now look at these research papers, and answer the following questions.
+  `;
+
+  const contents = [
+    {
+      role: 'user',
+      parts: [
+        {
+          fileData: {
+            fileUri:
+              'gs://cloud-samples-data/generative-ai/pdf/2312.11805v3.pdf',
+            mimeType: 'application/pdf',
+          },
+        },
+        {
+          fileData: {
+            fileUri: 'gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf',
+            mimeType: 'application/pdf',
+          },
+        },
+      ],
+    },
+  ];
+
+  const contentCache = await ai.caches.create({
+    model: 'gemini-2.5-flash',
+    config: {
+      contents: contents,
+      systemInstruction: systemInstruction,
+      displayName: 'example-cache',
+      ttl: '86400s',
+    },
+  });
+
+  console.log(contentCache);
+  console.log(contentCache.name);
+
+  return contentCache.name;
+}
+
+// [END googlegenaisdk_contentcache_create_with_txt_gcs_pdf]
+
+module.exports = {
+  generateContent,
+};

genai/content-cache/content-cache-delete.js

@@ -0,0 +1,50 @@
+// 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_contentcache_delete]
+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 generateContent(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION,
+  cacheName
+) {
+  const ai = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+    httpOptions: {
+      apiVersion: 'v1',
+    },
+  });
+
+  console.log('Removing cache');
+  const contentCache = await ai.caches.delete({
+    name: cacheName,
+  });
+
+  console.log(contentCache.text);
+
+  return contentCache;
+}
+// [END googlegenaisdk_contentcache_delete]
+
+module.exports = {
+  generateContent,
+};

genai/content-cache/content-cache-list.js

@@ -0,0 +1,56 @@
+// 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_contentcache_list]
+
+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 generateContent(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION
+) {
+  const ai = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+    httpOptions: {
+      apiVersion: 'v1',
+    },
+  });
+
+  const contentCacheList = await ai.caches.list();
+
+  // Access individual properties of a ContentCache object(s)
+  const contentCacheNames = [];
+  for (const contentCache of contentCacheList.pageInternal) {
+    console.log(
+      `Cache \`${contentCache.name}\` for model \`${contentCache.model}\``
+    );
+    console.log(`Last updated at: ${contentCache.updateTime}`);
+    console.log(`Expires at: ${contentCache.expireTime}`);
+    contentCacheNames.push(contentCache.name);
+  }
+  console.log();
+
+  return contentCacheNames;
+}
+// [END googlegenaisdk_contentcache_list]
+
+module.exports = {
+  generateContent,
+};

genai/content-cache/content-cache-update.js

@@ -0,0 +1,76 @@
+// 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_contentcache_update]
+const {GoogleGenAI} = require('@google/genai');
+const {DateTime} = require('luxon');
+
+const GOOGLE_CLOUD_PROJECT = process.env.GOOGLE_CLOUD_PROJECT;
+const GOOGLE_CLOUD_LOCATION = process.env.GOOGLE_CLOUD_LOCATION || 'global';
+
+
+async function generateContent(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION,
+  cacheName
+) {
+  const ai = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+    httpOptions: {
+      apiVersion: 'v1',
+    },
+  });
+
+  let contentCache = await ai.caches.get({
+    name: cacheName,
+  });
+
+  console.log('Expire time', contentCache.expireTime);
+
+  contentCache = await ai.caches.update({
+    name: cacheName,
+    config: {
+      ttl: '36000s',
+    },
+  });
+
+  const expireTime = DateTime.fromISO(contentCache.expireTime, {zone: 'utc'});
+  const now = DateTime.utc();
+  const timeDiff = expireTime.diff(now, ['seconds']);
+
+  console.log('Expire time (after update):', expireTime.toISO());
+  console.log('Expire time (in seconds):', Math.floor(timeDiff.seconds));
+
+  const nextWeekUtc = DateTime.utc().plus({days: 7});
+  console.log('Next week (UTC):', nextWeekUtc.toISO());
+
+  contentCache = await ai.caches.update({
+    name: cacheName,
+    config: {
+      expireTime: nextWeekUtc,
+    },
+  });
+
+  console.log('Expire time (after update):', contentCache.expireTime);
+  return contentCache;
+}
+// [END googlegenaisdk_contentcache_update]
+
+module.exports = {
+  generateContent,
+};

genai/content-cache/content-cache-use-with-txt.js

@@ -0,0 +1,55 @@
+// 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_contentcache_use_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 generateContent(
+  projectId = GOOGLE_CLOUD_PROJECT,
+  location = GOOGLE_CLOUD_LOCATION,
+  cacheName
+) {
+  const ai = new GoogleGenAI({
+    vertexai: true,
+    project: projectId,
+    location: location,
+    httpOptions: {
+      apiVersion: 'v1',
+    },
+  });
+
+  const response = await ai.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents: 'Summarize the pdfs',
+    config: {
+      cachedContent: cacheName,
+    },
+  });
+
+  console.log(response.text);
+
+  return response.text;
+}
+// [END googlegenaisdk_contentcache_use_with_txt]
+
+module.exports = {
+  generateContent,
+};

genai/package.json

@@ -15,6 +15,7 @@
   "dependencies": {
     "@google/genai": "^0.13.0",
     "axios": "^1.6.2",
+    "luxon": "^3.7.1",
     "supertest": "^7.0.0"
   },
   "devDependencies": {

genai/test/content-cache-create-use-update-delete.test.js

@@ -0,0 +1,39 @@
+'use strict';
+
+const {assert} = require('chai');
+const {describe, it} = require('mocha');
+
+const projectId = process.env.CAIP_PROJECT_ID;
+
+const createSample = require('../content-cache/content-cache-create-with-txt-gcs-pdf.js');
+const useSample = require('../content-cache/content-cache-use-with-txt.js');
+const updateSample = require('../content-cache/content-cache-update.js');
+const deleteSample = require('../content-cache/content-cache-delete.js');
+
+describe('content-cache-create-use-update-delete', function () {
+  this.timeout(60000);
+  let contentCacheName;
+
+  it('should create content cache', async () => {
+    contentCacheName = await createSample.generateContent(projectId);
+    assert.isString(contentCacheName);
+    assert.isAbove(contentCacheName.length, 0);
+  });
+
+  it('should update content cache', async () => {
+    await updateSample.generateContent(projectId, undefined, contentCacheName);
+  });
+
+  it('should use content cache', async () => {
+    const response = await useSample.generateContent(
+      projectId,
+      undefined,
+      contentCacheName
+    );
+    assert.isString(response);
+  });
+
+  it('should delete content cache', async () => {
+    await deleteSample.generateContent(projectId, undefined, contentCacheName);
+  });
+});