@assistant-ui/core 0.2.3 → 0.2.5

package/dist/runtime/base/base-thread-runtime-core.js

@@ -1,361 +1,353 @@
-import { getThreadMessageText } from "../../utils/text.js";
 import { generateId } from "../../utils/id.js";
-import { ExportedMessageRepository, MessageRepository, } from "../utils/message-repository.js";
+import { ExportedMessageRepository, MessageRepository } from "../utils/message-repository.js";
+import { getThreadMessageText } from "../../utils/text.js";
 import { DefaultThreadComposerRuntimeCore } from "./default-thread-composer-runtime-core.js";
 import { DefaultEditComposerRuntimeCore } from "./default-edit-composer-runtime-core.js";
-export class BaseThreadRuntimeCore {
-    _contextProvider;
-    _subscriptions = new Set();
-    _isInitialized = false;
-    repository = new MessageRepository();
-    _voiceMessages = [];
-    _voiceGeneration = 0;
-    _cachedMergedMessages = null;
-    _cachedVoiceGeneration = -1;
-    _cachedMergedBase = null;
-    _markVoiceMessagesDirty() {
-        this._voiceGeneration++;
-        this._cachedMergedMessages = null;
-    }
-    _getBaseMessages() {
-        return this.repository.getMessages();
-    }
-    get messages() {
-        if (this._voiceMessages.length === 0) {
-            return this._getBaseMessages();
-        }
-        const base = this._getBaseMessages();
-        if (this._cachedVoiceGeneration !== this._voiceGeneration ||
-            this._cachedMergedBase !== base) {
-            this._cachedMergedMessages = [...base, ...this._voiceMessages];
-            this._cachedVoiceGeneration = this._voiceGeneration;
-            this._cachedMergedBase = base;
-        }
-        return this._cachedMergedMessages;
-    }
-    get state() {
-        let mostRecentAssistantMessage;
-        for (const message of this.messages) {
-            if (message.role === "assistant") {
-                mostRecentAssistantMessage = message;
-            }
-        }
-        return mostRecentAssistantMessage?.metadata.unstable_state ?? null;
-    }
-    composer = new DefaultThreadComposerRuntimeCore(this);
-    constructor(_contextProvider) {
-        this._contextProvider = _contextProvider;
-    }
-    getModelContext() {
-        return this._contextProvider.getModelContext();
-    }
-    _editComposers = new Map();
-    getEditComposer(messageId) {
-        return this._editComposers.get(messageId);
-    }
-    beginEdit(messageId) {
-        if (this._editComposers.has(messageId))
-            throw new Error("Edit already in progress");
-        this._editComposers.set(messageId, new DefaultEditComposerRuntimeCore(this, () => this._editComposers.delete(messageId), this.repository.getMessage(messageId)));
-        this._notifySubscribers();
-    }
-    getMessageById(messageId) {
-        try {
-            return this.repository.getMessage(messageId);
-        }
-        catch {
-            // Check voice messages
-            const baseMessages = this.repository.getMessages();
-            const voiceIdx = this._voiceMessages.findIndex((m) => m.id === messageId);
-            if (voiceIdx !== -1) {
-                const parentId = voiceIdx > 0
-                    ? this._voiceMessages[voiceIdx - 1].id
-                    : (baseMessages.at(-1)?.id ?? null);
-                return {
-                    parentId,
-                    message: this._voiceMessages[voiceIdx],
-                    index: baseMessages.length + voiceIdx,
-                };
-            }
-            return undefined;
-        }
-    }
-    getBranches(messageId) {
-        if (this._voiceMessages.some((m) => m.id === messageId)) {
-            return [];
-        }
-        return this.repository.getBranches(messageId);
-    }
-    switchToBranch(branchId) {
-        this.repository.switchToBranch(branchId);
-        this._notifySubscribers();
-    }
-    _notifySubscribers() {
-        for (const callback of this._subscriptions)
-            callback();
-    }
-    _notifyEventSubscribers(event, payload) {
-        const subscribers = this._eventSubscribers.get(event);
-        if (!subscribers)
-            return;
-        for (const callback of subscribers)
-            callback(payload);
-    }
-    subscribe(callback) {
-        this._subscriptions.add(callback);
-        return () => this._subscriptions.delete(callback);
-    }
-    submitFeedback({ messageId, type }) {
-        const adapter = this.adapters?.feedback;
-        if (!adapter)
-            throw new Error("Feedback adapter not configured");
-        const { message, parentId } = this.repository.getMessage(messageId);
-        adapter.submit({ message, type });
-        if (message.role === "assistant") {
-            const updatedMessage = {
-                ...message,
-                metadata: {
-                    ...message.metadata,
-                    submittedFeedback: { type },
-                },
-            };
-            this.repository.addOrUpdateMessage(parentId, updatedMessage);
-        }
-        this._notifySubscribers();
-    }
-    _stopSpeaking;
-    speech;
-    speak(messageId) {
-        const adapter = this.adapters?.speech;
-        if (!adapter)
-            throw new Error("Speech adapter not configured");
-        const { message } = this.repository.getMessage(messageId);
-        this._stopSpeaking?.();
-        const utterance = adapter.speak(getThreadMessageText(message));
-        const unsub = utterance.subscribe(() => {
-            if (utterance.status.type === "ended") {
-                this._stopSpeaking = undefined;
-                this.speech = undefined;
-            }
-            else {
-                this.speech = { messageId, status: utterance.status };
-            }
-            this._notifySubscribers();
-        });
-        this.speech = { messageId, status: utterance.status };
-        this._notifySubscribers();
-        this._stopSpeaking = () => {
-            utterance.cancel();
-            unsub();
-            this.speech = undefined;
-            this._stopSpeaking = undefined;
-        };
-    }
-    stopSpeaking() {
-        if (!this._stopSpeaking)
-            throw new Error("No message is being spoken");
-        this._stopSpeaking();
-        this._notifySubscribers();
-    }
-    _voiceSession;
-    _voiceUnsubs = [];
-    voice;
-    _voiceVolume = 0;
-    _voiceVolumeSubscribers = new Set();
-    getVoiceVolume = () => this._voiceVolume;
-    subscribeVoiceVolume = (callback) => {
-        this._voiceVolumeSubscribers.add(callback);
-        return () => this._voiceVolumeSubscribers.delete(callback);
-    };
-    connectVoice() {
-        const adapter = this.adapters?.voice;
-        if (!adapter)
-            throw new Error("Voice adapter not configured");
-        this.disconnectVoice();
-        const session = adapter.connect({});
-        this._voiceSession = session;
-        const unsubs = [];
-        let currentMode = "listening";
-        this.voice = {
-            status: session.status,
-            isMuted: session.isMuted,
-            mode: currentMode,
-        };
-        this._voiceVolume = 0;
-        this._notifySubscribers();
-        unsubs.push(session.onStatusChange((status) => {
-            if (status.type === "ended") {
-                this._finishVoiceAssistantMessage();
-                this._voiceSession = undefined;
-                this.voice = undefined;
-            }
-            else {
-                this.voice = {
-                    status,
-                    isMuted: session.isMuted,
-                    mode: currentMode,
-                };
-            }
-            this._notifySubscribers();
-        }));
-        unsubs.push(session.onModeChange((mode) => {
-            currentMode = mode;
-            if (this.voice) {
-                this.voice = { ...this.voice, mode };
-                this._notifySubscribers();
-            }
-        }));
-        unsubs.push(session.onVolumeChange((volume) => {
-            this._voiceVolume = volume;
-            for (const cb of this._voiceVolumeSubscribers)
-                cb();
-        }));
-        unsubs.push(session.onTranscript((transcript) => {
-            this._handleVoiceTranscript(transcript);
-        }));
-        this._voiceUnsubs = unsubs;
-    }
-    _currentAssistantMsg = null;
-    _handleVoiceTranscript(transcript) {
-        this.ensureInitialized();
-        if (transcript.role === "user") {
-            this._finishVoiceAssistantMessage();
-            this._currentAssistantMsg = null;
-            if (transcript.isFinal) {
-                this._voiceMessages.push({
-                    id: generateId(),
-                    role: "user",
-                    content: [{ type: "text", text: transcript.text }],
-                    metadata: { custom: {} },
-                    createdAt: new Date(),
-                    status: { type: "complete", reason: "unknown" },
-                    attachments: [],
-                });
-                this._markVoiceMessagesDirty();
-                this._notifySubscribers();
-            }
-        }
-        else {
-            if (!this._currentAssistantMsg) {
-                this._currentAssistantMsg = {
-                    id: generateId(),
-                    role: "assistant",
-                    content: [{ type: "text", text: transcript.text }],
-                    metadata: {
-                        unstable_state: this.state,
-                        unstable_annotations: [],
-                        unstable_data: [],
-                        steps: [],
-                        custom: {},
-                    },
-                    status: { type: "running" },
-                    createdAt: new Date(),
-                };
-                this._voiceMessages.push(this._currentAssistantMsg);
-            }
-            else {
-                const idx = this._voiceMessages.indexOf(this._currentAssistantMsg);
-                if (idx === -1)
-                    return;
-                const updated = {
-                    ...this._currentAssistantMsg,
-                    content: [{ type: "text", text: transcript.text }],
-                    ...(transcript.isFinal
-                        ? { status: { type: "complete", reason: "stop" } }
-                        : {}),
-                };
-                this._voiceMessages[idx] = updated;
-                this._currentAssistantMsg = updated;
-            }
-            if (transcript.isFinal) {
-                this._currentAssistantMsg = null;
-            }
-            this._markVoiceMessagesDirty();
-            this._notifySubscribers();
-        }
-    }
-    _finishVoiceAssistantMessage() {
-        const last = this._voiceMessages.at(-1);
-        if (last?.role === "assistant" && last.status.type === "running") {
-            const idx = this._voiceMessages.length - 1;
-            this._voiceMessages[idx] = {
-                ...last,
-                status: { type: "complete", reason: "stop" },
-            };
-            this._markVoiceMessagesDirty();
-            this._notifySubscribers();
-        }
-    }
-    disconnectVoice() {
-        this._finishVoiceAssistantMessage();
-        this._currentAssistantMsg = null;
-        for (const unsub of this._voiceUnsubs)
-            unsub();
-        this._voiceUnsubs = [];
-        this._voiceSession?.disconnect();
-        this._voiceSession = undefined;
-        this.voice = undefined;
-        this._voiceVolume = 0;
-        for (const cb of this._voiceVolumeSubscribers)
-            cb();
-        this._voiceMessages = [];
-        this._markVoiceMessagesDirty();
-        this._notifySubscribers();
-    }
-    muteVoice() {
-        if (!this._voiceSession)
-            throw new Error("No active voice session");
-        this._voiceSession.mute();
-        this.voice = {
-            ...this.voice,
-            isMuted: true,
-        };
-        this._notifySubscribers();
-    }
-    unmuteVoice() {
-        if (!this._voiceSession)
-            throw new Error("No active voice session");
-        this._voiceSession.unmute();
-        this.voice = {
-            ...this.voice,
-            isMuted: false,
-        };
-        this._notifySubscribers();
-    }
-    ensureInitialized() {
-        if (!this._isInitialized) {
-            this._isInitialized = true;
-            this._notifyEventSubscribers("initialize", {});
-        }
-    }
-    export() {
-        return this.repository.export();
-    }
-    import(data) {
-        this.ensureInitialized();
-        this.repository.clear();
-        this.repository.import(data);
-        this._notifySubscribers();
-    }
-    reset(initialMessages) {
-        this.import(ExportedMessageRepository.fromArray(initialMessages ?? []));
-    }
-    _eventSubscribers = new Map();
-    unstable_on(event, callback) {
-        const wrapped = callback;
-        if (event === "modelContextUpdate") {
-            // provider.subscribe is `() => void`; pump the typed empty payload to the user callback.
-            return this._contextProvider.subscribe?.(() => wrapped({})) ?? (() => { });
-        }
-        let subscribers = this._eventSubscribers.get(event);
-        if (!subscribers) {
-            subscribers = new Set();
-            this._eventSubscribers.set(event, subscribers);
-        }
-        subscribers.add(wrapped);
-        return () => {
-            this._eventSubscribers.get(event)?.delete(wrapped);
-        };
-    }
-}
+//#region src/runtime/base/base-thread-runtime-core.ts
+var BaseThreadRuntimeCore = class {
+	_contextProvider;
+	_subscriptions = /* @__PURE__ */ new Set();
+	_isInitialized = false;
+	repository = new MessageRepository();
+	_voiceMessages = [];
+	_voiceGeneration = 0;
+	_cachedMergedMessages = null;
+	_cachedVoiceGeneration = -1;
+	_cachedMergedBase = null;
+	_markVoiceMessagesDirty() {
+		this._voiceGeneration++;
+		this._cachedMergedMessages = null;
+	}
+	_getBaseMessages() {
+		return this.repository.getMessages();
+	}
+	get messages() {
+		if (this._voiceMessages.length === 0) return this._getBaseMessages();
+		const base = this._getBaseMessages();
+		if (this._cachedVoiceGeneration !== this._voiceGeneration || this._cachedMergedBase !== base) {
+			this._cachedMergedMessages = [...base, ...this._voiceMessages];
+			this._cachedVoiceGeneration = this._voiceGeneration;
+			this._cachedMergedBase = base;
+		}
+		return this._cachedMergedMessages;
+	}
+	get state() {
+		let mostRecentAssistantMessage;
+		for (const message of this.messages) if (message.role === "assistant") mostRecentAssistantMessage = message;
+		return mostRecentAssistantMessage?.metadata.unstable_state ?? null;
+	}
+	composer = new DefaultThreadComposerRuntimeCore(this);
+	constructor(_contextProvider) {
+		this._contextProvider = _contextProvider;
+	}
+	getModelContext() {
+		return this._contextProvider.getModelContext();
+	}
+	_editComposers = /* @__PURE__ */ new Map();
+	getEditComposer(messageId) {
+		return this._editComposers.get(messageId);
+	}
+	beginEdit(messageId) {
+		if (this._editComposers.has(messageId)) throw new Error("Edit already in progress");
+		this._editComposers.set(messageId, new DefaultEditComposerRuntimeCore(this, () => this._editComposers.delete(messageId), this.repository.getMessage(messageId)));
+		this._notifySubscribers();
+	}
+	getMessageById(messageId) {
+		try {
+			return this.repository.getMessage(messageId);
+		} catch {
+			const baseMessages = this.repository.getMessages();
+			const voiceIdx = this._voiceMessages.findIndex((m) => m.id === messageId);
+			if (voiceIdx !== -1) return {
+				parentId: voiceIdx > 0 ? this._voiceMessages[voiceIdx - 1].id : baseMessages.at(-1)?.id ?? null,
+				message: this._voiceMessages[voiceIdx],
+				index: baseMessages.length + voiceIdx
+			};
+			return;
+		}
+	}
+	getBranches(messageId) {
+		if (this._voiceMessages.some((m) => m.id === messageId)) return [];
+		return this.repository.getBranches(messageId);
+	}
+	switchToBranch(branchId) {
+		this.repository.switchToBranch(branchId);
+		this._notifySubscribers();
+	}
+	_notifySubscribers() {
+		for (const callback of this._subscriptions) callback();
+	}
+	_notifyEventSubscribers(event, payload) {
+		const subscribers = this._eventSubscribers.get(event);
+		if (!subscribers) return;
+		for (const callback of subscribers) callback(payload);
+	}
+	subscribe(callback) {
+		this._subscriptions.add(callback);
+		return () => this._subscriptions.delete(callback);
+	}
+	submitFeedback({ messageId, type }) {
+		const adapter = this.adapters?.feedback;
+		if (!adapter) throw new Error("Feedback adapter not configured");
+		const { message, parentId } = this.repository.getMessage(messageId);
+		adapter.submit({
+			message,
+			type
+		});
+		if (message.role === "assistant") {
+			const updatedMessage = {
+				...message,
+				metadata: {
+					...message.metadata,
+					submittedFeedback: { type }
+				}
+			};
+			this.repository.addOrUpdateMessage(parentId, updatedMessage);
+		}
+		this._notifySubscribers();
+	}
+	_stopSpeaking;
+	speech;
+	speak(messageId) {
+		const adapter = this.adapters?.speech;
+		if (!adapter) throw new Error("Speech adapter not configured");
+		const { message } = this.repository.getMessage(messageId);
+		this._stopSpeaking?.();
+		const utterance = adapter.speak(getThreadMessageText(message));
+		const unsub = utterance.subscribe(() => {
+			if (utterance.status.type === "ended") {
+				this._stopSpeaking = void 0;
+				this.speech = void 0;
+			} else this.speech = {
+				messageId,
+				status: utterance.status
+			};
+			this._notifySubscribers();
+		});
+		this.speech = {
+			messageId,
+			status: utterance.status
+		};
+		this._notifySubscribers();
+		this._stopSpeaking = () => {
+			utterance.cancel();
+			unsub();
+			this.speech = void 0;
+			this._stopSpeaking = void 0;
+		};
+	}
+	stopSpeaking() {
+		if (!this._stopSpeaking) throw new Error("No message is being spoken");
+		this._stopSpeaking();
+		this._notifySubscribers();
+	}
+	_voiceSession;
+	_voiceUnsubs = [];
+	voice;
+	_voiceVolume = 0;
+	_voiceVolumeSubscribers = /* @__PURE__ */ new Set();
+	getVoiceVolume = () => this._voiceVolume;
+	subscribeVoiceVolume = (callback) => {
+		this._voiceVolumeSubscribers.add(callback);
+		return () => this._voiceVolumeSubscribers.delete(callback);
+	};
+	connectVoice() {
+		const adapter = this.adapters?.voice;
+		if (!adapter) throw new Error("Voice adapter not configured");
+		this.disconnectVoice();
+		const session = adapter.connect({});
+		this._voiceSession = session;
+		const unsubs = [];
+		let currentMode = "listening";
+		this.voice = {
+			status: session.status,
+			isMuted: session.isMuted,
+			mode: currentMode
+		};
+		this._voiceVolume = 0;
+		this._notifySubscribers();
+		unsubs.push(session.onStatusChange((status) => {
+			if (status.type === "ended") {
+				this._finishVoiceAssistantMessage();
+				this._voiceSession = void 0;
+				this.voice = void 0;
+			} else this.voice = {
+				status,
+				isMuted: session.isMuted,
+				mode: currentMode
+			};
+			this._notifySubscribers();
+		}));
+		unsubs.push(session.onModeChange((mode) => {
+			currentMode = mode;
+			if (this.voice) {
+				this.voice = {
+					...this.voice,
+					mode
+				};
+				this._notifySubscribers();
+			}
+		}));
+		unsubs.push(session.onVolumeChange((volume) => {
+			this._voiceVolume = volume;
+			for (const cb of this._voiceVolumeSubscribers) cb();
+		}));
+		unsubs.push(session.onTranscript((transcript) => {
+			this._handleVoiceTranscript(transcript);
+		}));
+		this._voiceUnsubs = unsubs;
+	}
+	_currentAssistantMsg = null;
+	_handleVoiceTranscript(transcript) {
+		this.ensureInitialized();
+		if (transcript.role === "user") {
+			this._finishVoiceAssistantMessage();
+			this._currentAssistantMsg = null;
+			if (transcript.isFinal) {
+				this._voiceMessages.push({
+					id: generateId(),
+					role: "user",
+					content: [{
+						type: "text",
+						text: transcript.text
+					}],
+					metadata: { custom: {} },
+					createdAt: /* @__PURE__ */ new Date(),
+					status: {
+						type: "complete",
+						reason: "unknown"
+					},
+					attachments: []
+				});
+				this._markVoiceMessagesDirty();
+				this._notifySubscribers();
+			}
+		} else {
+			if (!this._currentAssistantMsg) {
+				this._currentAssistantMsg = {
+					id: generateId(),
+					role: "assistant",
+					content: [{
+						type: "text",
+						text: transcript.text
+					}],
+					metadata: {
+						unstable_state: this.state,
+						unstable_annotations: [],
+						unstable_data: [],
+						steps: [],
+						custom: {}
+					},
+					status: { type: "running" },
+					createdAt: /* @__PURE__ */ new Date()
+				};
+				this._voiceMessages.push(this._currentAssistantMsg);
+			} else {
+				const idx = this._voiceMessages.indexOf(this._currentAssistantMsg);
+				if (idx === -1) return;
+				const updated = {
+					...this._currentAssistantMsg,
+					content: [{
+						type: "text",
+						text: transcript.text
+					}],
+					...transcript.isFinal ? { status: {
+						type: "complete",
+						reason: "stop"
+					} } : {}
+				};
+				this._voiceMessages[idx] = updated;
+				this._currentAssistantMsg = updated;
+			}
+			if (transcript.isFinal) this._currentAssistantMsg = null;
+			this._markVoiceMessagesDirty();
+			this._notifySubscribers();
+		}
+	}
+	_finishVoiceAssistantMessage() {
+		const last = this._voiceMessages.at(-1);
+		if (last?.role === "assistant" && last.status.type === "running") {
+			const idx = this._voiceMessages.length - 1;
+			this._voiceMessages[idx] = {
+				...last,
+				status: {
+					type: "complete",
+					reason: "stop"
+				}
+			};
+			this._markVoiceMessagesDirty();
+			this._notifySubscribers();
+		}
+	}
+	disconnectVoice() {
+		this._finishVoiceAssistantMessage();
+		this._currentAssistantMsg = null;
+		for (const unsub of this._voiceUnsubs) unsub();
+		this._voiceUnsubs = [];
+		this._voiceSession?.disconnect();
+		this._voiceSession = void 0;
+		this.voice = void 0;
+		this._voiceVolume = 0;
+		for (const cb of this._voiceVolumeSubscribers) cb();
+		this._voiceMessages = [];
+		this._markVoiceMessagesDirty();
+		this._notifySubscribers();
+	}
+	muteVoice() {
+		if (!this._voiceSession) throw new Error("No active voice session");
+		this._voiceSession.mute();
+		this.voice = {
+			...this.voice,
+			isMuted: true
+		};
+		this._notifySubscribers();
+	}
+	unmuteVoice() {
+		if (!this._voiceSession) throw new Error("No active voice session");
+		this._voiceSession.unmute();
+		this.voice = {
+			...this.voice,
+			isMuted: false
+		};
+		this._notifySubscribers();
+	}
+	ensureInitialized() {
+		if (!this._isInitialized) {
+			this._isInitialized = true;
+			this._notifyEventSubscribers("initialize", {});
+		}
+	}
+	export() {
+		return this.repository.export();
+	}
+	import(data) {
+		this.ensureInitialized();
+		this.repository.clear();
+		this.repository.import(data);
+		this._notifySubscribers();
+	}
+	reset(initialMessages) {
+		this.import(ExportedMessageRepository.fromArray(initialMessages ?? []));
+	}
+	_eventSubscribers = /* @__PURE__ */ new Map();
+	unstable_on(event, callback) {
+		const wrapped = callback;
+		if (event === "modelContextUpdate") return this._contextProvider.subscribe?.(() => wrapped({})) ?? (() => {});
+		let subscribers = this._eventSubscribers.get(event);
+		if (!subscribers) {
+			subscribers = /* @__PURE__ */ new Set();
+			this._eventSubscribers.set(event, subscribers);
+		}
+		subscribers.add(wrapped);
+		if (event === "initialize" && this._isInitialized) queueMicrotask(() => {
+			if (subscribers.has(wrapped)) wrapped({});
+		});
+		return () => {
+			this._eventSubscribers.get(event)?.delete(wrapped);
+		};
+	}
+};
+//#endregion
+export { BaseThreadRuntimeCore };
+
 //# sourceMappingURL=base-thread-runtime-core.js.map