@ai-sdk/alibaba 1.0.1 → 1.0.3

package/dist/index.mjs

@@ -3,14 +3,19 @@ import {
   NoSuchModelError
 } from "@ai-sdk/provider";
 import {
-  createJsonErrorResponseHandler,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   loadApiKey,
   withoutTrailingSlash,
   withUserAgentSuffix
 } from "@ai-sdk/provider-utils";
-import { z as z3 } from "zod/v4";
+import { z as z4 } from "zod/v4";
 
 // src/alibaba-chat-language-model.ts
+import {
+  getResponseMetadata,
+  mapOpenAICompatibleFinishReason,
+  prepareTools
+} from "@ai-sdk/openai-compatible/internal";
 import {
   InvalidResponseDataError
 } from "@ai-sdk/provider";
@@ -24,15 +29,10 @@ import {
   postJsonToApi
 } from "@ai-sdk/provider-utils";
 import { z as z2 } from "zod/v4";
-import {
-  getResponseMetadata,
-  mapOpenAICompatibleFinishReason,
-  prepareTools
-} from "@ai-sdk/openai-compatible/internal";
 
 // src/alibaba-chat-options.ts
 import { z } from "zod/v4";
-var alibabaProviderOptions = z.object({
+var alibabaLanguageModelOptions = z.object({
   /**
    * Enable thinking/reasoning mode for supported models.
    * When enabled, the model generates reasoning content before the response.
@@ -52,6 +52,23 @@ var alibabaProviderOptions = z.object({
   parallelToolCalls: z.boolean().optional()
 });
 
+// src/convert-alibaba-usage.ts
+import { convertOpenAICompatibleChatUsage } from "@ai-sdk/openai-compatible/internal";
+function convertAlibabaUsage(usage) {
+  var _a, _b, _c, _d;
+  const baseUsage = convertOpenAICompatibleChatUsage(usage);
+  const cacheWriteTokens = (_b = (_a = usage == null ? void 0 : usage.prompt_tokens_details) == null ? void 0 : _a.cache_creation_input_tokens) != null ? _b : 0;
+  const noCacheTokens = ((_c = baseUsage.inputTokens.total) != null ? _c : 0) - ((_d = baseUsage.inputTokens.cacheRead) != null ? _d : 0) - cacheWriteTokens;
+  return {
+    ...baseUsage,
+    inputTokens: {
+      ...baseUsage.inputTokens,
+      cacheWrite: cacheWriteTokens,
+      noCache: noCacheTokens
+    }
+  };
+}
+
 // src/convert-to-alibaba-chat-messages.ts
 import {
   UnsupportedFunctionalityError
@@ -197,23 +214,6 @@ function convertToAlibabaChatMessages({
   return messages;
 }
 
-// src/convert-alibaba-usage.ts
-import { convertOpenAICompatibleChatUsage } from "@ai-sdk/openai-compatible/internal";
-function convertAlibabaUsage(usage) {
-  var _a, _b, _c, _d;
-  const baseUsage = convertOpenAICompatibleChatUsage(usage);
-  const cacheWriteTokens = (_b = (_a = usage == null ? void 0 : usage.prompt_tokens_details) == null ? void 0 : _a.cache_creation_input_tokens) != null ? _b : 0;
-  const noCacheTokens = ((_c = baseUsage.inputTokens.total) != null ? _c : 0) - ((_d = baseUsage.inputTokens.cacheRead) != null ? _d : 0) - cacheWriteTokens;
-  return {
-    ...baseUsage,
-    inputTokens: {
-      ...baseUsage.inputTokens,
-      cacheWrite: cacheWriteTokens,
-      noCache: noCacheTokens
-    }
-  };
-}
-
 // src/get-cache-control.ts
 var MAX_CACHE_BREAKPOINTS = 4;
 function getCacheControl(providerMetadata) {
@@ -286,7 +286,7 @@ var AlibabaLanguageModel = class {
     const alibabaOptions = await parseProviderOptions({
       provider: "alibaba",
       providerOptions,
-      schema: alibabaProviderOptions
+      schema: alibabaLanguageModelOptions
     });
     if (frequencyPenalty != null) {
       warnings.push({ type: "unsupported", feature: "frequencyPenalty" });
@@ -677,24 +677,310 @@ var alibabaChatChunkSchema = z2.object({
   // Usage only appears in final chunk
 });
 
+// src/alibaba-video-model.ts
+import {
+  AISDKError
+} from "@ai-sdk/provider";
+import {
+  combineHeaders as combineHeaders2,
+  convertUint8ArrayToBase64,
+  createJsonErrorResponseHandler,
+  createJsonResponseHandler as createJsonResponseHandler2,
+  delay,
+  getFromApi,
+  lazySchema,
+  parseProviderOptions as parseProviderOptions2,
+  postJsonToApi as postJsonToApi2,
+  resolve,
+  zodSchema
+} from "@ai-sdk/provider-utils";
+import { z as z3 } from "zod/v4";
+var alibabaVideoModelOptionsSchema = lazySchema(
+  () => zodSchema(
+    z3.object({
+      negativePrompt: z3.string().nullish(),
+      audioUrl: z3.string().nullish(),
+      promptExtend: z3.boolean().nullish(),
+      shotType: z3.enum(["single", "multi"]).nullish(),
+      watermark: z3.boolean().nullish(),
+      audio: z3.boolean().nullish(),
+      referenceUrls: z3.array(z3.string()).nullish(),
+      pollIntervalMs: z3.number().positive().nullish(),
+      pollTimeoutMs: z3.number().positive().nullish()
+    }).passthrough()
+  )
+);
+var alibabaVideoErrorSchema = z3.object({
+  code: z3.string().nullish(),
+  message: z3.string(),
+  request_id: z3.string().nullish()
+});
+var alibabaVideoFailedResponseHandler = createJsonErrorResponseHandler({
+  errorSchema: alibabaVideoErrorSchema,
+  errorToMessage: (data) => data.message
+});
+var alibabaVideoCreateTaskSchema = z3.object({
+  output: z3.object({
+    task_status: z3.string(),
+    task_id: z3.string()
+  }).nullish(),
+  request_id: z3.string().nullish()
+});
+var alibabaVideoTaskStatusSchema = z3.object({
+  output: z3.object({
+    task_id: z3.string(),
+    task_status: z3.string(),
+    video_url: z3.string().nullish(),
+    submit_time: z3.string().nullish(),
+    scheduled_time: z3.string().nullish(),
+    end_time: z3.string().nullish(),
+    orig_prompt: z3.string().nullish(),
+    actual_prompt: z3.string().nullish(),
+    code: z3.string().nullish(),
+    message: z3.string().nullish()
+  }).nullish(),
+  usage: z3.object({
+    duration: z3.number().nullish(),
+    output_video_duration: z3.number().nullish(),
+    SR: z3.number().nullish(),
+    size: z3.string().nullish()
+  }).nullish(),
+  request_id: z3.string().nullish()
+});
+function detectMode(modelId) {
+  if (modelId.includes("-i2v")) return "i2v";
+  if (modelId.includes("-r2v")) return "r2v";
+  return "t2v";
+}
+var AlibabaVideoModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v3";
+    this.maxVideosPerCall = 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate(options) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    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 mode = detectMode(this.modelId);
+    const alibabaOptions = await parseProviderOptions2({
+      provider: "alibaba",
+      providerOptions: options.providerOptions,
+      schema: alibabaVideoModelOptionsSchema
+    });
+    const input = {};
+    if (options.prompt != null) {
+      input.prompt = options.prompt;
+    }
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.negativePrompt) != null) {
+      input.negative_prompt = alibabaOptions.negativePrompt;
+    }
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.audioUrl) != null) {
+      input.audio_url = alibabaOptions.audioUrl;
+    }
+    if (mode === "i2v" && options.image != null) {
+      if (options.image.type === "url") {
+        input.img_url = options.image.url;
+      } else {
+        const base64Data = typeof options.image.data === "string" ? options.image.data : convertUint8ArrayToBase64(options.image.data);
+        input.img_url = base64Data;
+      }
+    }
+    if (mode === "r2v" && (alibabaOptions == null ? void 0 : alibabaOptions.referenceUrls) != null) {
+      input.reference_urls = alibabaOptions.referenceUrls;
+    }
+    const parameters = {};
+    if (options.duration != null) {
+      parameters.duration = options.duration;
+    }
+    if (options.seed != null) {
+      parameters.seed = options.seed;
+    }
+    if (options.resolution != null) {
+      if (mode === "i2v") {
+        const resolutionMap = {
+          "1280x720": "720P",
+          "720x1280": "720P",
+          "960x960": "720P",
+          "1088x832": "720P",
+          "832x1088": "720P",
+          "1920x1080": "1080P",
+          "1080x1920": "1080P",
+          "1440x1440": "1080P",
+          "1632x1248": "1080P",
+          "1248x1632": "1080P",
+          "832x480": "480P",
+          "480x832": "480P",
+          "624x624": "480P"
+        };
+        parameters.resolution = resolutionMap[options.resolution] || options.resolution;
+      } else {
+        parameters.size = options.resolution.replace("x", "*");
+      }
+    }
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.promptExtend) != null) {
+      parameters.prompt_extend = alibabaOptions.promptExtend;
+    }
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.shotType) != null) {
+      parameters.shot_type = alibabaOptions.shotType;
+    }
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.watermark) != null) {
+      parameters.watermark = alibabaOptions.watermark;
+    }
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.audio) != null) {
+      parameters.audio = alibabaOptions.audio;
+    }
+    if (options.aspectRatio) {
+      warnings.push({
+        type: "unsupported",
+        feature: "aspectRatio",
+        details: "Alibaba video models use explicit size/resolution dimensions. Use the resolution option or providerOptions.alibaba for size control."
+      });
+    }
+    if (options.fps) {
+      warnings.push({
+        type: "unsupported",
+        feature: "fps",
+        details: "Alibaba video models do not support custom FPS."
+      });
+    }
+    if (options.n != null && options.n > 1) {
+      warnings.push({
+        type: "unsupported",
+        feature: "n",
+        details: "Alibaba video models only support generating 1 video per call."
+      });
+    }
+    const { value: createResponse } = await postJsonToApi2({
+      url: `${this.config.baseURL}/api/v1/services/aigc/video-generation/video-synthesis`,
+      headers: combineHeaders2(
+        await resolve(this.config.headers),
+        options.headers,
+        {
+          "X-DashScope-Async": "enable"
+        }
+      ),
+      body: {
+        model: this.modelId,
+        input,
+        parameters
+      },
+      successfulResponseHandler: createJsonResponseHandler2(
+        alibabaVideoCreateTaskSchema
+      ),
+      failedResponseHandler: alibabaVideoFailedResponseHandler,
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    const taskId = (_d = createResponse.output) == null ? void 0 : _d.task_id;
+    if (!taskId) {
+      throw new AISDKError({
+        name: "ALIBABA_VIDEO_GENERATION_ERROR",
+        message: `No task_id returned from Alibaba API. Response: ${JSON.stringify(createResponse)}`
+      });
+    }
+    const pollIntervalMs = (_e = alibabaOptions == null ? void 0 : alibabaOptions.pollIntervalMs) != null ? _e : 5e3;
+    const pollTimeoutMs = (_f = alibabaOptions == null ? void 0 : alibabaOptions.pollTimeoutMs) != null ? _f : 6e5;
+    const startTime = Date.now();
+    let finalResponse;
+    let responseHeaders;
+    while (true) {
+      await delay(pollIntervalMs, { abortSignal: options.abortSignal });
+      if (Date.now() - startTime > pollTimeoutMs) {
+        throw new AISDKError({
+          name: "ALIBABA_VIDEO_GENERATION_TIMEOUT",
+          message: `Video generation timed out after ${pollTimeoutMs}ms`
+        });
+      }
+      const { value: statusResponse, responseHeaders: pollHeaders } = await getFromApi({
+        url: `${this.config.baseURL}/api/v1/tasks/${taskId}`,
+        headers: combineHeaders2(
+          await resolve(this.config.headers),
+          options.headers
+        ),
+        successfulResponseHandler: createJsonResponseHandler2(
+          alibabaVideoTaskStatusSchema
+        ),
+        failedResponseHandler: alibabaVideoFailedResponseHandler,
+        abortSignal: options.abortSignal,
+        fetch: this.config.fetch
+      });
+      responseHeaders = pollHeaders;
+      const taskStatus = (_g = statusResponse.output) == null ? void 0 : _g.task_status;
+      if (taskStatus === "SUCCEEDED") {
+        finalResponse = statusResponse;
+        break;
+      }
+      if (taskStatus === "FAILED" || taskStatus === "CANCELED") {
+        throw new AISDKError({
+          name: "ALIBABA_VIDEO_GENERATION_FAILED",
+          message: `Video generation ${taskStatus.toLowerCase()}. Task ID: ${taskId}. ${(_i = (_h = statusResponse.output) == null ? void 0 : _h.message) != null ? _i : ""}`
+        });
+      }
+    }
+    const videoUrl = (_j = finalResponse == null ? void 0 : finalResponse.output) == null ? void 0 : _j.video_url;
+    if (!videoUrl) {
+      throw new AISDKError({
+        name: "ALIBABA_VIDEO_GENERATION_ERROR",
+        message: `No video URL in response. Task ID: ${taskId}`
+      });
+    }
+    return {
+      videos: [
+        {
+          type: "url",
+          url: videoUrl,
+          mediaType: "video/mp4"
+        }
+      ],
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      },
+      providerMetadata: {
+        alibaba: {
+          taskId,
+          videoUrl,
+          ...((_k = finalResponse == null ? void 0 : finalResponse.output) == null ? void 0 : _k.actual_prompt) ? { actualPrompt: finalResponse.output.actual_prompt } : {},
+          ...(finalResponse == null ? void 0 : finalResponse.usage) ? {
+            usage: {
+              duration: finalResponse.usage.duration,
+              outputVideoDuration: finalResponse.usage.output_video_duration,
+              resolution: finalResponse.usage.SR,
+              size: finalResponse.usage.size
+            }
+          } : {}
+        }
+      }
+    };
+  }
+};
+
 // src/version.ts
-var VERSION = "1.0.1";
+var VERSION = "1.0.3";
 
 // src/alibaba-provider.ts
-var alibabaErrorDataSchema = z3.object({
-  error: z3.object({
-    message: z3.string(),
-    code: z3.string().nullish(),
-    type: z3.string().nullish()
+var alibabaErrorDataSchema = z4.object({
+  error: z4.object({
+    message: z4.string(),
+    code: z4.string().nullish(),
+    type: z4.string().nullish()
   })
 });
-var alibabaFailedResponseHandler = createJsonErrorResponseHandler({
+var alibabaFailedResponseHandler = createJsonErrorResponseHandler2({
   errorSchema: alibabaErrorDataSchema,
   errorToMessage: (data) => data.error.message
 });
 function createAlibaba(options = {}) {
-  var _a;
+  var _a, _b;
   const baseURL = (_a = withoutTrailingSlash(options.baseURL)) != null ? _a : "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
+  const videoBaseURL = (_b = withoutTrailingSlash(options.videoBaseURL)) != null ? _b : "https://dashscope-intl.aliyuncs.com";
   const getHeaders = () => withUserAgentSuffix(
     {
       Authorization: `Bearer ${loadApiKey({
@@ -716,6 +1002,12 @@ function createAlibaba(options = {}) {
       includeUsage: (_a2 = options.includeUsage) != null ? _a2 : true
     });
   };
+  const createVideoModel = (modelId) => new AlibabaVideoModel(modelId, {
+    provider: "alibaba.video",
+    baseURL: videoBaseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -727,6 +1019,8 @@ function createAlibaba(options = {}) {
   provider.specificationVersion = "v3";
   provider.languageModel = createLanguageModel;
   provider.chatModel = createLanguageModel;
+  provider.video = createVideoModel;
+  provider.videoModel = createVideoModel;
   provider.imageModel = (modelId) => {
     throw new NoSuchModelError({ modelId, modelType: "imageModel" });
   };