ai 3.0.4 → 3.0.6

package/rsc/dist/{rsc-client.d.ts → rsc-client.d.mts}

@@ -1 +1 @@
-export { useAIState, useActions, useStreamableValue, useSyncUIState, useUIState } from './rsc-shared';
+export { useAIState, useActions, useStreamableValue, useSyncUIState, useUIState } from './rsc-shared.mjs';

package/rsc/dist/rsc-client.mjs

@@ -5,7 +5,7 @@ import {
   useAIState,
   useActions,
   useSyncUIState
-} from "./rsc-shared";
+} from "./rsc-shared.mjs";
 export {
   useAIState,
   useActions,

package/rsc/dist/rsc-client.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../rsc-client.ts"],"sourcesContent":["export {\n  useStreamableValue,\n  useUIState,\n  useAIState,\n  useActions,\n  useSyncUIState,\n} from './rsc-shared';\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;","names":[]}
+{"version":3,"sources":["../rsc-client.ts"],"sourcesContent":["export {\n  useStreamableValue,\n  useUIState,\n  useAIState,\n  useActions,\n  useSyncUIState,\n} from './rsc-shared.mjs';\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;","names":[]}

package/rsc/dist/{rsc-server.d.ts → rsc-server.d.mts}

@@ -2,7 +2,6 @@ import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ReactNode } from 'react';
 import OpenAI from 'openai';
 import { z } from 'zod';
-import './rsc-shared';
 
 type AIAction<T = any, R = any> = (...args: T[]) => Promise<R>;
 type AIActions<T = any, R = any> = Record<string, AIAction<T, R>>;

package/rsc/dist/rsc-server.mjs

@@ -745,7 +745,7 @@ function createFunctionCallTransformer(callbacks) {
                 type: "function",
                 func: {
                   name: tool.function.name,
-                  arguments: tool.function.arguments
+                  arguments: JSON.parse(tool.function.arguments)
                 }
               });
             }
@@ -842,6 +842,7 @@ function createFunctionCallTransformer(callbacks) {
 
 // rsc/constants.ts
 var STREAMABLE_VALUE_TYPE = Symbol.for("ui.streamable.value");
+var DEV_DEFAULT_STREAMABLE_WARNING_TIME = 15 * 1e3;
 
 // rsc/streamable.tsx
 import { Fragment as Fragment2, jsxs as jsxs2 } from "react/jsx-runtime";
@@ -849,27 +850,39 @@ function createStreamableUI(initialValue) {
   let currentValue = initialValue;
   let closed = false;
   let { row, resolve, reject } = createSuspensedChunk(initialValue);
-  function assertStream() {
+  function assertStream(method) {
     if (closed) {
-      throw new Error("UI stream is already closed.");
+      throw new Error(method + ": UI stream is already closed.");
     }
   }
+  let warningTimeout;
+  function warnUnclosedStream() {
+    if (process.env.NODE_ENV === "development") {
+      if (warningTimeout) {
+        clearTimeout(warningTimeout);
+      }
+      warningTimeout = setTimeout(() => {
+        console.warn(
+          "The streamable UI has been slow to update. This may be a bug or a performance issue or you forgot to call `.done()`."
+        );
+      }, DEV_DEFAULT_STREAMABLE_WARNING_TIME);
+    }
+  }
+  warnUnclosedStream();
   return {
     value: row,
     update(value) {
-      assertStream();
+      assertStream(".update()");
       const resolvable = createResolvablePromise();
-      resolve({ value, done: false, next: resolvable.promise });
+      currentValue = value;
+      resolve({ value: currentValue, done: false, next: resolvable.promise });
       resolve = resolvable.resolve;
       reject = resolvable.reject;
-      currentValue = value;
+      warnUnclosedStream();
     },
     append(value) {
-      assertStream();
+      assertStream(".append()");
       const resolvable = createResolvablePromise();
-      resolve({ value, done: false, next: resolvable.promise });
-      resolve = resolvable.resolve;
-      reject = resolvable.reject;
       if (typeof currentValue === "string" && typeof value === "string") {
         currentValue += value;
       } else {
@@ -878,14 +891,24 @@ function createStreamableUI(initialValue) {
           value
         ] });
       }
+      resolve({ value: currentValue, done: false, next: resolvable.promise });
+      resolve = resolvable.resolve;
+      reject = resolvable.reject;
+      warnUnclosedStream();
     },
     error(error) {
-      assertStream();
+      assertStream(".error()");
+      if (warningTimeout) {
+        clearTimeout(warningTimeout);
+      }
       closed = true;
       reject(error);
     },
     done(...args) {
-      assertStream();
+      assertStream(".done()");
+      if (warningTimeout) {
+        clearTimeout(warningTimeout);
+      }
       closed = true;
       if (args.length) {
         resolve({ value: args[0], done: true });
@@ -898,11 +921,25 @@ function createStreamableUI(initialValue) {
 function createStreamableValue(initialValue) {
   let closed = false;
   let { promise, resolve, reject } = createResolvablePromise();
-  function assertStream() {
+  function assertStream(method) {
     if (closed) {
-      throw new Error("Value stream is already closed.");
+      throw new Error(method + ": Value stream is already closed.");
     }
   }
+  let warningTimeout;
+  function warnUnclosedStream() {
+    if (process.env.NODE_ENV === "development") {
+      if (warningTimeout) {
+        clearTimeout(warningTimeout);
+      }
+      warningTimeout = setTimeout(() => {
+        console.warn(
+          "The streamable UI has been slow to update. This may be a bug or a performance issue or you forgot to call `.done()`."
+        );
+      }, DEV_DEFAULT_STREAMABLE_WARNING_TIME);
+    }
+  }
+  warnUnclosedStream();
   function createWrapped(val, initial) {
     if (initial) {
       return {
@@ -919,21 +956,28 @@ function createStreamableValue(initialValue) {
   return {
     value: createWrapped(initialValue, true),
     update(value) {
-      assertStream();
+      assertStream(".update()");
       const resolvePrevious = resolve;
       const resolvable = createResolvablePromise();
       promise = resolvable.promise;
       resolve = resolvable.resolve;
       reject = resolvable.reject;
       resolvePrevious(createWrapped(value));
+      warnUnclosedStream();
     },
     error(error) {
-      assertStream();
+      assertStream(".error()");
+      if (warningTimeout) {
+        clearTimeout(warningTimeout);
+      }
       closed = true;
       reject(error);
     },
     done(...args) {
-      assertStream();
+      assertStream(".done()");
+      if (warningTimeout) {
+        clearTimeout(warningTimeout);
+      }
       closed = true;
       if (args.length) {
         resolve({ curr: args[0] });
@@ -1075,7 +1119,7 @@ function render(options) {
 
 // rsc/provider.tsx
 import * as React2 from "react";
-import { InternalAIProvider } from "./rsc-shared";
+import { InternalAIProvider } from "./rsc-shared.mjs";
 import { jsx as jsx2 } from "react/jsx-runtime";
 async function innerAction({
   action,