ai 3.1.30 → 3.1.31

package/dist/index.mjs

@@ -4,6 +4,15 @@ var __export = (target, all) => {
     __defProp(target, name, { get: all[name], enumerable: true });
 };
 
+// streams/index.ts
+import {
+  formatStreamPart,
+  parseStreamPart,
+  readDataStream,
+  parseComplexResponse
+} from "@ai-sdk/ui-utils";
+import { generateId as generateId2, generateId as generateId3 } from "@ai-sdk/provider-utils";
+
 // core/util/retry-with-exponential-backoff.ts
 import { APICallError, RetryError } from "@ai-sdk/provider";
 import { getErrorMessage, isAbortError } from "@ai-sdk/provider-utils";
@@ -1446,15 +1455,7 @@ function prepareResponseHeaders(init, { contentType }) {
 
 // core/generate-text/run-tools-transformation.ts
 import { NoSuchToolError as NoSuchToolError2 } from "@ai-sdk/provider";
-
-// shared/generate-id.ts
-import { customAlphabet } from "nanoid/non-secure";
-var generateId = customAlphabet(
-  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  7
-);
-
-// core/generate-text/run-tools-transformation.ts
+import { generateId } from "@ai-sdk/ui-utils";
 function runToolsTransformation({
   tools,
   generatorStream
@@ -1998,282 +1999,6 @@ import {
   UnsupportedJSONSchemaError
 } from "@ai-sdk/provider";
 
-// shared/stream-parts.ts
-var textStreamPart = {
-  code: "0",
-  name: "text",
-  parse: (value) => {
-    if (typeof value !== "string") {
-      throw new Error('"text" parts expect a string value.');
-    }
-    return { type: "text", value };
-  }
-};
-var functionCallStreamPart = {
-  code: "1",
-  name: "function_call",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("function_call" in value) || typeof value.function_call !== "object" || value.function_call == null || !("name" in value.function_call) || !("arguments" in value.function_call) || typeof value.function_call.name !== "string" || typeof value.function_call.arguments !== "string") {
-      throw new Error(
-        '"function_call" parts expect an object with a "function_call" property.'
-      );
-    }
-    return {
-      type: "function_call",
-      value
-    };
-  }
-};
-var dataStreamPart = {
-  code: "2",
-  name: "data",
-  parse: (value) => {
-    if (!Array.isArray(value)) {
-      throw new Error('"data" parts expect an array value.');
-    }
-    return { type: "data", value };
-  }
-};
-var errorStreamPart = {
-  code: "3",
-  name: "error",
-  parse: (value) => {
-    if (typeof value !== "string") {
-      throw new Error('"error" parts expect a string value.');
-    }
-    return { type: "error", value };
-  }
-};
-var assistantMessageStreamPart = {
-  code: "4",
-  name: "assistant_message",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("id" in value) || !("role" in value) || !("content" in value) || typeof value.id !== "string" || typeof value.role !== "string" || value.role !== "assistant" || !Array.isArray(value.content) || !value.content.every(
-      (item) => item != null && typeof item === "object" && "type" in item && item.type === "text" && "text" in item && item.text != null && typeof item.text === "object" && "value" in item.text && typeof item.text.value === "string"
-    )) {
-      throw new Error(
-        '"assistant_message" parts expect an object with an "id", "role", and "content" property.'
-      );
-    }
-    return {
-      type: "assistant_message",
-      value
-    };
-  }
-};
-var assistantControlDataStreamPart = {
-  code: "5",
-  name: "assistant_control_data",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("threadId" in value) || !("messageId" in value) || typeof value.threadId !== "string" || typeof value.messageId !== "string") {
-      throw new Error(
-        '"assistant_control_data" parts expect an object with a "threadId" and "messageId" property.'
-      );
-    }
-    return {
-      type: "assistant_control_data",
-      value: {
-        threadId: value.threadId,
-        messageId: value.messageId
-      }
-    };
-  }
-};
-var dataMessageStreamPart = {
-  code: "6",
-  name: "data_message",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("role" in value) || !("data" in value) || typeof value.role !== "string" || value.role !== "data") {
-      throw new Error(
-        '"data_message" parts expect an object with a "role" and "data" property.'
-      );
-    }
-    return {
-      type: "data_message",
-      value
-    };
-  }
-};
-var toolCallsStreamPart = {
-  code: "7",
-  name: "tool_calls",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("tool_calls" in value) || typeof value.tool_calls !== "object" || value.tool_calls == null || !Array.isArray(value.tool_calls) || value.tool_calls.some(
-      (tc) => tc == null || typeof tc !== "object" || !("id" in tc) || typeof tc.id !== "string" || !("type" in tc) || typeof tc.type !== "string" || !("function" in tc) || tc.function == null || typeof tc.function !== "object" || !("arguments" in tc.function) || typeof tc.function.name !== "string" || typeof tc.function.arguments !== "string"
-    )) {
-      throw new Error(
-        '"tool_calls" parts expect an object with a ToolCallPayload.'
-      );
-    }
-    return {
-      type: "tool_calls",
-      value
-    };
-  }
-};
-var messageAnnotationsStreamPart = {
-  code: "8",
-  name: "message_annotations",
-  parse: (value) => {
-    if (!Array.isArray(value)) {
-      throw new Error('"message_annotations" parts expect an array value.');
-    }
-    return { type: "message_annotations", value };
-  }
-};
-var toolCallStreamPart = {
-  code: "9",
-  name: "tool_call",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("toolCallId" in value) || typeof value.toolCallId !== "string" || !("toolName" in value) || typeof value.toolName !== "string" || !("args" in value) || typeof value.args !== "object") {
-      throw new Error(
-        '"tool_call" parts expect an object with a "toolCallId", "toolName", and "args" property.'
-      );
-    }
-    return {
-      type: "tool_call",
-      value
-    };
-  }
-};
-var toolResultStreamPart = {
-  code: "a",
-  name: "tool_result",
-  parse: (value) => {
-    if (value == null || typeof value !== "object" || !("toolCallId" in value) || typeof value.toolCallId !== "string" || !("toolName" in value) || typeof value.toolName !== "string" || !("args" in value) || typeof value.args !== "object" || !("result" in value)) {
-      throw new Error(
-        '"tool_result" parts expect an object with a "toolCallId", "toolName", "args", and "result" property.'
-      );
-    }
-    return {
-      type: "tool_result",
-      value
-    };
-  }
-};
-var streamParts = [
-  textStreamPart,
-  functionCallStreamPart,
-  dataStreamPart,
-  errorStreamPart,
-  assistantMessageStreamPart,
-  assistantControlDataStreamPart,
-  dataMessageStreamPart,
-  toolCallsStreamPart,
-  messageAnnotationsStreamPart,
-  toolCallStreamPart,
-  toolResultStreamPart
-];
-var streamPartsByCode = {
-  [textStreamPart.code]: textStreamPart,
-  [functionCallStreamPart.code]: functionCallStreamPart,
-  [dataStreamPart.code]: dataStreamPart,
-  [errorStreamPart.code]: errorStreamPart,
-  [assistantMessageStreamPart.code]: assistantMessageStreamPart,
-  [assistantControlDataStreamPart.code]: assistantControlDataStreamPart,
-  [dataMessageStreamPart.code]: dataMessageStreamPart,
-  [toolCallsStreamPart.code]: toolCallsStreamPart,
-  [messageAnnotationsStreamPart.code]: messageAnnotationsStreamPart,
-  [toolCallStreamPart.code]: toolCallStreamPart,
-  [toolResultStreamPart.code]: toolResultStreamPart
-};
-var StreamStringPrefixes = {
-  [textStreamPart.name]: textStreamPart.code,
-  [functionCallStreamPart.name]: functionCallStreamPart.code,
-  [dataStreamPart.name]: dataStreamPart.code,
-  [errorStreamPart.name]: errorStreamPart.code,
-  [assistantMessageStreamPart.name]: assistantMessageStreamPart.code,
-  [assistantControlDataStreamPart.name]: assistantControlDataStreamPart.code,
-  [dataMessageStreamPart.name]: dataMessageStreamPart.code,
-  [toolCallsStreamPart.name]: toolCallsStreamPart.code,
-  [messageAnnotationsStreamPart.name]: messageAnnotationsStreamPart.code,
-  [toolCallStreamPart.name]: toolCallStreamPart.code,
-  [toolResultStreamPart.name]: toolResultStreamPart.code
-};
-var validCodes = streamParts.map((part) => part.code);
-var parseStreamPart = (line) => {
-  const firstSeparatorIndex = line.indexOf(":");
-  if (firstSeparatorIndex === -1) {
-    throw new Error("Failed to parse stream string. No separator found.");
-  }
-  const prefix = line.slice(0, firstSeparatorIndex);
-  if (!validCodes.includes(prefix)) {
-    throw new Error(`Failed to parse stream string. Invalid code ${prefix}.`);
-  }
-  const code = prefix;
-  const textValue = line.slice(firstSeparatorIndex + 1);
-  const jsonValue = JSON.parse(textValue);
-  return streamPartsByCode[code].parse(jsonValue);
-};
-function formatStreamPart(type, value) {
-  const streamPart = streamParts.find((part) => part.name === type);
-  if (!streamPart) {
-    throw new Error(`Invalid stream part type: ${type}`);
-  }
-  return `${streamPart.code}:${JSON.stringify(value)}
-`;
-}
-
-// shared/read-data-stream.ts
-var NEWLINE = "\n".charCodeAt(0);
-function concatChunks(chunks, totalLength) {
-  const concatenatedChunks = new Uint8Array(totalLength);
-  let offset = 0;
-  for (const chunk of chunks) {
-    concatenatedChunks.set(chunk, offset);
-    offset += chunk.length;
-  }
-  chunks.length = 0;
-  return concatenatedChunks;
-}
-async function* readDataStream(reader, {
-  isAborted
-} = {}) {
-  const decoder = new TextDecoder();
-  const chunks = [];
-  let totalLength = 0;
-  while (true) {
-    const { value } = await reader.read();
-    if (value) {
-      chunks.push(value);
-      totalLength += value.length;
-      if (value[value.length - 1] !== NEWLINE) {
-        continue;
-      }
-    }
-    if (chunks.length === 0) {
-      break;
-    }
-    const concatenatedChunks = concatChunks(chunks, totalLength);
-    totalLength = 0;
-    const streamParts2 = decoder.decode(concatenatedChunks, { stream: true }).split("\n").filter((line) => line !== "").map(parseStreamPart);
-    for (const streamPart of streamParts2) {
-      yield streamPart;
-    }
-    if (isAborted == null ? void 0 : isAborted()) {
-      reader.cancel();
-      break;
-    }
-  }
-}
-
-// shared/utils.ts
-function createChunkDecoder(complex) {
-  const decoder = new TextDecoder();
-  if (!complex) {
-    return function(chunk) {
-      if (!chunk)
-        return "";
-      return decoder.decode(chunk, { stream: true });
-    };
-  }
-  return function(chunk) {
-    const decoded = decoder.decode(chunk, { stream: true }).split("\n").filter((line) => line !== "");
-    return decoded.map(parseStreamPart).filter(Boolean);
-  };
-}
-var isStreamStringEqualToType = (type, value) => value.startsWith(`${StreamStringPrefixes[type]}:`) && value.endsWith("\n");
-
 // streams/ai-stream.ts
 import {
   createParser
@@ -2399,6 +2124,7 @@ function readableFromAsyncIterable(iterable) {
 }
 
 // streams/stream-data.ts
+import { formatStreamPart as formatStreamPart2 } from "@ai-sdk/ui-utils";
 var StreamData = class {
   constructor() {
     this.encoder = new TextEncoder();
@@ -2445,7 +2171,7 @@ var StreamData = class {
       throw new Error("Stream controller is not initialized.");
     }
     this.controller.enqueue(
-      this.encoder.encode(formatStreamPart("data", [value]))
+      this.encoder.encode(formatStreamPart2("data", [value]))
     );
   }
   appendMessageAnnotation(value) {
@@ -2456,7 +2182,7 @@ var StreamData = class {
       throw new Error("Stream controller is not initialized.");
     }
     this.controller.enqueue(
-      this.encoder.encode(formatStreamPart("message_annotations", [value]))
+      this.encoder.encode(formatStreamPart2("message_annotations", [value]))
     );
   }
 };
@@ -2466,7 +2192,7 @@ function createStreamDataTransformer() {
   return new TransformStream({
     transform: async (chunk, controller) => {
       const message = decoder.decode(chunk);
-      controller.enqueue(encoder.encode(formatStreamPart("text", message)));
+      controller.enqueue(encoder.encode(formatStreamPart2("text", message)));
     }
   });
 }
@@ -2520,6 +2246,9 @@ function AnthropicStream(res, cb) {
 }
 
 // streams/assistant-response.ts
+import {
+  formatStreamPart as formatStreamPart3
+} from "@ai-sdk/ui-utils";
 function AssistantResponse({ threadId, messageId }, process2) {
   const stream = new ReadableStream({
     async start(controller) {
@@ -2527,17 +2256,17 @@ function AssistantResponse({ threadId, messageId }, process2) {
       const textEncoder = new TextEncoder();
       const sendMessage = (message) => {
         controller.enqueue(
-          textEncoder.encode(formatStreamPart("assistant_message", message))
+          textEncoder.encode(formatStreamPart3("assistant_message", message))
         );
       };
       const sendDataMessage = (message) => {
         controller.enqueue(
-          textEncoder.encode(formatStreamPart("data_message", message))
+          textEncoder.encode(formatStreamPart3("data_message", message))
         );
       };
       const sendError = (errorMessage) => {
         controller.enqueue(
-          textEncoder.encode(formatStreamPart("error", errorMessage))
+          textEncoder.encode(formatStreamPart3("error", errorMessage))
         );
       };
       const forwardStream = async (stream2) => {
@@ -2548,7 +2277,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
             case "thread.message.created": {
               controller.enqueue(
                 textEncoder.encode(
-                  formatStreamPart("assistant_message", {
+                  formatStreamPart3("assistant_message", {
                     id: value.data.id,
                     role: "assistant",
                     content: [{ type: "text", text: { value: "" } }]
@@ -2562,7 +2291,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
               if ((content == null ? void 0 : content.type) === "text" && ((_b = content.text) == null ? void 0 : _b.value) != null) {
                 controller.enqueue(
                   textEncoder.encode(
-                    formatStreamPart("text", content.text.value)
+                    formatStreamPart3("text", content.text.value)
                   )
                 );
               }
@@ -2579,7 +2308,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
       };
       controller.enqueue(
         textEncoder.encode(
-          formatStreamPart("assistant_control_data", {
+          formatStreamPart3("assistant_control_data", {
             threadId,
             messageId
           })
@@ -2894,6 +2623,10 @@ function MistralStream(response, callbacks) {
 }
 
 // streams/openai-stream.ts
+import {
+  createChunkDecoder,
+  formatStreamPart as formatStreamPart4
+} from "@ai-sdk/ui-utils";
 function parseOpenAIStream() {
   const extract = chunkToText();
   return (data) => extract(JSON.parse(data));
@@ -3058,7 +2791,7 @@ function createFunctionCallTransformer(callbacks) {
       }
       if (!isFunctionStreamingIn) {
         controller.enqueue(
-          textEncoder.encode(formatStreamPart("text", message))
+          textEncoder.encode(formatStreamPart4("text", message))
         );
         return;
       } else {
@@ -3169,7 +2902,7 @@ function createFunctionCallTransformer(callbacks) {
           if (!functionResponse) {
             controller.enqueue(
               textEncoder.encode(
-                formatStreamPart(
+                formatStreamPart4(
                   payload.function_call ? "function_call" : "tool_calls",
                   // parse to prevent double-encoding:
                   JSON.parse(aggregatedResponse)
@@ -3179,7 +2912,7 @@ function createFunctionCallTransformer(callbacks) {
             return;
           } else if (typeof functionResponse === "string") {
             controller.enqueue(
-              textEncoder.encode(formatStreamPart("text", functionResponse))
+              textEncoder.encode(formatStreamPart4("text", functionResponse))
             );
             aggregatedFinalCompletionResponse = functionResponse;
             return;
@@ -3346,200 +3079,6 @@ function streamToResponse(res, response, init, data) {
   read();
 }
 
-// shared/parse-complex-response.ts
-function assignAnnotationsToMessage(message, annotations) {
-  if (!message || !annotations || !annotations.length)
-    return message;
-  return { ...message, annotations: [...annotations] };
-}
-async function parseComplexResponse({
-  reader,
-  abortControllerRef,
-  update,
-  onToolCall,
-  onFinish,
-  generateId: generateId2 = generateId,
-  getCurrentDate = () => /* @__PURE__ */ new Date()
-}) {
-  const createdAt = getCurrentDate();
-  const prefixMap = {
-    data: []
-  };
-  let message_annotations = void 0;
-  for await (const { type, value } of readDataStream(reader, {
-    isAborted: () => (abortControllerRef == null ? void 0 : abortControllerRef.current) === null
-  })) {
-    if (type === "text") {
-      if (prefixMap["text"]) {
-        prefixMap["text"] = {
-          ...prefixMap["text"],
-          content: (prefixMap["text"].content || "") + value
-        };
-      } else {
-        prefixMap["text"] = {
-          id: generateId2(),
-          role: "assistant",
-          content: value,
-          createdAt
-        };
-      }
-    }
-    if (type === "tool_call") {
-      if (prefixMap.text == null) {
-        prefixMap.text = {
-          id: generateId2(),
-          role: "assistant",
-          content: "",
-          createdAt
-        };
-      }
-      if (prefixMap.text.toolInvocations == null) {
-        prefixMap.text.toolInvocations = [];
-      }
-      prefixMap.text.toolInvocations.push(value);
-      if (onToolCall) {
-        const result = await onToolCall({ toolCall: value });
-        if (result != null) {
-          prefixMap.text.toolInvocations[prefixMap.text.toolInvocations.length - 1] = { ...value, result };
-        }
-      }
-    } else if (type === "tool_result") {
-      if (prefixMap.text == null) {
-        prefixMap.text = {
-          id: generateId2(),
-          role: "assistant",
-          content: "",
-          createdAt
-        };
-      }
-      if (prefixMap.text.toolInvocations == null) {
-        prefixMap.text.toolInvocations = [];
-      }
-      const toolInvocationIndex = prefixMap.text.toolInvocations.findIndex(
-        (invocation) => invocation.toolCallId === value.toolCallId
-      );
-      if (toolInvocationIndex !== -1) {
-        prefixMap.text.toolInvocations[toolInvocationIndex] = value;
-      } else {
-        prefixMap.text.toolInvocations.push(value);
-      }
-    }
-    let functionCallMessage = null;
-    if (type === "function_call") {
-      prefixMap["function_call"] = {
-        id: generateId2(),
-        role: "assistant",
-        content: "",
-        function_call: value.function_call,
-        name: value.function_call.name,
-        createdAt
-      };
-      functionCallMessage = prefixMap["function_call"];
-    }
-    let toolCallMessage = null;
-    if (type === "tool_calls") {
-      prefixMap["tool_calls"] = {
-        id: generateId2(),
-        role: "assistant",
-        content: "",
-        tool_calls: value.tool_calls,
-        createdAt
-      };
-      toolCallMessage = prefixMap["tool_calls"];
-    }
-    if (type === "data") {
-      prefixMap["data"].push(...value);
-    }
-    let responseMessage = prefixMap["text"];
-    if (type === "message_annotations") {
-      if (!message_annotations) {
-        message_annotations = [...value];
-      } else {
-        message_annotations.push(...value);
-      }
-      functionCallMessage = assignAnnotationsToMessage(
-        prefixMap["function_call"],
-        message_annotations
-      );
-      toolCallMessage = assignAnnotationsToMessage(
-        prefixMap["tool_calls"],
-        message_annotations
-      );
-      responseMessage = assignAnnotationsToMessage(
-        prefixMap["text"],
-        message_annotations
-      );
-    }
-    if (message_annotations == null ? void 0 : message_annotations.length) {
-      const messagePrefixKeys = [
-        "text",
-        "function_call",
-        "tool_calls"
-      ];
-      messagePrefixKeys.forEach((key) => {
-        if (prefixMap[key]) {
-          prefixMap[key].annotations = [...message_annotations];
-        }
-      });
-    }
-    const merged = [functionCallMessage, toolCallMessage, responseMessage].filter(Boolean).map((message) => ({
-      ...assignAnnotationsToMessage(message, message_annotations)
-    }));
-    update(merged, [...prefixMap["data"]]);
-  }
-  onFinish == null ? void 0 : onFinish(prefixMap);
-  return {
-    messages: [
-      prefixMap.text,
-      prefixMap.function_call,
-      prefixMap.tool_calls
-    ].filter(Boolean),
-    data: prefixMap.data
-  };
-}
-
-// streams/streaming-react-response.ts
-var experimental_StreamingReactResponse = class {
-  constructor(res, options) {
-    var _a, _b;
-    let resolveFunc = () => {
-    };
-    let next = new Promise((resolve) => {
-      resolveFunc = resolve;
-    });
-    const processedStream = (options == null ? void 0 : options.data) != null ? mergeStreams((_a = options == null ? void 0 : options.data) == null ? void 0 : _a.stream, res) : res;
-    let lastPayload = void 0;
-    parseComplexResponse({
-      reader: processedStream.getReader(),
-      update: (merged, data) => {
-        var _a2, _b2, _c;
-        const content = (_b2 = (_a2 = merged[0]) == null ? void 0 : _a2.content) != null ? _b2 : "";
-        const ui = ((_c = options == null ? void 0 : options.ui) == null ? void 0 : _c.call(options, { content, data })) || content;
-        const payload = { ui, content };
-        const resolvePrevious = resolveFunc;
-        const nextRow = new Promise((resolve) => {
-          resolveFunc = resolve;
-        });
-        resolvePrevious({
-          next: nextRow,
-          ...payload
-        });
-        lastPayload = payload;
-      },
-      generateId: (_b = options == null ? void 0 : options.generateId) != null ? _b : generateId,
-      onFinish: () => {
-        if (lastPayload !== void 0) {
-          resolveFunc({
-            next: null,
-            ...lastPayload
-          });
-        }
-      }
-    });
-    return next;
-  }
-};
-
 // streams/streaming-text-response.ts
 var StreamingTextResponse = class extends Response {
   constructor(res, init, data) {
@@ -3602,24 +3141,22 @@ export {
   convertDataContentToUint8Array,
   convertToCoreMessages,
   createCallbacksTransformer,
-  createChunkDecoder,
   createEventStreamTransformer,
   createStreamDataTransformer,
   embed,
   embedMany,
   experimental_AssistantResponse,
   experimental_StreamData,
-  experimental_StreamingReactResponse,
   experimental_generateObject,
   experimental_generateText,
   experimental_streamObject,
   experimental_streamText,
   formatStreamPart,
-  generateId,
+  generateId2 as generateId,
   generateObject,
   generateText,
-  isStreamStringEqualToType,
-  generateId as nanoid,
+  generateId3 as nanoid,
+  parseComplexResponse,
   parseStreamPart,
   readDataStream,
   readableFromAsyncIterable,