@epam/ai-dial-chat-hooks 1.2.0-dev.99 → 1.2.0-rc.0

package/conversation/useConversationStream/generation-resume.js

@@ -1,16 +1,40 @@
 import { safeDecodeURI as e } from "../../shared/string-utils.js";
-import { getConversationPath as t } from "./conversation-path.js";
-import { applyChunkToMessages as n } from "./apply-chunk.js";
-import { restoreBufferedMessage as r } from "./buffered-generation.js";
-import { MessageRole as i } from "@epam/ai-dial-chat-shared";
+import { DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as t, GenerationPersistenceError as n } from "../create-chat-stream-api.js";
+import { getConversationPath as r } from "./conversation-path.js";
+import { applyChunkToMessages as i } from "./apply-chunk.js";
+import { restoreBufferedMessage as a } from "./buffered-generation.js";
+import { BackgroundGenerationStatus as o, MessageRole as s } from "@epam/ai-dial-chat-shared";
 //#region src/conversation/useConversationStream/generation-resume.ts
-var a = (e) => {
+var c = (e) => {
 	let t = e.custom_content;
 	return !!e.content || e.responseId != null || !!t?.attachments?.length || !!t?.stages?.length || !!t?.annotations?.length || !!t?.form_schema || t?.state != null;
-}, o = (e) => {
+}, l = (e) => e.messages.findIndex((e) => e.backgroundGeneration?.status === o.Pending), u = (e) => {
+	if (l(e) !== -1) return !0;
 	let t = e.messages[e.messages.length - 1];
-	return !!t && t.role === i.Assistant && !a(t) && t.streamErrorMessage == null && !t.wasStoppedByUser;
-}, s = 300 * 1e3, c = "awaiting-resume", l = async (e, t) => {
+	return !!t && t.role === s.Assistant && !c(t) && t.streamErrorMessage == null && !t.wasStoppedByUser;
+}, d = 300 * 1e3, f = "awaiting-resume", p = [
+	1e3,
+	2e3,
+	4e3,
+	8e3,
+	16e3
+], m = (e) => new Promise((t) => {
+	let n = typeof window < "u", r = () => {
+		clearTimeout(i), n && window.removeEventListener("online", r), t();
+	}, i = setTimeout(r, e);
+	n && window.addEventListener("online", r);
+}), h = async (e, t, n) => {
+	for (let r = 0; r <= p.length; r++) {
+		if (t()) return null;
+		try {
+			let t = await e();
+			if (!n?.(t)) return t;
+		} catch {}
+		if (r === p.length) return null;
+		await m(p[r]);
+	}
+	return null;
+}, g = async (e, t) => {
 	let n = e.getReader(), r = new TextDecoder(), i = "";
 	try {
 		for (;;) {
@@ -39,88 +63,122 @@ var a = (e) => {
 	} catch {} finally {
 		n.releaseLock();
 	}
-}, u = ({ transport: i, setConversation: a, conversationRef: u, resumingPathsRef: d, bufferedGenerationsRef: f, addStreamingPath: p, removeStreamingPath: m, isPathDisplayed: h }) => (g, _) => {
-	if (!o(_)) return;
-	let v = t(g);
-	if (d.current.has(v)) return;
-	d.current.add(v), p(v);
-	let y = _.messages.length - 1, b = (e) => {
-		d.current.delete(v), m(v), f.current.delete(v), e && h(v) && (a(e), u.current = e);
-	}, x = async () => {
-		try {
-			b(await i.getConversation(e(g)));
+}, _ = ({ transport: o, setConversation: s, conversationRef: p, resumingPathsRef: m, bufferedGenerationsRef: h, addStreamingPath: _, removeStreamingPath: v, isPathDisplayed: y, generationPersistenceErrorMessage: b = t, frameScheduler: x, stoppedGenerationIdsRef: S, onReloadError: C, onReloadSuccess: w }) => (t, T, E = {}) => {
+	if (!u(T)) {
+		E.onSettled?.();
+		return;
+	}
+	let D = r(t);
+	if (!E.skipDedupe && m.current.has(D)) return;
+	m.current.add(D), _(D);
+	let O = l(T), k = O !== -1, A = k ? O : T.messages.length - 1, j = {
+		generationId: f,
+		messageIndex: A,
+		message: E.seedMessage ?? T.messages[A]
+	};
+	h.current.set(D, j);
+	let M = () => h.current.get(D) === j, N = k ? T.messages[A].backgroundGeneration?.generationId : void 0, P = () => N != null && (S?.current.has(N) ?? !1), F = (e, t = !1) => {
+		if (x?.flush(D), N != null && S?.current.delete(N), !M()) return;
+		m.current.delete(D), v(D);
+		let n = !k && e && e.messages.length - 1 === A && u(e) && c(j.message);
+		if (t || n) {
+			j.message = {
+				...j.message,
+				streamErrorMessage: b
+			}, y(D) && s((e) => {
+				if (!e || !M() || !y(D)) return e;
+				let t = a(e, j);
+				return p.current = t, t;
+			});
+			return;
+		}
+		h.current.delete(D), e && y(D) && (s(e), p.current = e);
+	}, I = async () => {
+		if (M()) try {
+			let n = await o.getConversation(e(t));
+			if (!M()) return;
+			w?.(D), F(n);
 		} catch {
-			b();
+			if (!M()) return;
+			x?.flush(D), m.current.delete(D), v(D), N != null && S?.current.delete(N), C?.(D, I, M);
 		}
-	}, S = (e) => {
-		let t = {
-			generationId: c,
-			messageIndex: y,
-			message: e
-		};
-		f.current.set(v, t), h(v) && a((e) => {
-			if (!e) return e;
-			let n = r(e, t);
-			return u.current = n, n;
-		});
-	}, C = (e) => {
-		let t = f.current.get(v);
-		if (t?.generationId === c) {
-			let r = n([t.message], 0, e);
-			r && (t.message = r[0]);
+	}, L = (e) => {
+		x?.cancel(D), M() && (j.message = e, y(D) && s((e) => {
+			if (!e || !M() || !y(D)) return e;
+			let t = a(e, j);
+			return p.current = t, t;
+		}));
+	}, R = () => s((e) => {
+		if (!e || !M() || !y(D)) return e;
+		let t = a(e, j);
+		return p.current = t, t;
+	}), z = (e) => {
+		if (!M() || P()) return;
+		let t = i([j.message], 0, e);
+		if (t && (j.message = t[0]), y(D)) {
+			if (x) {
+				x.schedule(D, R);
+				return;
+			}
+			R();
 		}
-		h(v) && a((e) => {
-			if (!e) return e;
-			let t = f.current.get(v);
-			if (t?.generationId !== c) return e;
-			let n = r(e, t);
-			return u.current = n, n;
-		});
-	}, w = async () => {
-		let t = new AbortController(), n;
+	}, B = async () => {
+		let n = new AbortController(), r;
 		try {
-			n = await i.watchConversation(v, t.signal);
+			r = await o.watchConversation(D, n.signal);
 		} catch {
-			await x();
+			await I();
 			return;
 		}
-		let r = !1, a = window.setTimeout(() => {
-			t.abort();
-		}, s);
 		try {
-			await l(n, async (t) => {
-				if (t?.action !== "UPDATE") return !1;
+			let i = await o.getConversation(e(t));
+			if (!u(i)) {
+				n.abort(), r.cancel().catch(() => void 0), F(i);
+				return;
+			}
+		} catch {}
+		let i = !1, a = window.setTimeout(() => {
+			n.abort();
+		}, d);
+		try {
+			await g(r, async (n) => {
+				if (n?.action !== "UPDATE") return !1;
 				try {
-					let t = await i.getConversation(e(g));
-					if (!o(t)) return b(t), r = !0, !0;
+					let n = await o.getConversation(e(t));
+					if (!u(n)) return F(n), i = !0, !0;
 				} catch {}
 				return !1;
 			});
 		} finally {
 			clearTimeout(a);
 		}
-		r || await x();
-	}, T = async () => {
+		i || await I();
+	}, V = async () => {
 		let e = new AbortController(), t;
 		try {
-			t = await i.attachToGeneration(v, e.signal);
+			t = await o.attachToGeneration(D, e.signal);
 		} catch {
 			return !1;
 		}
-		let n = !1;
-		return await l(t, (e) => {
+		let r = !1, i = !1;
+		return await g(t, (e) => {
+			if (!M()) return !0;
 			switch (e.type) {
-				case "snapshot": return S(e.message), !1;
-				case "chunk": return C(e.chunk), !1;
+				case "snapshot": return L(e.message), !1;
+				case "chunk": return z(e.chunk), !1;
 				case "done":
 				case "error":
-				case "stopped": return n = !0, !0;
+				case "stopped": return i = e.type === "error" && e.errorType === n.type, r = !0, !0;
 			}
-		}), n ? (await x(), !0) : !1;
+		}), r ? (i ? F(void 0, !0) : await I(), !0) : !M();
 	};
 	(async () => {
-		await T() || await w();
+		try {
+			await V() || await B();
+		} finally {
+			E.onSettled?.();
+		}
 	})();
 };
 //#endregion
-export { u as createResumeIfAwaitingGeneration, o as isAwaitingGenerationResume };
+export { _ as createResumeIfAwaitingGeneration, h as fetchConversationForRecovery, l as findPendingBackgroundMessageIndex, c as hasGeneratedPayload, u as isAwaitingGenerationResume };

package/conversation/useConversationStream/useConversationStream.js

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