@ai-sdk/alibaba 0.0.1

package/dist/index.mjs

@@ -0,0 +1,792 @@
+// src/alibaba-provider.ts
+import {
+  NoSuchModelError
+} from "@ai-sdk/provider";
+import {
+  createJsonErrorResponseHandler,
+  loadApiKey,
+  withoutTrailingSlash,
+  withUserAgentSuffix
+} from "@ai-sdk/provider-utils";
+import { z as z3 } from "zod/v4";
+
+// src/alibaba-chat-language-model.ts
+import {
+  getResponseMetadata,
+  mapOpenAICompatibleFinishReason
+} from "@ai-sdk/openai-compatible/internal";
+import {
+  InvalidResponseDataError
+} from "@ai-sdk/provider";
+import {
+  combineHeaders,
+  createEventSourceResponseHandler,
+  createJsonResponseHandler,
+  generateId,
+  isParsableJson,
+  parseProviderOptions,
+  postJsonToApi
+} from "@ai-sdk/provider-utils";
+import { z as z2 } from "zod/v4";
+
+// src/alibaba-chat-options.ts
+import { z } from "zod/v4";
+var alibabaLanguageModelOptions = z.object({
+  /**
+   * Enable thinking/reasoning mode for supported models.
+   * When enabled, the model generates reasoning content before the response.
+   *
+   * @default false
+   */
+  enableThinking: z.boolean().optional(),
+  /**
+   * Maximum number of reasoning tokens to generate.
+   */
+  thinkingBudget: z.number().positive().optional(),
+  /**
+   * Whether to enable parallel function calling during tool use.
+   *
+   * @default true
+   */
+  parallelToolCalls: z.boolean().optional()
+});
+
+// src/convert-alibaba-usage.ts
+function convertAlibabaUsage(usage) {
+  var _a, _b, _c, _d, _e, _f, _g;
+  return {
+    inputTokens: (_a = usage == null ? void 0 : usage.prompt_tokens) != null ? _a : void 0,
+    outputTokens: (_b = usage == null ? void 0 : usage.completion_tokens) != null ? _b : void 0,
+    totalTokens: (_c = usage == null ? void 0 : usage.total_tokens) != null ? _c : void 0,
+    reasoningTokens: (_e = (_d = usage == null ? void 0 : usage.completion_tokens_details) == null ? void 0 : _d.reasoning_tokens) != null ? _e : void 0,
+    cachedInputTokens: (_g = (_f = usage == null ? void 0 : usage.prompt_tokens_details) == null ? void 0 : _f.cached_tokens) != null ? _g : void 0
+  };
+}
+
+// src/convert-to-alibaba-chat-messages.ts
+import {
+  UnsupportedFunctionalityError
+} from "@ai-sdk/provider";
+import { convertToBase64 } from "@ai-sdk/provider-utils";
+function formatImageUrl({
+  data,
+  mediaType
+}) {
+  return data instanceof URL ? data.toString() : `data:${mediaType};base64,${convertToBase64(data)}`;
+}
+function convertToAlibabaChatMessages({
+  prompt,
+  cacheControlValidator
+}) {
+  const messages = [];
+  for (const { role, content, ...message } of prompt) {
+    switch (role) {
+      case "system": {
+        const cacheControl = cacheControlValidator == null ? void 0 : cacheControlValidator.getCacheControl(
+          message.providerOptions
+        );
+        if (cacheControl) {
+          messages.push({
+            role: "system",
+            content: [
+              {
+                type: "text",
+                text: content,
+                cache_control: cacheControl
+              }
+            ]
+          });
+        } else {
+          messages.push({ role: "system", content });
+        }
+        break;
+      }
+      case "user": {
+        if (content.length === 1 && content[0].type === "text") {
+          messages.push({
+            role: "user",
+            content: content[0].text
+          });
+          break;
+        }
+        messages.push({
+          role: "user",
+          content: content.map((part) => {
+            switch (part.type) {
+              case "text": {
+                return { type: "text", text: part.text };
+              }
+              case "file": {
+                if (part.mediaType.startsWith("image/")) {
+                  const mediaType = part.mediaType === "image/*" ? "image/jpeg" : part.mediaType;
+                  return {
+                    type: "image_url",
+                    image_url: {
+                      url: formatImageUrl({ data: part.data, mediaType })
+                    }
+                  };
+                } else {
+                  throw new UnsupportedFunctionalityError({
+                    functionality: "Only image file parts are supported"
+                  });
+                }
+              }
+            }
+          })
+        });
+        break;
+      }
+      case "assistant": {
+        let text = "";
+        const toolCalls = [];
+        for (const part of content) {
+          switch (part.type) {
+            case "text": {
+              text += part.text;
+              break;
+            }
+            case "tool-call": {
+              toolCalls.push({
+                id: part.toolCallId,
+                type: "function",
+                function: {
+                  name: part.toolName,
+                  arguments: JSON.stringify(part.input)
+                }
+              });
+              break;
+            }
+            case "reasoning": {
+              text += part.text;
+              break;
+            }
+          }
+        }
+        messages.push({
+          role: "assistant",
+          content: text || null,
+          tool_calls: toolCalls.length > 0 ? toolCalls : void 0
+        });
+        break;
+      }
+      case "tool": {
+        for (const toolResponse of content) {
+          const output = toolResponse.output;
+          let contentValue;
+          switch (output.type) {
+            case "text":
+            case "error-text":
+              contentValue = output.value;
+              break;
+            case "content":
+            case "json":
+            case "error-json":
+              contentValue = JSON.stringify(output.value);
+              break;
+          }
+          messages.push({
+            role: "tool",
+            tool_call_id: toolResponse.toolCallId,
+            content: contentValue
+          });
+        }
+        break;
+      }
+      default: {
+        const _exhaustiveCheck = role;
+        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
+      }
+    }
+  }
+  return messages;
+}
+
+// src/get-cache-control.ts
+var MAX_CACHE_BREAKPOINTS = 4;
+function getCacheControl(providerMetadata) {
+  var _a;
+  const alibaba2 = providerMetadata == null ? void 0 : providerMetadata.alibaba;
+  const cacheControlValue = (_a = alibaba2 == null ? void 0 : alibaba2.cacheControl) != null ? _a : alibaba2 == null ? void 0 : alibaba2.cache_control;
+  return cacheControlValue;
+}
+var CacheControlValidator = class {
+  constructor() {
+    this.breakpointCount = 0;
+    this.warnings = [];
+  }
+  getCacheControl(providerMetadata) {
+    const cacheControlValue = getCacheControl(providerMetadata);
+    if (!cacheControlValue) {
+      return void 0;
+    }
+    this.breakpointCount++;
+    if (this.breakpointCount > MAX_CACHE_BREAKPOINTS) {
+      this.warnings.push({
+        type: "other",
+        message: `cacheControl breakpoint limit: Maximum ${MAX_CACHE_BREAKPOINTS} cache breakpoints exceeded (found ${this.breakpointCount}). This breakpoint will be ignored.`
+      });
+      return void 0;
+    }
+    return cacheControlValue;
+  }
+  getWarnings() {
+    return this.warnings;
+  }
+};
+
+// src/alibaba-prepare-tools.ts
+import {
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
+} from "@ai-sdk/provider";
+function prepareTools({
+  tools,
+  toolChoice
+}) {
+  tools = (tools == null ? void 0 : tools.length) ? tools : void 0;
+  const toolWarnings = [];
+  if (tools == null) {
+    return { tools: void 0, toolChoice: void 0, toolWarnings };
+  }
+  const openaiCompatTools = [];
+  for (const tool of tools) {
+    if (tool.type === "provider-defined") {
+      toolWarnings.push({ type: "unsupported-tool", tool });
+    } else {
+      openaiCompatTools.push({
+        type: "function",
+        function: {
+          name: tool.name,
+          description: tool.description,
+          parameters: tool.inputSchema
+        }
+      });
+    }
+  }
+  if (toolChoice == null) {
+    return { tools: openaiCompatTools, toolChoice: void 0, toolWarnings };
+  }
+  const type = toolChoice.type;
+  switch (type) {
+    case "auto":
+    case "none":
+    case "required":
+      return { tools: openaiCompatTools, toolChoice: type, toolWarnings };
+    case "tool":
+      return {
+        tools: openaiCompatTools,
+        toolChoice: {
+          type: "function",
+          function: { name: toolChoice.toolName }
+        },
+        toolWarnings
+      };
+    default: {
+      const _exhaustiveCheck = type;
+      throw new UnsupportedFunctionalityError2({
+        functionality: `tool choice type: ${_exhaustiveCheck}`
+      });
+    }
+  }
+}
+
+// src/alibaba-chat-language-model.ts
+var AlibabaLanguageModel = class {
+  constructor(modelId, config) {
+    this.specificationVersion = "v2";
+    this.supportedUrls = {
+      "image/*": [/^https?:\/\/.*$/]
+    };
+    this.modelId = modelId;
+    this.config = config;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  /**
+   * Builds request arguments for Alibaba API call.
+   * Converts AI SDK options to Alibaba API format.
+   */
+  async getArgs({
+    prompt,
+    maxOutputTokens,
+    temperature,
+    topP,
+    topK,
+    frequencyPenalty,
+    presencePenalty,
+    stopSequences,
+    responseFormat,
+    seed,
+    providerOptions,
+    tools,
+    toolChoice
+  }) {
+    var _a;
+    const warnings = [];
+    const cacheControlValidator = new CacheControlValidator();
+    const alibabaOptions = await parseProviderOptions({
+      provider: "alibaba",
+      providerOptions,
+      schema: alibabaLanguageModelOptions
+    });
+    if (frequencyPenalty != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "frequencyPenalty"
+      });
+    }
+    const baseArgs = {
+      model: this.modelId,
+      max_tokens: maxOutputTokens,
+      temperature,
+      top_p: topP,
+      top_k: topK,
+      presence_penalty: presencePenalty,
+      stop: stopSequences,
+      seed,
+      response_format: (responseFormat == null ? void 0 : responseFormat.type) === "json" ? responseFormat.schema != null ? {
+        type: "json_schema",
+        json_schema: {
+          schema: responseFormat.schema,
+          name: (_a = responseFormat.name) != null ? _a : "response",
+          description: responseFormat.description
+        }
+      } : { type: "json_object" } : void 0,
+      // Alibaba-specific options
+      ...(alibabaOptions == null ? void 0 : alibabaOptions.enableThinking) != null ? { enable_thinking: alibabaOptions.enableThinking } : {},
+      ...(alibabaOptions == null ? void 0 : alibabaOptions.thinkingBudget) != null ? { thinking_budget: alibabaOptions.thinkingBudget } : {},
+      // Convert messages with cache control support
+      messages: convertToAlibabaChatMessages({
+        prompt,
+        cacheControlValidator
+      })
+    };
+    const {
+      tools: alibabaTools,
+      toolChoice: alibabaToolChoice,
+      toolWarnings
+    } = prepareTools({ tools, toolChoice });
+    warnings.push(...cacheControlValidator.getWarnings());
+    return {
+      args: {
+        ...baseArgs,
+        tools: alibabaTools,
+        tool_choice: alibabaToolChoice,
+        ...alibabaTools != null && (alibabaOptions == null ? void 0 : alibabaOptions.parallelToolCalls) !== void 0 ? { parallel_tool_calls: alibabaOptions.parallelToolCalls } : {}
+      },
+      warnings: [...warnings, ...toolWarnings]
+    };
+  }
+  async doGenerate(options) {
+    var _a, _b, _c;
+    const { args, warnings } = await this.getArgs(options);
+    const {
+      responseHeaders,
+      value: response,
+      rawValue: rawResponse
+    } = await postJsonToApi({
+      url: `${this.config.baseURL}/chat/completions`,
+      headers: combineHeaders(this.config.headers(), options.headers),
+      body: args,
+      failedResponseHandler: alibabaFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(
+        alibabaChatResponseSchema
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    const choice = response.choices[0];
+    const content = [];
+    const text = choice.message.content;
+    if (text != null && text.length > 0) {
+      content.push({ type: "text", text });
+    }
+    const reasoning = choice.message.reasoning_content;
+    if (reasoning != null && reasoning.length > 0) {
+      content.push({
+        type: "reasoning",
+        text: reasoning
+      });
+    }
+    if (choice.message.tool_calls != null) {
+      for (const toolCall of choice.message.tool_calls) {
+        content.push({
+          type: "tool-call",
+          toolCallId: toolCall.id,
+          toolName: toolCall.function.name,
+          input: toolCall.function.arguments
+        });
+      }
+    }
+    return {
+      content,
+      finishReason: mapOpenAICompatibleFinishReason(choice.finish_reason),
+      usage: convertAlibabaUsage(response.usage),
+      request: { body: JSON.stringify(args) },
+      response: {
+        ...getResponseMetadata(response),
+        headers: responseHeaders,
+        body: rawResponse
+      },
+      warnings,
+      providerMetadata: {
+        alibaba: {
+          cacheCreationInputTokens: (_c = (_b = (_a = response.usage) == null ? void 0 : _a.prompt_tokens_details) == null ? void 0 : _b.cache_creation_input_tokens) != null ? _c : null
+        }
+      }
+    };
+  }
+  async doStream(options) {
+    const { args, warnings } = await this.getArgs(options);
+    const body = {
+      ...args,
+      stream: true,
+      stream_options: this.config.includeUsage ? { include_usage: true } : void 0
+    };
+    const { responseHeaders, value: response } = await postJsonToApi({
+      url: `${this.config.baseURL}/chat/completions`,
+      headers: combineHeaders(this.config.headers(), options.headers),
+      body,
+      failedResponseHandler: alibabaFailedResponseHandler,
+      successfulResponseHandler: createEventSourceResponseHandler(
+        alibabaChatChunkSchema
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    let finishReason = "other";
+    let usage = void 0;
+    let isFirstChunk = true;
+    let activeText = false;
+    let activeReasoningId = null;
+    const toolCalls = [];
+    return {
+      stream: response.pipeThrough(
+        new TransformStream({
+          start(controller) {
+            controller.enqueue({ type: "stream-start", warnings });
+          },
+          transform(chunk, controller) {
+            var _a, _b, _c, _d;
+            if (options.includeRawChunks) {
+              controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
+            }
+            if (!chunk.success) {
+              controller.enqueue({ type: "error", error: chunk.error });
+              return;
+            }
+            const value = chunk.value;
+            if (isFirstChunk) {
+              isFirstChunk = false;
+              controller.enqueue({
+                type: "response-metadata",
+                ...getResponseMetadata(value)
+              });
+            }
+            if (value.usage != null) {
+              usage = value.usage;
+            }
+            if (value.choices.length === 0) {
+              return;
+            }
+            const choice = value.choices[0];
+            const delta = choice.delta;
+            if (delta.reasoning_content != null && delta.reasoning_content.length > 0) {
+              if (activeReasoningId == null) {
+                if (activeText) {
+                  controller.enqueue({ type: "text-end", id: "0" });
+                  activeText = false;
+                }
+                activeReasoningId = generateId();
+                controller.enqueue({
+                  type: "reasoning-start",
+                  id: activeReasoningId
+                });
+              }
+              controller.enqueue({
+                type: "reasoning-delta",
+                id: activeReasoningId,
+                delta: delta.reasoning_content
+              });
+            }
+            if (delta.content != null && delta.content.length > 0) {
+              if (activeReasoningId != null) {
+                controller.enqueue({
+                  type: "reasoning-end",
+                  id: activeReasoningId
+                });
+                activeReasoningId = null;
+              }
+              if (!activeText) {
+                controller.enqueue({ type: "text-start", id: "0" });
+                activeText = true;
+              }
+              controller.enqueue({
+                type: "text-delta",
+                id: "0",
+                delta: delta.content
+              });
+            }
+            if (delta.tool_calls != null) {
+              if (activeReasoningId != null) {
+                controller.enqueue({
+                  type: "reasoning-end",
+                  id: activeReasoningId
+                });
+                activeReasoningId = null;
+              }
+              if (activeText) {
+                controller.enqueue({ type: "text-end", id: "0" });
+                activeText = false;
+              }
+              for (const toolCallDelta of delta.tool_calls) {
+                const index = (_a = toolCallDelta.index) != null ? _a : toolCalls.length;
+                if (toolCalls[index] == null) {
+                  if (toolCallDelta.id == null) {
+                    throw new InvalidResponseDataError({
+                      data: toolCallDelta,
+                      message: `Expected 'id' to be a string.`
+                    });
+                  }
+                  if (((_b = toolCallDelta.function) == null ? void 0 : _b.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: (_c = toolCallDelta.function.arguments) != null ? _c : ""
+                    },
+                    hasFinished: false
+                  };
+                  const toolCall2 = toolCalls[index];
+                  if (toolCall2.function.arguments.length > 0) {
+                    controller.enqueue({
+                      type: "tool-input-delta",
+                      id: toolCall2.id,
+                      delta: toolCall2.function.arguments
+                    });
+                  }
+                  if (isParsableJson(toolCall2.function.arguments)) {
+                    controller.enqueue({
+                      type: "tool-input-end",
+                      id: toolCall2.id
+                    });
+                    controller.enqueue({
+                      type: "tool-call",
+                      toolCallId: toolCall2.id,
+                      toolName: toolCall2.function.name,
+                      input: toolCall2.function.arguments
+                    });
+                    toolCall2.hasFinished = true;
+                  }
+                  continue;
+                }
+                const toolCall = toolCalls[index];
+                if (toolCall.hasFinished) {
+                  continue;
+                }
+                if (((_d = toolCallDelta.function) == null ? void 0 : _d.arguments) != null) {
+                  toolCall.function.arguments += toolCallDelta.function.arguments;
+                  controller.enqueue({
+                    type: "tool-input-delta",
+                    id: toolCall.id,
+                    delta: toolCallDelta.function.arguments
+                  });
+                }
+                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;
+                }
+              }
+            }
+            if (choice.finish_reason != null) {
+              finishReason = mapOpenAICompatibleFinishReason(
+                choice.finish_reason
+              );
+            }
+          },
+          flush(controller) {
+            var _a, _b;
+            if (activeReasoningId != null) {
+              controller.enqueue({
+                type: "reasoning-end",
+                id: activeReasoningId
+              });
+            }
+            if (activeText) {
+              controller.enqueue({ type: "text-end", id: "0" });
+            }
+            controller.enqueue({
+              type: "finish",
+              finishReason,
+              usage: convertAlibabaUsage(usage),
+              providerMetadata: {
+                alibaba: {
+                  cacheCreationInputTokens: (_b = (_a = usage == null ? void 0 : usage.prompt_tokens_details) == null ? void 0 : _a.cache_creation_input_tokens) != null ? _b : null
+                }
+              }
+            });
+          }
+        })
+      ),
+      request: { body: JSON.stringify(body) },
+      response: { headers: responseHeaders }
+    };
+  }
+};
+var alibabaUsageSchema = z2.object({
+  prompt_tokens: z2.number(),
+  completion_tokens: z2.number(),
+  total_tokens: z2.number(),
+  prompt_tokens_details: z2.object({
+    cached_tokens: z2.number().nullish(),
+    cache_creation_input_tokens: z2.number().nullish()
+  }).nullish(),
+  completion_tokens_details: z2.object({
+    reasoning_tokens: z2.number().nullish()
+  }).nullish()
+});
+var alibabaChatResponseSchema = z2.object({
+  id: z2.string().nullish(),
+  created: z2.number().nullish(),
+  model: z2.string().nullish(),
+  choices: z2.array(
+    z2.object({
+      message: z2.object({
+        role: z2.literal("assistant").nullish(),
+        content: z2.string().nullish(),
+        reasoning_content: z2.string().nullish(),
+        // Alibaba thinking mode
+        tool_calls: z2.array(
+          z2.object({
+            id: z2.string(),
+            type: z2.literal("function"),
+            function: z2.object({
+              name: z2.string(),
+              arguments: z2.string()
+            })
+          })
+        ).nullish()
+      }),
+      finish_reason: z2.string().nullish(),
+      index: z2.number()
+    })
+  ),
+  usage: alibabaUsageSchema.nullish()
+});
+var alibabaChatChunkSchema = z2.object({
+  id: z2.string().nullish(),
+  created: z2.number().nullish(),
+  model: z2.string().nullish(),
+  choices: z2.array(
+    z2.object({
+      delta: z2.object({
+        role: z2.enum(["assistant"]).nullish(),
+        content: z2.string().nullish(),
+        reasoning_content: z2.string().nullish(),
+        // Alibaba thinking mode delta
+        tool_calls: z2.array(
+          z2.object({
+            index: z2.number().nullish(),
+            // Index for accumulating tool calls
+            id: z2.string().nullish(),
+            type: z2.literal("function").nullish(),
+            function: z2.object({
+              name: z2.string().nullish(),
+              arguments: z2.string().nullish()
+            }).nullish()
+          })
+        ).nullish()
+      }),
+      finish_reason: z2.string().nullish(),
+      index: z2.number()
+    })
+  ),
+  usage: alibabaUsageSchema.nullish()
+  // Usage only appears in final chunk
+});
+
+// src/version.ts
+var VERSION = "0.0.1";
+
+// src/alibaba-provider.ts
+var alibabaErrorDataSchema = z3.object({
+  error: z3.object({
+    message: z3.string(),
+    code: z3.string().nullish(),
+    type: z3.string().nullish()
+  })
+});
+var alibabaFailedResponseHandler = createJsonErrorResponseHandler({
+  errorSchema: alibabaErrorDataSchema,
+  errorToMessage: (data) => data.error.message
+});
+function createAlibaba(options = {}) {
+  var _a;
+  const baseURL = (_a = withoutTrailingSlash(options.baseURL)) != null ? _a : "https://dashscope-intl.aliyuncs.com/compatible-mode/v1";
+  const getHeaders = () => withUserAgentSuffix(
+    {
+      Authorization: `Bearer ${loadApiKey({
+        apiKey: options.apiKey,
+        environmentVariableName: "ALIBABA_API_KEY",
+        description: "Alibaba Cloud (DashScope)"
+      })}`,
+      ...options.headers
+    },
+    `ai-sdk/alibaba/${VERSION}`
+  );
+  const createLanguageModel = (modelId) => {
+    var _a2;
+    return new AlibabaLanguageModel(modelId, {
+      provider: "alibaba.chat",
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch,
+      includeUsage: (_a2 = options.includeUsage) != null ? _a2 : true
+    });
+  };
+  const provider = function(modelId) {
+    if (new.target) {
+      throw new Error(
+        "The Alibaba model function cannot be called with the new keyword."
+      );
+    }
+    return createLanguageModel(modelId);
+  };
+  provider.specificationVersion = "v2";
+  provider.languageModel = createLanguageModel;
+  provider.chatModel = createLanguageModel;
+  provider.imageModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "imageModel" });
+  };
+  provider.textEmbeddingModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
+  };
+  return provider;
+}
+var alibaba = createAlibaba();
+export {
+  VERSION,
+  alibaba,
+  createAlibaba
+};
+//# sourceMappingURL=index.mjs.map