@ai-sdk/alibaba 1.0.25 → 1.0.26

package/dist/index.mjs

@@ -700,75 +700,224 @@ var alibabaChatChunkSchema = z3.object({
   // Usage only appears in final chunk
 });
 
-// src/alibaba-video-model.ts
+// src/alibaba-embedding-model.ts
 import {
-  AISDKError
+  TooManyEmbeddingValuesForCallError,
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
 import {
   combineHeaders as combineHeaders2,
-  convertUint8ArrayToBase64,
   createJsonErrorResponseHandler as createJsonErrorResponseHandler2,
   createJsonResponseHandler as createJsonResponseHandler2,
+  parseProviderOptions as parseProviderOptions2,
+  postJsonToApi as postJsonToApi2
+} from "@ai-sdk/provider-utils";
+import { z as z5 } from "zod/v4";
+
+// src/alibaba-embedding-options.ts
+import { z as z4 } from "zod/v4";
+var alibabaEmbeddingModelOptions = z4.object({
+  /**
+   * Differentiates query text from document text for asymmetric retrieval tasks.
+   * Defaults to `document`.
+   */
+  textType: z4.enum(["query", "document"]).optional(),
+  /**
+   * The dimension of the output embedding vectors. Defaults to 1024.
+   *
+   * `text-embedding-v4` also supports 1536 and 2048 dimensions.
+   */
+  dimension: z4.number().optional(),
+  /**
+   * The output vector type. Defaults to `dense`.
+   *
+   * Sparse-only output is not supported by the AI SDK embedding interface,
+   * which requires dense number arrays.
+   */
+  outputType: z4.enum(["dense", "sparse", "dense&sparse"]).optional()
+});
+
+// src/alibaba-embedding-model.ts
+var alibabaEmbeddingFailedResponseHandler = createJsonErrorResponseHandler2({
+  errorSchema: z5.object({
+    code: z5.string().nullish(),
+    message: z5.string(),
+    request_id: z5.string().nullish()
+  }),
+  errorToMessage: (data) => data.message
+});
+var AlibabaEmbeddingModel = class {
+  constructor(modelId, config) {
+    this.specificationVersion = "v3";
+    this.maxEmbeddingsPerCall = 10;
+    this.supportsParallelCalls = false;
+    this.modelId = modelId;
+    this.config = config;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doEmbed({
+    values,
+    headers,
+    abortSignal,
+    providerOptions
+  }) {
+    if (values.length > this.maxEmbeddingsPerCall) {
+      throw new TooManyEmbeddingValuesForCallError({
+        provider: this.provider,
+        modelId: this.modelId,
+        maxEmbeddingsPerCall: this.maxEmbeddingsPerCall,
+        values
+      });
+    }
+    const alibabaOptions = await parseProviderOptions2({
+      provider: "alibaba",
+      providerOptions,
+      schema: alibabaEmbeddingModelOptions
+    });
+    if ((alibabaOptions == null ? void 0 : alibabaOptions.outputType) === "sparse") {
+      throw new UnsupportedFunctionalityError2({
+        functionality: "Alibaba embedding outputType 'sparse'",
+        message: "Alibaba embedding outputType 'sparse' is not supported because AI SDK embeddings require dense number arrays. Use 'dense' or 'dense&sparse' instead."
+      });
+    }
+    const {
+      responseHeaders,
+      value: response,
+      rawValue
+    } = await postJsonToApi2({
+      url: `${this.config.baseURL}/services/embeddings/text-embedding/text-embedding`,
+      headers: combineHeaders2(this.config.headers(), headers),
+      body: {
+        model: this.modelId,
+        input: {
+          texts: values
+        },
+        parameters: {
+          text_type: alibabaOptions == null ? void 0 : alibabaOptions.textType,
+          dimension: alibabaOptions == null ? void 0 : alibabaOptions.dimension,
+          output_type: alibabaOptions == null ? void 0 : alibabaOptions.outputType
+        }
+      },
+      failedResponseHandler: alibabaEmbeddingFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler2(
+        alibabaTextEmbeddingResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const sortedEmbeddings = response.output.embeddings.sort(
+      (a, b) => a.text_index - b.text_index
+    );
+    const sparseEmbeddings = sortedEmbeddings.map(
+      (item) => item.sparse_embedding == null ? void 0 : {
+        textIndex: item.text_index,
+        sparseEmbedding: item.sparse_embedding
+      }
+    ).filter((item) => item != null);
+    return {
+      warnings: [],
+      embeddings: sortedEmbeddings.map((item) => item.embedding),
+      usage: response.usage ? { tokens: response.usage.total_tokens } : void 0,
+      providerMetadata: sparseEmbeddings.length > 0 ? {
+        alibaba: {
+          sparseEmbeddings
+        }
+      } : void 0,
+      response: { headers: responseHeaders, body: rawValue }
+    };
+  }
+};
+var alibabaTextEmbeddingSparseEmbeddingSchema = z5.object({
+  index: z5.number(),
+  value: z5.number(),
+  token: z5.string().nullish()
+});
+var alibabaTextEmbeddingResponseSchema = z5.object({
+  output: z5.object({
+    embeddings: z5.array(
+      z5.object({
+        embedding: z5.array(z5.number()),
+        text_index: z5.number(),
+        sparse_embedding: z5.array(alibabaTextEmbeddingSparseEmbeddingSchema).nullish()
+      })
+    )
+  }),
+  usage: z5.object({
+    total_tokens: z5.number()
+  }).nullish()
+});
+
+// src/alibaba-video-model.ts
+import {
+  AISDKError
+} from "@ai-sdk/provider";
+import {
+  combineHeaders as combineHeaders3,
+  convertUint8ArrayToBase64,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler3,
+  createJsonResponseHandler as createJsonResponseHandler3,
   delay,
   getFromApi,
   lazySchema,
-  parseProviderOptions as parseProviderOptions2,
-  postJsonToApi as postJsonToApi2,
+  parseProviderOptions as parseProviderOptions3,
+  postJsonToApi as postJsonToApi3,
   resolve,
   zodSchema
 } from "@ai-sdk/provider-utils";
-import { z as z4 } from "zod/v4";
+import { z as z6 } from "zod/v4";
 var alibabaVideoModelOptionsSchema = lazySchema(
   () => zodSchema(
-    z4.object({
-      negativePrompt: z4.string().nullish(),
-      audioUrl: z4.string().nullish(),
-      promptExtend: z4.boolean().nullish(),
-      shotType: z4.enum(["single", "multi"]).nullish(),
-      watermark: z4.boolean().nullish(),
-      audio: z4.boolean().nullish(),
-      referenceUrls: z4.array(z4.string()).nullish(),
-      pollIntervalMs: z4.number().positive().nullish(),
-      pollTimeoutMs: z4.number().positive().nullish()
+    z6.object({
+      negativePrompt: z6.string().nullish(),
+      audioUrl: z6.string().nullish(),
+      promptExtend: z6.boolean().nullish(),
+      shotType: z6.enum(["single", "multi"]).nullish(),
+      watermark: z6.boolean().nullish(),
+      audio: z6.boolean().nullish(),
+      referenceUrls: z6.array(z6.string()).nullish(),
+      pollIntervalMs: z6.number().positive().nullish(),
+      pollTimeoutMs: z6.number().positive().nullish()
     }).passthrough()
   )
 );
-var alibabaVideoErrorSchema = z4.object({
-  code: z4.string().nullish(),
-  message: z4.string(),
-  request_id: z4.string().nullish()
+var alibabaVideoErrorSchema = z6.object({
+  code: z6.string().nullish(),
+  message: z6.string(),
+  request_id: z6.string().nullish()
 });
-var alibabaVideoFailedResponseHandler = createJsonErrorResponseHandler2({
+var alibabaVideoFailedResponseHandler = createJsonErrorResponseHandler3({
   errorSchema: alibabaVideoErrorSchema,
   errorToMessage: (data) => data.message
 });
-var alibabaVideoCreateTaskSchema = z4.object({
-  output: z4.object({
-    task_status: z4.string(),
-    task_id: z4.string()
+var alibabaVideoCreateTaskSchema = z6.object({
+  output: z6.object({
+    task_status: z6.string(),
+    task_id: z6.string()
   }).nullish(),
-  request_id: z4.string().nullish()
+  request_id: z6.string().nullish()
 });
-var alibabaVideoTaskStatusSchema = z4.object({
-  output: z4.object({
-    task_id: z4.string(),
-    task_status: z4.string(),
-    video_url: z4.string().nullish(),
-    submit_time: z4.string().nullish(),
-    scheduled_time: z4.string().nullish(),
-    end_time: z4.string().nullish(),
-    orig_prompt: z4.string().nullish(),
-    actual_prompt: z4.string().nullish(),
-    code: z4.string().nullish(),
-    message: z4.string().nullish()
+var alibabaVideoTaskStatusSchema = z6.object({
+  output: z6.object({
+    task_id: z6.string(),
+    task_status: z6.string(),
+    video_url: z6.string().nullish(),
+    submit_time: z6.string().nullish(),
+    scheduled_time: z6.string().nullish(),
+    end_time: z6.string().nullish(),
+    orig_prompt: z6.string().nullish(),
+    actual_prompt: z6.string().nullish(),
+    code: z6.string().nullish(),
+    message: z6.string().nullish()
   }).nullish(),
-  usage: z4.object({
-    duration: z4.number().nullish(),
-    output_video_duration: z4.number().nullish(),
-    SR: z4.number().nullish(),
-    size: z4.string().nullish()
+  usage: z6.object({
+    duration: z6.number().nullish(),
+    output_video_duration: z6.number().nullish(),
+    SR: z6.number().nullish(),
+    size: z6.string().nullish()
   }).nullish(),
-  request_id: z4.string().nullish()
+  request_id: z6.string().nullish()
 });
 function detectMode(modelId) {
   if (modelId.includes("-i2v")) return "i2v";
@@ -790,7 +939,7 @@ var AlibabaVideoModel = class {
     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({
+    const alibabaOptions = await parseProviderOptions3({
       provider: "alibaba",
       providerOptions: options.providerOptions,
       schema: alibabaVideoModelOptionsSchema
@@ -878,9 +1027,9 @@ var AlibabaVideoModel = class {
         details: "Alibaba video models only support generating 1 video per call."
       });
     }
-    const { value: createResponse } = await postJsonToApi2({
+    const { value: createResponse } = await postJsonToApi3({
       url: `${this.config.baseURL}/api/v1/services/aigc/video-generation/video-synthesis`,
-      headers: combineHeaders2(
+      headers: combineHeaders3(
         await resolve(this.config.headers),
         options.headers,
         {
@@ -892,7 +1041,7 @@ var AlibabaVideoModel = class {
         input,
         parameters
       },
-      successfulResponseHandler: createJsonResponseHandler2(
+      successfulResponseHandler: createJsonResponseHandler3(
         alibabaVideoCreateTaskSchema
       ),
       failedResponseHandler: alibabaVideoFailedResponseHandler,
@@ -921,11 +1070,11 @@ var AlibabaVideoModel = class {
       }
       const { value: statusResponse, responseHeaders: pollHeaders } = await getFromApi({
         url: `${this.config.baseURL}/api/v1/tasks/${taskId}`,
-        headers: combineHeaders2(
+        headers: combineHeaders3(
           await resolve(this.config.headers),
           options.headers
         ),
-        successfulResponseHandler: createJsonResponseHandler2(
+        successfulResponseHandler: createJsonResponseHandler3(
           alibabaVideoTaskStatusSchema
         ),
         failedResponseHandler: alibabaVideoFailedResponseHandler,
@@ -986,13 +1135,14 @@ var AlibabaVideoModel = class {
 };
 
 // src/version.ts
-var VERSION = true ? "1.0.25" : "0.0.0-test";
+var VERSION = true ? "1.0.26" : "0.0.0-test";
 
 // src/alibaba-provider.ts
 function createAlibaba(options = {}) {
-  var _a, _b;
+  var _a, _b, _c;
   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 embeddingBaseURL = (_c = withoutTrailingSlash(options.embeddingBaseURL)) != null ? _c : "https://dashscope-intl.aliyuncs.com/api/v1";
   const getHeaders = () => withUserAgentSuffix(
     {
       Authorization: `Bearer ${loadApiKey({
@@ -1014,6 +1164,12 @@ function createAlibaba(options = {}) {
       includeUsage: (_a2 = options.includeUsage) != null ? _a2 : true
     });
   };
+  const createEmbeddingModel = (modelId) => new AlibabaEmbeddingModel(modelId, {
+    provider: "alibaba.embedding",
+    baseURL: embeddingBaseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const createVideoModel = (modelId) => new AlibabaVideoModel(modelId, {
     provider: "alibaba.video",
     baseURL: videoBaseURL,
@@ -1031,14 +1187,15 @@ function createAlibaba(options = {}) {
   provider.specificationVersion = "v3";
   provider.languageModel = createLanguageModel;
   provider.chatModel = createLanguageModel;
+  provider.embedding = createEmbeddingModel;
+  provider.embeddingModel = createEmbeddingModel;
+  provider.textEmbedding = createEmbeddingModel;
+  provider.textEmbeddingModel = createEmbeddingModel;
   provider.video = createVideoModel;
   provider.videoModel = createVideoModel;
   provider.imageModel = (modelId) => {
     throw new NoSuchModelError({ modelId, modelType: "imageModel" });
   };
-  provider.embeddingModel = (modelId) => {
-    throw new NoSuchModelError({ modelId, modelType: "embeddingModel" });
-  };
   return provider;
 }
 var alibaba = createAlibaba();