@ai-sdk/xai 4.0.0-beta.21 → 4.0.0-beta.24

package/dist/index.js

@@ -35,7 +35,7 @@ module.exports = __toCommonJS(index_exports);
 
 // src/xai-provider.ts
 var import_provider7 = require("@ai-sdk/provider");
-var import_provider_utils17 = require("@ai-sdk/provider-utils");
+var import_provider_utils20 = require("@ai-sdk/provider-utils");
 
 // src/xai-chat-language-model.ts
 var import_provider3 = require("@ai-sdk/provider");
@@ -68,6 +68,17 @@ function convertToXaiChatMessages(prompt) {
                 return { type: "text", text: part.text };
               }
               case "file": {
+                if ((0, import_provider_utils.isProviderReference)(part.data)) {
+                  return {
+                    type: "file",
+                    file: {
+                      file_id: (0, import_provider_utils.resolveProviderReference)({
+                        reference: part.data,
+                        provider: "xai"
+                      })
+                    }
+                  };
+                }
                 if (part.mediaType.startsWith("image/")) {
                   const mediaType = part.mediaType === "image/*" ? "image/jpeg" : part.mediaType;
                   return {
@@ -1084,7 +1095,15 @@ async function convertToXaiResponsesInput({
               break;
             }
             case "file": {
-              if (block.mediaType.startsWith("image/")) {
+              if ((0, import_provider_utils5.isProviderReference)(block.data)) {
+                contentParts.push({
+                  type: "input_file",
+                  file_id: (0, import_provider_utils5.resolveProviderReference)({
+                    reference: block.data,
+                    provider: "xai"
+                  })
+                });
+              } else if (block.mediaType.startsWith("image/")) {
                 const mediaType = block.mediaType === "image/*" ? "image/jpeg" : block.mediaType;
                 const imageUrl = block.data instanceof URL ? block.data.toString() : `data:${mediaType};base64,${(0, import_provider_utils5.convertToBase64)(block.data)}`;
                 contentParts.push({ type: "input_image", image_url: imageUrl });
@@ -2753,33 +2772,174 @@ var xaiTools = {
 };
 
 // src/version.ts
-var VERSION = true ? "4.0.0-beta.21" : "0.0.0-test";
+var VERSION = true ? "4.0.0-beta.24" : "0.0.0-test";
 
-// src/xai-video-model.ts
-var import_provider6 = require("@ai-sdk/provider");
-var import_provider_utils16 = require("@ai-sdk/provider-utils");
-var import_v416 = require("zod/v4");
+// src/files/xai-files.ts
+var import_provider_utils17 = require("@ai-sdk/provider-utils");
 
-// src/xai-video-options.ts
+// src/files/xai-files-api.ts
 var import_provider_utils15 = require("@ai-sdk/provider-utils");
 var import_v415 = require("zod/v4");
-var xaiVideoModelOptionsSchema = (0, import_provider_utils15.lazySchema)(
+var xaiFilesResponseSchema = (0, import_provider_utils15.lazySchema)(
   () => (0, import_provider_utils15.zodSchema)(
     import_v415.z.object({
-      pollIntervalMs: import_v415.z.number().positive().nullish(),
-      pollTimeoutMs: import_v415.z.number().positive().nullish(),
-      resolution: import_v415.z.enum(["480p", "720p"]).nullish(),
-      videoUrl: import_v415.z.string().nullish()
+      id: import_v415.z.string(),
+      object: import_v415.z.string().nullish(),
+      bytes: import_v415.z.number().nullish(),
+      created_at: import_v415.z.number().nullish(),
+      filename: import_v415.z.string().nullish(),
+      purpose: import_v415.z.string().nullish(),
+      status: import_v415.z.string().nullish()
+    })
+  )
+);
+
+// src/files/xai-files-options.ts
+var import_provider_utils16 = require("@ai-sdk/provider-utils");
+var import_v416 = require("zod/v4");
+var xaiFilesOptionsSchema = (0, import_provider_utils16.lazySchema)(
+  () => (0, import_provider_utils16.zodSchema)(
+    import_v416.z.object({
+      teamId: import_v416.z.string().optional(),
+      filePath: import_v416.z.string().optional()
     }).passthrough()
   )
 );
 
+// src/files/xai-files.ts
+var XaiFiles = class {
+  constructor(config) {
+    this.config = config;
+    this.specificationVersion = "v4";
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async uploadFile({
+    data,
+    mediaType,
+    filename,
+    providerOptions
+  }) {
+    var _a, _b;
+    const xaiOptions = await (0, import_provider_utils17.parseProviderOptions)({
+      provider: "xai",
+      providerOptions,
+      schema: xaiFilesOptionsSchema
+    });
+    const fileBytes = data instanceof Uint8Array ? data : (0, import_provider_utils17.convertBase64ToUint8Array)(data);
+    const blob = new Blob([fileBytes], {
+      type: mediaType
+    });
+    const formData = new FormData();
+    if (filename != null) {
+      formData.append("file", blob, filename);
+    } else {
+      formData.append("file", blob);
+    }
+    if ((xaiOptions == null ? void 0 : xaiOptions.teamId) != null) {
+      formData.append("team_id", xaiOptions.teamId);
+    }
+    const { value: response } = await (0, import_provider_utils17.postFormDataToApi)({
+      url: `${this.config.baseURL}/files`,
+      headers: (0, import_provider_utils17.combineHeaders)(this.config.headers()),
+      formData,
+      failedResponseHandler: xaiFailedResponseHandler,
+      successfulResponseHandler: (0, import_provider_utils17.createJsonResponseHandler)(
+        xaiFilesResponseSchema
+      ),
+      fetch: this.config.fetch
+    });
+    return {
+      warnings: [],
+      providerReference: { xai: response.id },
+      ...((_a = response.filename) != null ? _a : filename) ? { filename: (_b = response.filename) != null ? _b : filename } : {},
+      ...mediaType != null ? { mediaType } : {},
+      providerMetadata: {
+        xai: {
+          ...response.filename != null ? { filename: response.filename } : {},
+          ...response.bytes != null ? { bytes: response.bytes } : {},
+          ...response.created_at != null ? { createdAt: response.created_at } : {}
+        }
+      }
+    };
+  }
+};
+
+// src/xai-video-model.ts
+var import_provider6 = require("@ai-sdk/provider");
+var import_provider_utils19 = require("@ai-sdk/provider-utils");
+var import_v418 = require("zod/v4");
+
+// src/xai-video-options.ts
+var import_provider_utils18 = require("@ai-sdk/provider-utils");
+var import_v417 = require("zod/v4");
+var nonEmptyStringSchema = import_v417.z.string().min(1);
+var resolutionSchema = import_v417.z.enum(["480p", "720p"]);
+var modeSchema = import_v417.z.enum(["edit-video", "extend-video", "reference-to-video"]);
+var baseFields = {
+  pollIntervalMs: import_v417.z.number().positive().nullish(),
+  pollTimeoutMs: import_v417.z.number().positive().nullish(),
+  resolution: resolutionSchema.nullish()
+};
+var editVideoSchema = import_v417.z.object({
+  ...baseFields,
+  mode: import_v417.z.literal("edit-video"),
+  videoUrl: nonEmptyStringSchema,
+  referenceImageUrls: import_v417.z.undefined().optional()
+});
+var extendVideoSchema = import_v417.z.object({
+  ...baseFields,
+  mode: import_v417.z.literal("extend-video"),
+  videoUrl: nonEmptyStringSchema,
+  referenceImageUrls: import_v417.z.undefined().optional()
+});
+var referenceToVideoSchema = import_v417.z.object({
+  ...baseFields,
+  mode: import_v417.z.literal("reference-to-video"),
+  referenceImageUrls: import_v417.z.array(nonEmptyStringSchema).min(1).max(7),
+  videoUrl: import_v417.z.undefined().optional()
+});
+var autoDetectSchema = import_v417.z.object({
+  ...baseFields,
+  mode: import_v417.z.undefined().optional(),
+  videoUrl: nonEmptyStringSchema.optional(),
+  referenceImageUrls: import_v417.z.array(nonEmptyStringSchema).min(1).max(7).optional()
+});
+var xaiVideoModelOptions = import_v417.z.union([
+  editVideoSchema,
+  extendVideoSchema,
+  referenceToVideoSchema,
+  autoDetectSchema
+]);
+var runtimeSchema = import_v417.z.object({
+  mode: modeSchema.optional(),
+  videoUrl: nonEmptyStringSchema.optional(),
+  referenceImageUrls: import_v417.z.array(nonEmptyStringSchema).min(1).max(7).optional(),
+  ...baseFields
+}).passthrough();
+var xaiVideoModelOptionsSchema = (0, import_provider_utils18.lazySchema)(
+  () => (0, import_provider_utils18.zodSchema)(runtimeSchema)
+);
+
 // src/xai-video-model.ts
 var RESOLUTION_MAP = {
   "1280x720": "720p",
   "854x480": "480p",
   "640x480": "480p"
 };
+function resolveVideoMode(options) {
+  if ((options == null ? void 0 : options.mode) != null) {
+    return options.mode;
+  }
+  if ((options == null ? void 0 : options.videoUrl) != null) {
+    return "edit-video";
+  }
+  if ((options == null ? void 0 : options.referenceImageUrls) != null && options.referenceImageUrls.length > 0) {
+    return "reference-to-video";
+  }
+  return void 0;
+}
 var XaiVideoModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -2794,12 +2954,15 @@ var XaiVideoModel = class {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
     const warnings = [];
-    const xaiOptions = await (0, import_provider_utils16.parseProviderOptions)({
+    const xaiOptions = await (0, import_provider_utils19.parseProviderOptions)({
       provider: "xai",
       providerOptions: options.providerOptions,
       schema: xaiVideoModelOptionsSchema
     });
-    const isEdit = (xaiOptions == null ? void 0 : xaiOptions.videoUrl) != null;
+    const effectiveMode = resolveVideoMode(xaiOptions);
+    const isEdit = effectiveMode === "edit-video";
+    const isExtension = effectiveMode === "extend-video";
+    const hasReferenceImages = effectiveMode === "reference-to-video";
     if (options.fps != null) {
       warnings.push({
         type: "unsupported",
@@ -2842,19 +3005,36 @@ var XaiVideoModel = class {
         details: "xAI video editing does not support custom resolution."
       });
     }
+    if (isExtension && options.aspectRatio != null) {
+      warnings.push({
+        type: "unsupported",
+        feature: "aspectRatio",
+        details: "xAI video extension does not support custom aspect ratio."
+      });
+    }
+    if (isExtension && ((xaiOptions == null ? void 0 : xaiOptions.resolution) != null || options.resolution != null)) {
+      warnings.push({
+        type: "unsupported",
+        feature: "resolution",
+        details: "xAI video extension does not support custom resolution."
+      });
+    }
     const body = {
       model: this.modelId,
       prompt: options.prompt
     };
-    if (!isEdit && options.duration != null) {
+    const allowDuration = !isEdit;
+    const allowAspectRatio = !isEdit && !isExtension;
+    const allowResolution = !isEdit && !isExtension;
+    if (allowDuration && options.duration != null) {
       body.duration = options.duration;
     }
-    if (!isEdit && options.aspectRatio != null) {
+    if (allowAspectRatio && options.aspectRatio != null) {
       body.aspect_ratio = options.aspectRatio;
     }
-    if (!isEdit && (xaiOptions == null ? void 0 : xaiOptions.resolution) != null) {
+    if (allowResolution && (xaiOptions == null ? void 0 : xaiOptions.resolution) != null) {
       body.resolution = xaiOptions.resolution;
-    } else if (!isEdit && options.resolution != null) {
+    } else if (allowResolution && options.resolution != null) {
       const mapped = RESOLUTION_MAP[options.resolution];
       if (mapped != null) {
         body.resolution = mapped;
@@ -2866,38 +3046,56 @@ var XaiVideoModel = class {
         });
       }
     }
-    if ((xaiOptions == null ? void 0 : xaiOptions.videoUrl) != null) {
+    if (isEdit) {
+      body.video = { url: xaiOptions.videoUrl };
+    }
+    if (isExtension) {
       body.video = { url: xaiOptions.videoUrl };
     }
     if (options.image != null) {
       if (options.image.type === "url") {
         body.image = { url: options.image.url };
       } else {
-        const base64Data = typeof options.image.data === "string" ? options.image.data : (0, import_provider_utils16.convertUint8ArrayToBase64)(options.image.data);
+        const base64Data = typeof options.image.data === "string" ? options.image.data : (0, import_provider_utils19.convertUint8ArrayToBase64)(options.image.data);
         body.image = {
           url: `data:${options.image.mediaType};base64,${base64Data}`
         };
       }
     }
+    if (hasReferenceImages) {
+      body.reference_images = xaiOptions.referenceImageUrls.map((url) => ({
+        url
+      }));
+    }
     if (xaiOptions != null) {
       for (const [key, value] of Object.entries(xaiOptions)) {
         if (![
+          "mode",
           "pollIntervalMs",
           "pollTimeoutMs",
           "resolution",
-          "videoUrl"
+          "videoUrl",
+          "referenceImageUrls"
         ].includes(key)) {
           body[key] = value;
         }
       }
     }
     const baseURL = (_d = this.config.baseURL) != null ? _d : "https://api.x.ai/v1";
-    const { value: createResponse } = await (0, import_provider_utils16.postJsonToApi)({
-      url: `${baseURL}/videos/${isEdit ? "edits" : "generations"}`,
-      headers: (0, import_provider_utils16.combineHeaders)(this.config.headers(), options.headers),
+    let endpoint;
+    if (isEdit) {
+      endpoint = `${baseURL}/videos/edits`;
+    } else if (isExtension) {
+      endpoint = `${baseURL}/videos/extensions`;
+    } else {
+      endpoint = `${baseURL}/videos/generations`;
+    }
+    const { value: createResponse } = await (0, import_provider_utils19.postJsonToApi)({
+      url: endpoint,
+      headers: (0, import_provider_utils19.combineHeaders)(this.config.headers(), options.headers),
       body,
       failedResponseHandler: xaiFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils16.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils19.createJsonResponseHandler)(
         xaiCreateVideoResponseSchema
       ),
       abortSignal: options.abortSignal,
@@ -2915,17 +3113,17 @@ var XaiVideoModel = class {
     const startTime = Date.now();
     let responseHeaders;
     while (true) {
-      await (0, import_provider_utils16.delay)(pollIntervalMs, { abortSignal: options.abortSignal });
+      await (0, import_provider_utils19.delay)(pollIntervalMs, { abortSignal: options.abortSignal });
       if (Date.now() - startTime > pollTimeoutMs) {
         throw new import_provider6.AISDKError({
           name: "XAI_VIDEO_GENERATION_TIMEOUT",
           message: `Video generation timed out after ${pollTimeoutMs}ms`
         });
       }
-      const { value: statusResponse, responseHeaders: pollHeaders } = await (0, import_provider_utils16.getFromApi)({
+      const { value: statusResponse, responseHeaders: pollHeaders } = await (0, import_provider_utils19.getFromApi)({
         url: `${baseURL}/videos/${requestId}`,
-        headers: (0, import_provider_utils16.combineHeaders)(this.config.headers(), options.headers),
-        successfulResponseHandler: (0, import_provider_utils16.createJsonResponseHandler)(
+        headers: (0, import_provider_utils19.combineHeaders)(this.config.headers(), options.headers),
+        successfulResponseHandler: (0, import_provider_utils19.createJsonResponseHandler)(
           xaiVideoStatusResponseSchema
         ),
         failedResponseHandler: xaiFailedResponseHandler,
@@ -2965,7 +3163,8 @@ var XaiVideoModel = class {
               requestId,
               videoUrl: statusResponse.video.url,
               ...statusResponse.video.duration != null ? { duration: statusResponse.video.duration } : {},
-              ...((_j = statusResponse.usage) == null ? void 0 : _j.cost_in_usd_ticks) != null ? { costInUsdTicks: statusResponse.usage.cost_in_usd_ticks } : {}
+              ...((_j = statusResponse.usage) == null ? void 0 : _j.cost_in_usd_ticks) != null ? { costInUsdTicks: statusResponse.usage.cost_in_usd_ticks } : {},
+              ...statusResponse.progress != null ? { progress: statusResponse.progress } : {}
             }
           }
         };
@@ -2976,34 +3175,45 @@ var XaiVideoModel = class {
           message: "Video generation request expired."
         });
       }
+      if (statusResponse.status === "failed") {
+        throw new import_provider6.AISDKError({
+          name: "XAI_VIDEO_GENERATION_FAILED",
+          message: "Video generation failed."
+        });
+      }
     }
   }
 };
-var xaiCreateVideoResponseSchema = import_v416.z.object({
-  request_id: import_v416.z.string().nullish()
+var xaiCreateVideoResponseSchema = import_v418.z.object({
+  request_id: import_v418.z.string().nullish()
 });
-var xaiVideoStatusResponseSchema = import_v416.z.object({
-  status: import_v416.z.string().nullish(),
-  video: import_v416.z.object({
-    url: import_v416.z.string(),
-    duration: import_v416.z.number().nullish(),
-    respect_moderation: import_v416.z.boolean().nullish()
+var xaiVideoStatusResponseSchema = import_v418.z.object({
+  status: import_v418.z.string().nullish(),
+  video: import_v418.z.object({
+    url: import_v418.z.string(),
+    duration: import_v418.z.number().nullish(),
+    respect_moderation: import_v418.z.boolean().nullish()
+  }).nullish(),
+  model: import_v418.z.string().nullish(),
+  usage: import_v418.z.object({
+    cost_in_usd_ticks: import_v418.z.number().nullish()
   }).nullish(),
-  model: import_v416.z.string().nullish(),
-  usage: import_v416.z.object({
-    cost_in_usd_ticks: import_v416.z.number().nullish()
+  progress: import_v418.z.number().nullish(),
+  error: import_v418.z.object({
+    code: import_v418.z.string().nullish(),
+    message: import_v418.z.string().nullish()
   }).nullish()
 });
 
 // src/xai-provider.ts
 function createXai(options = {}) {
   var _a;
-  const baseURL = (0, import_provider_utils17.withoutTrailingSlash)(
+  const baseURL = (0, import_provider_utils20.withoutTrailingSlash)(
     (_a = options.baseURL) != null ? _a : "https://api.x.ai/v1"
   );
-  const getHeaders = () => (0, import_provider_utils17.withUserAgentSuffix)(
+  const getHeaders = () => (0, import_provider_utils20.withUserAgentSuffix)(
     {
-      Authorization: `Bearer ${(0, import_provider_utils17.loadApiKey)({
+      Authorization: `Bearer ${(0, import_provider_utils20.loadApiKey)({
         apiKey: options.apiKey,
         environmentVariableName: "XAI_API_KEY",
         description: "xAI API key"
@@ -3017,7 +3227,7 @@ function createXai(options = {}) {
       provider: "xai.chat",
       baseURL,
       headers: getHeaders,
-      generateId: import_provider_utils17.generateId,
+      generateId: import_provider_utils20.generateId,
       fetch: options.fetch
     });
   };
@@ -3026,7 +3236,7 @@ function createXai(options = {}) {
       provider: "xai.responses",
       baseURL,
       headers: getHeaders,
-      generateId: import_provider_utils17.generateId,
+      generateId: import_provider_utils20.generateId,
       fetch: options.fetch
     });
   };
@@ -3046,6 +3256,12 @@ function createXai(options = {}) {
       fetch: options.fetch
     });
   };
+  const createFiles = () => new XaiFiles({
+    provider: "xai.files",
+    baseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const provider = (modelId) => createResponsesLanguageModel(modelId);
   provider.specificationVersion = "v4";
   provider.languageModel = createResponsesLanguageModel;
@@ -3059,6 +3275,7 @@ function createXai(options = {}) {
   provider.image = createImageModel;
   provider.videoModel = createVideoModel;
   provider.video = createVideoModel;
+  provider.files = createFiles;
   provider.tools = xaiTools;
   return provider;
 }