@ai-sdk/xai 4.0.0-beta.2 → 4.0.0-beta.4

package/docs/01-xai.mdx

@@ -840,6 +840,27 @@ const { images } = await generateImage({
 });
 ```
 
+#### Multi-Image Editing
+
+Combine or reference multiple input images (up to 3) in the prompt:
+
+```ts
+import { xai } from '@ai-sdk/xai';
+import { generateImage } from 'ai';
+import { readFileSync } from 'fs';
+
+const cat = readFileSync('./cat.png');
+const dog = readFileSync('./dog.png');
+
+const { images } = await generateImage({
+  model: xai.image('grok-imagine-image'),
+  prompt: {
+    text: 'Combine these two animals into a group photo',
+    images: [cat, dog],
+  },
+});
+```
+
 #### Style Transfer
 
 Apply artistic styles to an image:
@@ -859,7 +880,7 @@ const { images } = await generateImage({
 
 <Note>
   Input images can be provided as `Buffer`, `ArrayBuffer`, `Uint8Array`, or
-  base64-encoded strings. xAI only supports a single input image per request.
+  base64-encoded strings. Up to 3 input images are supported per request.
 </Note>
 
 ### Model-specific options

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/xai",
-  "version": "4.0.0-beta.2",
+  "version": "4.0.0-beta.4",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -29,8 +29,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/openai-compatible": "3.0.0-beta.2",
     "@ai-sdk/provider": "4.0.0-beta.0",
+    "@ai-sdk/openai-compatible": "3.0.0-beta.2",
     "@ai-sdk/provider-utils": "5.0.0-beta.1"
   },
   "devDependencies": {

package/src/xai-image-model.ts

@@ -27,7 +27,7 @@ interface XaiImageModelConfig {
 
 export class XaiImageModel implements ImageModelV3 {
   readonly specificationVersion = 'v3';
-  readonly maxImagesPerCall = 1;
+  readonly maxImagesPerCall = 3;
 
   get provider(): string {
     return this.config.provider;
@@ -84,19 +84,9 @@ export class XaiImageModel implements ImageModelV3 {
     });
 
     const hasFiles = files != null && files.length > 0;
-    let imageUrl: string | undefined;
-
-    if (hasFiles) {
-      imageUrl = convertImageModelFileToDataUri(files[0]);
-
-      if (files.length > 1) {
-        warnings.push({
-          type: 'other',
-          message:
-            'xAI only supports a single input image. Additional images are ignored.',
-        });
-      }
-    }
+    const imageUrls = hasFiles
+      ? files.map(file => convertImageModelFileToDataUri(file))
+      : [];
 
     const endpoint = hasFiles ? '/images/edits' : '/images/generations';
 
@@ -104,7 +94,7 @@ export class XaiImageModel implements ImageModelV3 {
       model: this.modelId,
       prompt,
       n,
-      response_format: 'url',
+      response_format: 'b64_json',
     };
 
     if (aspectRatio != null) {
@@ -127,8 +117,18 @@ export class XaiImageModel implements ImageModelV3 {
       body.resolution = xaiOptions.resolution;
     }
 
-    if (imageUrl != null) {
-      body.image = { url: imageUrl, type: 'image_url' };
+    if (xaiOptions?.quality != null) {
+      body.quality = xaiOptions.quality;
+    }
+
+    if (xaiOptions?.user != null) {
+      body.user = xaiOptions.user;
+    }
+
+    if (imageUrls.length === 1) {
+      body.image = { url: imageUrls[0], type: 'image_url' };
+    } else if (imageUrls.length > 1) {
+      body.images = imageUrls.map(url => ({ url, type: 'image_url' }));
     }
 
     const baseURL = this.config.baseURL ?? 'https://api.x.ai/v1';
@@ -145,12 +145,18 @@ export class XaiImageModel implements ImageModelV3 {
       fetch: this.config.fetch,
     });
 
-    const downloadedImages = await Promise.all(
-      response.data.map(image => this.downloadImage(image.url, abortSignal)),
-    );
+    const hasAllBase64 = response.data.every(image => image.b64_json != null);
+
+    const images = hasAllBase64
+      ? response.data.map(image => image.b64_json!)
+      : await Promise.all(
+          response.data.map(image =>
+            this.downloadImage(image.url!, abortSignal),
+          ),
+        );
 
     return {
-      images: downloadedImages,
+      images,
       warnings,
       response: {
         timestamp: currentDate,
@@ -164,6 +170,9 @@ export class XaiImageModel implements ImageModelV3 {
               ? { revisedPrompt: item.revised_prompt }
               : {}),
           })),
+          ...(response.usage?.cost_in_usd_ticks != null
+            ? { costInUsdTicks: response.usage.cost_in_usd_ticks }
+            : {}),
         },
       },
     };
@@ -187,8 +196,14 @@ export class XaiImageModel implements ImageModelV3 {
 const xaiImageResponseSchema = z.object({
   data: z.array(
     z.object({
-      url: z.string(),
+      url: z.string().nullish(),
+      b64_json: z.string().nullish(),
       revised_prompt: z.string().nullish(),
     }),
   ),
+  usage: z
+    .object({
+      cost_in_usd_ticks: z.number().nullish(),
+    })
+    .nullish(),
 });

package/src/xai-image-options.ts

@@ -5,6 +5,8 @@ export const xaiImageModelOptions = z.object({
   output_format: z.string().optional(),
   sync_mode: z.boolean().optional(),
   resolution: z.enum(['1k', '2k']).optional(),
+  quality: z.enum(['low', 'medium', 'high']).optional(),
+  user: z.string().optional(),
 });
 
 export type XaiImageModelOptions = z.infer<typeof xaiImageModelOptions>;