@assistant-ui/react 0.3.0 → 0.3.2

package/dist/index.js

@@ -210,11 +210,14 @@ var makeBaseComposer = (set) => ({
 });
 
 // src/context/stores/Composer.ts
-var makeComposerStore = (useThread, useThreadMessages, useThreadActions) => {
+var makeComposerStore = (useThreadMessages, useThreadActions) => {
   const focusListeners = /* @__PURE__ */ new Set();
   return (0, import_zustand3.create)()((set, get, store) => {
     return {
       ...makeBaseComposer(set, get, store),
+      get canCancel() {
+        return useThreadActions.getState().capabilities.cancel;
+      },
       isEditing: true,
       send: () => {
         const { setValue, value } = get();
@@ -226,10 +229,7 @@ var makeComposerStore = (useThread, useThreadMessages, useThreadActions) => {
         });
       },
       cancel: () => {
-        const thread = useThread.getState();
-        if (!thread.isRunning) return false;
         useThreadActions.getState().cancelRun();
-        return true;
       },
       focus: () => {
         for (const listener of focusListeners) {
@@ -279,6 +279,9 @@ var import_zustand6 = require("zustand");
 var makeThreadActionStore = (runtimeRef) => {
   return (0, import_zustand6.create)(
     () => Object.freeze({
+      get capabilities() {
+        return runtimeRef.current.capabilities;
+      },
       getBranches: (messageId) => runtimeRef.current.getBranches(messageId),
       switchToBranch: (branchId) => runtimeRef.current.switchToBranch(branchId),
       startRun: (parentId) => runtimeRef.current.startRun(parentId),
@@ -310,11 +313,7 @@ var ThreadProvider = ({
     const useThreadMessages = makeThreadMessagesStore(runtimeRef);
     const useThreadActions = makeThreadActionStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
-    const useComposer = makeComposerStore(
-      useThread,
-      useThreadMessages,
-      useThreadActions
-    );
+    const useComposer = makeComposerStore(useThreadMessages, useThreadActions);
     return {
       useThread,
       useThreadMessages,
@@ -687,7 +686,7 @@ var useBranchPickerPrevious = () => {
 var import_react21 = require("react");
 var useComposerCancel = () => {
   const { useComposer } = useComposerContext();
-  const disabled = useComposer((c) => !c.isEditing);
+  const disabled = useComposer((c) => !c.canCancel);
   const callback = (0, import_react21.useCallback)(() => {
     const { cancel } = useComposer.getState();
     cancel();
@@ -731,7 +730,7 @@ var useContentPartDisplay = () => {
       throw new Error(
         "This component can only be used inside ui content parts."
       );
-    return c.part.display;
+    return c;
   });
   return display;
 };
@@ -744,7 +743,7 @@ var useContentPartImage = () => {
       throw new Error(
         "ContentPartImage can only be used inside image content parts."
       );
-    return c.part.image;
+    return c;
   });
   return image;
 };
@@ -819,15 +818,18 @@ var useThreadSuggestion = ({
   autoSend
 }) => {
   const { useThread, useComposer } = useThreadContext();
+  const append = useAppendMessage();
   const disabled = useThread((t) => t.isRunning);
   const callback = (0, import_react24.useCallback)(() => {
     const thread = useThread.getState();
     const composer = useComposer.getState();
-    composer.setValue(prompt);
     if (autoSend && !thread.isRunning) {
-      composer.send();
+      append(prompt);
+      composer.setValue("");
+    } else {
+      composer.setValue(prompt);
     }
-  }, [useThread, useComposer, prompt, autoSend]);
+  }, [useThread, useComposer, autoSend, append, prompt]);
   if (disabled) return null;
   return callback;
 };
@@ -1095,7 +1097,7 @@ BranchPickerPrimitiveNumber.displayName = "BranchPickerPrimitive.Number";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
 var import_react_primitive6 = require("@radix-ui/react-primitive");
-var import_react38 = require("react");
+var import_react39 = require("react");
 
 // src/primitives/message/index.ts
 var message_exports = {};
@@ -1172,7 +1174,7 @@ var MessagePrimitiveIf = ({
 MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 
 // src/primitives/message/MessageContent.tsx
-var import_react37 = require("react");
+var import_react38 = require("react");
 
 // src/context/providers/ContentPartProvider.tsx
 var import_react34 = require("react");
@@ -1219,30 +1221,107 @@ var ContentPartProvider = ({
 
 // src/primitives/contentPart/ContentPartText.tsx
 var import_react_primitive4 = require("@radix-ui/react-primitive");
+var import_react36 = require("react");
+
+// src/utils/hooks/useSmooth.tsx
 var import_react35 = require("react");
+var TextStreamAnimator = class {
+  constructor(setText) {
+    this.setText = setText;
+  }
+  animationFrameId = null;
+  lastUpdateTime = Date.now();
+  decayFactor = 0.99;
+  targetText = "";
+  start() {
+    if (this.animationFrameId !== null) return;
+    this.animate();
+  }
+  stop() {
+    if (this.animationFrameId !== null) {
+      cancelAnimationFrame(this.animationFrameId);
+      this.animationFrameId = null;
+    }
+  }
+  animate = () => {
+    const currentTime = Date.now();
+    const deltaTime = currentTime - this.lastUpdateTime;
+    this.lastUpdateTime = currentTime;
+    this.setText((currentText) => {
+      const targetText = this.targetText;
+      if (currentText === targetText) {
+        this.animationFrameId = null;
+        return currentText;
+      }
+      const remainingChars = targetText.length - currentText.length;
+      const charsToAdd = Math.max(
+        1,
+        Math.floor(
+          remainingChars * (1 - Math.pow(this.decayFactor, deltaTime))
+        )
+      );
+      const newText = targetText.slice(0, currentText.length + charsToAdd);
+      this.animationFrameId = requestAnimationFrame(this.animate);
+      return newText;
+    });
+  };
+};
+var useSmooth = (text, smooth = false) => {
+  const [displayedText, setDisplayedText] = (0, import_react35.useState)(text);
+  const [animatorRef] = (0, import_react35.useState)(
+    new TextStreamAnimator(setDisplayedText)
+  );
+  (0, import_react35.useEffect)(() => {
+    if (!smooth) {
+      animatorRef.stop();
+      return;
+    }
+    if (!text.startsWith(animatorRef.targetText)) {
+      setDisplayedText(text);
+      animatorRef.targetText = text;
+      animatorRef.stop();
+      return;
+    }
+    animatorRef.targetText = text;
+    animatorRef.start();
+  }, [animatorRef, smooth, text]);
+  (0, import_react35.useEffect)(() => {
+    return () => {
+      animatorRef.stop();
+    };
+  }, [animatorRef]);
+  return smooth ? displayedText : text;
+};
+
+// src/primitives/contentPart/ContentPartText.tsx
 var import_jsx_runtime14 = require("react/jsx-runtime");
-var ContentPartPrimitiveText = (0, import_react35.forwardRef)((props, forwardedRef) => {
+var ContentPartPrimitiveText = (0, import_react36.forwardRef)(({ smooth, ...rest }, forwardedRef) => {
   const {
-    part: { text },
-    status
+    status,
+    part: { text }
   } = useContentPartText();
-  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(import_react_primitive4.Primitive.span, { "data-status": status, ...props, ref: forwardedRef, children: text });
+  const smoothText = useSmooth(text, smooth);
+  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(import_react_primitive4.Primitive.span, { "data-status": status, ...rest, ref: forwardedRef, children: smoothText });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
 
 // src/primitives/contentPart/ContentPartImage.tsx
 var import_react_primitive5 = require("@radix-ui/react-primitive");
-var import_react36 = require("react");
+var import_react37 = require("react");
 var import_jsx_runtime15 = require("react/jsx-runtime");
-var ContentPartPrimitiveImage = (0, import_react36.forwardRef)((props, forwardedRef) => {
-  const image = useContentPartImage();
+var ContentPartPrimitiveImage = (0, import_react37.forwardRef)((props, forwardedRef) => {
+  const {
+    part: { image }
+  } = useContentPartImage();
   return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(import_react_primitive5.Primitive.img, { src: image, ...props, ref: forwardedRef });
 });
 ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
 
 // src/primitives/contentPart/ContentPartDisplay.tsx
 var ContentPartPrimitiveDisplay = () => {
-  const display = useContentPartDisplay();
+  const {
+    part: { display }
+  } = useContentPartDisplay();
   return display ?? null;
 };
 ContentPartPrimitiveDisplay.displayName = "ContentPartPrimitive.Display";
@@ -1262,7 +1341,7 @@ var import_jsx_runtime16 = require("react/jsx-runtime");
 var defaultComponents = {
   Text: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)("p", { style: { whiteSpace: "pre-line" }, children: [
     /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveText, {}),
-    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveInProgress, { children: " \u25CF" })
+    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ (0, import_jsx_runtime16.jsx)("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
   ] }),
   Image: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveImage, {}),
   UI: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveDisplay, {}),
@@ -1310,7 +1389,7 @@ var MessageContentPartImpl = ({
 }) => {
   return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartProvider, { partIndex, children: /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(MessageContentPartComponent, { components }) });
 };
-var MessageContentPart = (0, import_react37.memo)(
+var MessageContentPart = (0, import_react38.memo)(
   MessageContentPartImpl,
   (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
 );
@@ -1341,7 +1420,7 @@ MessagePrimitiveInProgress.displayName = "MessagePrimitive.InProgress";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
 var import_jsx_runtime17 = require("react/jsx-runtime");
-var BranchPickerPrimitiveRoot = (0, import_react38.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
+var BranchPickerPrimitiveRoot = (0, import_react39.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
   return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(import_react_primitive6.Primitive.div, { ...rest, ref }) });
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";
@@ -1359,9 +1438,9 @@ __export(composer_exports, {
 // src/primitives/composer/ComposerRoot.tsx
 var import_primitive4 = require("@radix-ui/primitive");
 var import_react_primitive7 = require("@radix-ui/react-primitive");
-var import_react39 = require("react");
+var import_react40 = require("react");
 var import_jsx_runtime18 = require("react/jsx-runtime");
-var ComposerPrimitiveRoot = (0, import_react39.forwardRef)(({ onSubmit, ...rest }, forwardedRef) => {
+var ComposerPrimitiveRoot = (0, import_react40.forwardRef)(({ onSubmit, ...rest }, forwardedRef) => {
   const send = useComposerSend();
   const handleSubmit = (e) => {
     e.preventDefault();
@@ -1383,11 +1462,11 @@ ComposerPrimitiveRoot.displayName = "ComposerPrimitive.Root";
 var import_primitive5 = require("@radix-ui/primitive");
 var import_react_compose_refs2 = require("@radix-ui/react-compose-refs");
 var import_react_slot = require("@radix-ui/react-slot");
-var import_react40 = require("react");
+var import_react41 = require("react");
 var import_react_textarea_autosize = __toESM(require("react-textarea-autosize"));
 var import_react_use_escape_keydown = require("@radix-ui/react-use-escape-keydown");
 var import_jsx_runtime19 = require("react/jsx-runtime");
-var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
+var ComposerPrimitiveInput = (0, import_react41.forwardRef)(
   ({ autoFocus = false, asChild, disabled, onChange, onKeyDown, ...rest }, forwardedRef) => {
     const { useThread } = useThreadContext();
     const { useComposer, type } = useComposerContext();
@@ -1396,11 +1475,12 @@ var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
       return c.value;
     });
     const Component = asChild ? import_react_slot.Slot : import_react_textarea_autosize.default;
-    const textareaRef = (0, import_react40.useRef)(null);
+    const textareaRef = (0, import_react41.useRef)(null);
     const ref = (0, import_react_compose_refs2.useComposedRefs)(forwardedRef, textareaRef);
     (0, import_react_use_escape_keydown.useEscapeKeydown)((e) => {
       const composer = useComposer.getState();
-      if (composer.cancel()) {
+      if (composer.canCancel) {
+        composer.cancel();
         e.preventDefault();
       }
     });
@@ -1415,7 +1495,7 @@ var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
       }
     };
     const autoFocusEnabled = autoFocus && !disabled;
-    const focus = (0, import_react40.useCallback)(() => {
+    const focus = (0, import_react41.useCallback)(() => {
       const textarea = textareaRef.current;
       if (!textarea || !autoFocusEnabled) return;
       textarea.focus({ preventScroll: true });
@@ -1424,7 +1504,7 @@ var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
         textareaRef.current.value.length
       );
     }, [autoFocusEnabled]);
-    (0, import_react40.useEffect)(() => focus(), [focus]);
+    (0, import_react41.useEffect)(() => focus(), [focus]);
     useOnComposerFocus(() => {
       if (type === "new") {
         focus();
@@ -1451,10 +1531,10 @@ var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
 ComposerPrimitiveInput.displayName = "ComposerPrimitive.Input";
 
 // src/primitives/composer/ComposerSend.tsx
-var import_react41 = require("react");
+var import_react42 = require("react");
 var import_react_primitive8 = require("@radix-ui/react-primitive");
 var import_jsx_runtime20 = require("react/jsx-runtime");
-var ComposerPrimitiveSend = (0, import_react41.forwardRef)(({ disabled, ...rest }, ref) => {
+var ComposerPrimitiveSend = (0, import_react42.forwardRef)(({ disabled, ...rest }, ref) => {
   const { useComposer } = useComposerContext();
   const hasValue = useComposer((c) => c.isEditing && c.value.length > 0);
   return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
@@ -1508,9 +1588,9 @@ __export(thread_exports, {
 
 // src/primitives/thread/ThreadRoot.tsx
 var import_react_primitive9 = require("@radix-ui/react-primitive");
-var import_react42 = require("react");
+var import_react43 = require("react");
 var import_jsx_runtime21 = require("react/jsx-runtime");
-var ThreadPrimitiveRoot = (0, import_react42.forwardRef)((props, ref) => {
+var ThreadPrimitiveRoot = (0, import_react43.forwardRef)((props, ref) => {
   return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(import_react_primitive9.Primitive.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
@@ -1537,18 +1617,18 @@ ThreadPrimitiveIf.displayName = "ThreadPrimitive.If";
 // src/primitives/thread/ThreadViewport.tsx
 var import_react_compose_refs4 = require("@radix-ui/react-compose-refs");
 var import_react_primitive10 = require("@radix-ui/react-primitive");
-var import_react46 = require("react");
+var import_react47 = require("react");
 
 // src/primitive-hooks/thread/useThreadViewportAutoScroll.tsx
 var import_react_compose_refs3 = require("@radix-ui/react-compose-refs");
-var import_react45 = require("react");
+var import_react46 = require("react");
 
 // src/utils/hooks/useOnResizeContent.tsx
 var import_react_use_callback_ref2 = require("@radix-ui/react-use-callback-ref");
-var import_react43 = require("react");
+var import_react44 = require("react");
 var useOnResizeContent = (callback) => {
   const callbackRef = (0, import_react_use_callback_ref2.useCallbackRef)(callback);
-  const refCallback = (0, import_react43.useCallback)(
+  const refCallback = (0, import_react44.useCallback)(
     (el) => {
       const resizeObserver = new ResizeObserver(() => {
         callbackRef();
@@ -1585,11 +1665,11 @@ var useOnResizeContent = (callback) => {
 
 // src/utils/hooks/useOnScrollToBottom.tsx
 var import_react_use_callback_ref3 = require("@radix-ui/react-use-callback-ref");
-var import_react44 = require("react");
+var import_react45 = require("react");
 var useOnScrollToBottom = (callback) => {
   const callbackRef = (0, import_react_use_callback_ref3.useCallbackRef)(callback);
   const { useViewport } = useThreadContext();
-  (0, import_react44.useEffect)(() => {
+  (0, import_react45.useEffect)(() => {
     return useViewport.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -1600,11 +1680,11 @@ var useOnScrollToBottom = (callback) => {
 var useThreadViewportAutoScroll = ({
   autoScroll = true
 }) => {
-  const divRef = (0, import_react45.useRef)(null);
+  const divRef = (0, import_react46.useRef)(null);
   const { useViewport } = useThreadContext();
-  const firstRenderRef = (0, import_react45.useRef)(true);
-  const lastScrollTop = (0, import_react45.useRef)(0);
-  const isScrollingToBottomRef = (0, import_react45.useRef)(false);
+  const firstRenderRef = (0, import_react46.useRef)(true);
+  const lastScrollTop = (0, import_react46.useRef)(0);
+  const isScrollingToBottomRef = (0, import_react46.useRef)(false);
   const scrollToBottom = () => {
     const div = divRef.current;
     if (!div || !autoScroll) return;
@@ -1651,7 +1731,7 @@ var useThreadViewportAutoScroll = ({
 
 // src/primitives/thread/ThreadViewport.tsx
 var import_jsx_runtime22 = require("react/jsx-runtime");
-var ThreadPrimitiveViewport = (0, import_react46.forwardRef)(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
+var ThreadPrimitiveViewport = (0, import_react47.forwardRef)(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
   const autoScrollRef = useThreadViewportAutoScroll({
     autoScroll
   });
@@ -1661,10 +1741,10 @@ var ThreadPrimitiveViewport = (0, import_react46.forwardRef)(({ autoScroll, onSc
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
 // src/primitives/thread/ThreadMessages.tsx
-var import_react48 = require("react");
+var import_react49 = require("react");
 
 // src/context/providers/MessageProvider.tsx
-var import_react47 = require("react");
+var import_react48 = require("react");
 var import_zustand12 = require("zustand");
 
 // src/context/stores/EditComposer.ts
@@ -1674,20 +1754,19 @@ var makeEditComposerStore = ({
   onSend
 }) => (0, import_zustand10.create)()((set, get, store) => ({
   ...makeBaseComposer(set, get, store),
+  canCancel: false,
   isEditing: false,
   edit: () => {
     const value = onEdit();
-    set({ isEditing: true, value });
+    set({ isEditing: true, canCancel: true, value });
   },
   send: () => {
     const value = get().value;
-    set({ isEditing: false });
+    set({ isEditing: false, canCancel: false });
     onSend(value);
   },
   cancel: () => {
-    if (!get().isEditing) return false;
-    set({ isEditing: false });
-    return true;
+    set({ isEditing: false, canCancel: false });
   }
 }));
 
@@ -1727,7 +1806,7 @@ var syncMessage = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = (0, import_react47.useState)(() => {
+  const [context] = (0, import_react48.useState)(() => {
     const useMessage = (0, import_zustand12.create)(() => ({}));
     const useMessageUtils = makeMessageUtilsStore();
     const useEditComposer = makeEditComposerStore({
@@ -1764,7 +1843,7 @@ var useMessageContext2 = (messageIndex) => {
     );
     return { useMessage, useMessageUtils, useEditComposer };
   });
-  (0, import_react47.useEffect)(() => {
+  (0, import_react48.useEffect)(() => {
     return useThreadMessages.subscribe((thread) => {
       syncMessage(
         thread,
@@ -1806,11 +1885,11 @@ var ThreadMessageImpl = ({
     /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(AssistantMessage, {}) })
   ] });
 };
-var ThreadMessage = (0, import_react48.memo)(
+var ThreadMessage = (0, import_react49.memo)(
   ThreadMessageImpl,
-  (prev, next) => prev.messageIndex === next.messageIndex && prev.components.UserMessage === next.components.UserMessage && prev.components.EditComposer === next.components.EditComposer && prev.components.AssistantMessage === next.components.AssistantMessage
+  (prev, next) => prev.messageIndex === next.messageIndex && prev.components.Message === next.components.Message && prev.components.UserMessage === next.components.UserMessage && prev.components.EditComposer === next.components.EditComposer && prev.components.AssistantMessage === next.components.AssistantMessage
 );
-var ThreadPrimitiveMessages = ({
+var ThreadPrimitiveMessagesImpl = ({
   components
 }) => {
   const { useThreadMessages } = useThreadContext();
@@ -1828,7 +1907,11 @@ var ThreadPrimitiveMessages = ({
     );
   });
 };
-ThreadPrimitiveMessages.displayName = "ThreadPrimitive.Messages";
+ThreadPrimitiveMessagesImpl.displayName = "ThreadPrimitive.Messages";
+var ThreadPrimitiveMessages = (0, import_react49.memo)(
+  ThreadPrimitiveMessagesImpl,
+  (prev, next) => prev.components?.Message === next.components?.Message && prev.components?.UserMessage === next.components?.UserMessage && prev.components?.EditComposer === next.components?.EditComposer && prev.components?.AssistantMessage === next.components?.AssistantMessage
+);
 
 // src/primitives/thread/ThreadScrollToBottom.tsx
 var ThreadPrimitiveScrollToBottom = createActionButton(
@@ -1843,7 +1926,7 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtime/local/useLocalRuntime.tsx
-var import_react49 = require("react");
+var import_react50 = require("react");
 
 // src/runtime/utils/idUtils.tsx
 var import_non_secure = require("nanoid/non-secure");
@@ -2012,6 +2095,9 @@ var BaseAssistantRuntime = class {
     this._thread = _thread;
     this._unsubscribe = this._thread.subscribe(this.subscriptionHandler);
   }
+  get capabilities() {
+    return this._thread.capabilities;
+  }
   _unsubscribe;
   get thread() {
     return this._thread;
@@ -2084,6 +2170,12 @@ var LocalRuntime = class extends BaseAssistantRuntime {
     );
   }
 };
+var CAPABILITIES = Object.freeze({
+  edit: true,
+  reload: true,
+  cancel: true,
+  copy: true
+});
 var LocalThreadRuntime = class {
   constructor(_configProviders, adapter) {
     this._configProviders = _configProviders;
@@ -2092,6 +2184,7 @@ var LocalThreadRuntime = class {
   _subscriptions = /* @__PURE__ */ new Set();
   abortController = null;
   repository = new MessageRepository();
+  capabilities = CAPABILITIES;
   get messages() {
     return this.repository.getMessages();
   }
@@ -2175,8 +2268,8 @@ var LocalThreadRuntime = class {
 
 // src/runtime/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter) => {
-  const [runtime] = (0, import_react49.useState)(() => new LocalRuntime(adapter));
-  (0, import_react49.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react50.useState)(() => new LocalRuntime(adapter));
+  (0, import_react50.useInsertionEffect)(() => {
     runtime.adapter = adapter;
   });
   return runtime;
@@ -2187,7 +2280,8 @@ var internal_exports = {};
 __export(internal_exports, {
   BaseAssistantRuntime: () => BaseAssistantRuntime,
   MessageRepository: () => MessageRepository,
-  ProxyConfigProvider: () => ProxyConfigProvider
+  ProxyConfigProvider: () => ProxyConfigProvider,
+  useSmooth: () => useSmooth
 });
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {