@assistant-ui/core 0.2.22 → 0.2.23

package/src/runtime/api/message-part-runtime.ts

@@ -40,11 +40,18 @@ export class MessagePartRuntimeImpl implements MessagePartRuntime {
     return this.contentBinding.path;
   }
 
+  private contentBinding: MessagePartSnapshotBinding;
+  private messageApi: MessageStateBinding | undefined;
+  private threadApi: ThreadRuntimeCoreBinding | undefined;
+
   constructor(
-    private contentBinding: MessagePartSnapshotBinding,
-    private messageApi?: MessageStateBinding,
-    private threadApi?: ThreadRuntimeCoreBinding,
+    contentBinding: MessagePartSnapshotBinding,
+    messageApi?: MessageStateBinding,
+    threadApi?: ThreadRuntimeCoreBinding,
   ) {
+    this.contentBinding = contentBinding;
+    this.messageApi = messageApi;
+    this.threadApi = threadApi;
     this.__internal_bindMethods();
   }
 

package/src/runtime/api/message-runtime.ts

@@ -134,10 +134,15 @@ export class MessageRuntimeImpl implements MessageRuntime {
     return this._core.path;
   }
 
+  private _core: MessageStateBinding;
+  private _threadBinding: ThreadRuntimeCoreBinding;
+
   constructor(
-    private _core: MessageStateBinding,
-    private _threadBinding: ThreadRuntimeCoreBinding,
+    _core: MessageStateBinding,
+    _threadBinding: ThreadRuntimeCoreBinding,
   ) {
+    this._core = _core;
+    this._threadBinding = _threadBinding;
     this.composer = new EditComposerRuntimeImpl(
       new NestedSubscriptionSubject({
         path: {

package/src/runtime/api/thread-list-item-runtime.ts

@@ -66,10 +66,15 @@ export class ThreadListItemRuntimeImpl implements ThreadListItemRuntime {
     return this._core.path;
   }
 
+  private _core: ThreadListItemStateBinding;
+  private _threadListBinding: ThreadListRuntimeCoreBinding;
+
   constructor(
-    private _core: ThreadListItemStateBinding,
-    private _threadListBinding: ThreadListRuntimeCoreBinding,
+    _core: ThreadListItemStateBinding,
+    _threadListBinding: ThreadListRuntimeCoreBinding,
   ) {
+    this._core = _core;
+    this._threadListBinding = _threadListBinding;
     this.__internal_bindMethods();
   }
 

package/src/runtime/api/thread-list-runtime.ts

@@ -98,13 +98,21 @@ export type ThreadListRuntimeCoreBinding = ThreadListRuntimeCore;
 
 export class ThreadListRuntimeImpl implements ThreadListRuntime {
   private _getState;
+  private _core: ThreadListRuntimeCoreBinding;
+  private _runtimeFactory: new (
+    binding: ThreadRuntimeCoreBinding,
+    threadListItemBinding: ThreadListItemRuntimeBinding,
+  ) => ThreadRuntime;
+
   constructor(
-    private _core: ThreadListRuntimeCoreBinding,
-    private _runtimeFactory: new (
+    _core: ThreadListRuntimeCoreBinding,
+    _runtimeFactory: new (
       binding: ThreadRuntimeCoreBinding,
       threadListItemBinding: ThreadListItemRuntimeBinding,
     ) => ThreadRuntime = ThreadRuntimeImpl,
   ) {
+    this._core = _core;
+    this._runtimeFactory = _runtimeFactory;
     const stateBinding = new LazyMemoizeSubject({
       path: {},
       getState: () => getThreadListState(_core),

package/src/runtime/base/base-thread-runtime-core.ts

@@ -113,7 +113,11 @@ export abstract class BaseThreadRuntimeCore implements ThreadRuntimeCore {
 
   public readonly composer = new DefaultThreadComposerRuntimeCore(this);
 
-  constructor(private readonly _contextProvider: ModelContextProvider) {}
+  private readonly _contextProvider: ModelContextProvider;
+
+  constructor(_contextProvider: ModelContextProvider) {
+    this._contextProvider = _contextProvider;
+  }
 
   public getModelContext() {
     return this._contextProvider.getModelContext();

package/src/runtime/base/default-edit-composer-runtime-core.ts

@@ -31,8 +31,18 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
   private _nonTextPassthrough: readonly ThreadMessage["content"][number][];
   private _parentId: string | null;
   private _sourceId: string | null;
+  private runtime: ThreadRuntimeCore & {
+    adapters?:
+      | {
+          attachments?: AttachmentAdapter | undefined;
+          dictation?: DictationAdapter | undefined;
+        }
+      | undefined;
+  };
+  private endEditCallback: () => void;
+
   constructor(
-    private runtime: ThreadRuntimeCore & {
+    runtime: ThreadRuntimeCore & {
       adapters?:
         | {
             attachments?: AttachmentAdapter | undefined;
@@ -40,10 +50,12 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
           }
         | undefined;
     },
-    private endEditCallback: () => void,
+    endEditCallback: () => void,
     { parentId, message }: { parentId: string | null; message: ThreadMessage },
   ) {
     super();
+    this.runtime = runtime;
+    this.endEditCallback = endEditCallback;
     this._parentId = parentId;
     this._sourceId = message.id;
     this._previousText = getThreadMessageText(message);

package/src/runtime/base/default-thread-composer-runtime-core.ts

@@ -46,8 +46,17 @@ export class DefaultThreadComposerRuntimeCore
     return this.runtime.adapters?.dictation;
   }
 
+  private runtime: Omit<ThreadRuntimeCore, "composer"> & {
+    adapters?:
+      | {
+          attachments?: AttachmentAdapter | undefined;
+          dictation?: DictationAdapter | undefined;
+        }
+      | undefined;
+  };
+
   constructor(
-    private runtime: Omit<ThreadRuntimeCore, "composer"> & {
+    runtime: Omit<ThreadRuntimeCore, "composer"> & {
       adapters?:
         | {
             attachments?: AttachmentAdapter | undefined;
@@ -57,6 +66,7 @@ export class DefaultThreadComposerRuntimeCore
     },
   ) {
     super();
+    this.runtime = runtime;
     this.connect();
   }
 

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/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";