@assistant-ui/react 0.10.50 → 0.11.1

package/dist/legacy-runtime/runtime-cores/remote-thread-list/index.js

@@ -0,0 +1,10 @@
+// src/legacy-runtime/runtime-cores/remote-thread-list/index.ts
+import { useRemoteThreadListRuntime } from "./useRemoteThreadListRuntime.js";
+import { InMemoryThreadListAdapter } from "./adapter/in-memory.js";
+import { useCloudThreadListAdapter } from "./adapter/cloud.js";
+export {
+  InMemoryThreadListAdapter as unstable_InMemoryThreadListAdapter,
+  useCloudThreadListAdapter as unstable_useCloudThreadListAdapter,
+  useRemoteThreadListRuntime as unstable_useRemoteThreadListRuntime
+};
+//# sourceMappingURL=index.js.map

package/dist/legacy-runtime/runtime-cores/remote-thread-list/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/legacy-runtime/runtime-cores/remote-thread-list/index.ts"],"sourcesContent":["export { useRemoteThreadListRuntime as unstable_useRemoteThreadListRuntime } from \"./useRemoteThreadListRuntime\";\nexport type { RemoteThreadListAdapter as unstable_RemoteThreadListAdapter } from \"./types\";\n\nexport { InMemoryThreadListAdapter as unstable_InMemoryThreadListAdapter } from \"./adapter/in-memory\";\nexport { useCloudThreadListAdapter as unstable_useCloudThreadListAdapter } from \"./adapter/cloud\";\n"],"mappings":";AAAA,SAAuC,kCAA2C;AAGlF,SAAsC,iCAA0C;AAChF,SAAsC,iCAA0C;","names":[]}

package/dist/legacy-runtime/runtime-cores/remote-thread-list/types.d.ts

@@ -0,0 +1,32 @@
+import { ComponentType, PropsWithChildren } from "react";
+import { AssistantRuntime } from "../../runtime";
+import { AssistantStream } from "assistant-stream";
+import { ThreadMessage } from "../../../types";
+export type RemoteThreadInitializeResponse = {
+    remoteId: string;
+    externalId: string | undefined;
+};
+export type RemoteThreadMetadata = {
+    readonly status: "regular" | "archived";
+    readonly remoteId: string;
+    readonly externalId?: string | undefined;
+    readonly title?: string | undefined;
+};
+export type RemoteThreadListResponse = {
+    threads: RemoteThreadMetadata[];
+};
+export type RemoteThreadListAdapter = {
+    list(): Promise<RemoteThreadListResponse>;
+    rename(remoteId: string, newTitle: string): Promise<void>;
+    archive(remoteId: string): Promise<void>;
+    unarchive(remoteId: string): Promise<void>;
+    delete(remoteId: string): Promise<void>;
+    initialize(threadId: string): Promise<RemoteThreadInitializeResponse>;
+    generateTitle(remoteId: string, unstable_messages: readonly ThreadMessage[]): Promise<AssistantStream>;
+    unstable_Provider?: ComponentType<PropsWithChildren>;
+};
+export type RemoteThreadListOptions = {
+    runtimeHook: () => AssistantRuntime;
+    adapter: RemoteThreadListAdapter;
+};
+//# sourceMappingURL=types.d.ts.map

package/dist/legacy-runtime/runtime-cores/remote-thread-list/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/legacy-runtime/runtime-cores/remote-thread-list/types.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,MAAM,MAAM,8BAA8B,GAAG;IAC3C,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;CAChC,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,CAAC,MAAM,EAAE,SAAS,GAAG,UAAU,CAAC;IACxC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,OAAO,EAAE,oBAAoB,EAAE,CAAC;CACjC,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG;IACpC,IAAI,IAAI,OAAO,CAAC,wBAAwB,CAAC,CAAC;IAE1C,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,8BAA8B,CAAC,CAAC;IACtE,aAAa,CACX,QAAQ,EAAE,MAAM,EAChB,iBAAiB,EAAE,SAAS,aAAa,EAAE,GAC1C,OAAO,CAAC,eAAe,CAAC,CAAC;IAE5B,iBAAiB,CAAC,EAAE,aAAa,CAAC,iBAAiB,CAAC,CAAC;CACtD,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG;IACpC,WAAW,EAAE,MAAM,gBAAgB,CAAC;IACpC,OAAO,EAAE,uBAAuB,CAAC;CAClC,CAAC"}

