fixing failed tests

Author: Guiners

genai/image-generation/imggen-canny-ctrl-type-with-txt-img.js

@@ -46,6 +46,7 @@ async function generateImage(
 
   const response = await client.models.editImage({
     model: 'imagen-3.0-capability-001',
+    // The '[1]' in the prompt corresponds to the controlReferenceImage.referenceId above.
     prompt: 'a watercolor painting of a red car[1] driving on a road',
     referenceImages: [controlReferenceImage],
     config: {

genai/image-generation/imggen-subj-refer-ctrl-refer-with-txt-imgs.js

@@ -60,8 +60,7 @@ async function generateImage(
 
   const response = await client.models.editImage({
     model: 'imagen-3.0-capability-001',
-    prompt:
-      `a portrait of a woman[1] in the pose of the control image[2]in a watercolor style by a professional artist,
+    prompt: `a portrait of a woman[1] in the pose of the control image[2]in a watercolor style by a professional artist,
         light and low-contrast stokes, bright pastel colors, a warm atmosphere, clean background, grainy paper,
         bold visible brushstrokes, patchy details`,
     referenceImages: [subjectReferenceImage, controlReferenceImage],

genai/test/imggen-raw-reference-with-txt-img.test.js

@@ -24,15 +24,24 @@ const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
 describe('imggen-raw-reference-with-txt-img', () => {
   it('should return an array of generated image URIs', async function () {
-    this.timeout(180000);
-    this.retries(4);
+    this.timeout(600000);
+    this.retries(3);
+
     const output = await createOutputGcsUri();
-    console.log(output.uri);
-    await delay(this.test);
-    const generatedFileNames = await sample.generateImage(
-      output.uri,
-      projectId
-    );
-    assert(generatedFileNames.length > 0);
+    console.log('Output GCS URI:', output.uri);
+
+    try {
+      await delay(this.test);
+      const generatedFileNames = await sample.generateImage(
+        output.uri,
+        projectId
+      );
+      console.log('Generated files:', generatedFileNames);
+
+      assert(generatedFileNames.length > 0);
+    } catch (err) {
+      console.error('Image generation failed:', err);
+      throw err;
+    }
   });
 });

genai/test/imggen-scribble-ctrl-type-with-txt-img.test.js

@@ -24,15 +24,24 @@ const {createOutputGcsUri} = require('./imggen-util');
 
 describe('imggen-scribble-ctrl-type-with-txt-img', async () => {
   it('should generate images from a text prompt with control reference image', async function () {
-    this.timeout(180000);
-    this.retries(4);
+    this.timeout(600000);
+    this.retries(3);
+
     const output = await createOutputGcsUri();
-    console.log(output.uri);
-    await delay(this.test);
-    const generatedFileNames = await sample.generateImage(
-      output.uri,
-      projectId
-    );
-    assert(generatedFileNames.length > 0);
+    console.log('Output GCS URI:', output.uri);
+
+    try {
+      await delay(this.test);
+      const generatedFileNames = await sample.generateImage(
+        output.uri,
+        projectId
+      );
+      console.log('Generated files:', generatedFileNames);
+
+      assert(generatedFileNames.length > 0);
+    } catch (err) {
+      console.error('Image generation failed:', err);
+      throw err;
+    }
   });
 });