size

packages/ai/core/generate-image/generate-image.test.ts

@@ -19,13 +19,15 @@ describe('generateImage', () => {
         },
       }),
       prompt,
+      size: '1024x1024',
       headers: { 'custom-request-header': 'request-header-value' },
       abortSignal,
     });
 
     expect(capturedArgs).toStrictEqual({
-      prompt,
       n: 1,
+      prompt,
+      size: '1024x1024',
       headers: { 'custom-request-header': 'request-header-value' },
       abortSignal,
     });

packages/ai/core/generate-image/generate-image.ts

@@ -18,6 +18,7 @@ export async function generateImage({
   model,
   prompt,
   n,
+  size,
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
@@ -37,6 +38,11 @@ Number of images to generate.
    */
   n?: number;
 
+  /**
+Size of the images to generate. Must have the format `{width}x{height}`.
+   */
+  size?: `${number}x${number}`;
+
   /**
 Maximum number of retries per embedding model call. Set to 0 to disable retries.
 
@@ -63,6 +69,7 @@ Only applicable for HTTP-based providers.
       n: n ?? 1,
       abortSignal,
       headers,
+      size,
     }),
   );
 

packages/openai/src/openai-image-model.ts

@@ -28,6 +28,7 @@ export class OpenAIImageModel implements ImageModelV1 {
   async doGenerate({
     prompt,
     n,
+    size,
     headers,
     abortSignal,
   }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<
@@ -43,7 +44,7 @@ export class OpenAIImageModel implements ImageModelV1 {
         model: this.modelId,
         prompt,
         n,
-        // TODO size
+        size,
         // TODO passthrough provider options
         response_format: 'b64_json',
       },

packages/provider/src/image-model/v1/image-model-v1.ts

@@ -39,6 +39,11 @@ Number of images to generate.
      */
     n: number;
 
+    /**
+Size of the images to generate. Must have the format `{width}x{height}`.
+     */
+    size: `${number}x${number}` | undefined;
+
     /**
 Abort signal for cancelling the operation.
      */