package/dist/legacy-runtime/runtime-cores/remote-thread-list/types.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.d.ts

@@ -0,0 +1,4 @@
+import { RemoteThreadListOptions } from "./types";
+import { AssistantRuntimeImpl } from "../../../internal";
+export declare const useRemoteThreadListRuntime: (options: RemoteThreadListOptions) => AssistantRuntimeImpl;
+//# sourceMappingURL=useRemoteThreadListRuntime.d.ts.map

package/dist/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useRemoteThreadListRuntime.d.ts","sourceRoot":"","sources":["../../../../src/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.tsx"],"names":[],"mappings":"AAKA,OAAO,EAAE,uBAAuB,EAAE,MAAM,SAAS,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAsBzD,eAAO,MAAM,0BAA0B,GACrC,SAAS,uBAAuB,yBAQjC,CAAC"}

package/dist/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.js

@@ -0,0 +1,32 @@
+"use client";
+
+// src/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.tsx
+import { useState, useEffect, useMemo } from "react";
+import { BaseAssistantRuntimeCore } from "../core/BaseAssistantRuntimeCore.js";
+import { RemoteThreadListThreadListRuntimeCore } from "./RemoteThreadListThreadListRuntimeCore.js";
+import { AssistantRuntimeImpl } from "../../../internal.js";
+var RemoteThreadListRuntimeCore = class extends BaseAssistantRuntimeCore {
+  threads;
+  constructor(options) {
+    super();
+    this.threads = new RemoteThreadListThreadListRuntimeCore(
+      options,
+      this._contextProvider
+    );
+  }
+  get RenderComponent() {
+    return this.threads.__internal_RenderComponent;
+  }
+};
+var useRemoteThreadListRuntime = (options) => {
+  const [runtime] = useState(() => new RemoteThreadListRuntimeCore(options));
+  useEffect(() => {
+    runtime.threads.__internal_setOptions(options);
+    runtime.threads.__internal_load();
+  }, [runtime, options]);
+  return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);
+};
+export {
+  useRemoteThreadListRuntime
+};
+//# sourceMappingURL=useRemoteThreadListRuntime.js.map

package/dist/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/legacy-runtime/runtime-cores/remote-thread-list/useRemoteThreadListRuntime.tsx"],"sourcesContent":["\"use client\";\n\nimport { useState, useEffect, useMemo } from \"react\";\nimport { BaseAssistantRuntimeCore } from \"../core/BaseAssistantRuntimeCore\";\nimport { RemoteThreadListThreadListRuntimeCore } from \"./RemoteThreadListThreadListRuntimeCore\";\nimport { RemoteThreadListOptions } from \"./types\";\nimport { AssistantRuntimeImpl } from \"../../../internal\";\nimport { AssistantRuntimeCore } from \"../core/AssistantRuntimeCore\";\n\nclass RemoteThreadListRuntimeCore\n  extends BaseAssistantRuntimeCore\n  implements AssistantRuntimeCore\n{\n  public readonly threads;\n\n  constructor(options: RemoteThreadListOptions) {\n    super();\n    this.threads = new RemoteThreadListThreadListRuntimeCore(\n      options,\n      this._contextProvider,\n    );\n  }\n\n  public get RenderComponent() {\n    return this.threads.__internal_RenderComponent;\n  }\n}\n\nexport const useRemoteThreadListRuntime = (\n  options: RemoteThreadListOptions,\n) => {\n  const [runtime] = useState(() => new RemoteThreadListRuntimeCore(options));\n  useEffect(() => {\n    runtime.threads.__internal_setOptions(options);\n    runtime.threads.__internal_load();\n  }, [runtime, options]);\n  return useMemo(() => new AssistantRuntimeImpl(runtime), [runtime]);\n};\n"],"mappings":";;;AAEA,SAAS,UAAU,WAAW,eAAe;AAC7C,SAAS,gCAAgC;AACzC,SAAS,6CAA6C;AAEtD,SAAS,4BAA4B;AAGrC,IAAM,8BAAN,cACU,yBAEV;AAAA,EACkB;AAAA,EAEhB,YAAY,SAAkC;AAC5C,UAAM;AACN,SAAK,UAAU,IAAI;AAAA,MACjB;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAEA,IAAW,kBAAkB;AAC3B,WAAO,KAAK,QAAQ;AAAA,EACtB;AACF;AAEO,IAAM,6BAA6B,CACxC,YACG;AACH,QAAM,CAAC,OAAO,IAAI,SAAS,MAAM,IAAI,4BAA4B,OAAO,CAAC;AACzE,YAAU,MAAM;AACd,YAAQ,QAAQ,sBAAsB,OAAO;AAC7C,YAAQ,QAAQ,gBAAgB;AAAA,EAClC,GAAG,CAAC,SAAS,OAAO,CAAC;AACrB,SAAO,QAAQ,MAAM,IAAI,qBAAqB,OAAO,GAAG,CAAC,OAAO,CAAC;AACnE;","names":[]}

