@epam/ai-dial-chat-hooks 1.3.0-dev.32 → 1.3.0-dev.321

package/conversation/useConversationStream/useConversationStream.js

@@ -1,336 +1,345 @@
 import { safeDecodeURI as e } from "../../shared/string-utils.js";
-import { DEFAULT_GENERATION_CONFLICT_MESSAGE as t, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as n, GenerationConflictError as r, GenerationPersistenceError as i, StreamInterruptedError as a, StreamUpstreamError as o } from "../create-chat-stream-api.js";
-import { getConversationPath as s } from "./conversation-path.js";
-import { applyChunkToMessages as c } from "./apply-chunk.js";
-import { restoreBufferedMessage as l } from "./buffered-generation.js";
-import { createResumeIfAwaitingGeneration as ee, fetchConversationForRecovery as te, findPendingBackgroundMessageIndex as u, hasGeneratedPayload as ne, isAwaitingGenerationResume as d } from "./generation-resume.js";
+import { DEFAULT_GENERATION_CONFLICT_MESSAGE as t, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as n, GenerationConflictError as r, GenerationPersistenceError as i, StopCompletionError as a, StreamInterruptedError as o, StreamUpstreamError as s } from "../create-chat-stream-api.js";
+import { getConversationPath as c } from "./conversation-path.js";
+import { applyChunkToMessages as l } from "./apply-chunk.js";
+import { restoreBufferedMessage as u } from "./buffered-generation.js";
+import { createResumeIfAwaitingGeneration as ee, fetchConversationForRecovery as te, findPendingBackgroundMessageIndex as d, hasGeneratedPayload as ne, isAwaitingGenerationResume as f } from "./generation-resume.js";
 import { createFrameScheduler as re } from "./frame-scheduler.js";
-import { MessageRole as ie, generateUUID as ae } from "@epam/ai-dial-chat-shared";
-import { useCallback as f, useEffect as p, useMemo as oe, useRef as m, useState as h } from "react";
-import { SendCompletionDtoModeEnum as g } from "@epam/ai-dial-chat-api-client";
+import { SendCompletionDtoModeEnum as ie } from "@epam/ai-dial-chat-api-client";
+import { MessageRole as ae, generateUUID as oe } from "@epam/ai-dial-chat-shared";
+import { useCallback as p, useEffect as m, useMemo as h, useRef as g, useState as _ } from "react";
 //#region src/conversation/useConversationStream/useConversationStream.ts
