fixing failed tests

Author: Guiners

genai/bounding-box/boundingbox-with-txt-img.js

@@ -15,16 +15,19 @@
   "dependencies": {
     "@google/genai": "^0.13.0",
     "axios": "^1.6.2",
-    "supertest": "^7.0.0",
     "canvas": "^3.1.0",
-    "node-fetch": "^2.7.0"
+    "google-auth-library": "^10.5.0",
+    "luxon": "^3.7.2",
+    "node-fetch": "^2.7.0",
+    "openai": "^6.9.1",
+    "supertest": "^7.0.0"
   },
   "devDependencies": {
     "c8": "^10.0.0",
     "chai": "^4.5.0",
     "mocha": "^10.0.0",
+    "proxyquire": "^2.1.3",
     "sinon": "^18.0.0",
-    "uuid": "^10.0.0",
-    "proxyquire": "^2.1.3"
+    "uuid": "^10.0.0"
   }
 }

genai/output-folder/image.png

@@ -63,17 +63,17 @@ async function generateWithSafetySettings(
     },
   });
 
-  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);
-  }
+  // 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:
   //

genai/package.json

@@ -16,12 +16,16 @@
 
 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.',
     };

genai/safety/safety-with-txt.js

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

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

@@ -19,10 +19,13 @@ const {describe, it} = require('mocha');
 
 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 () {
     this.timeout(50000);
+    this.retries(4);
+    await delay(this.test);
     const output = await sample.generateWithProvisionedThroughput(projectId);
     assert(output.length > 0);
   });

genai/test/boundingbox-with-txt-img.test.js

@@ -19,11 +19,15 @@ const {describe, it} = require('mocha');
 
 const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../safety/safety-with-txt.js');
+const {delay} = require('./util');
 
 describe('safety-with-txt', () => {
   it('should call generateContentStream with safety instructions', async function () {
     this.timeout(50000);
+    this.retries(4);
+    await delay(this.test);
     const output = await sample.generateWithSafetySettings(projectId);
-    assert(output > 0);
+    console.log('output', output);
+    assert.isObject(output);
   });
 });