package/dist/legacy-runtime/runtime-cores/utils/MessageRepository.d.ts

@@ -0,0 +1,151 @@
+import type { ThreadMessage } from "../../../types";
+import { ThreadMessageLike } from "../external-store";
+/**
+ * Represents a message item that can be exported from the repository.
+ */
+export type ExportedMessageRepositoryItem = {
+    /** The message data */
+    message: ThreadMessage;
+    /** ID of the parent message, or null for root messages */
+    parentId: string | null;
+};
+/**
+ * Represents the entire repository state for export/import.
+ */
+export type ExportedMessageRepository = {
+    /** ID of the head message, or null/undefined if no head */
+    headId?: string | null;
+    /** Array of all messages with their parent references */
+    messages: Array<{
+        message: ThreadMessage;
+        parentId: string | null;
+    }>;
+};
+/**
+ * Utility functions for working with exported message repositories.
+ */
+export declare const ExportedMessageRepository: {
+    /**
+     * Converts an array of messages to an ExportedMessageRepository format.
+     * Creates parent-child relationships based on the order of messages in the array.
+     *
+     * @param messages - Array of message-like objects to convert
+     * @returns ExportedMessageRepository with parent-child relationships established
+     */
+    fromArray: (messages: readonly ThreadMessageLike[]) => ExportedMessageRepository;
+};
+/**
+ * A repository that manages a tree of messages with branching capabilities.
+ * Supports operations like adding, updating, and deleting messages, as well as
+ * managing multiple conversation branches.
+ */
+export declare class MessageRepository {
+    /** Map of message IDs to repository message objects */
+    private messages;
+    /** Reference to the current head (most recent) message in the active branch */
+    private head;
+    /** Root node of the tree structure */
+    private root;
+    /**
+     * Performs link/unlink operations between messages in the tree.
+     *
+     * @param newParent - The new parent message, or null
+     * @param child - The child message to operate on
+     * @param operation - The type of operation to perform:
+     *   - "cut": Remove the child from its current parent
+     *   - "link": Add the child to a new parent
+     *   - "relink": Both cut and link operations
+     */
+    private performOp;
+    /** Cached array of messages in the current active branch, from root to head */
+    private _messages;
+    /**
+     * Gets the ID of the current head message.
+     * @returns The ID of the head message, or null if no messages exist
+     */
+    get headId(): string | null;
+    /**
+     * Gets all messages in the current active branch, from root to head.
+     * @returns Array of messages in the current branch
+     */
+    getMessages(): readonly ThreadMessage[];
+    /**
+     * Adds a new message or updates an existing one in the repository.
+     * If the message ID already exists, the message is updated and potentially relinked to a new parent.
+     * If the message is new, it's added as a child of the specified parent.
+     *
+     * @param parentId - ID of the parent message, or null for root messages
+     * @param message - The message to add or update
+     * @throws Error if the parent message is not found
+     */
+    addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
+    /**
+     * Gets a message and its parent ID by message ID.
+     *
+     * @param messageId - ID of the message to retrieve
+     * @returns Object containing the message and its parent ID
+     * @throws Error if the message is not found
+     */
+    getMessage(messageId: string): {
+        parentId: string | null;
+        message: ThreadMessage;
+    };
+    /**
+     * Adds an optimistic message to the repository.
+     * An optimistic message is a temporary placeholder that will be replaced by a real message later.
+     *
+     * @param parentId - ID of the parent message, or null for root messages
+     * @param message - The core message to convert to an optimistic message
+     * @returns The generated optimistic ID
+     */
+    appendOptimisticMessage(parentId: string | null, message: ThreadMessageLike): string;
+    /**
+     * Deletes a message from the repository and relinks its children.
+     *
+     * @param messageId - ID of the message to delete
+     * @param replacementId - Optional ID of the message to become the new parent of the children,
+     *                       undefined means use the deleted message's parent,
+     *                       null means use the root
+     * @throws Error if the message or replacement is not found
+     */
+    deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
+    /**
+     * Gets all branch IDs (sibling messages) at the level of a specified message.
+     *
+     * @param messageId - ID of the message to find branches for
+     * @returns Array of message IDs representing branches
+     * @throws Error if the message is not found
+     */
+    getBranches(messageId: string): string[];
+    /**
+     * Switches the active branch to the one containing the specified message.
+     *
+     * @param messageId - ID of the message in the branch to switch to
+     * @throws Error if the branch is not found
+     */
+    switchToBranch(messageId: string): void;
+    /**
+     * Resets the head to a specific message or null.
+     *
+     * @param messageId - ID of the message to set as head, or null to clear the head
+     * @throws Error if the message is not found
+     */
+    resetHead(messageId: string | null): void;
+    /**
+     * Clears all messages from the repository.
+     */
+    clear(): void;
+    /**
+     * Exports the repository state for persistence.
+     *
+     * @returns Exportable repository state
+     */
+    export(): ExportedMessageRepository;
+    /**
+     * Imports repository state from an exported repository.
+     *
+     * @param repository - The exported repository state to import
+     */
+    import({ headId, messages }: ExportedMessageRepository): void;
+}
+//# sourceMappingURL=MessageRepository.d.ts.map

