ai 3.1.30 → 3.1.32

package/rsc/dist/rsc-server.mjs

@@ -3,8 +3,6 @@ import { AsyncLocalStorage } from "async_hooks";
 import * as jsondiffpatch from "jsondiffpatch";
 
 // rsc/utils.tsx
-import { Suspense } from "react";
-import { Fragment, jsx, jsxs } from "react/jsx-runtime";
 function createResolvablePromise() {
   let resolve, reject;
   const promise = new Promise((res, rej) => {
@@ -17,34 +15,6 @@ function createResolvablePromise() {
     reject
   };
 }
-var R = [
-  async ({
-    c,
-    // current
-    n
-    // next
-  }) => {
-    const chunk = await n;
-    if (chunk.done) {
-      return chunk.value;
-    }
-    if (chunk.append) {
-      return /* @__PURE__ */ jsxs(Fragment, { children: [
-        c,
-        /* @__PURE__ */ jsx(Suspense, { fallback: chunk.value, children: /* @__PURE__ */ jsx(R, { c: chunk.value, n: chunk.next }) })
-      ] });
-    }
-    return /* @__PURE__ */ jsx(Suspense, { fallback: chunk.value, children: /* @__PURE__ */ jsx(R, { c: chunk.value, n: chunk.next }) });
-  }
-][0];
-function createSuspensedChunk(initialValue) {
-  const { promise, resolve, reject } = createResolvablePromise();
-  return {
-    row: /* @__PURE__ */ jsx(Suspense, { fallback: initialValue, children: /* @__PURE__ */ jsx(R, { c: initialValue, n: promise }) }),
-    resolve,
-    reject
-  };
-}
 var isFunction = (x) => typeof x === "function";
 var consumeStream = async (stream) => {
   const reader = stream.getReader();
@@ -175,6 +145,7 @@ function getMutableAIState(...args) {
 }
 
 // rsc/streamable.tsx
+import { isValidElement } from "react";
 import zodToJsonSchema2 from "zod-to-json-schema";
 
 // core/util/retry-with-exponential-backoff.ts
@@ -537,238 +508,6 @@ function prepareToolsAndToolChoice({
   };
 }
 
-// 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/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);
-  };
-}
-
 // streams/ai-stream.ts
 import {
   createParser
@@ -894,6 +633,7 @@ function readableFromAsyncIterable(iterable) {
 }
 
 // streams/stream-data.ts
+import { formatStreamPart } from "@ai-sdk/ui-utils";
 function createStreamDataTransformer() {
   const encoder = new TextEncoder();
   const decoder = new TextDecoder();
@@ -906,6 +646,10 @@ function createStreamDataTransformer() {
 }
 
 // streams/openai-stream.ts
+import {
+  createChunkDecoder,
+  formatStreamPart as formatStreamPart2
+} from "@ai-sdk/ui-utils";
 function parseOpenAIStream() {
   const extract = chunkToText();
   return (data) => extract(JSON.parse(data));
@@ -1070,7 +814,7 @@ function createFunctionCallTransformer(callbacks) {
       }
       if (!isFunctionStreamingIn) {
         controller.enqueue(
-          textEncoder.encode(formatStreamPart("text", message))
+          textEncoder.encode(formatStreamPart2("text", message))
         );
         return;
       } else {
@@ -1181,7 +925,7 @@ function createFunctionCallTransformer(callbacks) {
           if (!functionResponse) {
             controller.enqueue(
               textEncoder.encode(
-                formatStreamPart(
+                formatStreamPart2(
                   payload.function_call ? "function_call" : "tool_calls",
                   // parse to prevent double-encoding:
                   JSON.parse(aggregatedResponse)
@@ -1191,7 +935,7 @@ function createFunctionCallTransformer(callbacks) {
             return;
           } else if (typeof functionResponse === "string") {
             controller.enqueue(
-              textEncoder.encode(formatStreamPart("text", functionResponse))
+              textEncoder.encode(formatStreamPart2("text", functionResponse))
             );
             aggregatedFinalCompletionResponse = functionResponse;
             return;
@@ -1228,10 +972,11 @@ var STREAMABLE_VALUE_TYPE = Symbol.for("ui.streamable.value");
 var DEV_DEFAULT_STREAMABLE_WARNING_TIME = 15 * 1e3;
 
 // rsc/streamable.tsx
+import { InternalStreamableUIClient } from "./rsc-shared.mjs";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
 function createStreamableUI(initialValue) {
-  let currentValue = initialValue;
+  const innerStreamable = createStreamableValue(initialValue);
   let closed = false;
-  let { row, resolve, reject } = createSuspensedChunk(initialValue);
   function assertStream(method) {
     if (closed) {
       throw new Error(method + ": UI stream is already closed.");
@@ -1252,28 +997,16 @@ function createStreamableUI(initialValue) {
   }
   warnUnclosedStream();
   const streamable2 = {
-    value: row,
+    value: /* @__PURE__ */ jsx(InternalStreamableUIClient, { s: innerStreamable.value }),
     update(value) {
       assertStream(".update()");
-      if (value === currentValue) {
-        warnUnclosedStream();
-        return streamable2;
-      }
-      const resolvable = createResolvablePromise();
-      currentValue = value;
-      resolve({ value: currentValue, done: false, next: resolvable.promise });
-      resolve = resolvable.resolve;
-      reject = resolvable.reject;
+      innerStreamable.update(value);
       warnUnclosedStream();
       return streamable2;
     },
     append(value) {
       assertStream(".append()");
-      const resolvable = createResolvablePromise();
-      currentValue = value;
-      resolve({ value, done: false, append: true, next: resolvable.promise });
-      resolve = resolvable.resolve;
-      reject = resolvable.reject;
+      innerStreamable.append(value);
       warnUnclosedStream();
       return streamable2;
     },
@@ -1283,7 +1016,7 @@ function createStreamableUI(initialValue) {
         clearTimeout(warningTimeout);
       }
       closed = true;
-      reject(error);
+      innerStreamable.error(error);
       return streamable2;
     },
     done(...args) {
@@ -1293,10 +1026,10 @@ function createStreamableUI(initialValue) {
       }
       closed = true;
       if (args.length) {
-        resolve({ value: args[0], done: true });
+        innerStreamable.done(args[0]);
         return streamable2;
       }
-      resolve({ value: currentValue, done: true });
+      innerStreamable.done();
       return streamable2;
     }
   };
@@ -1416,25 +1149,38 @@ function createStreamableValueImpl(initialValue) {
     },
     append(value) {
       assertStream(".append()");
-      if (typeof currentValue !== "string" && typeof currentValue !== "undefined") {
+      if (typeof value !== "string" && !isValidElement(value)) {
         throw new Error(
-          `.append(): The current value is not a string. Received: ${typeof currentValue}`
+          `.append(): The value type can't be appended to the stream. Received: ${typeof value}`
         );
       }
-      if (typeof value !== "string") {
+      if (typeof currentValue === "undefined") {
+        currentPatchValue = void 0;
+        currentValue = value;
+      } else if (typeof currentValue === "string") {
+        if (typeof value === "string") {
+          currentPatchValue = [0, value];
+          currentValue = currentValue + value;
+        } else {
+          currentPatchValue = [1, value];
+          currentValue = /* @__PURE__ */ jsxs(Fragment, { children: [
+            currentValue,
+            value
+          ] });
+        }
+      } else if (isValidElement(currentValue)) {
+        currentPatchValue = [1, value];
+        currentValue = /* @__PURE__ */ jsxs(Fragment, { children: [
+          currentValue,
+          value
+        ] });
+      } else {
         throw new Error(
-          `.append(): The value is not a string. Received: ${typeof value}`
+          `.append(): The current value doesn't support appending data. Type: ${typeof currentValue}`
         );
       }
       const resolvePrevious = resolvable.resolve;
       resolvable = createResolvablePromise();
-      if (typeof currentValue === "string") {
-        currentPatchValue = [0, value];
-        currentValue = currentValue + value;
-      } else {
-        currentPatchValue = void 0;
-        currentValue = value;
-      }
       currentPromise = resolvable.promise;
       resolvePrevious(createWrapped());
       warnUnclosedStream();
@@ -1800,7 +1546,7 @@ async function streamUI({
 }
 
 // rsc/provider.tsx
-import * as React2 from "react";
+import * as React from "react";
 import { InternalAIProvider } from "./rsc-shared.mjs";
 import { jsx as jsx2 } from "react/jsx-runtime";
 async function innerAction({
@@ -1839,7 +1585,7 @@ function createAI({
   const wrappedSyncUIState = onGetUIState ? wrapAction(onGetUIState, {}) : void 0;
   const AI = async (props) => {
     var _a, _b;
-    if ("useState" in React2) {
+    if ("useState" in React) {
       throw new Error(
         "This component can only be used inside Server Components."
       );