@assistant-ui/core 0.2.4 → 0.2.5

package/dist/runtime/utils/message-repository.js

@@ -1,274 +1,220 @@
 import { generateId, generateOptimisticId } from "../../utils/id.js";
 import { getAutoStatus } from "./auto-status.js";
 import { fromThreadMessageLike } from "./thread-message-like.js";
-export const ExportedMessageRepository = {
-    fromArray: (messages) => {
-        const conv = messages.map((m) => fromThreadMessageLike(m, generateId(), getAutoStatus(false, false, false, false, undefined)));
-        return {
-            messages: conv.map((m, idx) => ({
-                parentId: idx > 0 ? conv[idx - 1].id : null,
-                message: m,
-            })),
-        };
-    },
-    fromBranchableArray: (items, options) => {
-        const fallbackStatus = getAutoStatus(false, false, false, false, undefined);
-        return {
-            ...(options?.headId !== undefined
-                ? { headId: options.headId }
-                : undefined),
-            messages: items.map(({ message, parentId }) => {
-                if (!message.id) {
-                    throw new Error("ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.");
-                }
-                return {
-                    parentId,
-                    message: fromThreadMessageLike(message, message.id, fallbackStatus),
-                };
-            }),
-        };
-    },
+//#region src/runtime/utils/message-repository.ts
+const ExportedMessageRepository = {
+	fromArray: (messages) => {
+		const conv = messages.map((m) => fromThreadMessageLike(m, generateId(), getAutoStatus(false, false, false, false, void 0)));
+		return { messages: conv.map((m, idx) => ({
+			parentId: idx > 0 ? conv[idx - 1].id : null,
+			message: m
+		})) };
+	},
+	fromBranchableArray: (items, options) => {
+		const fallbackStatus = getAutoStatus(false, false, false, false, void 0);
+		return {
+			...options?.headId !== void 0 ? { headId: options.headId } : void 0,
+			messages: items.map(({ message, parentId }) => {
+				if (!message.id) throw new Error("ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.");
+				return {
+					parentId,
+					message: fromThreadMessageLike(message, message.id, fallbackStatus)
+				};
+			})
+		};
+	}
 };
 const findHead = (message) => {
-    if (message.next)
-        return findHead(message.next);
-    if ("current" in message)
-        return message;
-    return null;
+	if (message.next) return findHead(message.next);
+	if ("current" in message) return message;
+	return null;
 };
-class CachedValue {
-    func;
-    _value = null;
-    constructor(func) {
-        this.func = func;
-    }
-    get value() {
-        if (this._value === null) {
-            this._value = this.func();
-        }
-        return this._value;
-    }
-    dirty() {
-        this._value = null;
-    }
-}
-export class MessageRepository {
-    messages = new Map();
-    head = null;
-    root = {
-        children: [],
-        next: null,
-    };
-    updateLevels(message, newLevel) {
-        message.level = newLevel;
-        for (const childId of message.children) {
-            const childMessage = this.messages.get(childId);
-            if (childMessage) {
-                this.updateLevels(childMessage, newLevel + 1);
-            }
-        }
-    }
-    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 === undefined) {
-                    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;
-            const newLevel = newParent ? newParent.level + 1 : 0;
-            this.updateLevels(child, newLevel);
-        }
-    }
-    _messages = new CachedValue(() => {
-        const messages = new Array((this.head?.level ?? -1) + 1);
-        for (let current = this.head; current; current = current.prev) {
-            messages[current.level] = current.current;
-        }
-        return messages;
-    });
-    get headId() {
-        return this.head?.current.id ?? null;
-    }
-    getMessages(headId) {
-        if (headId === undefined || headId === this.head?.current.id) {
-            return this._messages.value;
-        }
-        const headMessage = this.messages.get(headId);
-        if (!headMessage) {
-            throw new Error("MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.");
-        }
-        const messages = new Array(headMessage.level + 1);
-        for (let current = headMessage; current; current = current.prev) {
-            messages[current.level] = current.current;
-        }
-        return messages;
-    }
-    addOrUpdateMessage(parentId, message) {
-        const existingItem = this.messages.get(message.id);
-        const prev = parentId ? this.messages.get(parentId) : null;
-        if (prev === undefined)
-            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();
-    }
-    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,
-            index: message.level,
-        };
-    }
-    appendOptimisticMessage(parentId, message) {
-        let optimisticId;
-        do {
-            optimisticId = generateOptimisticId();
-        } while (this.messages.has(optimisticId));
-        this.addOrUpdateMessage(parentId, fromThreadMessageLike(message, optimisticId, { type: "running" }));
-        return optimisticId;
-    }
-    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 === undefined
-            ? message.prev
-            : replacementId === null
-                ? null
-                : this.messages.get(replacementId);
-        if (replacement === undefined)
-            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();
-    }
-    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;
-    }
-    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();
-    }
-    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;
-            }
-            else {
-                this.root.next = current;
-            }
-        }
-        this._messages.dirty();
-    }
-    clear() {
-        this.messages.clear();
-        this.head = null;
-        this.root = {
-            children: [],
-            next: null,
-        };
-        this._messages.dirty();
-    }
-    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);
-    }
-}
+var CachedValue = class {
+	func;
+	_value = null;
+	constructor(func) {
+		this.func = func;
+	}
+	get value() {
+		if (this._value === null) this._value = this.func();
+		return this._value;
+	}
+	dirty() {
+		this._value = null;
+	}
+};
+var MessageRepository = class {
+	messages = /* @__PURE__ */ new Map();
+	head = null;
+	root = {
+		children: [],
+		next: null
+	};
+	updateLevels(message, newLevel) {
+		message.level = newLevel;
+		for (const childId of message.children) {
+			const childMessage = this.messages.get(childId);
+			if (childMessage) this.updateLevels(childMessage, newLevel + 1);
+		}
+	}
+	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;
+			const newLevel = newParent ? newParent.level + 1 : 0;
+			this.updateLevels(child, newLevel);
+		}
+	}
+	_messages = new CachedValue(() => {
+		const messages = new Array((this.head?.level ?? -1) + 1);
+		for (let current = this.head; current; current = current.prev) messages[current.level] = current.current;
+		return messages;
+	});
+	get headId() {
+		return this.head?.current.id ?? null;
+	}
+	getMessages(headId) {
+		if (headId === void 0 || headId === this.head?.current.id) return this._messages.value;
+		const headMessage = this.messages.get(headId);
+		if (!headMessage) throw new Error("MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.");
+		const messages = new Array(headMessage.level + 1);
+		for (let current = headMessage; current; current = current.prev) messages[current.level] = current.current;
+		return messages;
+	}
+	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();
+	}
+	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,
+			index: message.level
+		};
+	}
+	appendOptimisticMessage(parentId, message) {
+		let optimisticId;
+		do
+			optimisticId = generateOptimisticId();
+		while (this.messages.has(optimisticId));
+		this.addOrUpdateMessage(parentId, fromThreadMessageLike(message, optimisticId, { type: "running" }));
+		return optimisticId;
+	}
+	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();
+	}
+	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;
+	}
+	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();
+	}
+	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;
+		else this.root.next = current;
+		this._messages.dirty();
+	}
+	clear() {
+		this.messages.clear();
+		this.head = null;
+		this.root = {
+			children: [],
+			next: null
+		};
+		this._messages.dirty();
+	}
+	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);
+	}
+};
+//#endregion
+export { ExportedMessageRepository, MessageRepository };
+
 //# sourceMappingURL=message-repository.js.map