package/dist/legacy-runtime/runtime-cores/utils/MessageRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MessageRepository.d.ts","sourceRoot":"","sources":["../../../../src/legacy-runtime/runtime-cores/utils/MessageRepository.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAEpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AA0BtD;;GAEG;AACH,MAAM,MAAM,6BAA6B,GAAG;IAC1C,uBAAuB;IACvB,OAAO,EAAE,aAAa,CAAC;IACvB,0DAA0D;IAC1D,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,yBAAyB,GAAG;IACtC,2DAA2D;IAC3D,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,yDAAyD;IACzD,QAAQ,EAAE,KAAK,CAAC;QACd,OAAO,EAAE,aAAa,CAAC;QACvB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;KACzB,CAAC,CAAC;CACJ,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,yBAAyB;IACpC;;;;;;OAMG;0BAES,SAAS,iBAAiB,EAAE,KACrC,yBAAyB;CAgB7B,CAAC;AA6CF;;;;GAIG;AACH,qBAAa,iBAAiB;IAC5B,uDAAuD;IACvD,OAAO,CAAC,QAAQ,CAAwC;IACxD,+EAA+E;IAC/E,OAAO,CAAC,IAAI,CAAkC;IAC9C,sCAAsC;IACtC,OAAO,CAAC,IAAI,CAGV;IAEF;;;;;;;;;OASG;IACH,OAAO,CAAC,SAAS;IA4DjB,+EAA+E;IAC/E,OAAO,CAAC,SAAS,CAMd;IAEH;;;OAGG;IACH,IAAI,MAAM,kBAET;IAED;;;OAGG;IACH,WAAW;IAIX;;;;;;;;OAQG;IACH,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,EAAE,aAAa;IAmClE;;;;;;OAMG;IACH,UAAU,CAAC,SAAS,EAAE,MAAM;;;;IAa5B;;;;;;;OAOG;IACH,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,EAAE,iBAAiB;IAc3E;;;;;;;;OAQG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS;IAsC1E;;;;;;OAMG;IACH,WAAW,CAAC,SAAS,EAAE,MAAM;IAW7B;;;;;OAKG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM;IAehC;;;;;OAKG;IACH,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IA0ClC;;OAEG;IACH,KAAK,IAAI,IAAI;IAUb;;;;OAIG;IACH,MAAM,IAAI,yBAAyB;IAkBnC;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,yBAAyB;CAQvD"}

package/dist/legacy-runtime/runtime-cores/utils/MessageRepository.js

