@assistant-ui/core 0.2.21 → 0.2.23

package/src/runtime/utils/message-repository.ts

@@ -89,7 +89,11 @@ const findHead = (
 class CachedValue<T> {
   private _value: T | null = null;
 
-  constructor(private func: () => T) {}
+  private func: () => T;
+
+  constructor(func: () => T) {
+    this.func = func;
+  }
 
   get value() {
     if (this._value === null) {
@@ -131,6 +135,20 @@ export class MessageRepository {
 
     if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
 
+    if (operation !== "cut") {
+      for (
+        let current: RepositoryMessage | null = 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.",
+          );
+        }
+      }
+    }
+
     if (operation !== "link") {
       parentOrRoot.children = parentOrRoot.children.filter(
         (m) => m !== child.current.id,
@@ -149,18 +167,6 @@ export class MessageRepository {
     }
 
     if (operation !== "cut") {
-      for (
-        let current: RepositoryMessage | null = 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,

package/src/runtime/utils/thread-message-like.ts

@@ -266,6 +266,7 @@ export const fromThreadMessageLike = (
         })),
         metadata: {
           custom: metadata?.custom ?? {},
+          ...(metadata?.isOptimistic && { isOptimistic: true }),
         },
       } satisfies ThreadUserMessage;
 

package/src/runtimes/external-store/external-store-thread-list-runtime-core.ts

@@ -60,10 +60,13 @@ export class ExternalStoreThreadListRuntimeCore implements ThreadListRuntimeCore
     return this._mainThreadId;
   }
 
+  private threadFactory: ExternalStoreThreadFactory;
+
   constructor(
     adapter: ExternalStoreThreadListAdapter = {},
-    private threadFactory: ExternalStoreThreadFactory,
+    threadFactory: ExternalStoreThreadFactory,
   ) {
+    this.threadFactory = threadFactory;
     this.__internal_setAdapter(adapter, true);
   }
 

package/src/runtimes/external-store/external-store-thread-runtime-core.ts

@@ -251,6 +251,21 @@ export class ExternalStoreThreadRuntimeCore
             return newMessage;
           });
 
+      const seenIds = new Set<string>();
+      const deduped: ThreadMessage[] = [];
+      for (let i = messages.length - 1; i >= 0; i--) {
+        const message = messages[i]!;
+        if (seenIds.has(message.id)) {
+          console.warn(
+            `ExternalStoreThreadRuntimeCore: duplicate message id "${message.id}" in the provided messages array; keeping the last occurrence.`,
+          );
+          continue;
+        }
+        seenIds.add(message.id);
+        deduped.push(message);
+      }
+      if (deduped.length !== messages.length) messages = deduped.reverse();
+
       for (let i = 0; i < messages.length; i++) {
         const message = messages[i]!;
         const parent = messages[i - 1];

package/src/runtimes/internal.ts

@@ -37,3 +37,6 @@ export type {
   RemoteThreadInitializeResponse,
   RemoteThreadListOptions,
 } from "./remote-thread-list/types";
+
+// Tool Invocations
+export { ToolInvocationTracker } from "./tool-invocations/ToolInvocationTracker";