@assistant-ui/core 0.3.17 → 0.3.19

package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.js

@@ -1,14 +1,16 @@
+import { nullProtoRecord } from "../../utils/record.js";
 import { BaseSubscribable, WritableSubscribable } from "../../subscribable/subscribable.js";
+import { invokeUserCallback } from "../../utils/invoke-user-callback.js";
 import { OptimisticState } from "../../runtimes/remote-thread-list/optimistic-state.js";
 import { EMPTY_THREAD_CORE } from "../../runtimes/remote-thread-list/empty-thread-core.js";
-import { classifyThreads, createEmptyRemoteThreadState, createThreadMappingId, getThreadData, normalizeCursor, preserveMidLoadTransitions, seedNewThread, statusSnapshot, updateStatusReducer } from "../../runtimes/remote-thread-list/remote-thread-state.js";
+import { classifyThreads, createEmptyRemoteThreadState, createThreadMappingId, getThreadData, normalizeCursor, preserveMidLoadTransitions, promoteNewThreadReducer, reconcileInitializedThread, seedNewThread, statusSnapshot, updateStatusReducer } from "../../runtimes/remote-thread-list/remote-thread-state.js";
 import { useSubscribable } from "../../store/runtime-clients/useSubscribable.js";
 import { ThreadListAdapterChangedError } from "../../runtimes/remote-thread-list/adapter-changed.js";
 import { useStableRuntimeAdapters } from "./useRuntimeAdapters.js";
-import { isTitleSourceMessage } from "../../runtimes/remote-thread-list/title.js";
+import { applyTitleStream, isTitleSourceMessage } from "../../runtimes/remote-thread-list/title.js";
+import { clearThreadTitleState, finishThreadTitleRename, runThreadTitleGeneration, startThreadTitleRename } from "../../runtimes/remote-thread-list/title-generation.js";
 import { RuntimeAdapterProvider } from "./RuntimeAdapterProvider.js";
 import { RemoteThreadListHookInstanceManager } from "./RemoteThreadListHookInstanceManager.js";
-import { AssistantMessageStream } from "assistant-stream";
 import { Fragment, useEffect, useId } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
 import { jsx, jsxs } from "react/jsx-runtime";
@@ -28,6 +30,7 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 	_staleThreadIdsOnReplace;
 	_switchGeneration = 0;
 	_switchTask;
+	_titleStates = /* @__PURE__ */ new Map();
 	_mainThreadId;
 	_state = new OptimisticState(EMPTY_REMOTE_STATE);
 	_useAdaptersProvider = ({ children }) => {
@@ -65,7 +68,8 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 					if (generation !== this._loadGeneration) return state;
 					return {
 						...state,
-						isLoading: true
+						isLoading: true,
+						loadError: void 0
 					};
 				},
 				then: (state, l) => {
@@ -73,7 +77,10 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 					if (this._replaceListOnNextLoad) {
 						this._replaceListOnNextLoad = false;
 						replacedList = true;
-						return this._replaceWithThreads(state, l.threads, normalizeCursor(l.nextCursor));
+						return this._replaceWithThreads({
+							...state,
+							loadError: void 0
+						}, l.threads, normalizeCursor(l.nextCursor));
 					}
 					const fresh = classifyThreads(l.threads, {
 						threadIds: [],
@@ -84,6 +91,7 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 					const merged = {
 						...state,
 						isLoading: false,
+						loadError: void 0,
 						cursor: normalizeCursor(l.nextCursor),
 						threadIds: fresh.threadIds,
 						archivedThreadIds: fresh.archivedThreadIds,
@@ -99,13 +107,17 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 				if (!this._replaceListOnNextLoad) {
 					this._state.update({
 						...this._state.baseValue,
-						isLoading: false
+						isLoading: false,
+						loadError: error
 					});
 					return;
 				}
 				this._replaceListOnNextLoad = false;
 				replacedList = true;
-				this._state.update(this._replaceWithThreads(this._state.baseValue, [], void 0));
+				this._state.update(this._replaceWithThreads({
+					...this._state.baseValue,
+					loadError: error
+				}, [], void 0));
 			}).then(() => {
 				if (!replacedList) return;
 				const threadId = this._options.threadId;
@@ -197,8 +209,10 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 			this._staleThreadIdsOnReplace = new Set(Object.values(this._state.baseValue.threadData).filter((item) => item.status !== "new").map((item) => item.id));
 			this._state.update({
 				...this._state.baseValue,
-				cursor: void 0
+				cursor: void 0,
+				loadError: void 0
 			});
+			this._titleStates.clear();
 		}
 		if (controlledThreadIdChanged) this._switchToThreadFromProp(options.threadId).catch(() => {});
 	}
@@ -211,8 +225,8 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 	_replaceWithThreads(state, threads, cursor) {
 		const carried = [];
 		if (state.newThreadId) {
-			const mappingId = state.threadIdMap[state.newThreadId];
-			const draft = mappingId ? state.threadData[mappingId] : void 0;
+			const mappingId = Object.hasOwn(state.threadIdMap, state.newThreadId) ? state.threadIdMap[state.newThreadId] : void 0;
+			const draft = mappingId && Object.hasOwn(state.threadData, mappingId) ? state.threadData[mappingId] : void 0;
 			if (draft?.status === "new") carried.push(draft);
 		}
 		const stale = this._staleThreadIdsOnReplace;
@@ -225,12 +239,12 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 		const seed = {
 			threadIds: [],
 			archivedThreadIds: [],
-			threadIdMap: {},
-			threadData: {}
+			threadIdMap: nullProtoRecord(),
+			threadData: nullProtoRecord()
 		};
 		for (const item of carried) {
 			const mappingId = createThreadMappingId(item.id);
-			if (seed.threadData[mappingId]) continue;
+			if (Object.hasOwn(seed.threadData, mappingId)) continue;
 			seed.threadIdMap[item.id] = mappingId;
 			if (item.remoteId !== void 0) seed.threadIdMap[item.remoteId] = mappingId;
 			seed.threadData[mappingId] = item;
@@ -238,7 +252,8 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 		const { threadIds, archivedThreadIds, threadIdMap, threadData } = classifyThreads(threads, seed);
 		for (const item of carried) {
 			if (item.remoteId === void 0) continue;
-			const current = threadData[createThreadMappingId(item.id)];
+			const currentMappingId = createThreadMappingId(item.id);
+			const current = Object.hasOwn(threadData, currentMappingId) ? threadData[currentMappingId] : void 0;
 			if (current === void 0) continue;
 			if (current.status === "regular" && !threadIds.includes(current.id)) threadIds.push(current.id);
 			else if (current.status === "archived" && !archivedThreadIds.includes(current.id)) archivedThreadIds.push(current.id);
@@ -251,7 +266,7 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 			archivedThreadIds,
 			threadIdMap,
 			threadData,
-			newThreadId: state.newThreadId !== void 0 && threadIdMap[state.newThreadId] === void 0 ? void 0 : state.newThreadId
+			newThreadId: state.newThreadId !== void 0 && !Object.hasOwn(threadIdMap, state.newThreadId) ? void 0 : state.newThreadId
 		};
 		if (getThreadData(nextState, this._mainThreadId) === void 0) {
 			const preservedDraft = nextState.newThreadId;
@@ -304,6 +319,9 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 	get isLoading() {
 		return this._state.value.isLoading;
 	}
+	get loadError() {
+		return this._state.value.loadError;
+	}
 	get isLoadingMore() {
 		return this._state.value.isLoadingMore;
 	}
@@ -331,7 +349,7 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 		const threadId = this._mainThreadRemoteId;
 		if (this._lastNotifiedThreadId === threadId) return;
 		this._lastNotifiedThreadId = threadId;
-		if (emit) this._options.onThreadIdChange?.(threadId);
+		if (emit) invokeUserCallback("assistant-ui", "onThreadIdChange", this._options.onThreadIdChange, threadId);
 	}
 	getMainThreadRuntimeCore() {
 		const result = this._hookManager.getThreadRuntimeCore(this._mainThreadId);
@@ -473,65 +491,28 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 				externalId
 			};
 		}
+		this._requireAdapterGeneration(adapterGeneration);
+		const initializeTask = adapter.initialize(threadId);
+		let removedMappingId;
 		const { remoteId, externalId } = await this._state.optimisticUpdate({
-			execute: () => {
-				this._requireAdapterGeneration(adapterGeneration);
-				return adapter.initialize(threadId);
-			},
-			optimistic: (state) => {
-				return updateStatusReducer(state, threadId, "regular");
-			},
-			loading: (state, task) => {
-				const mappingId = createThreadMappingId(threadId);
-				return {
-					...state,
-					threadData: {
-						...state.threadData,
-						[mappingId]: {
-							...state.threadData[mappingId],
-							initializeTask: task
-						}
-					}
-				};
-			},
+			execute: () => initializeTask,
+			optimistic: (state) => promoteNewThreadReducer(state, threadId, initializeTask),
 			then: (state, { remoteId, externalId }) => {
 				if (adapterGeneration !== this._adapterGeneration) return state;
-				const data = getThreadData(state, threadId);
-				if (!data) return state;
-				const mappingId = createThreadMappingId(threadId);
-				const listedMappingId = state.threadIdMap[remoteId];
-				const orphan = listedMappingId !== void 0 && listedMappingId !== mappingId ? state.threadData[listedMappingId] : void 0;
-				const threadData = { ...state.threadData };
-				if (orphan !== void 0) delete threadData[listedMappingId];
-				threadData[mappingId] = {
-					...data,
-					initializeTask: Promise.resolve({
-						remoteId,
-						externalId
-					}),
-					remoteId,
-					externalId
-				};
-				const rewire = (ids) => orphan === void 0 ? ids : ids.filter((id) => id !== orphan.id);
-				return {
-					...state,
-					threadIds: rewire(state.threadIds),
-					archivedThreadIds: rewire(state.archivedThreadIds),
-					threadIdMap: {
-						...state.threadIdMap,
-						[remoteId]: mappingId
-					},
-					threadData
-				};
+				const reconciliation = reconcileInitializedThread(state, threadId, remoteId, externalId, threadId);
+				removedMappingId = reconciliation.removedMappingId;
+				if (removedMappingId === this._mainThreadId) this._mainThreadId = reconciliation.survivorMappingId;
+				return reconciliation.state;
 			}
 		});
 		this._requireAdapterGeneration(adapterGeneration);
+		if (removedMappingId !== void 0) this._hookManager.stopThreadRuntime(removedMappingId);
 		return {
 			remoteId,
 			externalId
 		};
 	};
-	generateTitle = async (threadId) => {
+	generateTitle = async (threadId, options) => {
 		this._requireAdapterSettled();
 		const adapter = this._options.adapter;
 		const adapterGeneration = this._adapterGeneration;
@@ -543,16 +524,58 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 		const runtimeCore = this._hookManager.getThreadRuntimeCore(data.id);
 		if (!runtimeCore) return;
 		const messages = runtimeCore.messages.filter(isTitleSourceMessage);
-		const stream = await adapter.generateTitle(remoteId, messages);
-		const messageStream = AssistantMessageStream.fromAssistantStream(stream);
-		for await (const result of messageStream) {
-			if (adapterGeneration !== this._adapterGeneration) return;
-			const newTitle = result.parts.filter((c) => c.type === "text")[0]?.text;
-			await this._state.optimisticUpdate({
-				execute: async () => {},
+		await runThreadTitleGeneration({
+			states: this._titleStates,
+			threadId: data.id,
+			automatic: options?.automatic === true,
+			generate: async (onTitle) => {
+				const stream = await adapter.generateTitle(remoteId, messages);
+				this._requireAdapterGeneration(adapterGeneration);
+				await applyTitleStream(stream, onTitle);
+			},
+			rename: async (title) => {
+				this._requireAdapterGeneration(adapterGeneration);
+				await adapter.rename(remoteId, title);
+			},
+			applyTitle: async (title) => {
+				await this._state.optimisticUpdate({
+					execute: async () => {},
+					optimistic: (state) => {
+						if (adapterGeneration !== this._adapterGeneration) return state;
+						const currentData = getThreadData(state, data.id);
+						if (!currentData) return state;
+						return {
+							...state,
+							threadData: {
+								...state.threadData,
+								[currentData.id]: {
+									...currentData,
+									title
+								}
+							}
+						};
+					}
+				});
+			}
+		});
+	};
+	async rename(threadIdOrRemoteId, newTitle) {
+		this._requireAdapterSettled();
+		const adapter = this._options.adapter;
+		const adapterGeneration = this._adapterGeneration;
+		const data = this.getItemById(threadIdOrRemoteId);
+		if (!data) throw threadNotFoundError(threadIdOrRemoteId, "renaming it");
+		if (data.status === "new") throw threadStatusError(threadIdOrRemoteId, data.status, "be renamed");
+		const claim = startThreadTitleRename(this._titleStates, data.id, newTitle);
+		try {
+			const result = await this._state.optimisticUpdate({
+				execute: async () => {
+					const { remoteId } = await data.initializeTask;
+					this._requireAdapterGeneration(adapterGeneration);
+					return adapter.rename(remoteId, newTitle);
+				},
 				optimistic: (state) => {
-					if (adapterGeneration !== this._adapterGeneration) return state;
-					const currentData = getThreadData(state, data.id);
+					const currentData = getThreadData(state, threadIdOrRemoteId);
 					if (!currentData) return state;
 					return {
 						...state,
@@ -566,36 +589,12 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 					};
 				}
 			});
+			finishThreadTitleRename(this._titleStates, data.id, claim, true);
+			return result;
+		} catch (error) {
+			finishThreadTitleRename(this._titleStates, data.id, claim, false);
+			throw error;
 		}
-	};
-	async rename(threadIdOrRemoteId, newTitle) {
-		this._requireAdapterSettled();
-		const adapter = this._options.adapter;
-		const adapterGeneration = this._adapterGeneration;
-		const data = this.getItemById(threadIdOrRemoteId);
-		if (!data) throw threadNotFoundError(threadIdOrRemoteId, "renaming it");
-		if (data.status === "new") throw threadStatusError(threadIdOrRemoteId, data.status, "be renamed");
-		return this._state.optimisticUpdate({
-			execute: async () => {
-				const { remoteId } = await data.initializeTask;
-				this._requireAdapterGeneration(adapterGeneration);
-				return adapter.rename(remoteId, newTitle);
-			},
-			optimistic: (state) => {
-				const data = getThreadData(state, threadIdOrRemoteId);
-				if (!data) return state;
-				return {
-					...state,
-					threadData: {
-						...state.threadData,
-						[data.id]: {
-							...data,
-							title: newTitle
-						}
-					}
-				};
-			}
-		});
 	}
 	async updateCustom(threadIdOrRemoteId, custom) {
 		this._requireAdapterSettled();
@@ -695,8 +694,7 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 		if (data.status !== "regular" && data.status !== "archived") throw threadStatusError(threadIdOrRemoteId, data.status, "be deleted");
 		await this._ensureThreadIsNotMain(data.id);
 		this._requireAdapterGeneration(adapterGeneration);
-		this._hookManager.stopThreadRuntime(data.id);
-		return this._state.optimisticUpdate({
+		const result = await this._state.optimisticUpdate({
 			execute: async () => {
 				const { remoteId } = await data.initializeTask;
 				this._requireAdapterGeneration(adapterGeneration);
@@ -706,6 +704,9 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
 				return updateStatusReducer(state, data.id, "deleted");
 			}
 		});
+		this._hookManager.stopThreadRuntime(data.id);
+		clearThreadTitleState(this._titleStates, data.id);
+		return result;
 	}
 	__internal_dispose() {
 		this._hookManager.__internal_dispose();