@ai-sdk/xai 4.0.53 → 4.0.54

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/xai",
-  "version": "4.0.53",
+  "version": "4.0.54",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,

package/src/responses/xai-responses-batch.ts

@@ -20,6 +20,7 @@ import {
   getFromApi,
   lazySchema,
   normalizeBatchRequestCounts,
+  parseProviderOptions,
   postFormDataToApi,
   postJsonToApi,
   safeValidateTypes,
@@ -48,6 +49,24 @@ const xaiBatchEndpoint = '/v1/responses';
 const xaiBatchName = 'ai-sdk-text-batch';
 const xaiBatchResultsPageSize = 1000;
 
+const xaiBatchProviderOptionsSchema = lazySchema(() =>
+  zodSchema(
+    z.object({
+      /**
+       * TTL in seconds for the uploaded batch input file, measured from
+       * upload time. xAI accepts integers between 3600 (1 hour) and
+       * 2592000 (30 days) inclusive. Without it the file has no expiry.
+       */
+      inputFileExpiresAfter: z
+        .number()
+        .int()
+        .min(3600)
+        .max(2_592_000)
+        .optional(),
+    }),
+  ),
+);
+
 type XaiBatchRequest = Parameters<
   BatchLanguageModelV4['experimental_doStartBatch']
 >[0]['requests'][number];
@@ -144,6 +163,12 @@ class XaiResponsesBatch {
             },
           ];
 
+    const batchOptions = await parseProviderOptions({
+      provider: 'xai',
+      providerOptions: options.providerOptions,
+      schema: xaiBatchProviderOptionsSchema,
+    });
+
     for (const request of options.requests) {
       const preparedRequest = await this.options.prepareRequest(request);
 
@@ -166,6 +191,14 @@ class XaiResponsesBatch {
     const file = new Blob(fileParts, { type: 'application/jsonl' });
     fileParts.length = 0;
     const formData = new FormData();
+    // xAI rejects uploads where expires_after arrives after the file part,
+    // so all fields precede the file.
+    if (batchOptions?.inputFileExpiresAfter != null) {
+      formData.append(
+        'expires_after',
+        String(batchOptions.inputFileExpiresAfter),
+      );
+    }
     formData.append('file', file, filename);
 
     const headers = combineHeaders(
@@ -203,6 +236,18 @@ class XaiResponsesBatch {
     return {
       batchId: batch.batch_id,
       ...convertXaiBatchStatus(batch),
+      providerMetadata: {
+        xai: {
+          inputFileId: uploadedFile.id,
+          ...(uploadedFile.expires_at != null
+            ? {
+                inputFileExpiresAt: new Date(
+                  uploadedFile.expires_at * 1000,
+                ).toISOString(),
+              }
+            : {}),
+        },
+      },
       warnings,
     };
   }