fixing failed tests

Author: Guiners

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

@@ -21,12 +21,11 @@ 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', async () => {
-  it('should return an object containing embeddings and metadata', async () => {
+describe('embeddings-docretrieval-with-txt', () => {
+  it('should return an object containing embeddings and metadata', async function () {
     this.retries(4);
     await delay(this.test);
-    const generatedFileNames =
-      await sample.generateEmbeddingsForRetrieval(projectId);
-    assert.containsAllKeys(generatedFileNames, ['embeddings', 'metadata']);
+    const result = await sample.generateEmbeddingsForRetrieval(projectId);
+    assert.containsAllKeys(result, ['embeddings', 'metadata']);
   });
 });

genai/test/provisionedthroughput-with-txt.test.js

@@ -21,8 +21,8 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../provisioned-throughput/provisionedthroughput-with-txt.js');
 const {delay} = require('./util');
 
-describe('provisionedthroughput-with-txt', () => {
-  it('should return provisioned throughput result', async function () {
+describe('provisioned-throughput-with-txt', () => {
+  it('should return provisioned throughput result', async () => {
     this.timeout(50000);
     this.retries(4);
     await delay(this.test);

genai/test/tools-google-search-with-txt.test.js

@@ -22,7 +22,7 @@ const sample = require('../tools/tools-google-search-with-txt.js');
 
 describe('tools-google-search-with-txt', () => {
   it('should generate answer to a question in prompt using google search', async function () {
-    this.timeout(10000);
+    this.timeout(100000);
     const output = await sample.generateGoogleSearch(projectId);
     assert(output.length > 0);
   });