codereview fix

Author: Guiners

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

@@ -57,11 +57,11 @@ async function generateImage(
       outputGcsUri: outputGcsUri,
     },
   });
-  console.log(response.generatedImages);
 
+  console.log(response.generatedImages[0].image.gcsUri);
   // Example response:
   //  gs://your-bucket/your-prefix
-  return response.generatedImages;
+  return response.generatedImages[0].image.gcsUri;
 }
 // [END googlegenaisdk_imggen_canny_ctrl_type_with_txt_img]
 

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

@@ -22,6 +22,7 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../image-generation/imggen-canny-ctrl-type-with-txt-img');
 const {delay} = require('./util');
 const {createOutputGcsUri} = require('./imggen-util');
+
 describe('imggen-canny-ctrl-type-with-txt-img', () => {
   it('should return an array of generated image URIs', async function () {
     this.timeout(180000);

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

@@ -22,10 +22,11 @@ const projectId = process.env.CAIP_PROJECT_ID;
 const sample = require('../image-generation/imggen-raw-reference-with-txt-img');
 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(600000);
-    this.retries(3);
+    // this.retries(3);
 
     const output = await createOutputGcsUri();
     console.log('Output GCS URI:', output.uri);

genai/test/imggen-util.js

@@ -15,8 +15,9 @@
 const {Storage} = require('@google-cloud/storage');
 const {format} = require('date-fns');
 
-const gcsOutputBucket = 'nodejs-docs-samples-tests';
-
+// const gcsOutputBucket = 'nodejs-docs-samples-tests';
+const gcsOutputBucket =
+    'example-cloud-bucket-5439474/nodejs-docs-samples-tests';
 module.exports.createOutputGcsUri = async function () {
   const prefix = `text_output/${format(new Date(), 'yyyy-MM-dd-HH-mm-ss')}`;
   const gcsUri = `gs://${gcsOutputBucket}/${prefix}`;