@assistant-ui/react 0.0.18 → 0.0.20

package/dist/index.js

@@ -173,18 +173,22 @@ var ThreadViewport = (0, import_react5.forwardRef)(({ autoScroll = true, onScrol
   const ref = (0, import_react_compose_refs.useComposedRefs)(forwardedRef, divRef);
   const { useViewport } = useAssistantContext();
   const firstRenderRef = (0, import_react5.useRef)(true);
+  const isScrollingToBottomRef = (0, import_react5.useRef)(false);
   const lastScrollTop = (0, import_react5.useRef)(0);
   const scrollToBottom = () => {
     const div = messagesEndRef.current;
     if (!div || !autoScroll) return;
     const behavior = firstRenderRef.current ? "instant" : "auto";
     firstRenderRef.current = false;
-    useViewport.setState({ isAtBottom: true });
+    isScrollingToBottomRef.current = true;
     div.scrollIntoView({ behavior });
   };
   useOnResizeContent(divRef, () => {
-    if (!useViewport.getState().isAtBottom) return;
-    scrollToBottom();
+    if (!isScrollingToBottomRef.current && !useViewport.getState().isAtBottom) {
+      handleScroll();
+    } else {
+      scrollToBottom();
+    }
   });
   useOnScrollToBottom(() => {
     scrollToBottom();
@@ -196,6 +200,7 @@ var ThreadViewport = (0, import_react5.forwardRef)(({ autoScroll = true, onScrol
     const newIsAtBottom = div.scrollHeight - div.scrollTop <= div.clientHeight;
     if (!newIsAtBottom && lastScrollTop.current < div.scrollTop) {
     } else if (newIsAtBottom !== isAtBottom) {
+      isScrollingToBottomRef.current = false;
       useViewport.setState({ isAtBottom: newIsAtBottom });
     }
     lastScrollTop.current = div.scrollTop;
@@ -841,6 +846,7 @@ __export(branchPicker_exports, {
 });
 
 // src/actions/useGoToNextBranch.tsx
+var import_react21 = require("react");
 var useGoToNextBranch = () => {
   const { useThread } = useAssistantContext();
   const { useMessage, useComposer } = useMessageContext();
@@ -848,20 +854,21 @@ var useGoToNextBranch = () => {
     [useMessage, useComposer],
     (m, c) => c.isEditing || m.branches.indexOf(m.message.id) + 1 >= m.branches.length
   );
-  if (disabled) return null;
-  return () => {
+  const callback = (0, import_react21.useCallback)(() => {
     const { message, branches } = useMessage.getState();
     useThread.getState().switchToBranch(branches[branches.indexOf(message.id) + 1]);
-  };
+  }, [useMessage, useThread]);
+  if (disabled) return null;
+  return callback;
 };
 
 // src/utils/createActionButton.tsx
 var import_primitive8 = require("@radix-ui/primitive");
 var import_react_primitive10 = require("@radix-ui/react-primitive");
-var import_react21 = require("react");
+var import_react22 = require("react");
 var import_jsx_runtime16 = require("react/jsx-runtime");
 var createActionButton = (useActionButton) => {
-  return (0, import_react21.forwardRef)(
+  return (0, import_react22.forwardRef)(
     (props, forwardedRef) => {
       const onClick = useActionButton(props);
       return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
@@ -882,6 +889,7 @@ var createActionButton = (useActionButton) => {
 var BranchPickerNext = createActionButton(useGoToNextBranch);
 
 // src/actions/useGoToPreviousBranch.tsx
+var import_react23 = require("react");
 var useGoToPreviousBranch = () => {
   const { useThread } = useAssistantContext();
   const { useMessage, useComposer } = useMessageContext();
@@ -889,13 +897,12 @@ var useGoToPreviousBranch = () => {
     [useMessage, useComposer],
     (m, c) => c.isEditing || m.branches.indexOf(m.message.id) <= 0
   );
-  if (disabled) return null;
-  return () => {
+  const callback = (0, import_react23.useCallback)(() => {
     const { message, branches } = useMessage.getState();
-    useThread.getState().switchToBranch(
-      branches[branches.indexOf(message.id) - 1]
-    );
-  };
+    useThread.getState().switchToBranch(branches[branches.indexOf(message.id) - 1]);
+  }, [useMessage, useThread]);
+  if (disabled) return null;
+  return callback;
 };
 
 // src/primitives/branchPicker/BranchPickerPrevious.tsx
@@ -919,9 +926,9 @@ var BranchPickerNumber = () => {
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
 var import_react_primitive11 = require("@radix-ui/react-primitive");
-var import_react22 = require("react");
+var import_react24 = require("react");
 var import_jsx_runtime19 = require("react/jsx-runtime");
-var BranchPickerRoot = (0, import_react22.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
+var BranchPickerRoot = (0, import_react24.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
   return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(MessageIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(import_react_primitive11.Primitive.div, { ...rest, ref }) });
 });
 
@@ -936,9 +943,9 @@ __export(actionBar_exports, {
 
 // src/primitives/actionBar/ActionBarRoot.tsx
 var import_react_primitive12 = require("@radix-ui/react-primitive");
-var import_react23 = require("react");
+var import_react25 = require("react");
 var import_jsx_runtime20 = require("react/jsx-runtime");
-var ActionBarRoot = (0, import_react23.forwardRef)(({ hideWhenRunning, autohide, autohideFloat, ...rest }, ref) => {
+var ActionBarRoot = (0, import_react25.forwardRef)(({ hideWhenRunning, autohide, autohideFloat, ...rest }, ref) => {
   const { useThread } = useAssistantContext();
   const { useMessage } = useMessageContext();
   const hideAndfloatStatus = useCombinedStore(
@@ -965,6 +972,7 @@ var ActionBarRoot = (0, import_react23.forwardRef)(({ hideWhenRunning, autohide,
 });
 
 // src/actions/useCopyMessage.tsx
+var import_react26 = require("react");
 var useCopyMessage = ({ copiedDuration = 3e3 }) => {
   const { useMessage, useComposer } = useMessageContext();
   const hasCopyableContent = useCombinedStore(
@@ -973,21 +981,23 @@ var useCopyMessage = ({ copiedDuration = 3e3 }) => {
       return c.isEditing || m.message.content.some((c2) => c2.type === "text");
     }
   );
-  if (!hasCopyableContent) return null;
-  return () => {
+  const callback = (0, import_react26.useCallback)(() => {
     const { isEditing, value: composerValue } = useComposer.getState();
     const { message, setIsCopied } = useMessage.getState();
     const valueToCopy = isEditing ? composerValue : getMessageText(message);
     navigator.clipboard.writeText(valueToCopy);
     setIsCopied(true);
     setTimeout(() => setIsCopied(false), copiedDuration);
-  };
+  }, [useComposer, useMessage, copiedDuration]);
+  if (!hasCopyableContent) return null;
+  return callback;
 };
 
 // src/primitives/actionBar/ActionBarCopy.tsx
 var ActionBarCopy = createActionButton(useCopyMessage);
 
 // src/actions/useReloadMessage.tsx
+var import_react27 = require("react");
 var useReloadMessage = () => {
   const { useThread, useViewport } = useAssistantContext();
   const { useMessage } = useMessageContext();
@@ -995,29 +1005,32 @@ var useReloadMessage = () => {
     [useThread, useMessage],
     (t, m) => t.isRunning || m.message.role !== "assistant"
   );
-  if (disabled) return null;
-  return () => {
+  const callback = (0, import_react27.useCallback)(() => {
     const { parentId } = useMessage.getState();
     useThread.getState().startRun(parentId);
     useViewport.getState().scrollToBottom();
-  };
+  }, [useMessage, useThread, useViewport]);
+  if (disabled) return null;
+  return callback;
 };
 
 // src/primitives/actionBar/ActionBarReload.tsx
 var ActionBarReload = createActionButton(useReloadMessage);
 
 // src/actions/useBeginMessageEdit.tsx
+var import_react28 = require("react");
 var useBeginMessageEdit = () => {
   const { useMessage, useComposer } = useMessageContext();
   const disabled = useCombinedStore(
     [useMessage, useComposer],
     (m, c) => m.message.role !== "user" || c.isEditing
   );
-  if (disabled) return null;
-  return () => {
+  const callback = (0, import_react28.useCallback)(() => {
     const { edit } = useComposer.getState();
     edit();
-  };
+  }, [useComposer]);
+  if (disabled) return null;
+  return callback;
 };
 
 // src/primitives/actionBar/ActionBarEdit.tsx
@@ -1031,10 +1044,10 @@ __export(contentPart_exports, {
 });
 
 // src/adapters/vercel/VercelAIAssistantProvider.tsx
-var import_react26 = require("react");
+var import_react31 = require("react");
 
-// src/adapters/vercel/useDummyAIAssistantContext.tsx
-var import_react24 = require("react");
+// src/adapters/core/AssistantProvider.tsx
+var import_react29 = require("react");
 var import_zustand5 = require("zustand");
 
 // src/utils/context/stores/ViewportStore.tsx
@@ -1057,41 +1070,79 @@ var makeViewportStore = () => {
   }));
 };
 
-// src/adapters/vercel/useDummyAIAssistantContext.tsx
-var makeDummyThreadStore = () => {
-  return (0, import_zustand5.create)(() => ({
-    messages: [],
-    isRunning: false,
-    getBranches: () => {
-      return [];
-    },
-    switchToBranch: () => {
-      throw new Error("Not implemented");
-    },
-    append: () => {
-      throw new Error("Not implemented");
-    },
-    startRun: () => {
-      throw new Error("Not implemented");
-    },
-    cancelRun: () => {
-      throw new Error("Not implemented");
-    }
+// src/adapters/core/AssistantProvider.tsx
+var import_jsx_runtime21 = require("react/jsx-runtime");
+var makeThreadStore = (runtimeRef) => {
+  const useThread = (0, import_zustand5.create)(() => ({
+    messages: runtimeRef.current.messages,
+    isRunning: runtimeRef.current.isRunning,
+    getBranches: (messageId) => runtimeRef.current.getBranches(messageId),
+    switchToBranch: (branchId) => runtimeRef.current.switchToBranch(branchId),
+    startRun: (parentId) => runtimeRef.current.startRun(parentId),
+    append: (message) => runtimeRef.current.append(message),
+    cancelRun: () => runtimeRef.current.cancelRun()
   }));
+  const onRuntimeUpdate = () => {
+    useThread.setState({
+      messages: runtimeRef.current.messages,
+      isRunning: runtimeRef.current.isRunning
+    });
+  };
+  return {
+    useThread,
+    onRuntimeUpdate
+  };
 };
-var useDummyAIAssistantContext = () => {
-  const [context] = (0, import_react24.useState)(() => {
-    const useThread = makeDummyThreadStore();
+var AssistantProvider = ({ children, runtime }) => {
+  const runtimeRef = (0, import_react29.useRef)(runtime);
+  (0, import_react29.useInsertionEffect)(() => {
+    runtimeRef.current = runtime;
+  });
+  const [{ context, onRuntimeUpdate }] = (0, import_react29.useState)(() => {
+    const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
     const useViewport = makeViewportStore();
     const useComposer = makeThreadComposerStore(useThread);
-    return { useThread, useViewport, useComposer };
+    return {
+      context: {
+        useViewport,
+        useThread,
+        useComposer
+      },
+      onRuntimeUpdate: onRuntimeUpdate2
+    };
   });
-  return context;
+  (0, import_react29.useEffect)(() => {
+    onRuntimeUpdate();
+    return runtime.subscribe(onRuntimeUpdate);
+  }, [onRuntimeUpdate, runtime]);
+  return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(AssistantContext.Provider, { value: context, children });
 };
 
-// src/adapters/vercel/useVercelAIThreadState.tsx
-var import_react_use_callback_ref3 = require("@radix-ui/react-use-callback-ref");
-var import_react25 = require("react");
+// src/adapters/core/vercel-use-chat/useVercelUseChatRuntime.tsx
+var import_react30 = require("react");
+
+// src/adapters/ThreadMessageConverter.ts
+var ThreadMessageConverter = class {
+  cache = /* @__PURE__ */ new WeakMap();
+  convertMessages(converter, messages) {
+    return messages.map((m) => {
+      const cached = this.cache.get(m);
+      const newMessage = converter(m, cached);
+      this.cache.set(m, newMessage);
+      return newMessage;
+    });
+  }
+};
+
+// src/adapters/vercel/VercelThreadMessage.tsx
+var symbolInnerMessage = Symbol("innerMessage");
+var symbolInnerRSCMessage = Symbol("innerRSCMessage");
+var getVercelMessage = (message) => {
+  return message[symbolInnerMessage];
+};
+var getVercelRSCMessage = (message) => {
+  return message[symbolInnerRSCMessage];
+};
 
 // src/adapters/idUtils.tsx
 var import_non_secure = require("nanoid/non-secure");
@@ -1253,30 +1304,104 @@ var MessageRepository = class {
   }
 };
 
-// src/adapters/ThreadMessageConverter.ts
-var ThreadMessageConverter = class {
-  cache = /* @__PURE__ */ new WeakMap();
-  convertMessages(converter, messages) {
-    return messages.map((m) => {
-      const cached = this.cache.get(m);
-      const newMessage = converter(m, cached);
-      this.cache.set(m, newMessage);
-      return newMessage;
+// src/adapters/core/vercel-use-chat/VercelUseChatRuntime.tsx
+var sliceMessagesUntil = (messages, messageId) => {
+  if (messageId == null) return [];
+  const messageIdx = messages.findIndex((m) => m.id === messageId);
+  if (messageIdx === -1)
+    throw new Error(
+      "useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui."
+    );
+  return messages.slice(0, messageIdx + 1);
+};
+var hasUpcomingMessage = (isRunning, messages) => {
+  return isRunning && messages[messages.length - 1]?.role !== "assistant";
+};
+var VercelUseChatRuntime = class {
+  constructor(vercel) {
+    this.vercel = vercel;
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  repository = new MessageRepository();
+  assistantOptimisticId = null;
+  messages = [];
+  isRunning = false;
+  getBranches(messageId) {
+    return this.repository.getBranches(messageId);
+  }
+  switchToBranch(branchId) {
+    this.repository.switchToBranch(branchId);
+    this.vercel.setMessages(
+      this.repository.getMessages().map(getVercelMessage).filter((m) => m != null)
+    );
+  }
+  async append(message) {
+    if (message.content.length !== 1 || message.content[0]?.type !== "text")
+      throw new Error("Only text content is supported by Vercel AI SDK.");
+    const newMessages = sliceMessagesUntil(
+      this.vercel.messages,
+      message.parentId
+    );
+    this.vercel.setMessages(newMessages);
+    await this.vercel.append({
+      role: "user",
+      content: message.content[0].text
     });
   }
+  async startRun(parentId) {
+    const reloadMaybe = "reload" in this.vercel ? this.vercel.reload : void 0;
+    if (!reloadMaybe)
+      throw new Error(
+        "Reload is not supported by Vercel AI SDK's useAssistant."
+      );
+    const newMessages = sliceMessagesUntil(this.vercel.messages, parentId);
+    this.vercel.setMessages(newMessages);
+    await reloadMaybe();
+  }
+  cancelRun() {
+    const lastMessage = this.vercel.messages.at(-1);
+    this.vercel.stop();
+    if (lastMessage?.role === "user") {
+      this.vercel.setInput(lastMessage.content);
+    }
+  }
+  updateData(isRunning, vm) {
+    for (let i = 0; i < vm.length; i++) {
+      const message = vm[i];
+      const parent = vm[i - 1];
+      this.repository.addOrUpdateMessage(parent?.id ?? null, message);
+    }
+    if (this.assistantOptimisticId) {
+      this.repository.deleteMessage(this.assistantOptimisticId, null);
+      this.assistantOptimisticId = null;
+    }
+    if (hasUpcomingMessage(isRunning, vm)) {
+      this.assistantOptimisticId = this.repository.appendOptimisticMessage(
+        vm.at(-1)?.id ?? null,
+        {
+          role: "assistant",
+          content: [{ type: "text", text: "" }]
+        }
+      );
+    }
+    this.repository.resetHead(
+      this.assistantOptimisticId ?? vm.at(-1)?.id ?? null
+    );
+    this.messages = this.repository.getMessages();
+    this.isRunning = isRunning;
+    for (const callback of this._subscriptions) callback();
+  }
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
 };
 
-// src/adapters/vercel/VercelThreadMessage.tsx
-var symbolInnerMessage = Symbol("innerMessage");
-var symbolInnerRSCMessage = Symbol("innerRSCMessage");
-var getVercelMessage = (message) => {
-  return message[symbolInnerMessage];
-};
-var getVercelRSCMessage = (message) => {
-  return message[symbolInnerRSCMessage];
+// src/adapters/core/vercel-use-chat/useVercelUseChatRuntime.tsx
+var getIsRunning = (vercel) => {
+  if ("isLoading" in vercel) return vercel.isLoading;
+  return vercel.status === "in_progress";
 };
-
-// src/adapters/vercel/useVercelAIThreadState.tsx
 var vercelToThreadMessage = (message, status) => {
   const common = {
     id: message.id,
@@ -1313,28 +1438,14 @@ var vercelToThreadMessage = (message, status) => {
       );
   }
 };
-var sliceMessagesUntil = (messages, messageId) => {
-  if (messageId == null) return [];
-  const messageIdx = messages.findIndex((m) => m.id === messageId);
-  if (messageIdx === -1)
-    throw new Error(
-      "useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui."
-    );
-  return messages.slice(0, messageIdx + 1);
-};
-var hasUpcomingMessage = (isRunning, messages) => {
-  return isRunning && messages[messages.length - 1]?.role !== "assistant";
-};
-var getIsRunning = (vercel) => {
-  if ("isLoading" in vercel) return vercel.isLoading;
-  return vercel.status === "in_progress";
-};
-var useVercelAIThreadState = (vercel) => {
-  const [data] = (0, import_react25.useState)(() => new MessageRepository());
+var useVercelUseChatRuntime = (vercel) => {
+  const [runtime] = (0, import_react30.useState)(() => new VercelUseChatRuntime(vercel));
+  (0, import_react30.useInsertionEffect)(() => {
+    runtime.vercel = vercel;
+  });
   const isRunning = getIsRunning(vercel);
-  const converter = (0, import_react25.useMemo)(() => new ThreadMessageConverter(), []);
-  const assistantOptimisticIdRef = (0, import_react25.useRef)(null);
-  const messages = (0, import_react25.useMemo)(() => {
+  const converter = (0, import_react30.useMemo)(() => new ThreadMessageConverter(), []);
+  const messages = (0, import_react30.useMemo)(() => {
     const lastMessageId = vercel.messages.at(-1)?.id;
     const convertCallback = (message, cache) => {
       const status = lastMessageId === message.id && isRunning ? "in_progress" : "done";
@@ -1342,113 +1453,102 @@ var useVercelAIThreadState = (vercel) => {
         return cache;
       return vercelToThreadMessage(message, status);
     };
-    const vm = converter.convertMessages(convertCallback, vercel.messages);
-    for (let i = 0; i < vm.length; i++) {
-      const message = vm[i];
-      const parent = vm[i - 1];
-      data.addOrUpdateMessage(parent?.id ?? null, message);
-    }
-    if (assistantOptimisticIdRef.current) {
-      data.deleteMessage(assistantOptimisticIdRef.current, null);
-      assistantOptimisticIdRef.current = null;
-    }
-    if (hasUpcomingMessage(isRunning, vm)) {
-      assistantOptimisticIdRef.current = data.appendOptimisticMessage(
-        vm.at(-1)?.id ?? null,
-        {
-          role: "assistant",
-          content: [{ type: "text", text: "" }]
-        }
-      );
-    }
-    data.resetHead(assistantOptimisticIdRef.current ?? vm.at(-1)?.id ?? null);
-    return data.getMessages();
-  }, [converter, data, isRunning, vercel.messages]);
-  const getBranches2 = (0, import_react25.useCallback)(
-    (messageId) => {
-      return data.getBranches(messageId);
-    },
-    [data]
-  );
-  const switchToBranch2 = (0, import_react_use_callback_ref3.useCallbackRef)((messageId) => {
-    data.switchToBranch(messageId);
-    vercel.setMessages(
-      data.getMessages().map(getVercelMessage).filter((m) => m != null)
-    );
-  });
-  const startRun = (0, import_react_use_callback_ref3.useCallbackRef)(async (parentId) => {
-    const reloadMaybe = "reload" in vercel ? vercel.reload : void 0;
-    if (!reloadMaybe)
-      throw new Error(
-        "Reload is not supported by Vercel AI SDK's useAssistant."
-      );
-    const newMessages = sliceMessagesUntil(vercel.messages, parentId);
-    vercel.setMessages(newMessages);
-    await reloadMaybe();
-  });
-  const append = (0, import_react_use_callback_ref3.useCallbackRef)(async (message) => {
-    if (message.content.length !== 1 || message.content[0]?.type !== "text")
-      throw new Error("Only text content is supported by Vercel AI SDK.");
-    const newMessages = sliceMessagesUntil(vercel.messages, message.parentId);
-    vercel.setMessages(newMessages);
-    await vercel.append({
-      role: "user",
-      content: message.content[0].text
-    });
-  });
-  const cancelRun2 = (0, import_react_use_callback_ref3.useCallbackRef)(() => {
-    const lastMessage = vercel.messages.at(-1);
-    vercel.stop();
-    if (lastMessage?.role === "user") {
-      vercel.setInput(lastMessage.content);
-    }
-  });
-  return (0, import_react25.useMemo)(
-    () => ({
-      isRunning,
-      messages,
-      getBranches: getBranches2,
-      switchToBranch: switchToBranch2,
-      append,
-      startRun,
-      cancelRun: cancelRun2
-    }),
-    [
-      isRunning,
-      messages,
-      getBranches2,
-      switchToBranch2,
-      append,
-      startRun,
-      cancelRun2
-    ]
-  );
+    return converter.convertMessages(convertCallback, vercel.messages);
+  }, [isRunning, vercel.messages, converter]);
+  (0, import_react30.useEffect)(() => {
+    runtime.updateData(isRunning, messages);
+  }, [runtime, isRunning, messages]);
+  return runtime;
 };
 
 // src/adapters/vercel/VercelAIAssistantProvider.tsx
-var import_jsx_runtime21 = require("react/jsx-runtime");
+var import_jsx_runtime22 = require("react/jsx-runtime");
 var VercelAIAssistantProvider = ({
   children,
   ...rest
 }) => {
-  const context = useDummyAIAssistantContext();
   const vercel = "chat" in rest ? rest.chat : rest.assistant;
-  const threadState = useVercelAIThreadState(vercel);
-  (0, import_react26.useMemo)(() => {
-    context.useThread.setState(threadState, true);
-  }, [context, threadState]);
-  (0, import_react26.useMemo)(() => {
-    context.useComposer.setState({
+  const runtime = useVercelUseChatRuntime(vercel);
+  return /* @__PURE__ */ (0, import_jsx_runtime22.jsxs)(AssistantProvider, { runtime, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ComposerSync, { vercel }),
+    children
+  ] });
+};
+var ComposerSync = ({
+  vercel
+}) => {
+  const { useComposer } = useAssistantContext();
+  (0, import_react31.useEffect)(() => {
+    useComposer.setState({
       value: vercel.input,
       setValue: vercel.setInput
     });
-  }, [context, vercel.input, vercel.setInput]);
-  return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(AssistantContext.Provider, { value: context, children });
+  }, [useComposer, vercel.input, vercel.setInput]);
+  return null;
 };
 
-// src/adapters/vercel/VercelRSCAssistantProvider.tsx
-var import_react27 = require("react");
-var import_jsx_runtime22 = require("react/jsx-runtime");
+// src/adapters/core/vercel-rsc/useVercelRSCRuntime.tsx
+var import_react32 = require("react");
+
+// src/adapters/core/vercel-rsc/VercelRSCRuntime.tsx
+var EMPTY_BRANCHES = Object.freeze([]);
+var VercelRSCRuntime = class {
+  constructor(adapter) {
+    this.adapter = adapter;
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  isRunning = false;
+  messages = [];
+  withRunning = (callback) => {
+    this.isRunning = true;
+    return callback.finally(() => {
+      this.isRunning = false;
+    });
+  };
+  getBranches() {
+    return EMPTY_BRANCHES;
+  }
+  switchToBranch() {
+    throw new Error(
+      "Branch switching is not supported by VercelRSCAssistantProvider."
+    );
+  }
+  async append(message) {
+    if (message.parentId !== (this.messages.at(-1)?.id ?? null)) {
+      if (!this.adapter.edit)
+        throw new Error(
+          "Message editing is not enabled, please provide an edit callback to VercelRSCAssistantProvider."
+        );
+      await this.withRunning(this.adapter.edit(message));
+    } else {
+      await this.withRunning(this.adapter.append(message));
+    }
+  }
+  async startRun(parentId) {
+    if (!this.adapter.reload)
+      throw new Error(
+        "Message reloading is not enabled, please provide a reload callback to VercelRSCAssistantProvider."
+      );
+    await this.withRunning(this.adapter.reload(parentId));
+  }
+  cancelRun() {
+    if (process.env["NODE_ENV"] === "development") {
+      console.warn(
+        "Run cancellation is not supported by VercelRSCAssistantProvider."
+      );
+    }
+  }
+  updateData(messages) {
+    this.messages = messages;
+    for (const callback of this._subscriptions) callback();
+  }
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+};
+
+// src/adapters/core/vercel-rsc/useVercelRSCRuntime.tsx
 var vercelToThreadMessage2 = (converter, rawMessage) => {
   const message = converter(rawMessage);
   return {
@@ -1460,220 +1560,61 @@ var vercelToThreadMessage2 = (converter, rawMessage) => {
     [symbolInnerRSCMessage]: rawMessage
   };
 };
-var EMPTY_BRANCHES = [];
-var getBranches = () => {
-  return EMPTY_BRANCHES;
-};
-var switchToBranch = () => {
-  throw new Error(
-    "Branch switching is not supported by VercelRSCAssistantProvider."
-  );
-};
-var cancelRun = () => {
-  if (process.env["NODE_ENV"] === "development") {
-    console.warn(
-      "Run cancellation is not supported by VercelRSCAssistantProvider."
-    );
-  }
-};
-var VercelRSCAssistantProvider = ({
-  children,
-  convertMessage,
-  messages: vercelMessages,
-  append: appendCallback,
-  edit,
-  reload
-}) => {
-  const context = useDummyAIAssistantContext();
-  const [isRunning, setIsRunning] = (0, import_react27.useState)(false);
-  const withRunning = (0, import_react27.useCallback)((callback) => {
-    setIsRunning(true);
-    return callback.finally(() => setIsRunning(false));
-  }, []);
-  const [converter, convertCallback] = (0, import_react27.useMemo)(() => {
-    const rscConverter = convertMessage ?? ((m) => m);
+var useVercelRSCRuntime = (adapter) => {
+  const [runtime] = (0, import_react32.useState)(() => new VercelRSCRuntime(adapter));
+  (0, import_react32.useInsertionEffect)(() => {
+    runtime.adapter = adapter;
+  });
+  const [converter, convertCallback] = (0, import_react32.useMemo)(() => {
+    const rscConverter = adapter.convertMessage ?? ((m) => m);
     const convertCallback2 = (m, cache) => {
       if (cache) return cache;
       return vercelToThreadMessage2(rscConverter, m);
     };
     return [new ThreadMessageConverter(), convertCallback2];
-  }, [convertMessage]);
-  const messages = (0, import_react27.useMemo)(() => {
-    return converter.convertMessages(convertCallback, vercelMessages);
-  }, [converter, convertCallback, vercelMessages]);
-  const append = (0, import_react27.useCallback)(
-    async (message) => {
-      if (message.parentId !== (context.useThread.getState().messages.at(-1)?.id ?? null)) {
-        if (!edit)
-          throw new Error(
-            "Message editing is not enabled, please provide an edit callback to VercelRSCAssistantProvider."
-          );
-        await withRunning(edit(message));
-      } else {
-        await withRunning(appendCallback(message));
-      }
-    },
-    [context, withRunning, appendCallback, edit]
-  );
-  const startRun = (0, import_react27.useCallback)(
-    async (parentId) => {
-      if (!reload)
-        throw new Error(
-          "Message reloading is not enabled, please provide a reload callback to VercelRSCAssistantProvider."
-        );
-      await withRunning(reload(parentId));
-    },
-    [withRunning, reload]
-  );
-  (0, import_react27.useMemo)(() => {
-    context.useThread.setState(
-      {
-        messages,
-        isRunning,
-        getBranches,
-        switchToBranch,
-        append,
-        startRun,
-        cancelRun
-      },
-      true
+  }, [adapter.convertMessage]);
+  (0, import_react32.useEffect)(() => {
+    runtime.updateData(
+      converter.convertMessages(convertCallback, adapter.messages)
     );
-  }, [context, messages, isRunning, append, startRun]);
-  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(AssistantContext.Provider, { value: context, children });
-};
-
-// src/adapters/core/utils/useAssistantContext.tsx
-var import_react28 = require("react");
-var import_zustand6 = require("zustand");
-
-// src/adapters/core/utils/AssistantMessageRepository.tsx
-var AssistantMessageRepository = class {
-  constructor(flushCallback) {
-    this.flushCallback = flushCallback;
-  }
-  repository = new MessageRepository();
-  getBranches(messageId) {
-    return this.repository.getBranches(messageId);
-  }
-  withModifications(callback) {
-    const res = callback(this.repository);
-    this.flushCallback(this.repository.getMessages());
-    return res;
-  }
-};
-
-// src/adapters/core/utils/useAssistantContext.tsx
-var makeThreadStore = (runtimeRef) => {
-  const repository = new AssistantMessageRepository((messages) => {
-    useThread.setState({ messages });
-  });
-  const useThread = (0, import_zustand6.create)(() => ({
-    messages: [],
-    isRunning: false,
-    getBranches: (messageId) => repository.getBranches(messageId),
-    switchToBranch: (branchId) => {
-      repository.withModifications((repository2) => {
-        repository2.switchToBranch(branchId);
-      });
-    },
-    startRun: async (parentId) => {
-      const optimisticId = repository.withModifications((repository2) => {
-        const optimisticId2 = repository2.appendOptimisticMessage(parentId, {
-          role: "assistant",
-          content: [{ type: "text", text: "" }]
-        });
-        repository2.resetHead(optimisticId2);
-        return optimisticId2;
-      });
-      const { id } = await runtimeRef.current.startRun(parentId);
-      repository.withModifications((repository2) => {
-        repository2.deleteMessage(optimisticId, id);
-      });
-    },
-    append: async (message) => {
-      const [parentOptimisticId, optimisticId] = repository.withModifications(
-        (repository2) => {
-          const parentOptimisticId2 = repository2.appendOptimisticMessage(
-            message.parentId,
-            {
-              role: "user",
-              content: message.content
-            }
-          );
-          const optimisticId2 = repository2.appendOptimisticMessage(
-            parentOptimisticId2,
-            {
-              role: "assistant",
-              content: [{ type: "text", text: "" }]
-            }
-          );
-          repository2.resetHead(optimisticId2);
-          return [parentOptimisticId2, optimisticId2];
-        }
-      );
-      const { parentId, id } = await runtimeRef.current.append(message);
-      repository.withModifications((repository2) => {
-        repository2.deleteMessage(parentOptimisticId, parentId);
-        repository2.deleteMessage(optimisticId, id);
-      });
-    },
-    cancelRun: () => runtimeRef.current.cancelRun()
-  }));
-  const onNewMessage = (parentId, message) => {
-    repository.withModifications((repository2) => {
-      repository2.addOrUpdateMessage(parentId, message);
-    });
-  };
-  const onRunningChange = (isRunning) => {
-    useThread.setState({ isRunning });
-  };
-  return {
-    useThread,
-    onNewMessage,
-    onRunningChange
-  };
-};
-var useAssistantContext2 = (runtime) => {
-  const runtimeRef = (0, import_react28.useRef)(runtime);
-  runtimeRef.current = runtime;
-  const [{ context, onNewMessage, onRunningChange }] = (0, import_react28.useState)(() => {
-    const { useThread, onNewMessage: onNewMessage2, onRunningChange: onRunningChange2 } = makeThreadStore(runtimeRef);
-    const useViewport = makeViewportStore();
-    const useComposer = makeThreadComposerStore(useThread);
-    return {
-      context: { useViewport, useThread, useComposer },
-      onNewMessage: onNewMessage2,
-      onRunningChange: onRunningChange2
-    };
-  });
-  (0, import_react28.useEffect)(() => {
-    return runtime.subscribeToMessageUpdates(onNewMessage);
-  }, [runtime, onNewMessage]);
-  (0, import_react28.useEffect)(() => {
-    return runtime.subscribeToStatusUpdates(onRunningChange);
-  }, [runtime, onRunningChange]);
-  return context;
+  }, [runtime, converter, convertCallback, adapter.messages]);
+  return runtime;
 };
 
-// src/adapters/core/AssistantProvider.tsx
+// src/adapters/vercel/VercelRSCAssistantProvider.tsx
 var import_jsx_runtime23 = require("react/jsx-runtime");
-var AssistantProvider = ({ children, runtime }) => {
-  const context = useAssistantContext2(runtime);
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantContext.Provider, { value: context, children });
+var VercelRSCAssistantProvider = ({
+  children,
+  ...adapter
+}) => {
+  const runtime = useVercelRSCRuntime(adapter);
+  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantProvider, { runtime, children });
 };
 
 // src/adapters/core/local/useLocalRuntime.tsx
-var import_react29 = require("react");
+var import_react33 = require("react");
 
 // src/adapters/core/local/LocalRuntime.tsx
 var LocalRuntime = class {
   constructor(adapter) {
     this.adapter = adapter;
   }
-  _messageUpdateCallbacks = /* @__PURE__ */ new Set();
-  _statusUpdateCallbacks = /* @__PURE__ */ new Set();
+  _subscriptions = /* @__PURE__ */ new Set();
   abortController = null;
   repository = new MessageRepository();
+  get messages() {
+    return this.repository.getMessages();
+  }
+  get isRunning() {
+    return this.abortController != null;
+  }
+  getBranches(messageId) {
+    return this.repository.getBranches(messageId);
+  }
+  switchToBranch(branchId) {
+    this.repository.switchToBranch(branchId);
+    this.notifySubscribers();
+  }
   async append(message) {
     const userMessageId = generateId();
     const userMessage = {
@@ -1682,9 +1623,8 @@ var LocalRuntime = class {
       content: message.content,
       createdAt: /* @__PURE__ */ new Date()
     };
-    this.addOrUpdateMessage(message.parentId, userMessage);
-    const { id } = await this.startRun(userMessageId);
-    return { parentId: userMessageId, id };
+    this.repository.addOrUpdateMessage(message.parentId, userMessage);
+    await this.startRun(userMessageId);
   }
   async startRun(parentId) {
     const id = generateId();
@@ -1697,59 +1637,52 @@ var LocalRuntime = class {
       content: [{ type: "text", text: "" }],
       createdAt: /* @__PURE__ */ new Date()
     };
-    this.addOrUpdateMessage(parentId, message);
-    void this.run(parentId, messages, message);
-    return { id };
-  }
-  addOrUpdateMessage(parentId, message) {
-    const clone = { ...message };
-    this.repository.addOrUpdateMessage(parentId, clone);
-    for (const callback of this._messageUpdateCallbacks)
-      callback(parentId, clone);
-  }
-  async run(parentId, messages, message) {
-    this.cancelRun();
-    for (const callback of this._statusUpdateCallbacks) callback(true);
+    this.repository.addOrUpdateMessage(parentId, { ...message });
+    this.abortController?.abort();
     this.abortController = new AbortController();
+    this.notifySubscribers();
     try {
       await this.adapter.run({
         messages,
         abortSignal: this.abortController.signal,
         onUpdate: ({ content }) => {
           message.content = content;
-          this.addOrUpdateMessage(parentId, message);
+          this.repository.addOrUpdateMessage(parentId, { ...message });
+          this.notifySubscribers();
         }
       });
       message.status = "done";
-      this.addOrUpdateMessage(parentId, message);
+      this.repository.addOrUpdateMessage(parentId, { ...message });
     } catch (e) {
       message.status = "error";
-      this.addOrUpdateMessage(parentId, message);
+      this.repository.addOrUpdateMessage(parentId, { ...message });
       console.error(e);
     } finally {
-      this.cancelRun();
+      this.abortController = null;
+      this.notifySubscribers();
     }
   }
   cancelRun() {
     if (!this.abortController) return;
     this.abortController.abort();
     this.abortController = null;
-    for (const callback of this._statusUpdateCallbacks) callback(false);
+    this.notifySubscribers();
   }
-  subscribeToMessageUpdates(callback) {
-    this._messageUpdateCallbacks.add(callback);
-    return () => this._messageUpdateCallbacks.delete(callback);
+  notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
   }
-  subscribeToStatusUpdates(callback) {
-    this._statusUpdateCallbacks.add(callback);
-    return () => this._statusUpdateCallbacks.delete(callback);
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
   }
 };
 
 // src/adapters/core/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter) => {
-  const [runtime] = (0, import_react29.useState)(() => new LocalRuntime(adapter));
-  runtime.adapter = adapter;
+  const [runtime] = (0, import_react33.useState)(() => new LocalRuntime(adapter));
+  (0, import_react33.useInsertionEffect)(() => {
+    runtime.adapter = adapter;
+  });
   return runtime;
 };