Nest imageFormat in outputOptions

Author: dlarocque

packages/vertexai/src/requests/request-helpers.test.ts

@@ -217,10 +217,9 @@ describe('request formatting methods', () => {
 
       // Parameters without default values should be undefined
       expect(body.parameters.storageUri).to.be.undefined;
-      expect(body.parameters.compressionQuality).to.be.undefined;
-      expect(body.parameters.negativePrompt).to.be.undefined;
       expect(body.parameters.storageUri).to.be.undefined;
-      expect(body.parameters.mimeType).to.be.undefined;
+      expect(body.parameters.outputOptions).to.be.undefined;
+      expect(body.parameters.negativePrompt).to.be.undefined;
       expect(body.parameters.aspectRatio).to.be.undefined;
       expect(body.parameters.addWatermark).to.be.undefined;
       expect(body.parameters.safetyFilterLevel).to.be.undefined;
@@ -250,8 +249,10 @@ describe('request formatting methods', () => {
     console.log(body);
     expect(body.parameters).deep.equal({
       sampleCount: numberOfImages,
-      mimeType: imageFormat.mimeType,
-      compressionQuality: imageFormat.compressionQuality,
+      outputOptions: {
+        mimeType: imageFormat.mimeType,
+        compressionQuality: imageFormat.compressionQuality
+      },
       addWatermark,
       negativePrompt,
       safetyFilterLevel: safetySettings.safetyFilterLevel,

packages/vertexai/src/requests/request-helpers.ts

@@ -156,7 +156,7 @@ export function createPredictRequestBody(
       negativePrompt,
       sampleCount: numberOfImages,
       aspectRatio,
-      ...imageFormat,
+      outputOptions: imageFormat,
       addWatermark,
       safetyFilterLevel,
       personGeneration: personFilterLevel,

packages/vertexai/src/types/imagen/internal.ts

@@ -101,8 +101,10 @@ export interface PredictRequestBody {
   parameters: {
     sampleCount: number; // Maps to numberOfImages
     aspectRatio?: string;
-    mimeType?: string;
-    compressionQuality?: number;
+    outputOptions?: {
+      mimeType: string;
+      compressionQuality?: number;
+    };
     negativePrompt?: string;
     storageUri?: string; // Maps to gcsURI
     addWatermark?: boolean;