@ai-sdk/alibaba 1.0.24 → 1.0.26

package/src/alibaba-provider.ts

@@ -1,4 +1,5 @@
 import {
+  type EmbeddingModelV3,
   type Experimental_VideoModelV3,
   type LanguageModelV3,
   NoSuchModelError,
@@ -12,6 +13,8 @@ import {
 } from '@ai-sdk/provider-utils';
 import { AlibabaLanguageModel } from './alibaba-chat-language-model';
 import type { AlibabaChatModelId } from './alibaba-chat-options';
+import { AlibabaEmbeddingModel } from './alibaba-embedding-model';
+import type { AlibabaEmbeddingModelId } from './alibaba-embedding-options';
 import { AlibabaVideoModel } from './alibaba-video-model';
 import type { AlibabaVideoModelId } from './alibaba-video-settings';
 import { VERSION } from './version';
@@ -34,6 +37,30 @@ export interface AlibabaProvider extends ProviderV3 {
    */
   chatModel(modelId: AlibabaChatModelId): LanguageModelV3;
 
+  /**
+   * Creates a model for text embeddings.
+   */
+  embedding(modelId: AlibabaEmbeddingModelId): EmbeddingModelV3;
+
+  /**
+   * Creates a model for text embeddings.
+   */
+  embeddingModel(modelId: AlibabaEmbeddingModelId): EmbeddingModelV3;
+
+  /**
+   * Creates a model for text embeddings.
+   *
+   * @deprecated Use `embedding` instead.
+   */
+  textEmbedding(modelId: AlibabaEmbeddingModelId): EmbeddingModelV3;
+
+  /**
+   * Creates a model for text embeddings.
+   *
+   * @deprecated Use `embeddingModel` instead.
+   */
+  textEmbeddingModel(modelId: AlibabaEmbeddingModelId): EmbeddingModelV3;
+
   /**
    * Creates a model for video generation.
    */
@@ -59,6 +86,13 @@ export interface AlibabaProviderSettings {
    */
   videoBaseURL?: string;
 
+  /**
+   * Use a different URL prefix for embedding API calls.
+   * The embedding API uses the DashScope native endpoint (not the OpenAI-compatible endpoint).
+   * The default prefix is `https://dashscope-intl.aliyuncs.com/api/v1`.
+   */
+  embeddingBaseURL?: string;
+
   /**
    * API key that is being sent using the `Authorization` header.
    * It defaults to the `ALIBABA_API_KEY` environment variable.
@@ -99,6 +133,10 @@ export function createAlibaba(
     withoutTrailingSlash(options.videoBaseURL) ??
     'https://dashscope-intl.aliyuncs.com';
 
+  const embeddingBaseURL =
+    withoutTrailingSlash(options.embeddingBaseURL) ??
+    'https://dashscope-intl.aliyuncs.com/api/v1';
+
   const getHeaders = () =>
     withUserAgentSuffix(
       {
@@ -121,6 +159,14 @@ export function createAlibaba(
       includeUsage: options.includeUsage ?? true,
     });
 
+  const createEmbeddingModel = (modelId: AlibabaEmbeddingModelId) =>
+    new AlibabaEmbeddingModel(modelId, {
+      provider: 'alibaba.embedding',
+      baseURL: embeddingBaseURL,
+      headers: getHeaders,
+      fetch: options.fetch,
+    });
+
   const createVideoModel = (modelId: AlibabaVideoModelId) =>
     new AlibabaVideoModel(modelId, {
       provider: 'alibaba.video',
@@ -142,6 +188,10 @@ export function createAlibaba(
   provider.specificationVersion = 'v3' as const;
   provider.languageModel = createLanguageModel;
   provider.chatModel = createLanguageModel;
+  provider.embedding = createEmbeddingModel;
+  provider.embeddingModel = createEmbeddingModel;
+  provider.textEmbedding = createEmbeddingModel;
+  provider.textEmbeddingModel = createEmbeddingModel;
   provider.video = createVideoModel;
   provider.videoModel = createVideoModel;
 
@@ -149,10 +199,6 @@ export function createAlibaba(
     throw new NoSuchModelError({ modelId, modelType: 'imageModel' });
   };
 
-  provider.embeddingModel = (modelId: string) => {
-    throw new NoSuchModelError({ modelId, modelType: 'embeddingModel' });
-  };
-
   return provider;
 }
 

package/src/index.ts

@@ -5,6 +5,10 @@ export type {
   AlibabaLanguageModelOptions as AlibabaProviderOptions,
 } from './alibaba-chat-options';
 export type { AlibabaCacheControl } from './alibaba-chat-prompt';
+export type {
+  AlibabaEmbeddingModelId,
+  AlibabaEmbeddingModelOptions,
+} from './alibaba-embedding-options';
 export {
   type AlibabaProvider,
   type AlibabaProviderSettings,