@ai-sdk/alibaba 2.0.0-beta.3 → 2.0.0-beta.31

package/src/alibaba-chat-language-model.ts

@@ -4,7 +4,6 @@ import {
   prepareTools,
 } from '@ai-sdk/openai-compatible/internal';
 import {
-  InvalidResponseDataError,
   type LanguageModelV4,
   type LanguageModelV4CallOptions,
   type LanguageModelV4Content,
@@ -19,10 +18,16 @@ import {
   createEventSourceResponseHandler,
   createJsonResponseHandler,
   generateId,
-  isParsableJson,
+  type InferSchema,
+  isCustomReasoning,
+  mapReasoningToProviderBudget,
   parseProviderOptions,
   postJsonToApi,
   type ParseResult,
+  serializeModelOptions,
+  StreamingToolCallTracker,
+  WORKFLOW_SERIALIZE,
+  WORKFLOW_DESERIALIZE,
 } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 import {
@@ -50,6 +55,20 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
 
   private readonly config: AlibabaConfig;
 
+  static [WORKFLOW_SERIALIZE](model: AlibabaLanguageModel) {
+    return serializeModelOptions({
+      modelId: model.modelId,
+      config: model.config,
+    });
+  }
+
+  static [WORKFLOW_DESERIALIZE](options: {
+    modelId: AlibabaChatModelId;
+    config: AlibabaConfig;
+  }) {
+    return new AlibabaLanguageModel(options.modelId, options.config);
+  }
+
   constructor(modelId: AlibabaChatModelId, config: AlibabaConfig) {
     this.modelId = modelId;
     this.config = config;
@@ -78,6 +97,7 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
     stopSequences,
     responseFormat,
     seed,
+    reasoning,
     providerOptions,
     tools,
     toolChoice,
@@ -121,13 +141,11 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
             : { type: 'json_object' }
           : undefined,
 
-      // Alibaba-specific options
-      ...(alibabaOptions?.enableThinking != null
-        ? { enable_thinking: alibabaOptions.enableThinking }
-        : {}),
-      ...(alibabaOptions?.thinkingBudget != null
-        ? { thinking_budget: alibabaOptions.thinkingBudget }
-        : {}),
+      ...resolveAlibabaThinking({
+        reasoning,
+        alibabaOptions,
+        warnings,
+      }),
 
       // Convert messages with cache control support
       messages: convertToAlibabaChatMessages({
@@ -170,7 +188,7 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
       rawValue: rawResponse,
     } = await postJsonToApi({
       url: `${this.config.baseURL}/chat/completions`,
-      headers: combineHeaders(this.config.headers(), options.headers),
+      headers: combineHeaders(this.config.headers?.(), options.headers),
       body: args,
       failedResponseHandler: alibabaFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(
@@ -203,7 +221,7 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
       for (const toolCall of choice.message.tool_calls) {
         content.push({
           type: 'tool-call',
-          toolCallId: toolCall.id,
+          toolCallId: toolCall.id ?? generateId(),
           toolName: toolCall.function.name,
           input: toolCall.function.arguments!,
         });
@@ -241,7 +259,7 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
 
     const { responseHeaders, value: response } = await postJsonToApi({
       url: `${this.config.baseURL}/chat/completions`,
-      headers: combineHeaders(this.config.headers(), options.headers),
+      headers: combineHeaders(this.config.headers?.(), options.headers),
       body,
       failedResponseHandler: alibabaFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
@@ -262,13 +280,7 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
     let activeText = false;
     let activeReasoningId: string | null = null;
 
-    // Track tool calls for accumulation across chunks
-    const toolCalls: Array<{
-      id: string;
-      type: 'function';
-      function: { name: string; arguments: string };
-      hasFinished: boolean;
-    }> = [];
+    const toolCallTracker = new StreamingToolCallTracker({ generateId });
 
     return {
       stream: response.pipeThrough(
@@ -381,106 +393,10 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
               }
 
               for (const toolCallDelta of delta.tool_calls) {
-                const index = toolCallDelta.index ?? toolCalls.length;
-
-                // New tool call - first chunk with id and name
-                if (toolCalls[index] == null) {
-                  if (toolCallDelta.id == null) {
-                    throw new InvalidResponseDataError({
-                      data: toolCallDelta,
-                      message: `Expected 'id' to be a string.`,
-                    });
-                  }
-
-                  if (toolCallDelta.function?.name == null) {
-                    throw new InvalidResponseDataError({
-                      data: toolCallDelta,
-                      message: `Expected 'function.name' to be a string.`,
-                    });
-                  }
-
-                  controller.enqueue({
-                    type: 'tool-input-start',
-                    id: toolCallDelta.id,
-                    toolName: toolCallDelta.function.name,
-                  });
-
-                  toolCalls[index] = {
-                    id: toolCallDelta.id,
-                    type: 'function',
-                    function: {
-                      name: toolCallDelta.function.name,
-                      arguments: toolCallDelta.function.arguments ?? '',
-                    },
-                    hasFinished: false,
-                  };
-
-                  const toolCall = toolCalls[index];
-
-                  // Send initial delta if arguments started
-                  if (toolCall.function.arguments.length > 0) {
-                    controller.enqueue({
-                      type: 'tool-input-delta',
-                      id: toolCall.id,
-                      delta: toolCall.function.arguments,
-                    });
-                  }
-
-                  // Check if already complete (some providers send full tool call at once)
-                  if (isParsableJson(toolCall.function.arguments)) {
-                    controller.enqueue({
-                      type: 'tool-input-end',
-                      id: toolCall.id,
-                    });
-
-                    controller.enqueue({
-                      type: 'tool-call',
-                      toolCallId: toolCall.id,
-                      toolName: toolCall.function.name,
-                      input: toolCall.function.arguments,
-                    });
-
-                    toolCall.hasFinished = true;
-                  }
-
-                  continue;
-                }
-
-                // Existing tool call - accumulate arguments
-                const toolCall = toolCalls[index];
-
-                if (toolCall.hasFinished) {
-                  continue;
-                }
-
-                // Append arguments if not null (skip arguments: null chunks)
-                if (toolCallDelta.function?.arguments != null) {
-                  toolCall.function.arguments +=
-                    toolCallDelta.function.arguments;
-
-                  controller.enqueue({
-                    type: 'tool-input-delta',
-                    id: toolCall.id,
-                    delta: toolCallDelta.function.arguments,
-                  });
-                }
-
-                // Check if tool call is now complete
-                if (isParsableJson(toolCall.function.arguments)) {
-                  controller.enqueue({
-                    type: 'tool-input-end',
-                    id: toolCall.id,
-                  });
-
-                  controller.enqueue({
-                    type: 'tool-call',
-                    toolCallId: toolCall.id,
-                    toolName: toolCall.function.name,
-                    input: toolCall.function.arguments,
-                  });
-
-                  toolCall.hasFinished = true;
-                }
+                toolCallTracker.processDelta(
+                  toolCallDelta,
+                  controller.enqueue.bind(controller),
+                );
               }
             }
 
@@ -505,6 +421,8 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
               controller.enqueue({ type: 'text-end', id: '0' });
             }
 
+            toolCallTracker.flush(controller.enqueue.bind(controller));
+
             controller.enqueue({
               type: 'finish',
               finishReason,
@@ -519,6 +437,50 @@ export class AlibabaLanguageModel implements LanguageModelV4 {
   }
 }
 
+function resolveAlibabaThinking({
+  reasoning,
+  alibabaOptions,
+  warnings,
+}: {
+  reasoning: LanguageModelV4CallOptions['reasoning'];
+  alibabaOptions: InferSchema<typeof alibabaLanguageModelOptions> | undefined;
+  warnings: SharedV4Warning[];
+}): { enable_thinking?: boolean; thinking_budget?: number } {
+  if (
+    alibabaOptions?.enableThinking != null ||
+    alibabaOptions?.thinkingBudget != null
+  ) {
+    return {
+      ...(alibabaOptions.enableThinking != null
+        ? { enable_thinking: alibabaOptions.enableThinking }
+        : {}),
+      ...(alibabaOptions.thinkingBudget != null
+        ? { thinking_budget: alibabaOptions.thinkingBudget }
+        : {}),
+    };
+  }
+
+  if (!isCustomReasoning(reasoning)) {
+    return {};
+  }
+
+  if (reasoning === 'none') {
+    return { enable_thinking: false };
+  }
+
+  const thinkingBudget = mapReasoningToProviderBudget({
+    reasoning,
+    maxOutputTokens: 16384,
+    maxReasoningBudget: 16384,
+    warnings,
+  });
+
+  return {
+    enable_thinking: true,
+    ...(thinkingBudget != null ? { thinking_budget: thinkingBudget } : {}),
+  };
+}
+
 /**
  * Reference for schemas below:
  * https://www.alibabacloud.com/help/en/model-studio/qwen-api-via-openai-chat-completions

package/src/alibaba-config.ts

@@ -3,7 +3,7 @@ import type { FetchFunction } from '@ai-sdk/provider-utils';
 export interface AlibabaConfig {
   provider: string;
   baseURL: string;
-  headers: () => Record<string, string | undefined>;
+  headers?: () => Record<string, string | undefined>;
   fetch?: FetchFunction;
   includeUsage?: boolean;
 }

package/src/convert-to-alibaba-chat-messages.ts

@@ -3,7 +3,7 @@ import {
   type LanguageModelV4Prompt,
   UnsupportedFunctionalityError,
 } from '@ai-sdk/provider';
-import { convertToBase64 } from '@ai-sdk/provider-utils';
+import { convertToBase64, isProviderReference } from '@ai-sdk/provider-utils';
 import type { AlibabaChatPrompt } from './alibaba-chat-prompt';
 import type { CacheControlValidator } from './get-cache-control';
 
@@ -53,26 +53,13 @@ export function convertToAlibabaChatMessages({
       }
 
       case 'user': {
-        const isSinglePart = content.length === 1;
-
-        if (
-          isSinglePart &&
-          content[0].type === 'text' &&
-          !messageCacheControl
-        ) {
-          messages.push({
-            role: 'user',
-            content: content[0].text,
-          });
-          break;
-        }
-
         messages.push({
           role: 'user',
-          content: content.map(part => {
-            const partCacheControl = isSinglePart
-              ? messageCacheControl
-              : cacheControlValidator?.getCacheControl(part.providerOptions);
+          content: content.map((part, index) => {
+            const isLastPart = index === content.length - 1;
+            const partCacheControl =
+              cacheControlValidator?.getCacheControl(part.providerOptions) ??
+              (isLastPart ? messageCacheControl : undefined);
 
             switch (part.type) {
               case 'text': {
@@ -86,6 +73,12 @@ export function convertToAlibabaChatMessages({
               }
 
               case 'file': {
+                if (isProviderReference(part.data)) {
+                  throw new UnsupportedFunctionalityError({
+                    functionality: 'file parts with provider references',
+                  });
+                }
+
                 if (part.mediaType.startsWith('image/')) {
                   const mediaType =
                     part.mediaType === 'image/*'
@@ -163,17 +156,15 @@ export function convertToAlibabaChatMessages({
           r => r.type !== 'tool-approval-response',
         );
 
-        const isSinglePart = toolResponses.length === 1;
-
         for (let i = 0; i < toolResponses.length; i++) {
           const toolResponse = toolResponses[i];
           const output = toolResponse.output;
+          const isLastPart = i === toolResponses.length - 1;
 
-          const partCacheControl = isSinglePart
-            ? messageCacheControl
-            : cacheControlValidator?.getCacheControl(
-                (toolResponse as any).providerOptions,
-              );
+          const partCacheControl =
+            cacheControlValidator?.getCacheControl(
+              toolResponse.providerOptions,
+            ) ?? (isLastPart ? messageCacheControl : undefined);
 
           let contentValue: string;
           switch (output.type) {

package/dist/index.d.mts

@@ -1,119 +0,0 @@
-import { z } from 'zod/v4';
-import { FetchFunction } from '@ai-sdk/provider-utils';
-import { ProviderV4, LanguageModelV4, Experimental_VideoModelV4 } from '@ai-sdk/provider';
-
-type AlibabaChatModelId = 'qwen3-max' | 'qwen3-max-preview' | 'qwen-plus' | 'qwen-plus-latest' | 'qwen-flash' | 'qwen-turbo' | 'qwen-turbo-latest' | 'qwen3-235b-a22b' | 'qwen3-32b' | 'qwen3-30b-a3b' | 'qwen3-14b' | 'qwen3-next-80b-a3b-thinking' | 'qwen3-235b-a22b-thinking-2507' | 'qwen3-30b-a3b-thinking-2507' | 'qwq-plus' | 'qwq-plus-latest' | 'qwq-32b' | 'qwen-coder' | 'qwen3-coder-plus' | 'qwen3-coder-flash' | (string & {});
-declare const alibabaLanguageModelOptions: z.ZodObject<{
-    enableThinking: z.ZodOptional<z.ZodBoolean>;
-    thinkingBudget: z.ZodOptional<z.ZodNumber>;
-    parallelToolCalls: z.ZodOptional<z.ZodBoolean>;
-}, z.core.$strip>;
-type AlibabaLanguageModelOptions = z.infer<typeof alibabaLanguageModelOptions>;
-
-type AlibabaCacheControl = {
-    type: string;
-};
-
-type AlibabaVideoModelId = 'wan2.6-t2v' | 'wan2.5-t2v-preview' | 'wan2.6-i2v' | 'wan2.6-i2v-flash' | 'wan2.6-r2v' | 'wan2.6-r2v-flash' | (string & {});
-
-interface AlibabaProvider extends ProviderV4 {
-    (modelId: AlibabaChatModelId): LanguageModelV4;
-    /**
-     * Creates a model for text generation.
-     */
-    languageModel(modelId: AlibabaChatModelId): LanguageModelV4;
-    /**
-     * Creates a chat model for text generation.
-     */
-    chatModel(modelId: AlibabaChatModelId): LanguageModelV4;
-    /**
-     * Creates a model for video generation.
-     */
-    video(modelId: AlibabaVideoModelId): Experimental_VideoModelV4;
-    /**
-     * Creates a model for video generation.
-     */
-    videoModel(modelId: AlibabaVideoModelId): Experimental_VideoModelV4;
-}
-interface AlibabaProviderSettings {
-    /**
-     * Use a different URL prefix for API calls, e.g. to use proxy servers or regional endpoints.
-     * The default prefix is `https://dashscope-intl.aliyuncs.com/compatible-mode/v1`.
-     */
-    baseURL?: string;
-    /**
-     * Use a different URL prefix for video generation API calls.
-     * The video API uses the DashScope native endpoint (not the OpenAI-compatible endpoint).
-     * The default prefix is `https://dashscope-intl.aliyuncs.com`.
-     */
-    videoBaseURL?: string;
-    /**
-     * API key that is being sent using the `Authorization` header.
-     * It defaults to the `ALIBABA_API_KEY` environment variable.
-     */
-    apiKey?: string;
-    /**
-     * Custom headers to include in the requests.
-     */
-    headers?: Record<string, string>;
-    /**
-     * Custom fetch implementation. You can use it as a middleware to intercept requests,
-     * or to provide a custom fetch implementation for e.g. testing.
-     */
-    fetch?: FetchFunction;
-    /**
-     * Include usage information in streaming responses.
-     * When enabled, token usage will be included in the final chunk.
-     *
-     * @default true
-     */
-    includeUsage?: boolean;
-}
-/**
- * Create an Alibaba Cloud (Qwen) provider instance.
- */
-declare function createAlibaba(options?: AlibabaProviderSettings): AlibabaProvider;
-declare const alibaba: AlibabaProvider;
-
-type AlibabaVideoModelOptions = {
-    /** Negative prompt to specify what to avoid (max 500 chars). */
-    negativePrompt?: string | null;
-    /** URL to audio file for audio-video sync (WAV/MP3, 3-30s, max 15MB). */
-    audioUrl?: string | null;
-    /** Enable prompt extension/rewriting for better generation. Defaults to true. */
-    promptExtend?: boolean | null;
-    /** Shot type: 'single' for single-shot or 'multi' for multi-shot narrative. */
-    shotType?: 'single' | 'multi' | null;
-    /** Whether to add watermark to generated video. Defaults to false. */
-    watermark?: boolean | null;
-    /** Enable audio generation (for I2V/R2V models). */
-    audio?: boolean | null;
-    /**
-     * Reference URLs for reference-to-video mode.
-     * Array of URLs to images (0-5) and/or videos (0-3), max 5 total.
-     * Use character identifiers (character1, character2) in prompts to reference them.
-     */
-    referenceUrls?: string[] | null;
-    /** Polling interval in milliseconds. Defaults to 5000 (5 seconds). */
-    pollIntervalMs?: number | null;
-    /** Maximum wait time in milliseconds for video generation. Defaults to 600000 (10 minutes). */
-    pollTimeoutMs?: number | null;
-    [key: string]: unknown;
-};
-
-type AlibabaUsage = {
-    prompt_tokens?: number | null;
-    completion_tokens?: number | null;
-    total_tokens?: number | null;
-    prompt_tokens_details?: {
-        cached_tokens?: number | null;
-        cache_creation_input_tokens?: number | null;
-    } | null;
-    completion_tokens_details?: {
-        reasoning_tokens?: number | null;
-    } | null;
-};
-
-declare const VERSION: string;
-
-export { type AlibabaCacheControl, type AlibabaChatModelId, type AlibabaLanguageModelOptions, type AlibabaProvider, type AlibabaLanguageModelOptions as AlibabaProviderOptions, type AlibabaProviderSettings, type AlibabaUsage, type AlibabaVideoModelId, type AlibabaVideoModelOptions, type AlibabaVideoModelOptions as AlibabaVideoProviderOptions, VERSION, alibaba, createAlibaba };