@assistant-ui/react 0.4.0 → 0.4.2

package/dist/index.js

@@ -907,21 +907,30 @@ var ActionBarPrimitiveRoot = (0, import_react25.forwardRef)(({ hideWhenRunning,
 ActionBarPrimitiveRoot.displayName = "ActionBarPrimitive.Root";
 
 // src/utils/createActionButton.tsx
-var import_primitive = require("@radix-ui/primitive");
-var import_react_primitive2 = require("@radix-ui/react-primitive");
 var import_react26 = require("react");
+var import_react_primitive2 = require("@radix-ui/react-primitive");
+var import_primitive = require("@radix-ui/primitive");
 var import_jsx_runtime5 = require("react/jsx-runtime");
-var createActionButton = (displayName, useActionButton) => {
+var createActionButton = (displayName, useActionButton, forwardProps = []) => {
   const ActionButton = (0, import_react26.forwardRef)((props, forwardedRef) => {
-    const callback = useActionButton(props);
+    const forwardedProps = {};
+    const primitiveProps = {};
+    Object.keys(props).forEach((key) => {
+      if (forwardProps.includes(key)) {
+        forwardedProps[key] = props[key];
+      } else {
+        primitiveProps[key] = props[key];
+      }
+    });
+    const callback = useActionButton(forwardedProps);
     return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(
       import_react_primitive2.Primitive.button,
       {
         type: "button",
         disabled: !callback,
-        ...props,
+        ...primitiveProps,
         ref: forwardedRef,
-        onClick: (0, import_primitive.composeEventHandlers)(props.onClick, () => {
+        onClick: (0, import_primitive.composeEventHandlers)(primitiveProps.onClick, () => {
           callback?.();
         })
       }
@@ -934,7 +943,8 @@ var createActionButton = (displayName, useActionButton) => {
 // src/primitives/actionBar/ActionBarCopy.tsx
 var ActionBarPrimitiveCopy = createActionButton(
   "ActionBarPrimitive.Copy",
-  useActionBarCopy
+  useActionBarCopy,
+  ["copiedDuration"]
 );
 
 // src/primitives/actionBar/ActionBarReload.tsx
@@ -1244,7 +1254,8 @@ var import_react36 = require("react");
 // src/utils/hooks/useSmooth.tsx
 var import_react35 = require("react");
 var TextStreamAnimator = class {
-  constructor(setText) {
+  constructor(currentText, setText) {
+    this.currentText = currentText;
     this.setText = setText;
   }
   animationFrameId = null;
@@ -1265,33 +1276,31 @@ var TextStreamAnimator = class {
     const currentTime = Date.now();
     const deltaTime = currentTime - this.lastUpdateTime;
     let timeToConsume = deltaTime;
-    this.setText((currentText) => {
-      const targetText = this.targetText;
-      if (currentText === targetText) {
-        this.animationFrameId = null;
-        return currentText;
-      }
-      const remainingChars = targetText.length - currentText.length;
-      const baseTimePerChar = Math.min(5, 250 / remainingChars);
-      let charsToAdd = 0;
-      while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
-        charsToAdd++;
-        timeToConsume -= baseTimePerChar;
-      }
+    const remainingChars = this.targetText.length - this.currentText.length;
+    const baseTimePerChar = Math.min(5, 250 / remainingChars);
+    let charsToAdd = 0;
+    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
+      charsToAdd++;
+      timeToConsume -= baseTimePerChar;
+    }
+    if (charsToAdd !== remainingChars) {
       this.animationFrameId = requestAnimationFrame(this.animate);
-      if (charsToAdd === 0) {
-        return currentText;
-      }
-      const newText = targetText.slice(0, currentText.length + charsToAdd);
-      this.lastUpdateTime = currentTime - timeToConsume;
-      return newText;
-    });
+    } else {
+      this.animationFrameId = null;
+    }
+    if (charsToAdd === 0) return;
+    this.currentText = this.targetText.slice(
+      0,
+      this.currentText.length + charsToAdd
+    );
+    this.lastUpdateTime = currentTime - timeToConsume;
+    this.setText(this.currentText);
   };
 };
 var useSmooth = (text, smooth = false) => {
   const [displayedText, setDisplayedText] = (0, import_react35.useState)(text);
   const [animatorRef] = (0, import_react35.useState)(
-    new TextStreamAnimator(setDisplayedText)
+    new TextStreamAnimator(text, setDisplayedText)
   );
   (0, import_react35.useEffect)(() => {
     if (!smooth) {
@@ -1300,6 +1309,7 @@ var useSmooth = (text, smooth = false) => {
     }
     if (!text.startsWith(animatorRef.targetText)) {
       setDisplayedText(text);
+      animatorRef.currentText = text;
       animatorRef.targetText = text;
       animatorRef.stop();
       return;
@@ -1951,7 +1961,8 @@ var ThreadPrimitiveScrollToBottom = createActionButton(
 // src/primitives/thread/ThreadSuggestion.tsx
 var ThreadPrimitiveSuggestion = createActionButton(
   "ThreadPrimitive.Suggestion",
-  useThreadSuggestion
+  useThreadSuggestion,
+  ["prompt", "autoSend", "method"]
 );
 
 // src/runtimes/local/useLocalRuntime.tsx