@assistant-ui/react 0.5.22 → 0.5.24

package/dist/index.mjs

@@ -2144,6 +2144,13 @@ var MessageRepository = class {
       }
     }
     if (operation !== "cut") {
+      for (let current = newParent; current; current = current.prev) {
+        if (current.current.id === child.current.id) {
+          throw new Error(
+            "MessageRepository(performOp/link): A message with the same id already exists in the parent tree. This error occurs if the same message id is found multiple times. This is likely an internal bug in assistant-ui."
+          );
+        }
+      }
       newParentOrRoot.children = [
         ...newParentOrRoot.children,
         child.current.id
@@ -2273,6 +2280,25 @@ var MessageRepository = class {
       }
     }
   }
+  export() {
+    const exportItems = [];
+    for (const [, message] of this.messages) {
+      exportItems.push({
+        message: message.current,
+        parentId: message.prev?.current.id ?? null
+      });
+    }
+    return {
+      headId: this.head?.current.id ?? null,
+      messages: exportItems
+    };
+  }
+  import({ headId, messages }) {
+    for (const { message, parentId } of messages) {
+      this.addOrUpdateMessage(parentId, message);
+    }
+    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);
+  }
 };
 
 // src/ui/base/tooltip-icon-button.tsx
@@ -2831,6 +2857,13 @@ var LocalThreadRuntime = class {
       this.performRoundtrip(parentId, message);
     }
   }
+  export() {
+    return this.repository.export();
+  }
+  import(data) {
+    this.repository.import(data);
+    this.notifySubscribers();
+  }
 };
 
 // src/runtimes/local/LocalRuntime.tsx
@@ -2867,18 +2900,12 @@ var useLocalRuntime = (adapter, options) => {
   return runtime;
 };
 
-// src/runtimes/external-store/ExternalStoreThreadRuntime.tsx
-import { create as create15 } from "zustand";
-
 // src/runtimes/external-store/getExternalStoreMessage.tsx
 var symbolInnerMessage = Symbol("innerMessage");
 var getExternalStoreMessage = (message) => {
   return message[symbolInnerMessage];
 };
 
-// src/runtimes/external-store/useExternalStoreSync.tsx
-import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect4, useMemo as useMemo4, useRef as useRef6 } from "react";
-
 // src/runtimes/external-store/ThreadMessageConverter.ts
 var ThreadMessageConverter = class {
   cache = /* @__PURE__ */ new WeakMap();
@@ -2899,6 +2926,7 @@ var AUTO_STATUS_COMPLETE = Object.freeze({
   type: "complete",
   reason: "unknown"
 });
+var isAutoStatus = (status) => status === AUTO_STATUS_RUNNING || status === AUTO_STATUS_COMPLETE;
 var getAutoStatus = (isLast, isRunning) => isLast && isRunning ? AUTO_STATUS_RUNNING : AUTO_STATUS_COMPLETE;
 
 // src/runtimes/external-store/ThreadMessageLike.tsx
@@ -2969,106 +2997,114 @@ var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
   }
 };
 