-var se = 2e4, ce = 401, le = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof o ? e.message : "", _ = (e) => {
+var se = 2e4, ce = 401, v = 404, y = 2e3, b = 100, le = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof s ? e.message : "", x = (e) => {
 	if (!e) return;
-	let t = u(e);
+	let t = d(e);
 	return t === -1 ? void 0 : e.messages[t].backgroundGeneration?.generationId;
-}, v = ({ conversationId: o, state: { setConversation: v, conversationRef: y }, transport: b, generation: { startGeneration: ue, completeGeneration: x }, channel: S, overlay: C, onStopError: w, generationConflictMessage: de = t, generationPersistenceErrorMessage: T = n, onStreamError: E, batchChunksPerFrame: D = !1 }) => {
-	let O = m(E);
-	p(() => {
-		O.current = E;
-	}, [E]);
-	let fe = m(D);
-	p(() => {
-		fe.current = D;
-	}, [D]);
-	let [k] = h(re), [A, j] = h(() => /* @__PURE__ */ new Set()), [M, N] = h(() => /* @__PURE__ */ new Set()), P = m(/* @__PURE__ */ new Map()), F = m(/* @__PURE__ */ new Set()), I = m(/* @__PURE__ */ new Map()), L = m(/* @__PURE__ */ new Set()), pe = m(/* @__PURE__ */ new Map()), R = m(!0), z = m(/* @__PURE__ */ new Map()), [B, V] = h(z.current), H = m(/* @__PURE__ */ new Set()), [me, U] = h(H.current);
-	p(() => (R.current = !0, () => {
-		R.current = !1;
+}, S = ({ conversationId: s, state: { setConversation: S, conversationRef: C }, transport: w, generation: { startGeneration: ue, completeGeneration: de }, channel: T, overlay: E, onStopError: fe, generationConflictMessage: pe = t, generationPersistenceErrorMessage: D = n, onStreamError: O, batchChunksPerFrame: k = !1 }) => {
+	let A = g(O);
+	m(() => {
+		A.current = O;
+	}, [O]);
+	let me = g(k);
+	m(() => {
+		me.current = k;
+	}, [k]);
+	let [j] = _(re), [M, N] = _(() => /* @__PURE__ */ new Set()), P = g(M);
+	P.current = M;
+	let [F, he] = _(() => /* @__PURE__ */ new Set()), I = g(/* @__PURE__ */ new Map()), L = g(/* @__PURE__ */ new Map()), R = g(/* @__PURE__ */ new Set()), z = g(/* @__PURE__ */ new Map()), B = g(/* @__PURE__ */ new Set()), V = g(/* @__PURE__ */ new Set()), ge = g(/* @__PURE__ */ new Map()), H = g(!0), U = g(/* @__PURE__ */ new Map()), [_e, ve] = _(U.current), W = g(/* @__PURE__ */ new Set()), [ye, be] = _(W.current);
+	m(() => (H.current = !0, () => {
+		H.current = !1;
 	}), []);
-	let W = f((e) => {
-		if (!z.current.has(e)) return;
-		let t = new Map(z.current);
-		t.delete(e), z.current = t, R.current && V(t);
-	}, []), G = f((e, t, n) => {
-		if (!R.current || !n()) return;
-		let r = new Map(z.current);
+	let G = p((e) => {
+		if (!U.current.has(e)) return;
+		let t = new Map(U.current);
+		t.delete(e), U.current = t, H.current && ve(t);
+	}, []), K = p((e, t, n) => {
+		if (!H.current || !n()) return;
+		let r = new Map(U.current);
 		r.set(e, {
 			retry: t,
 			isCurrent: n
-		}), z.current = r, V(r);
-	}, []), he = f(async () => {
-		if (!o) return;
-		let e = z.current.get(s(o));
-		if (!(!e || !e.isCurrent() || H.current.has(e))) {
-			H.current.add(e), U(new Set(H.current));
+		}), U.current = r, ve(r);
+	}, []), xe = p(async () => {
+		if (!s) return;
+		let e = U.current.get(c(s));
+		if (e && e.isCurrent() && !W.current.has(e)) {
+			W.current.add(e), be(new Set(W.current));
 			try {
 				await e.retry();
 			} finally {
-				H.current.delete(e), R.current && U(new Set(H.current));
+				W.current.delete(e), H.current && be(new Set(W.current));
 			}
 		}
-	}, [o]), K = m(o);
-	p(() => (K.current = o, () => k.cancelAll()), [o, k]);
-	let q = f((e) => R.current && s(K.current ?? "") === e, []), J = f((e) => {
-		W(e), j((t) => {
+	}, [s]), Se = g(s);
+	m(() => (Se.current = s, () => j.cancelAll()), [s, j]);
+	let q = p((e) => H.current && c(Se.current ?? "") === e, []), J = p((e) => {
+		G(e), N((t) => {
 			if (t.has(e)) return t;
 			let n = new Set(t);
 			return n.add(e), n;
 		});
-	}, [W]), Y = f((e) => {
-		j((t) => {
+	}, [G]), Y = p((e) => {
+		N((t) => {
 			if (!t.has(e)) return t;
 			let n = new Set(t);
 			return n.delete(e), n;
 		});
-	}, []), X = f((e, t) => {
-		N((n) => {
+	}, []), X = p((e, t) => {
+		he((n) => {
 			if (n.has(e) === t) return n;
 			let r = new Set(n);
 			return t ? r.add(e) : r.delete(e), r;
 		});
-	}, []), Z = f((e, t) => {
-		P.current.get(e) === t && (P.current.delete(e), X(e, !1));
-	}, [X]), Q = oe(() => ee({
-		transport: b,
-		setConversation: v,
-		conversationRef: y,
-		resumingPathsRef: F,
-		bufferedGenerationsRef: I,
+	}, []), Ce = p((e, t) => {
+		I.current.get(e) === t && (I.current.delete(e), X(e, !1));
+	}, [X]), we = p((e, t) => {
+		I.current.has(e) || (L.current.set(e, t), X(e, !0));
+	}, [X]), Te = p((e, t) => {
+		L.current.get(e) === t && (L.current.delete(e), I.current.has(e) || X(e, !1));
+	}, [X]), Z = h(() => ee({
+		transport: w,
+		setConversation: S,
+		conversationRef: C,
+		resumingPathsRef: R,
+		bufferedGenerationsRef: z,
 		addStreamingPath: J,
 		removeStreamingPath: Y,
 		isPathDisplayed: q,
-		generationPersistenceErrorMessage: T,
-		frameScheduler: D ? k : void 0,
-		stoppedGenerationIdsRef: L,
-		onReloadError: G,
-		onReloadSuccess: W
+		generationPersistenceErrorMessage: D,
+		frameScheduler: k ? j : void 0,
+		stoppedGenerationIdsRef: B,
+		rememberResumedGenerationId: we,
+		forgetResumedGenerationId: Te,
+		onReloadError: K,
+		onReloadSuccess: G
 	}), [
 		J,
 		Y,
 		q,
-		b,
-		T,
+		w,
 		D,
 		k,
-		G,
-		W
-	]), ge = f((t, n, o, ee, re, f, p = g.Append, oe = {}) => {
-		let m = f ?? ae(), h = s(t);
-		k.flush(h), F.current.delete(h), P.current.set(h, m), pe.current.set(h, m), X(h, !0);
+		j,
+		we,
+		Te,
+		K,
+		G
+	]), Ee = p((t, n, a, s, ee, re, p = ie.Append, m = {}) => {
+		let h = re ?? oe(), g = c(t);
+		j.flush(g), R.current.delete(g), I.current.set(g, h), ge.current.set(g, h), X(g, !0);
 		let _;
-		p === g.Regenerate ? _ = o : p === g.Edit && (_ = o - 1);
-		let w = () => pe.current.get(h) !== m, E = ue(h, m), D = y.current?.messages[o];
-		D ? I.current.set(h, {
-			generationId: m,
-			messageIndex: o,
-			message: D
-		}) : I.current.delete(h), J(h), C?.notifyGenerationStart?.(), S?.ensureConnected();
-		let A = () => {
-			w() || Y(h), Z(h, m), x(h, m), S?.notifyGenerationSettled?.();
-		}, j = () => {
-			if (w()) return;
-			let e = I.current.get(h);
-			!e || e.generationId !== m || (e.message = {
+		p === ie.Regenerate ? _ = a : p === ie.Edit && (_ = a - 1);
+		let v = () => ge.current.get(g) !== h, y = ue(g, h), b = C.current?.messages[a];
+		b ? z.current.set(g, {
+			generationId: h,
+			messageIndex: a,
+			message: b
+		}) : z.current.delete(g), J(g), E?.notifyGenerationStart?.(), T?.ensureConnected();
+		let x = () => {
+			v() || Y(g), Ce(g, h), de(g, h), T?.notifyGenerationSettled?.();
+		}, fe = () => {
+			if (v()) return;
+			let e = z.current.get(g);
+			e && e.generationId === h && (e.message = {
 				...e.message,
-				streamErrorMessage: T
-			}, O.current?.(new i()), q(h) && v((t) => {
-				if (!t || w() || !q(h)) return t;
-				let n = l(t, e);
-				return y.current = n, n;
+				streamErrorMessage: D
+			}, A.current?.(new i()), q(g) && S((t) => {
+				if (!t || v() || !q(g)) return t;
+				let n = u(t, e);
+				return C.current = n, n;
 			}));
-		}, M = I.current.get(h), N = () => R.current && !w() && I.current.get(h) === M, z = async () => {
-			if (N()) {
-				if (!q(h)) {
-					M && I.current.delete(h);
-					return;
-				}
-				try {
-					let n = await b.getConversation(e(t));
-					if (!N()) return;
-					if (W(h), !q(h)) {
-						M && I.current.delete(h);
+		}, O = z.current.get(g), k = () => H.current && !v() && z.current.get(g) === O, M = async () => {
+			if (k()) {
+				if (!q(g)) O && z.current.delete(g);
+				else try {
+					let n = await w.getConversation(e(t));
+					if (!k()) return;
+					if (G(g), !q(g)) {
+						O && z.current.delete(g);
 						return;
 					}
-					if (u(n) !== -1) {
-						Q(t, n, { seedMessage: M?.message });
+					if (d(n) !== -1) {
+						Z(t, n, { seedMessage: O?.message });
 						return;
 					}
-					if (M && M.messageIndex === n.messages.length - 1 && d(n) && ne(M.message)) {
-						j();
+					if (O && O.messageIndex === n.messages.length - 1 && f(n) && ne(O.message)) {
+						fe();
 						return;
 					}
-					M && I.current.delete(h), v(n), y.current = n;
+					O && z.current.delete(g), S(n), C.current = n;
 				} catch {
-					N() && G(h, z, N);
+					k() && K(g, M, k);
 				}
 			}
-		}, B = {
-			signal: E.signal,
+		}, N = {
+			signal: y.signal,
 			onChunk: (e) => {
-				if (w()) return;
-				let t = I.current.get(h);
-				if (t?.generationId === m) {
-					let n = c([t.message], 0, e);
+				if (v()) return;
+				let t = z.current.get(g);
+				if (t?.generationId === h) {
+					let n = l([t.message], 0, e);
 					n && (t.message = n[0]);
 				}
-				if (q(h)) {
-					if (fe.current && t?.generationId === m) {
-						k.schedule(h, () => {
-							let e = I.current.get(h);
-							if (w() || !q(h) || e?.generationId !== m) return;
-							let t = { ...e };
-							v((e) => {
-								if (!e) return e;
-								let n = l(e, t);
-								return y.current = n, n;
-							});
-						});
-						return;
-					}
-					v((t) => {
-						if (!t) return t;
-						let n = I.current.get(h), r;
-						if (n?.generationId === m) r = l(t, n);
-						else {
-							let n = c(t.messages, o, e);
-							if (!n) return t;
-							r = {
-								...t,
-								messages: n
-							};
-						}
-						return y.current = r, r;
+				q(g) && (me.current && t?.generationId === h ? j.schedule(g, () => {
+					let e = z.current.get(g);
+					if (v() || !q(g) || e?.generationId !== h) return;
+					let t = { ...e };
+					S((e) => {
+						if (!e) return e;
+						let n = u(e, t);
+						return C.current = n, n;
 					});
-				}
+				}) : S((t) => {
+					if (!t) return t;
+					let n = z.current.get(g), r;
+					if (n?.generationId === h) r = u(t, n);
+					else {
+						let n = l(t.messages, a, e);
+						if (!n) return t;
+						r = {
+							...t,
+							messages: n
+						};
+					}
+					return C.current = r, r;
+				}));
 			},
 			onComplete: async () => {
-				k.flush(h), A(), L.current.has(m) ? L.current.delete(m) : w() || C?.notifyGenerationEnd?.(), await z();
+				j.flush(g), x(), B.current.has(h) ? B.current.delete(h) : v() || E?.notifyGenerationEnd?.(), await M();
 			},
 			onError: (e) => {
-				if (k.flush(h), O.current?.(e), e instanceof a && !w()) {
-					he(e);
-					return;
-				}
-				if (e instanceof r && oe.resumeOnConflict && !w()) {
-					K(e);
-					return;
-				}
-				V(e);
-			}
-		}, V = (e) => {
-			let t = I.current.get(h), n = t?.generationId === m ? t : void 0;
-			if (n && !(e instanceof i) && I.current.delete(h), A(), n && b.saveConversation && e instanceof i && e.status === ce) {
-				me(e, n, b.saveConversation.bind(b));
-				return;
+				j.flush(g), A.current?.(e), e instanceof o && !v() ? V(e) : e instanceof r && m.resumeOnConflict && !v() ? U(e) : P(e);
 			}
-			H(e, n);
-		}, H = (e, t) => {
+		}, P = (e) => {
+			let t = z.current.get(g), n = t?.generationId === h ? t : void 0;
+			n && !(e instanceof i) && z.current.delete(g), x(), n && w.saveConversation && e instanceof i && e.status === ce ? he(e, n, w.saveConversation.bind(w)) : F(e, n);
+		}, F = (e, t) => {
 			if (t && e instanceof i && (t.message = {
 				...t.message,
-				streamErrorMessage: T
-			}), w() || !q(h)) return;
-			let n = le(e, de, T);
-			v((e) => {
-				if (!e || w() || !q(h)) return e;
-				let r = t?.generationId === m ? l(e, t) : e, i = {
+				streamErrorMessage: D
+			}), v() || !q(g)) return;
+			let n = le(e, pe, D);
+			S((e) => {
+				if (!e || v() || !q(g)) return e;
+				let r = t?.generationId === h ? u(e, t) : e, i = {
 					...r,
-					messages: r.messages.map((e, t) => t === o ? {
+					messages: r.messages.map((e, t) => t === a ? {
 						...e,
 						streamErrorMessage: n
 					} : e)
 				};
-				return y.current = i, i;
+				return C.current = i, i;
 			});
-		}, me = async (n, r, i) => {
-			let a = () => R.current && !w() && I.current.get(h) === r;
+		}, he = async (n, r, i) => {
+			let a = () => H.current && !v() && z.current.get(g) === r;
 			try {
-				let o = await b.getConversation(e(t));
+				let o = await w.getConversation(e(t));
 				if (!a()) return;
-				if (r.messageIndex !== o.messages.length - 1 || !d(o)) {
-					H(n, r);
+				if (r.messageIndex !== o.messages.length - 1 || !f(o)) {
+					F(n, r);
 					return;
 				}
-				let s = await i(h, l(o, r));
-				if (!a() || (I.current.delete(h), !q(h))) return;
-				v(s), y.current = s;
+				let s = n.terminalMarker ? {
+					...r,
+					message: {
+						...r.message,
+						...n.terminalMarker
+					}
+				} : r, c = await i(g, u(o, s));
+				if (!a() || (z.current.delete(g), !q(g))) return;
+				S(c), C.current = c;
 			} catch {
-				a() && H(n, r);
+				a() && F(n, r);
 			}
-		}, U = () => {
-			A(), L.current.has(m) ? L.current.delete(m) : w() || C?.notifyGenerationEnd?.();
-		}, he = async (n) => {
-			F.current.add(h), ge(await te(() => b.getConversation(e(t)), w), n, !1);
-		}, K = async (n) => {
-			P.current.get(h) === m && X(h, !1), F.current.add(h), ge(await te(() => b.getConversation(e(t)), w, (e) => e.messages.length === o && e.messages.at(-1)?.role === ie.User), n, !0);
-		}, ge = (e, n, r) => {
-			if (w()) {
-				A();
+		}, L = () => {
+			x(), B.current.has(h) ? B.current.delete(h) : v() || E?.notifyGenerationEnd?.();
+		}, V = async (n) => {
+			R.current.add(g);
+			let r = await te(() => w.getConversation(e(t)), v);
+			_e(r, n, !1);
+		}, U = async (n) => {
+			I.current.get(g) === h && X(g, !1), R.current.add(g);
+			let r = await te(() => w.getConversation(e(t)), v, (e) => e.messages.length === a && e.messages.at(-1)?.role === ae.User);
+			_e(r, n, !0);
+		}, _e = (e, n, r) => {
+			if (v()) {
+				x();
 				return;
 			}
-			let i = e?.messages[e.messages.length - 1], a = e != null && e.messages.length - 1 === o && i?.role === ie.Assistant, s = r && e != null && u(e) !== -1;
-			if (!e || !a && !s) {
-				F.current.delete(h), V(n);
+			let i = e?.messages[e.messages.length - 1], o = e != null && e.messages.length - 1 === a && i?.role === ae.Assistant, s = r && e != null && d(e) !== -1;
+			if (!e || !o && !s) {
+				R.current.delete(g), P(n);
 				return;
 			}
-			let c = I.current.get(h), l = c?.generationId === m ? c : void 0;
-			if (d(e)) {
-				Q(t, e, {
-					seedMessage: a ? l?.message : void 0,
-					skipDedupe: !0,
-					onSettled: U
-				});
-				return;
-			}
-			F.current.delete(h), l && I.current.delete(h), q(h) && (v(e), y.current = e), U();
+			let c = z.current.get(g), l = c?.generationId === h ? c : void 0;
+			f(e) ? Z(t, e, {
+				seedMessage: o ? l?.message : void 0,
+				skipDedupe: !0,
+				onSettled: L
+			}) : (R.current.delete(g), l && z.current.delete(g), q(g) && (S(e), C.current = e), L());
 		};
 		(async () => {
 			try {
-				let e = S?.channelId ?? await S?.waitForChannel(se) ?? void 0, t = L.current.has(m);
-				if (E.signal.aborted || w() || t) {
-					I.current.get(h)?.generationId === m && I.current.delete(h), w() || Y(h), Z(h, m), x(h, m), S?.notifyGenerationSettled?.(), t ? L.current.delete(m) : w() || C?.notifyGenerationEnd?.();
+				let e = T?.channelId ?? await T?.waitForChannel(se) ?? void 0, t = B.current.has(h);
+				if (y.signal.aborted || v() || t) {
+					z.current.get(g)?.generationId === h && z.current.delete(g), v() || Y(g), Ce(g, h), de(g, h), T?.notifyGenerationSettled?.(), t ? B.current.delete(h) : v() || E?.notifyGenerationEnd?.();
 					return;
 				}
-				b.streamCompletion(h, n, ee, B, re, m, p, _, e);
+				w.streamCompletion(g, n, s, N, ee, h, p, _, e);
 			} catch (e) {
-				B.onError(e instanceof Error ? e : Error(String(e)));
+				N.onError(e instanceof Error ? e : Error(String(e)));
 			}
 		})();
 	}, [
 		ue,
-		x,
+		de,
 		J,
 		Y,
 		X,
-		Z,
+		Ce,
 		q,
-		S?.channelId,
-		S?.ensureConnected,
-		S?.waitForChannel,
-		S?.notifyGenerationSettled,
-		C,
-		b,
-		Q,
-		de,
-		T,
-		k,
-		G,
-		W
-	]), _e = f((e, t) => {
-		let n = s(e), r = I.current.get(n);
-		return z.current.get(n)?.isCurrent() && r && t.messages.length - 1 >= r.messageIndex && t.messages[r.messageIndex].role === ie.Assistant && !d(t) ? (W(n), I.current.delete(n), t) : r ? l(t, r) : t;
-	}, [W]), ve = f(() => {
-		if (!o) return;
-		let e = s(o), t = P.current.get(e) ?? _(y.current);
-		if (!t) return;
-		L.current.add(t), k.flush(e), C?.notifyStopGenerating?.();
-		let n = I.current.get(e)?.message.content;
-		b.stopCompletion({
-			generationId: t,
-			path: e,
-			content: n
-		}).catch((e) => {
-			let t = e instanceof Error ? e : Error(String(e));
-			w?.(t);
-		});
-	}, [
-		o,
-		y,
-		k,
+		T?.channelId,
+		T?.ensureConnected,
+		T?.waitForChannel,
+		T?.notifyGenerationSettled,
+		E,
 		w,
+		Z,
+		pe,
+		D,
+		j,
+		K,
+		G
+	]), De = p((e, t) => {
+		let n = c(e), r = z.current.get(n);
+		return U.current.get(n)?.isCurrent() && r && t.messages.length - 1 >= r.messageIndex && t.messages[r.messageIndex].role === ae.Assistant && !f(t) ? (G(n), z.current.delete(n), t) : r ? u(t, r) : t;
+	}, [G]), Oe = p(() => {
+		if (!s) return;
+		let e = c(s), t = I.current.get(e), n = t ?? x(C.current) ?? L.current.get(e);
+		if (!n || V.current.has(n)) return;
+		V.current.add(n), B.current.add(n), j.flush(e), E?.notifyStopGenerating?.();
+		let r = z.current.get(e)?.message.content, i = () => I.current.get(e) === n || t == null && P.current.has(e), o = async () => {
+			let e = y / b;
+			for (let t = 0; t < e; t++) {
+				if (!i() || !H.current) return !0;
+				await new Promise((e) => setTimeout(e, b));
+			}
+			return !i();
+		};
+		(async () => {
+			try {
+				await w.stopCompletion({
+					generationId: n,
+					path: e,
+					content: r
+				});
+			} catch (e) {
+				let t = e instanceof Error ? e : Error(String(e));
+				if (t instanceof a && t.status === v && await o()) return;
+				V.current.delete(n), fe?.(t);
+			}
+		})();
+	}, [
+		s,
 		C,
-		b
-	]), ye = f((e, t) => Q(e, t), [Q]), be = o != null && A.has(s(o)), $ = o == null ? null : s(o), xe = $ != null && (M.has($) || be && _(y.current) != null), Se = $ ? B.get($) : void 0, Ce = Se?.isCurrent() ?? !1;
+		j,
+		fe,
+		E,
+		w
+	]), ke = p((e, t) => {
+		I.current.has(c(e)) || Z(e, t, { trackAttachGenerationId: !0 });
+	}, [Z]), Ae = s != null && M.has(c(s)), Q = s == null ? null : c(s), je = Q != null && (F.has(Q) || Ae && x(C.current) != null), $ = Q ? _e.get(Q) : void 0, Me = $?.isCurrent() ?? !1;
 	return {
-		hasConversationReloadError: Ce,
-		isReloadingConversation: !!Se && Ce && me.has(Se),
-		retryConversationReload: he,
-		startStream: ge,
-		handleStop: ve,
-		resumeIfAwaitingGeneration: ye,
-		restoreBufferedGeneration: _e,
-		isStreaming: be,
-		canStopStreaming: xe
+		hasConversationReloadError: Me,
+		isReloadingConversation: !!$ && Me && ye.has($),
+		retryConversationReload: xe,
+		startStream: Ee,
+		handleStop: Oe,
+		resumeIfAwaitingGeneration: ke,
+		restoreBufferedGeneration: De,
+		isStreaming: Ae,
+		canStopStreaming: je
 	};
 };
 //#endregion
-export { v as useConversationStream };
+export { S as useConversationStream };

package/conversation/useConversationTitleEnrichmentController.d.ts

@@ -0,0 +1,64 @@
+import { Dispatch, SetStateAction } from 'react';
+/** Completion outcome of one incremental request. */
+export declare enum ConversationTitleRequestEnd {
+    Complete = "complete",
+    Budget = "budget",
+    Unsupported = "unsupported",
+    Stopped = "stopped"
+}
+/** Minimal list row required for title reconciliation. */
+export interface ConversationTitleItem {
+    /** Exact resource identity. */
+    id: string;
+    /** Current display title. */
+    title: string;
+    /** Whether the title still needs enrichment. */
+    isTitlePending?: boolean;
+}
+/** Outcome of an incremental transport call. */
+export interface ConversationTitleTransportResult {
+    /** Completion or fallback outcome. */
+    end: ConversationTitleRequestEnd;
+    /** IDs with a terminal outcome, including missing titles. */
+    settledIds: readonly string[];
+}
+/** Host transports and scheduling policies. Keep this object stable. */
+export interface ConversationTitleEnrichmentOptions {
+    /** Fetches one JSON batch. */
+    getTitles: (ids: string[], signal: AbortSignal) => Promise<{
+        items: {
+            id: string;
+            title: string;
+        }[];
+    }>;
+    /** Optional incremental transport; omitted for JSON-only hosts. */
+    streamTitles?: (ids: string[], signal: AbortSignal, onTitle: (id: string, title: string) => void) => Promise<ConversationTitleTransportResult>;
+    /** Maximum IDs in each JSON batch; must be positive. */
+    batchSize: number;
+    /** Maximum IDs in each incremental request; must be positive. */
+    maxIds: number;
+    /** Maximum encoded JSON-array bytes per incremental request. */
+    maxBytes: number;
+    /** Maximum delay before buffered titles flush. */
+    flushMs: number;
+    /** Maximum buffered title count; must be positive. */
+    maxPendingUpdates: number;
+    /** Duration for which local writes override remote snapshots. */
+    protectionMs: number;
+    /** Clock for local protection; defaults to Date.now. */
+    now?: () => number;
+}
+/** Resolves list titles without blocking the initial history response. */
+export declare const useConversationTitleEnrichmentController: <T extends ConversationTitleItem>(identityScope: unknown, setConversations: Dispatch<SetStateAction<T[]>>, options: ConversationTitleEnrichmentOptions) => {
+    applyConversationList: (items: T[], requestedAtRevision: number) => void;
+    getTitleRevision: () => number;
+    rememberTitle: (id: string, title: string, options?: {
+        isWriting?: boolean;
+        protect?: boolean;
+    }) => void;
+    forgetTitle: (id: string, isPending?: boolean) => void;
+    restoreTitle: (id: string, title: string) => void;
+    setVisibleConversationIds: (viewKey: string, ids: string[] | null) => void;
+    isUpdatingTitles: boolean;
+};
+//# sourceMappingURL=useConversationTitleEnrichmentController.d.ts.map