package/dist/runtime/utils/message-repository.js.map

@@ -1 +1 @@
-{"version":3,"file":"message-repository.js","sourceRoot":"","sources":["../../../src/runtime/utils/message-repository.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,oBAAoB,EAAE,0BAAuB;AAElE,OAAO,EAAE,aAAa,EAAE,yBAAsB;AAC9C,OAAO,EAAE,qBAAqB,EAAE,iCAA8B;AAiB9D,MAAM,CAAC,MAAM,yBAAyB,GAAG;IACvC,SAAS,EAAE,CACT,QAAsC,EACX,EAAE;QAC7B,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC9B,qBAAqB,CACnB,CAAC,EACD,UAAU,EAAE,EACZ,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,CACrD,CACF,CAAC;QAEF,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC9B,QAAQ,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI;gBAC5C,OAAO,EAAE,CAAC;aACX,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,mBAAmB,EAAE,CACnB,KAGG,EACH,OAAoC,EACT,EAAE;QAC7B,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;QAC5E,OAAO;YACL,GAAG,CAAC,OAAO,EAAE,MAAM,KAAK,SAAS;gBAC/B,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;gBAC5B,CAAC,CAAC,SAAS,CAAC;YACd,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE;gBAC5C,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;oBAChB,MAAM,IAAI,KAAK,CACb,0FAA0F,CAC3F,CAAC;gBACJ,CAAC;gBACD,OAAO;oBACL,QAAQ;oBACR,OAAO,EAAE,qBAAqB,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE,cAAc,CAAC;iBACpE,CAAC;YACJ,CAAC,CAAC;SACH,CAAC;IACJ,CAAC;CACF,CAAC;AAaF,MAAM,QAAQ,GAAG,CACf,OAA6C,EACnB,EAAE;IAC5B,IAAI,OAAO,CAAC,IAAI;QAAE,OAAO,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,IAAI,SAAS,IAAI,OAAO;QAAE,OAAO,OAAO,CAAC;IACzC,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF,MAAM,WAAW;IAGK;IAFZ,MAAM,GAAa,IAAI,CAAC;IAEhC,YAAoB,IAAa;QAAb,SAAI,GAAJ,IAAI,CAAS;IAAG,CAAC;IAErC,IAAI,KAAK;QACP,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;CACF;AAED,MAAM,OAAO,iBAAiB;IACpB,QAAQ,GAAG,IAAI,GAAG,EAA6B,CAAC;IAChD,IAAI,GAA6B,IAAI,CAAC;IACtC,IAAI,GAAqB;QAC/B,QAAQ,EAAE,EAAE;QACZ,IAAI,EAAE,IAAI;KACX,CAAC;IAEM,YAAY,CAAC,OAA0B,EAAE,QAAgB;QAC/D,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC;QACzB,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvC,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAChD,IAAI,YAAY,EAAE,CAAC;gBACjB,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CACf,SAAmC,EACnC,KAAwB,EACxB,SAAoC;QAEpC,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QAC7C,MAAM,eAAe,GAAG,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC;QAE/C,IAAI,SAAS,KAAK,QAAQ,IAAI,YAAY,KAAK,eAAe;YAAE,OAAO;QAEvE,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;YACzB,YAAY,CAAC,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC,MAAM,CAClD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,KAAK,CAAC,OAAO,CAAC,EAAE,CAC9B,CAAC;YAEF,IAAI,YAAY,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;gBAChC,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAChD,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBACnE,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAC3B,MAAM,IAAI,KAAK,CACb,uHAAuH,CACxH,CAAC;gBACJ,CAAC;gBACD,YAAY,CAAC,IAAI,GAAG,QAAQ,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;YACxB,KACE,IAAI,OAAO,GAA6B,SAAS,EACjD,OAAO,EACP,OAAO,GAAG,OAAO,CAAC,IAAI,EACtB,CAAC;gBACD,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;oBAC5C,MAAM,IAAI,KAAK,CACb,oNAAoN,CACrN,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,eAAe,CAAC,QAAQ,GAAG;gBACzB,GAAG,eAAe,CAAC,QAAQ;gBAC3B,KAAK,CAAC,OAAO,CAAC,EAAE;aACjB,CAAC;YAEF,IAAI,QAAQ,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,IAAI,IAAI,eAAe,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBACnE,eAAe,CAAC,IAAI,GAAG,KAAK,CAAC;YAC/B,CAAC;YAED,KAAK,CAAC,IAAI,GAAG,SAAS,CAAC;YAEvB,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACrD,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,SAAS,GAAG,IAAI,WAAW,CAA2B,GAAG,EAAE;QACjE,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACxE,KAAK,IAAI,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;YAC9D,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC;IACvC,CAAC;IAED,WAAW,CAAC,MAAe;QACzB,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC;YAC7D,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;QAC9B,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,yGAAyG,CAC1G,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAgB,WAAW,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACjE,KACE,IAAI,OAAO,GAA6B,WAAW,EACnD,OAAO,EACP,OAAO,GAAG,OAAO,CAAC,IAAI,EACtB,CAAC;YACD,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,kBAAkB,CAAC,QAAuB,EAAE,OAAsB;QAChE,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3D,IAAI,IAAI,KAAK,SAAS;YACpB,MAAM,IAAI,KAAK,CACb,kHAAkH,CACnH,CAAC;QAEJ,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,OAAO,GAAG,OAAO,CAAC;YAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;YAC7C,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAsB;YACjC,IAAI;YACJ,OAAO,EAAE,OAAO;YAChB,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,EAAE;YACZ,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;SACjC,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACvC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAEtC,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC;QACtB,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,UAAU,CAAC,SAAiB;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YACV,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;QAEJ,OAAO;YACL,QAAQ,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI;YAC1C,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,QAAuB,EAAE,OAA0B;QACzE,IAAI,YAAoB,CAAC;QACzB,GAAG,CAAC;YACF,YAAY,GAAG,oBAAoB,EAAE,CAAC;QACxC,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE;QAE1C,IAAI,CAAC,kBAAkB,CACrB,QAAQ,EACR,qBAAqB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAClE,CAAC;QAEF,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,aAAyC;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAE7C,IAAI,CAAC,OAAO;YACV,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;QAEJ,MAAM,WAAW,GACf,aAAa,KAAK,SAAS;YACzB,CAAC,CAAC,OAAO,CAAC,IAAI;YACd,CAAC,CAAC,aAAa,KAAK,IAAI;gBACtB,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACzC,IAAI,WAAW,KAAK,SAAS;YAC3B,MAAM,IAAI,KAAK,CACb,0GAA0G,CAC3G,CAAC;QAEJ,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC9C,IAAI,CAAC,YAAY;gBACf,MAAM,IAAI,KAAK,CACb,4GAA4G,CAC7G,CAAC;YACJ,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEhC,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC1B,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YACV,MAAM,IAAI,KAAK,CACb,oGAAoG,CACrG,CAAC;QAEJ,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QAC/C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,cAAc,CAAC,SAAiB;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YACV,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;QAEJ,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QAC7C,UAAU,CAAC,IAAI,GAAG,OAAO,CAAC;QAE1B,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;QAE9B,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,SAAS,CAAC,SAAwB;QAChC,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YACV,MAAM,IAAI,KAAK,CACb,iGAAiG,CAClG,CAAC;QAEJ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,iBAAiB,GAAG,CAAC,GAAsB,EAAE,EAAE;gBACnD,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;oBACnC,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAChD,IAAI,YAAY,EAAE,CAAC;wBACjB,iBAAiB,CAAC,YAAY,CAAC,CAAC;wBAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC,CAAC;YACF,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAE3B,OAAO,CAAC,QAAQ,GAAG,EAAE,CAAC;YACtB,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;QACtB,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC;QACpB,KACE,IAAI,OAAO,GAA6B,OAAO,EAC/C,OAAO,EACP,OAAO,GAAG,OAAO,CAAC,IAAI,EACtB,CAAC;YACD,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;gBACjB,OAAO,CAAC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC;YAC9B,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG;YACV,QAAQ,EAAE,EAAE;YACZ,IAAI,EAAE,IAAI;SACX,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,MAAM;QACJ,MAAM,WAAW,GAA0C,EAAE,CAAC;QAE9D,KAAK,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxC,WAAW,CAAC,IAAI,CAAC;gBACf,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,QAAQ,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI;aAC3C,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI;YACrC,QAAQ,EAAE,WAAW;SACtB,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,EAAE,MAAM,EAAE,QAAQ,EAA6B;QACpD,KAAK,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC7C,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC;IAChE,CAAC;CACF"}
+{"version":3,"file":"message-repository.js","names":[],"sources":["../../../src/runtime/utils/message-repository.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { generateId, generateOptimisticId } from \"../../utils/id\";\nimport type { ThreadMessageLike } from \"./thread-message-like\";\nimport { getAutoStatus } from \"./auto-status\";\nimport { fromThreadMessageLike } from \"./thread-message-like\";\n\nexport type ExportedMessageRepositoryItem = {\n  message: ThreadMessage;\n  parentId: string | null;\n  runConfig?: RunConfig;\n};\n\nexport type ExportedMessageRepository = {\n  headId?: string | null;\n  messages: Array<{\n    message: ThreadMessage;\n    parentId: string | null;\n    runConfig?: RunConfig;\n  }>;\n};\n\nexport const ExportedMessageRepository = {\n  fromArray: (\n    messages: readonly ThreadMessageLike[],\n  ): ExportedMessageRepository => {\n    const conv = messages.map((m) =>\n      fromThreadMessageLike(\n        m,\n        generateId(),\n        getAutoStatus(false, false, false, false, undefined),\n      ),\n    );\n\n    return {\n      messages: conv.map((m, idx) => ({\n        parentId: idx > 0 ? conv[idx - 1]!.id : null,\n        message: m,\n      })),\n    };\n  },\n\n  fromBranchableArray: (\n    items: readonly {\n      message: ThreadMessageLike;\n      parentId: string | null;\n    }[],\n    options?: { headId?: string | null },\n  ): ExportedMessageRepository => {\n    const fallbackStatus = getAutoStatus(false, false, false, false, undefined);\n    return {\n      ...(options?.headId !== undefined\n        ? { headId: options.headId }\n        : undefined),\n      messages: items.map(({ message, parentId }) => {\n        if (!message.id) {\n          throw new Error(\n            \"ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.\",\n          );\n        }\n        return {\n          parentId,\n          message: fromThreadMessageLike(message, message.id, fallbackStatus),\n        };\n      }),\n    };\n  },\n};\n\ntype RepositoryParent = {\n  children: string[];\n  next: RepositoryMessage | null;\n};\n\ntype RepositoryMessage = RepositoryParent & {\n  prev: RepositoryMessage | null;\n  current: ThreadMessage;\n  level: number;\n};\n\nconst findHead = (\n  message: RepositoryMessage | RepositoryParent,\n): RepositoryMessage | null => {\n  if (message.next) return findHead(message.next);\n  if (\"current\" in message) return message;\n  return null;\n};\n\nclass CachedValue<T> {\n  private _value: T | null = null;\n\n  constructor(private func: () => T) {}\n\n  get value() {\n    if (this._value === null) {\n      this._value = this.func();\n    }\n    return this._value;\n  }\n\n  dirty() {\n    this._value = null;\n  }\n}\n\nexport class MessageRepository {\n  private messages = new Map<string, RepositoryMessage>();\n  private head: RepositoryMessage | null = null;\n  private root: RepositoryParent = {\n    children: [],\n    next: null,\n  };\n\n  private updateLevels(message: RepositoryMessage, newLevel: number) {\n    message.level = newLevel;\n    for (const childId of message.children) {\n      const childMessage = this.messages.get(childId);\n      if (childMessage) {\n        this.updateLevels(childMessage, newLevel + 1);\n      }\n    }\n  }\n\n  private performOp(\n    newParent: RepositoryMessage | null,\n    child: RepositoryMessage,\n    operation: \"cut\" | \"link\" | \"relink\",\n  ) {\n    const parentOrRoot = child.prev ?? this.root;\n    const newParentOrRoot = newParent ?? this.root;\n\n    if (operation === \"relink\" && parentOrRoot === newParentOrRoot) return;\n\n    if (operation !== \"link\") {\n      parentOrRoot.children = parentOrRoot.children.filter(\n        (m) => m !== child.current.id,\n      );\n\n      if (parentOrRoot.next === child) {\n        const fallbackId = parentOrRoot.children.at(-1);\n        const fallback = fallbackId ? this.messages.get(fallbackId) : null;\n        if (fallback === undefined) {\n          throw new Error(\n            \"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n        parentOrRoot.next = fallback;\n      }\n    }\n\n    if (operation !== \"cut\") {\n      for (\n        let current: RepositoryMessage | null = newParent;\n        current;\n        current = current.prev\n      ) {\n        if (current.current.id === child.current.id) {\n          throw new Error(\n            \"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.\",\n          );\n        }\n      }\n\n      newParentOrRoot.children = [\n        ...newParentOrRoot.children,\n        child.current.id,\n      ];\n\n      if (findHead(child) === this.head || newParentOrRoot.next === null) {\n        newParentOrRoot.next = child;\n      }\n\n      child.prev = newParent;\n\n      const newLevel = newParent ? newParent.level + 1 : 0;\n      this.updateLevels(child, newLevel);\n    }\n  }\n\n  private _messages = new CachedValue<readonly ThreadMessage[]>(() => {\n    const messages = new Array<ThreadMessage>((this.head?.level ?? -1) + 1);\n    for (let current = this.head; current; current = current.prev) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  });\n\n  get headId() {\n    return this.head?.current.id ?? null;\n  }\n\n  getMessages(headId?: string) {\n    if (headId === undefined || headId === this.head?.current.id) {\n      return this._messages.value;\n    }\n\n    const headMessage = this.messages.get(headId);\n    if (!headMessage) {\n      throw new Error(\n        \"MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.\",\n      );\n    }\n\n    const messages = new Array<ThreadMessage>(headMessage.level + 1);\n    for (\n      let current: RepositoryMessage | null = headMessage;\n      current;\n      current = current.prev\n    ) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  }\n\n  addOrUpdateMessage(parentId: string | null, message: ThreadMessage) {\n    const existingItem = this.messages.get(message.id);\n    const prev = parentId ? this.messages.get(parentId) : null;\n    if (prev === undefined)\n      throw new Error(\n        \"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (existingItem) {\n      existingItem.current = message;\n      this.performOp(prev, existingItem, \"relink\");\n      this._messages.dirty();\n      return;\n    }\n\n    const newItem: RepositoryMessage = {\n      prev,\n      current: message,\n      next: null,\n      children: [],\n      level: prev ? prev.level + 1 : 0,\n    };\n\n    this.messages.set(message.id, newItem);\n    this.performOp(prev, newItem, \"link\");\n\n    if (this.head === prev) {\n      this.head = newItem;\n    }\n\n    this._messages.dirty();\n  }\n\n  getMessage(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    return {\n      parentId: message.prev?.current.id ?? null,\n      message: message.current,\n      index: message.level,\n    };\n  }\n\n  appendOptimisticMessage(parentId: string | null, message: ThreadMessageLike) {\n    let optimisticId: string;\n    do {\n      optimisticId = generateOptimisticId();\n    } while (this.messages.has(optimisticId));\n\n    this.addOrUpdateMessage(\n      parentId,\n      fromThreadMessageLike(message, optimisticId, { type: \"running\" }),\n    );\n\n    return optimisticId;\n  }\n\n  deleteMessage(messageId: string, replacementId?: string | null | undefined) {\n    const message = this.messages.get(messageId);\n\n    if (!message)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const replacement =\n      replacementId === undefined\n        ? message.prev\n        : replacementId === null\n          ? null\n          : this.messages.get(replacementId);\n    if (replacement === undefined)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    for (const child of message.children) {\n      const childMessage = this.messages.get(child);\n      if (!childMessage)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.\",\n        );\n      this.performOp(replacement, childMessage, \"relink\");\n    }\n\n    this.performOp(null, message, \"cut\");\n    this.messages.delete(messageId);\n\n    if (this.head === message) {\n      this.head = findHead(replacement ?? this.root);\n    }\n\n    this._messages.dirty();\n  }\n\n  getBranches(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const { children } = message.prev ?? this.root;\n    return children;\n  }\n\n  switchToBranch(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const prevOrRoot = message.prev ?? this.root;\n    prevOrRoot.next = message;\n\n    this.head = findHead(message);\n\n    this._messages.dirty();\n  }\n\n  resetHead(messageId: string | null) {\n    if (messageId === null) {\n      this.clear();\n      return;\n    }\n\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (message.children.length > 0) {\n      const deleteDescendants = (msg: RepositoryMessage) => {\n        for (const childId of msg.children) {\n          const childMessage = this.messages.get(childId);\n          if (childMessage) {\n            deleteDescendants(childMessage);\n            this.messages.delete(childId);\n          }\n        }\n      };\n      deleteDescendants(message);\n\n      message.children = [];\n      message.next = null;\n    }\n\n    this.head = message;\n    for (\n      let current: RepositoryMessage | null = message;\n      current;\n      current = current.prev\n    ) {\n      if (current.prev) {\n        current.prev.next = current;\n      } else {\n        this.root.next = current;\n      }\n    }\n\n    this._messages.dirty();\n  }\n\n  clear(): void {\n    this.messages.clear();\n    this.head = null;\n    this.root = {\n      children: [],\n      next: null,\n    };\n    this._messages.dirty();\n  }\n\n  export(): ExportedMessageRepository {\n    const exportItems: ExportedMessageRepository[\"messages\"] = [];\n\n    for (const [, message] of this.messages) {\n      exportItems.push({\n        message: message.current,\n        parentId: message.prev?.current.id ?? null,\n      });\n    }\n\n    return {\n      headId: this.head?.current.id ?? null,\n      messages: exportItems,\n    };\n  }\n\n  import({ headId, messages }: ExportedMessageRepository) {\n    for (const { message, parentId } of messages) {\n      this.addOrUpdateMessage(parentId, message);\n    }\n\n    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);\n  }\n}\n"],"mappings":";;;;AAsBA,MAAa,4BAA4B;CACvC,YACE,aAC8B;EAC9B,MAAM,OAAO,SAAS,KAAK,MACzB,sBACE,GACA,WAAW,GACX,cAAc,OAAO,OAAO,OAAO,OAAO,KAAA,CAAS,CACrD,CACF;EAEA,OAAO,EACL,UAAU,KAAK,KAAK,GAAG,SAAS;GAC9B,UAAU,MAAM,IAAI,KAAK,MAAM,GAAI,KAAK;GACxC,SAAS;EACX,EAAE,EACJ;CACF;CAEA,sBACE,OAIA,YAC8B;EAC9B,MAAM,iBAAiB,cAAc,OAAO,OAAO,OAAO,OAAO,KAAA,CAAS;EAC1E,OAAO;GACL,GAAI,SAAS,WAAW,KAAA,IACpB,EAAE,QAAQ,QAAQ,OAAO,IACzB,KAAA;GACJ,UAAU,MAAM,KAAK,EAAE,SAAS,eAAe;IAC7C,IAAI,CAAC,QAAQ,IACX,MAAM,IAAI,MACR,0FACF;IAEF,OAAO;KACL;KACA,SAAS,sBAAsB,SAAS,QAAQ,IAAI,cAAc;IACpE;GACF,CAAC;EACH;CACF;AACF;AAaA,MAAM,YACJ,YAC6B;CAC7B,IAAI,QAAQ,MAAM,OAAO,SAAS,QAAQ,IAAI;CAC9C,IAAI,aAAa,SAAS,OAAO;CACjC,OAAO;AACT;AAEA,IAAM,cAAN,MAAqB;CAGC;CAFpB,SAA2B;CAE3B,YAAY,MAAuB;EAAf,KAAA,OAAA;CAAgB;CAEpC,IAAI,QAAQ;EACV,IAAI,KAAK,WAAW,MAClB,KAAK,SAAS,KAAK,KAAK;EAE1B,OAAO,KAAK;CACd;CAEA,QAAQ;EACN,KAAK,SAAS;CAChB;AACF;AAEA,IAAa,oBAAb,MAA+B;CAC7B,2BAAmB,IAAI,IAA+B;CACtD,OAAyC;CACzC,OAAiC;EAC/B,UAAU,CAAC;EACX,MAAM;CACR;CAEA,aAAqB,SAA4B,UAAkB;EACjE,QAAQ,QAAQ;EAChB,KAAK,MAAM,WAAW,QAAQ,UAAU;GACtC,MAAM,eAAe,KAAK,SAAS,IAAI,OAAO;GAC9C,IAAI,cACF,KAAK,aAAa,cAAc,WAAW,CAAC;EAEhD;CACF;CAEA,UACE,WACA,OACA,WACA;EACA,MAAM,eAAe,MAAM,QAAQ,KAAK;EACxC,MAAM,kBAAkB,aAAa,KAAK;EAE1C,IAAI,cAAc,YAAY,iBAAiB,iBAAiB;EAEhE,IAAI,cAAc,QAAQ;GACxB,aAAa,WAAW,aAAa,SAAS,QAC3C,MAAM,MAAM,MAAM,QAAQ,EAC7B;GAEA,IAAI,aAAa,SAAS,OAAO;IAC/B,MAAM,aAAa,aAAa,SAAS,GAAG,EAAE;IAC9C,MAAM,WAAW,aAAa,KAAK,SAAS,IAAI,UAAU,IAAI;IAC9D,IAAI,aAAa,KAAA,GACf,MAAM,IAAI,MACR,uHACF;IAEF,aAAa,OAAO;GACtB;EACF;EAEA,IAAI,cAAc,OAAO;GACvB,KACE,IAAI,UAAoC,WACxC,SACA,UAAU,QAAQ,MAElB,IAAI,QAAQ,QAAQ,OAAO,MAAM,QAAQ,IACvC,MAAM,IAAI,MACR,oNACF;GAIJ,gBAAgB,WAAW,CACzB,GAAG,gBAAgB,UACnB,MAAM,QAAQ,EAChB;GAEA,IAAI,SAAS,KAAK,MAAM,KAAK,QAAQ,gBAAgB,SAAS,MAC5D,gBAAgB,OAAO;GAGzB,MAAM,OAAO;GAEb,MAAM,WAAW,YAAY,UAAU,QAAQ,IAAI;GACnD,KAAK,aAAa,OAAO,QAAQ;EACnC;CACF;CAEA,YAAoB,IAAI,kBAA4C;EAClE,MAAM,WAAW,IAAI,OAAsB,KAAK,MAAM,SAAS,MAAM,CAAC;EACtE,KAAK,IAAI,UAAU,KAAK,MAAM,SAAS,UAAU,QAAQ,MACvD,SAAS,QAAQ,SAAS,QAAQ;EAEpC,OAAO;CACT,CAAC;CAED,IAAI,SAAS;EACX,OAAO,KAAK,MAAM,QAAQ,MAAM;CAClC;CAEA,YAAY,QAAiB;EAC3B,IAAI,WAAW,KAAA,KAAa,WAAW,KAAK,MAAM,QAAQ,IACxD,OAAO,KAAK,UAAU;EAGxB,MAAM,cAAc,KAAK,SAAS,IAAI,MAAM;EAC5C,IAAI,CAAC,aACH,MAAM,IAAI,MACR,yGACF;EAGF,MAAM,WAAW,IAAI,MAAqB,YAAY,QAAQ,CAAC;EAC/D,KACE,IAAI,UAAoC,aACxC,SACA,UAAU,QAAQ,MAElB,SAAS,QAAQ,SAAS,QAAQ;EAEpC,OAAO;CACT;CAEA,mBAAmB,UAAyB,SAAwB;EAClE,MAAM,eAAe,KAAK,SAAS,IAAI,QAAQ,EAAE;EACjD,MAAM,OAAO,WAAW,KAAK,SAAS,IAAI,QAAQ,IAAI;EACtD,IAAI,SAAS,KAAA,GACX,MAAM,IAAI,MACR,kHACF;EAEF,IAAI,cAAc;GAChB,aAAa,UAAU;GACvB,KAAK,UAAU,MAAM,cAAc,QAAQ;GAC3C,KAAK,UAAU,MAAM;GACrB;EACF;EAEA,MAAM,UAA6B;GACjC;GACA,SAAS;GACT,MAAM;GACN,UAAU,CAAC;GACX,OAAO,OAAO,KAAK,QAAQ,IAAI;EACjC;EAEA,KAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;EACrC,KAAK,UAAU,MAAM,SAAS,MAAM;EAEpC,IAAI,KAAK,SAAS,MAChB,KAAK,OAAO;EAGd,KAAK,UAAU,MAAM;CACvB;CAEA,WAAW,WAAmB;EAC5B,MAAM,UAAU,KAAK,SAAS,IAAI,SAAS;EAC3C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,sGACF;EAEF,OAAO;GACL,UAAU,QAAQ,MAAM,QAAQ,MAAM;GACtC,SAAS,QAAQ;GACjB,OAAO,QAAQ;EACjB;CACF;CAEA,wBAAwB,UAAyB,SAA4B;EAC3E,IAAI;EACJ;GACE,eAAe,qBAAqB;SAC7B,KAAK,SAAS,IAAI,YAAY;EAEvC,KAAK,mBACH,UACA,sBAAsB,SAAS,cAAc,EAAE,MAAM,UAAU,CAAC,CAClE;EAEA,OAAO;CACT;CAEA,cAAc,WAAmB,eAA2C;EAC1E,MAAM,UAAU,KAAK,SAAS,IAAI,SAAS;EAE3C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,sGACF;EAEF,MAAM,cACJ,kBAAkB,KAAA,IACd,QAAQ,OACR,kBAAkB,OAChB,OACA,KAAK,SAAS,IAAI,aAAa;EACvC,IAAI,gBAAgB,KAAA,GAClB,MAAM,IAAI,MACR,0GACF;EAEF,KAAK,MAAM,SAAS,QAAQ,UAAU;GACpC,MAAM,eAAe,KAAK,SAAS,IAAI,KAAK;GAC5C,IAAI,CAAC,cACH,MAAM,IAAI,MACR,4GACF;GACF,KAAK,UAAU,aAAa,cAAc,QAAQ;EACpD;EAEA,KAAK,UAAU,MAAM,SAAS,KAAK;EACnC,KAAK,SAAS,OAAO,SAAS;EAE9B,IAAI,KAAK,SAAS,SAChB,KAAK,OAAO,SAAS,eAAe,KAAK,IAAI;EAG/C,KAAK,UAAU,MAAM;CACvB;CAEA,YAAY,WAAmB;EAC7B,MAAM,UAAU,KAAK,SAAS,IAAI,SAAS;EAC3C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,oGACF;EAEF,MAAM,EAAE,aAAa,QAAQ,QAAQ,KAAK;EAC1C,OAAO;CACT;CAEA,eAAe,WAAmB;EAChC,MAAM,UAAU,KAAK,SAAS,IAAI,SAAS;EAC3C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,sGACF;EAEF,MAAM,aAAa,QAAQ,QAAQ,KAAK;EACxC,WAAW,OAAO;EAElB,KAAK,OAAO,SAAS,OAAO;EAE5B,KAAK,UAAU,MAAM;CACvB;CAEA,UAAU,WAA0B;EAClC,IAAI,cAAc,MAAM;GACtB,KAAK,MAAM;GACX;EACF;EAEA,MAAM,UAAU,KAAK,SAAS,IAAI,SAAS;EAC3C,IAAI,CAAC,SACH,MAAM,IAAI,MACR,iGACF;EAEF,IAAI,QAAQ,SAAS,SAAS,GAAG;GAC/B,MAAM,qBAAqB,QAA2B;IACpD,KAAK,MAAM,WAAW,IAAI,UAAU;KAClC,MAAM,eAAe,KAAK,SAAS,IAAI,OAAO;KAC9C,IAAI,cAAc;MAChB,kBAAkB,YAAY;MAC9B,KAAK,SAAS,OAAO,OAAO;KAC9B;IACF;GACF;GACA,kBAAkB,OAAO;GAEzB,QAAQ,WAAW,CAAC;GACpB,QAAQ,OAAO;EACjB;EAEA,KAAK,OAAO;EACZ,KACE,IAAI,UAAoC,SACxC,SACA,UAAU,QAAQ,MAElB,IAAI,QAAQ,MACV,QAAQ,KAAK,OAAO;OAEpB,KAAK,KAAK,OAAO;EAIrB,KAAK,UAAU,MAAM;CACvB;CAEA,QAAc;EACZ,KAAK,SAAS,MAAM;EACpB,KAAK,OAAO;EACZ,KAAK,OAAO;GACV,UAAU,CAAC;GACX,MAAM;EACR;EACA,KAAK,UAAU,MAAM;CACvB;CAEA,SAAoC;EAClC,MAAM,cAAqD,CAAC;EAE5D,KAAK,MAAM,GAAG,YAAY,KAAK,UAC7B,YAAY,KAAK;GACf,SAAS,QAAQ;GACjB,UAAU,QAAQ,MAAM,QAAQ,MAAM;EACxC,CAAC;EAGH,OAAO;GACL,QAAQ,KAAK,MAAM,QAAQ,MAAM;GACjC,UAAU;EACZ;CACF;CAEA,OAAO,EAAE,QAAQ,YAAuC;EACtD,KAAK,MAAM,EAAE,SAAS,cAAc,UAClC,KAAK,mBAAmB,UAAU,OAAO;EAG3C,KAAK,UAAU,UAAU,SAAS,GAAG,EAAE,GAAG,QAAQ,MAAM,IAAI;CAC9D;AACF"}