@@ -0,0 +1,362 @@
+// src/legacy-runtime/runtime-cores/utils/MessageRepository.tsx
+import { generateId, generateOptimisticId } from "../../../utils/idUtils.js";
+import { getAutoStatus } from "../external-store/auto-status.js";
+import { fromThreadMessageLike } from "../external-store/ThreadMessageLike.js";
+var ExportedMessageRepository = {
+  /**
+   * Converts an array of messages to an ExportedMessageRepository format.
+   * Creates parent-child relationships based on the order of messages in the array.
+   *
+   * @param messages - Array of message-like objects to convert
+   * @returns ExportedMessageRepository with parent-child relationships established
+   */
+  fromArray: (messages) => {
+    const conv = messages.map(
+      (m) => fromThreadMessageLike(
+        m,
+        generateId(),
+        getAutoStatus(false, false, false)
+      )
+    );
+    return {
+      messages: conv.map((m, idx) => ({
+        parentId: idx > 0 ? conv[idx - 1].id : null,
+        message: m
+      }))
+    };
+  }
+};
+var findHead = (message) => {
+  if (message.next) return findHead(message.next);
+  if ("current" in message) return message;
+  return null;
+};
+var CachedValue = class {
+  /**
+   * @param func - The function that computes the cached value
+   */
+  constructor(func) {
+    this.func = func;
+  }
+  _value = null;
+  /**
+   * Gets the cached value, computing it if necessary.
+   */
+  get value() {
+    if (this._value === null) {
+      this._value = this.func();
+    }
+    return this._value;
+  }
+  /**
+   * Invalidates the cache, forcing recomputation on next access.
+   */
+  dirty() {
+    this._value = null;
+  }
+};
+var MessageRepository = class {
+  /** Map of message IDs to repository message objects */
+  messages = /* @__PURE__ */ new Map();
+  /** Reference to the current head (most recent) message in the active branch */
+  head = null;
+  /** Root node of the tree structure */
+  root = {
+    children: [],
+    next: null
+  };
+  /**
+   * Performs link/unlink operations between messages in the tree.
+   *
+   * @param newParent - The new parent message, or null
+   * @param child - The child message to operate on
+   * @param operation - The type of operation to perform:
+   *   - "cut": Remove the child from its current parent
+   *   - "link": Add the child to a new parent
+   *   - "relink": Both cut and link operations
+   */
+  performOp(newParent, child, operation) {
+    const parentOrRoot = child.prev ?? this.root;
+    const newParentOrRoot = newParent ?? this.root;
+    if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
+    if (operation !== "link") {
+      parentOrRoot.children = parentOrRoot.children.filter(
+        (m) => m !== child.current.id
+      );
+      if (parentOrRoot.next === child) {
+        const fallbackId = parentOrRoot.children.at(-1);
+        const fallback = fallbackId ? this.messages.get(fallbackId) : null;
+        if (fallback === void 0) {
+          throw new Error(
+            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
+          );
+        }
+        parentOrRoot.next = fallback;
+      }
+    }
+    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
+      ];
+      if (findHead(child) === this.head || newParentOrRoot.next === null) {
+        newParentOrRoot.next = child;
+      }
+      child.prev = newParent;
+    }
+  }
+  /** Cached array of messages in the current active branch, from root to head */
+  _messages = new CachedValue(() => {
+    const messages = new Array(this.head?.level ?? 0);
+    for (let current = this.head; current; current = current.prev) {
+      messages[current.level] = current.current;
+    }
+    return messages;
+  });
+  /**
+   * Gets the ID of the current head message.
+   * @returns The ID of the head message, or null if no messages exist
+   */
+  get headId() {
+    return this.head?.current.id ?? null;
+  }
+  /**
+   * Gets all messages in the current active branch, from root to head.
+   * @returns Array of messages in the current branch
+   */
+  getMessages() {
+    return this._messages.value;
+  }
+  /**
+   * Adds a new message or updates an existing one in the repository.
+   * If the message ID already exists, the message is updated and potentially relinked to a new parent.
+   * If the message is new, it's added as a child of the specified parent.
+   *
+   * @param parentId - ID of the parent message, or null for root messages
+   * @param message - The message to add or update
+   * @throws Error if the parent message is not found
+   */
+  addOrUpdateMessage(parentId, message) {
+    const existingItem = this.messages.get(message.id);
+    const prev = parentId ? this.messages.get(parentId) : null;
+    if (prev === void 0)
+      throw new Error(
+        "MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
+      );
+    if (existingItem) {
+      existingItem.current = message;
+      this.performOp(prev, existingItem, "relink");
+      this._messages.dirty();
+      return;
+    }
+    const newItem = {
+      prev,
+      current: message,
+      next: null,
+      children: [],
+      level: prev ? prev.level + 1 : 0
+    };
+    this.messages.set(message.id, newItem);
+    this.performOp(prev, newItem, "link");
+    if (this.head === prev) {
+      this.head = newItem;
+    }
+    this._messages.dirty();
+  }
+  /**
+   * Gets a message and its parent ID by message ID.
+   *
+   * @param messageId - ID of the message to retrieve
+   * @returns Object containing the message and its parent ID
+   * @throws Error if the message is not found
+   */
+  getMessage(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui."
+      );
+    return {
+      parentId: message.prev?.current.id ?? null,
+      message: message.current
+    };
+  }
+  /**
+   * Adds an optimistic message to the repository.
+   * An optimistic message is a temporary placeholder that will be replaced by a real message later.
+   *
+   * @param parentId - ID of the parent message, or null for root messages
+   * @param message - The core message to convert to an optimistic message
+   * @returns The generated optimistic ID
+   */
+  appendOptimisticMessage(parentId, message) {
+    let optimisticId;
+    do {
+      optimisticId = generateOptimisticId();
+    } while (this.messages.has(optimisticId));
+    this.addOrUpdateMessage(
+      parentId,
+      fromThreadMessageLike(message, optimisticId, { type: "running" })
+    );
+    return optimisticId;
+  }
+  /**
+   * Deletes a message from the repository and relinks its children.
+   *
+   * @param messageId - ID of the message to delete
+   * @param replacementId - Optional ID of the message to become the new parent of the children,
+   *                       undefined means use the deleted message's parent,
+   *                       null means use the root
+   * @throws Error if the message or replacement is not found
+   */
+  deleteMessage(messageId, replacementId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui."
+      );
+    const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
+    if (replacement === void 0)
+      throw new Error(
+        "MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
+      );
+    for (const child of message.children) {
+      const childMessage = this.messages.get(child);
+      if (!childMessage)
+        throw new Error(
+          "MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
+        );
+      this.performOp(replacement, childMessage, "relink");
+    }
+    this.performOp(null, message, "cut");
+    this.messages.delete(messageId);
+    if (this.head === message) {
+      this.head = findHead(replacement ?? this.root);
+    }
+    this._messages.dirty();
+  }
+  /**
+   * Gets all branch IDs (sibling messages) at the level of a specified message.
+   *
+   * @param messageId - ID of the message to find branches for
+   * @returns Array of message IDs representing branches
+   * @throws Error if the message is not found
+   */
+  getBranches(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
+      );
+    const { children } = message.prev ?? this.root;
+    return children;
+  }
+  /**
+   * Switches the active branch to the one containing the specified message.
+   *
+   * @param messageId - ID of the message in the branch to switch to
+   * @throws Error if the branch is not found
+   */
+  switchToBranch(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
+      );
+    const prevOrRoot = message.prev ?? this.root;
+    prevOrRoot.next = message;
+    this.head = findHead(message);
+    this._messages.dirty();
+  }
+  /**
+   * Resets the head to a specific message or null.
+   *
+   * @param messageId - ID of the message to set as head, or null to clear the head
+   * @throws Error if the message is not found
+   */
+  resetHead(messageId) {
+    if (messageId === null) {
+      this.clear();
+      return;
+    }
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
+      );
+    if (message.children.length > 0) {
+      const deleteDescendants = (msg) => {
+        for (const childId of msg.children) {
+          const childMessage = this.messages.get(childId);
+          if (childMessage) {
+            deleteDescendants(childMessage);
+            this.messages.delete(childId);
+          }
+        }
+      };
+      deleteDescendants(message);
+      message.children = [];
+      message.next = null;
+    }
+    this.head = message;
+    for (let current = message; current; current = current.prev) {
+      if (current.prev) {
+        current.prev.next = current;
+      }
+    }
+    this._messages.dirty();
+  }
+  /**
+   * Clears all messages from the repository.
+   */
+  clear() {
+    this.messages.clear();
+    this.head = null;
+    this.root = {
+      children: [],
+      next: null
+    };
+    this._messages.dirty();
+  }
+  /**
+   * Exports the repository state for persistence.
+   *
+   * @returns Exportable repository state
+   */
+  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
+    };
+  }
+  /**
+   * Imports repository state from an exported repository.
+   *
+   * @param repository - The exported repository state to import
+   */
+  import({ headId, messages }) {
+    for (const { message, parentId } of messages) {
+      this.addOrUpdateMessage(parentId, message);
+    }
+    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);
+  }
+};
+export {
+  ExportedMessageRepository,
+  MessageRepository
+};
+//# sourceMappingURL=MessageRepository.js.map