-// src/runtimes/external-store/useExternalStoreSync.tsx
-var useExternalStoreSync = (adapter, updateData) => {
-  const adapterRef = useRef6(adapter);
-  useInsertionEffect4(() => {
-    adapterRef.current = adapter;
-  });
-  const [converter, convertCallback] = useMemo4(() => {
-    const converter2 = adapter.convertMessage ?? ((m) => m);
-    const convertCallback2 = (cache, m, idx) => {
-      const autoStatus = getAutoStatus(
-        adapterRef.current.messages.at(-1) === m,
-        adapterRef.current.isRunning ?? false
-      );
-      if (cache && (cache.role !== "assistant" || cache.status === autoStatus))
-        return cache;
-      const newMessage = fromThreadMessageLike(
-        converter2(m, idx),
-        idx.toString(),
-        autoStatus
-      );
-      newMessage[symbolInnerMessage] = m;
-      return newMessage;
-    };
-    return [new ThreadMessageConverter(), convertCallback2];
-  }, [adapter.convertMessage]);
-  useEffect11(() => {
-    updateData(
-      adapter.isDisabled ?? false,
-      adapter.isRunning ?? false,
-      converter.convertMessages(adapter.messages, convertCallback)
-    );
-  }, [
-    updateData,
-    converter,
-    convertCallback,
-    adapter.isDisabled,
-    adapter.isRunning,
-    adapter.messages
-  ]);
-};
-
 // src/runtimes/external-store/ExternalStoreThreadRuntime.tsx
 var hasUpcomingMessage = (isRunning, messages) => {
   return isRunning && messages[messages.length - 1]?.role !== "assistant";
 };
 var ExternalStoreThreadRuntime = class {
-  constructor(store) {
-    this.store = store;
-    this.updateData(
-      store.isDisabled ?? false,
-      store.isRunning ?? false,
-      store.messages
-    );
-    this.useStore = create15(() => ({
-      store
-    }));
-  }
   _subscriptions = /* @__PURE__ */ new Set();
   repository = new MessageRepository();
   assistantOptimisticId = null;
-  useStore;
   get capabilities() {
     return {
-      switchToBranch: this.store.setMessages !== void 0,
-      edit: this.store.onEdit !== void 0,
-      reload: this.store.onReload !== void 0,
-      cancel: this.store.onCancel !== void 0,
-      copy: this.store.onCopy !== null
+      switchToBranch: this._store.setMessages !== void 0,
+      edit: this._store.onEdit !== void 0,
+      reload: this._store.onReload !== void 0,
+      cancel: this._store.onCancel !== void 0,
+      copy: this._store.onCopy !== null
     };
   }
   messages = [];
   isDisabled = false;
   isRunning = false;
+  converter = new ThreadMessageConverter();
+  _store;
+  constructor(store) {
+    this.store = store;
+  }
+  set store(store) {
+    const oldStore = this._store;
+    if (oldStore) {
+      if (oldStore.convertMessage !== store.convertMessage) {
+        this.converter = new ThreadMessageConverter();
+      } else if (oldStore.isDisabled === store.isDisabled && oldStore.isRunning === store.isRunning && oldStore.messages === store.messages) {
+        return;
+      }
+    }
+    this._store = store;
+    const isRunning = store.isRunning ?? false;
+    const isDisabled = store.isDisabled ?? false;
+    const convertCallback = (cache, m, idx) => {
+      if (!store.convertMessage) return m;
+      const isLast = idx === store.messages.length - 1;
+      const autoStatus = getAutoStatus(isLast, isRunning);
+      if (cache && (cache.role !== "assistant" || !isAutoStatus(cache.status) || cache.status === autoStatus))
+        return cache;
+      const newMessage = fromThreadMessageLike(
+        store.convertMessage(m, idx),
+        idx.toString(),
+        autoStatus
+      );
+      newMessage[symbolInnerMessage] = m;
+      return newMessage;
+    };
+    const messages = this.converter.convertMessages(
+      store.messages,
+      convertCallback
+    );
+    for (let i = 0; i < messages.length; i++) {
+      const message = messages[i];
+      const parent = messages[i - 1];
+      this.repository.addOrUpdateMessage(parent?.id ?? null, message);
+    }
+    if (this.assistantOptimisticId) {
+      this.repository.deleteMessage(this.assistantOptimisticId);
+      this.assistantOptimisticId = null;
+    }
+    if (hasUpcomingMessage(isRunning, messages)) {
+      this.assistantOptimisticId = this.repository.appendOptimisticMessage(
+        messages.at(-1)?.id ?? null,
+        {
+          role: "assistant",
+          content: []
+        }
+      );
+    }
+    this.repository.resetHead(
+      this.assistantOptimisticId ?? messages.at(-1)?.id ?? null
+    );
+    this.messages = this.repository.getMessages();
+    this.isDisabled = isDisabled;
+    this.isRunning = isRunning;
+    for (const callback of this._subscriptions) callback();
+  }
   getBranches(messageId) {
     return this.repository.getBranches(messageId);
   }
   switchToBranch(branchId) {
-    if (!this.store.setMessages)
+    if (!this._store.setMessages)
       throw new Error("Runtime does not support switching branches.");
     this.repository.switchToBranch(branchId);
     this.updateMessages(this.repository.getMessages());
   }
   async append(message) {
     if (message.parentId !== (this.messages.at(-1)?.id ?? null)) {
-      if (!this.store.onEdit)
+      if (!this._store.onEdit)
         throw new Error("Runtime does not support editing messages.");
-      await this.store.onEdit(message);
+      await this._store.onEdit(message);
     } else {
-      await this.store.onNew(message);
+      await this._store.onNew(message);
     }
   }
   async startRun(parentId) {
-    if (!this.store.onReload)
+    if (!this._store.onReload)
       throw new Error("Runtime does not support reloading messages.");
-    await this.store.onReload(parentId);
+    await this._store.onReload(parentId);
   }
   cancelRun() {
-    if (!this.store.onCancel)
+    if (!this._store.onCancel)
       throw new Error("Runtime does not support cancelling runs.");
-    this.store.onCancel();
+    this._store.onCancel();
     if (this.assistantOptimisticId) {
       this.repository.deleteMessage(this.assistantOptimisticId);
       this.assistantOptimisticId = null;
@@ -3083,51 +3119,14 @@ var ExternalStoreThreadRuntime = class {
     return () => this._subscriptions.delete(callback);
   }
   updateMessages = (messages) => {
-    this.store.setMessages?.(
+    this._store.setMessages?.(
       messages.flatMap(getExternalStoreMessage).filter((m) => m != null)
     );
   };
-  onStoreUpdated() {
-    if (this.useStore.getState().store !== this.store) {
-      this.useStore.setState({ store: this.store });
-    }
-  }
-  updateData = (isDisabled, 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);
-      this.assistantOptimisticId = null;
-    }
-    if (hasUpcomingMessage(isRunning, vm)) {
-      this.assistantOptimisticId = this.repository.appendOptimisticMessage(
-        vm.at(-1)?.id ?? null,
-        {
-          role: "assistant",
-          content: []
-        }
-      );
-    }
-    this.repository.resetHead(
-      this.assistantOptimisticId ?? vm.at(-1)?.id ?? null
-    );
-    this.messages = this.repository.getMessages();
-    this.isDisabled = isDisabled;
-    this.isRunning = isRunning;
-    for (const callback of this._subscriptions) callback();
-  };
-  unstable_synchronizer = () => {
-    const { store } = this.useStore();
-    useExternalStoreSync(store, this.updateData);
-    return null;
-  };
   addToolResult(options) {
-    if (!this.store.onAddToolResult)
+    if (!this._store.onAddToolResult)
       throw new Error("Runtime does not support tool results.");
-    this.store.onAddToolResult(options);
+    this._store.onAddToolResult(options);
   }
 };
 
@@ -3140,9 +3139,6 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
   set store(store) {
     this.thread.store = store;
   }
-  onStoreUpdated() {
-    return this.thread.onStoreUpdated();
-  }
   getModelConfig() {
     return this._proxyConfigProvider.getModelConfig();
   }
@@ -3163,15 +3159,12 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
 };
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
-import { useEffect as useEffect12, useInsertionEffect as useInsertionEffect5, useState as useState10 } from "react";
+import { useInsertionEffect as useInsertionEffect4, useState as useState10 } from "react";
 var useExternalStoreRuntime = (store) => {
   const [runtime] = useState10(() => new ExternalStoreRuntime(store));
-  useInsertionEffect5(() => {
+  useInsertionEffect4(() => {
     runtime.store = store;
   });
-  useEffect12(() => {
-    runtime.onStoreUpdated();
-  });
   return runtime;
 };