ai 3.1.31 → 3.1.33

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "3.1.31",
+  "version": "3.1.33",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -98,8 +98,8 @@
     "tsup": "^7.2.0",
     "typescript": "5.1.3",
     "zod": "3.23.8",
-    "@vercel/ai-tsconfig": "0.0.0",
-    "eslint-config-vercel-ai": "0.0.0"
+    "eslint-config-vercel-ai": "0.0.0",
+    "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {
     "openai": "^4.42.0",
@@ -149,12 +149,13 @@
     "lint": "eslint \"./**/*.ts*\"",
     "type-check": "tsc --noEmit",
     "prettier-check": "prettier --check \"./**/*.ts*\"",
-    "test": "pnpm test:node && pnpm test:edge && pnpm test:ui",
+    "test": "pnpm test:node && pnpm test:edge && pnpm test:ui && pnpm test:e2e",
     "test:edge": "vitest --config vitest.edge.config.js --run",
     "test:node": "vitest --config vitest.node.config.js --run",
     "test:node:core": "pnpm vitest --config vitest.node.config.js --run ./core/",
     "test:ui": "pnpm test:ui:react",
     "test:ui:react": "vitest --config vitest.ui.react.config.js --run",
+    "test:e2e": "playwright test",
     "test:edge:watch": "vitest --config vitest.edge.config.js",
     "test:node:watch": "vitest --config vitest.node.config.js",
     "test:ui:react:watch": "vitest --config vitest.ui.react.config.js"

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
@@ -1001,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.");
@@ -1025,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;
     },
@@ -1056,7 +1016,7 @@ function createStreamableUI(initialValue) {
         clearTimeout(warningTimeout);
       }
       closed = true;
-      reject(error);
+      innerStreamable.error(error);
       return streamable2;
     },
     done(...args) {
@@ -1066,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;
     }
   };
@@ -1189,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();
@@ -1573,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({
@@ -1612,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."
       );