@epam/ai-dial-chat-hooks 1.3.0-dev.356 → 1.3.0-dev.358

package/conversation/useConversationStream/useConversationStream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useConversationStream.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useConversationStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EACL,KAAK,YAAY,EAEjB,KAAK,oBAAoB,EAEzB,KAAK,WAAW,EACjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,KAAK,QAAQ,EACb,KAAK,gBAAgB,EACrB,KAAK,cAAc,EAMpB,MAAM,OAAO,CAAC;AA6Cf,gFAAgF;AAChF,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;IACtC,UAAU,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IAChC,MAAM,EAAE,WAAW,CAAC;CACrB;AAED;;;;GAIG;AACH,MAAM,WAAW,2BAA2B;IAC1C,6GAA6G;IAC7G,gBAAgB,CACd,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,GAAG,SAAS,EAC3B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,uBAAuB,EAChC,aAAa,CAAC,EAAE,oBAAoB,EACpC,YAAY,CAAC,EAAE,MAAM,EACrB,IAAI,CAAC,EAAE,yBAAyB,EAChC,YAAY,CAAC,EAAE,MAAM,EACrB,eAAe,CAAC,EAAE,MAAM,GACvB,IAAI,CAAC;IACR;;;OAGG;IACH,cAAc,CAAC,MAAM,EAAE;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClB,uFAAuF;IACvF,iBAAiB,CACf,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IACvC;;;;;;;OAOG;IACH,kBAAkB,CAChB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IACvC,4EAA4E;IAC5E,eAAe,CACb,cAAc,EAAE,MAAM,EACtB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,YAAY,CAAC,CAAC;IACzB;;;;;OAKG;IACH,gBAAgB,CAAC,CACf,IAAI,EAAE,MAAM,EACZ,YAAY,EAAE,YAAY,GACzB,OAAO,CAAC,YAAY,CAAC,CAAC;CAC1B;AAED,6FAA6F;AAC7F,MAAM,WAAW,+BAA+B;IAC9C,iGAAiG;IACjG,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,eAAe,CAAC;IACzE,wGAAwG;IACxG,kBAAkB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,IAAI,CAAC;CAClE;AAED,yFAAyF;AACzF,MAAM,WAAW,yBAAyB;IACxC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,eAAe,EAAE,MAAM,IAAI,CAAC;IAC5B,oKAAoK;IACpK,cAAc,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC/D,+HAA+H;IAC/H,uBAAuB,CAAC,EAAE,MAAM,IAAI,CAAC;CACtC;AAED,6GAA6G;AAC7G,MAAM,WAAW,iCAAiC;IAChD,qBAAqB,CAAC,EAAE,MAAM,IAAI,CAAC;IACnC,mBAAmB,CAAC,EAAE,MAAM,IAAI,CAAC;IACjC,oBAAoB,CAAC,EAAE,MAAM,IAAI,CAAC;CACnC;AAED,+FAA+F;AAC/F,MAAM,WAAW,yBAAyB;IACxC,eAAe,EAAE,QAAQ,CAAC,cAAc,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC;IAC/D,eAAe,EAAE,gBAAgB,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC;CACxD;AAED,oDAAoD;AACpD,MAAM,WAAW,2BAA2B;IAC1C,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,KAAK,EAAE,yBAAyB,CAAC;IACjC,SAAS,EAAE,2BAA2B,CAAC;IACvC,UAAU,EAAE,+BAA+B,CAAC;IAC5C,OAAO,CAAC,EAAE,yBAAyB,CAAC;IACpC,OAAO,CAAC,EAAE,iCAAiC,CAAC;IAC5C,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACrC;;;;;OAKG;IACH,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,iFAAiF;IACjF,iCAAiC,CAAC,EAAE,MAAM,CAAC;IAC3C;;;;;OAKG;IACH,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACvC;;;;;OAKG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAqBD,4EAA4E;AAC5E,MAAM,WAAW,kBAAkB;IACjC;;;;;;OAMG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,qDAAqD;AACrD,MAAM,WAAW,2BAA2B;IAC1C,WAAW,EAAE,CACX,cAAc,EAAE,MAAM,EACtB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,EACb,aAAa,CAAC,EAAE,oBAAoB,EACpC,YAAY,CAAC,EAAE,MAAM,EACrB,IAAI,CAAC,EAAE,yBAAyB,EAChC,OAAO,CAAC,EAAE,kBAAkB,KACzB,IAAI,CAAC;IACV,UAAU,EAAE,MAAM,IAAI,CAAC;IACvB,0BAA0B,EAAE,CAC1B,qBAAqB,EAAE,MAAM,EAC7B,YAAY,EAAE,YAAY,KACvB,IAAI,CAAC;IACV,+GAA+G;IAC/G,yBAAyB,EAAE,CACzB,qBAAqB,EAAE,MAAM,EAC7B,YAAY,EAAE,YAAY,KACvB,YAAY,CAAC;IAClB,WAAW,EAAE,OAAO,CAAC;IACrB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,iEAAiE;IACjE,0BAA0B,EAAE,OAAO,CAAC;IACpC,iDAAiD;IACjD,uBAAuB,EAAE,OAAO,CAAC;IACjC,yEAAyE;IACzE,uBAAuB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9C;AAkBD;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,GAAI,mQAYnC,2BAA2B,KAAG,2BAo+BhC,CAAC"}
+{"version":3,"file":"useConversationStream.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useConversationStream.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAEV,2BAA2B,EAC3B,2BAA2B,EAC5B,MAAM,6BAA6B,CAAC;AAYrC,YAAY,EACV,+BAA+B,EAC/B,yBAAyB,EACzB,yBAAyB,EACzB,iCAAiC,EACjC,2BAA2B,EAC3B,kBAAkB,EAClB,uBAAuB,EACvB,2BAA2B,EAC3B,2BAA2B,GAC5B,MAAM,6BAA6B,CAAC;AAErC;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,GAAI,mQAYnC,2BAA2B,KAAG,2BAqRhC,CAAC"}

package/conversation/useConversationStream/useConversationStream.js

@@ -1,345 +1,147 @@
-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, 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 { 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";
+import { DEFAULT_GENERATION_CONFLICT_MESSAGE as e, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as t } from "../create-chat-stream-api.js";
+import { getConversationPath as n } from "./conversation-path.js";
+import { restoreBufferedMessage as r } from "./buffered-generation.js";
+import { createResumeIfAwaitingGeneration as i, isAwaitingGenerationResume as a } from "./generation-resume.js";
+import { createFrameScheduler as o } from "./frame-scheduler.js";
+import { findResumedBackgroundGenerationId as s } from "./stream-errors.js";
+import { runStreamSession as ee } from "./stream-session.js";
+import { useConversationReloadErrors as te } from "./useConversationReloadErrors.js";
+import { useStopGeneration as c } from "./useStopGeneration.js";
+import { useStreamPathState as l } from "./useStreamPathState.js";
+import { MessageRole as u } from "@epam/ai-dial-chat-shared";
+import { useCallback as d, useEffect as f, useMemo as p, useRef as m, useState as ne } from "react";
 //#region src/conversation/useConversationStream/useConversationStream.ts
-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 = d(e);
-	return t === -1 ? void 0 : e.messages[t].backgroundGeneration?.generationId;
-}, 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;
+var h = ({ conversationId: h, state: { setConversation: g, conversationRef: _ }, transport: v, generation: { startGeneration: y, completeGeneration: b }, channel: x, overlay: S, onStopError: re, generationConflictMessage: C = e, generationPersistenceErrorMessage: w = t, onStreamError: T, batchChunksPerFrame: E = !1 }) => {
+	let D = m(T);
+	f(() => {
+		D.current = T;
+	}, [T]);
+	let O = m(E);
+	f(() => {
+		O.current = E;
+	}, [E]);
+	let [k] = ne(o), A = m(!0);
+	f(() => (A.current = !0, () => {
+		A.current = !1;
 	}), []);
-	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
-		}), 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 {
-				W.current.delete(e), H.current && be(new Set(W.current));
-			}
-		}
-	}, [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;
-		});
-	}, [G]), Y = p((e) => {
-		N((t) => {
-			if (!t.has(e)) return t;
-			let n = new Set(t);
-			return n.delete(e), 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;
-		});
-	}, []), 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: D,
-		frameScheduler: k ? j : void 0,
-		stoppedGenerationIdsRef: B,
-		rememberResumedGenerationId: we,
-		forgetResumedGenerationId: Te,
-		onReloadError: K,
-		onReloadSuccess: G
+	let { reloadFailuresRef: j, reloadFailures: M, retrying: N, clearReloadError: P, reportReloadError: F, retryConversationReload: ie } = te({
+		conversationId: h,
+		mountedRef: A
+	}), { streamingPaths: ae, streamingPathsRef: oe, stoppablePaths: se, activeGenerationIdsRef: I, resumedGenerationIdsRef: ce, resumingPathsRef: L, bufferedGenerationsRef: R, stoppedGenerationIdsRef: z, stopRequestedIdsRef: B, latestGenerationIdsRef: V, isPathDisplayed: H, addStreamingPath: U, removeStreamingPath: W, setPathStoppable: G, releaseActiveGeneration: K, rememberResumedGenerationId: q, forgetResumedGenerationId: J } = l({
+		conversationId: h,
+		frameScheduler: k,
+		mountedRef: A,
+		clearReloadError: P
+	}), Y = p(() => i({
+		transport: v,
+		setConversation: g,
+		conversationRef: _,
+		resumingPathsRef: L,
+		bufferedGenerationsRef: R,
+		addStreamingPath: U,
+		removeStreamingPath: W,
+		isPathDisplayed: H,
+		generationPersistenceErrorMessage: w,
+		frameScheduler: E ? k : void 0,
+		stoppedGenerationIdsRef: z,
+		rememberResumedGenerationId: q,
+		forgetResumedGenerationId: J,
+		onReloadError: F,
+		onReloadSuccess: P
 	}), [
-		J,
-		Y,
-		q,
+		U,
+		W,
+		H,
+		v,
 		w,
-		D,
+		E,
 		k,
-		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 === 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: 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;
-			}));
-		}, 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 (d(n) !== -1) {
-						Z(t, n, { seedMessage: O?.message });
-						return;
-					}
-					if (O && O.messageIndex === n.messages.length - 1 && f(n) && ne(O.message)) {
-						fe();
-						return;
-					}
-					O && z.current.delete(g), S(n), C.current = n;
-				} catch {
-					k() && K(g, M, k);
-				}
-			}
-		}, N = {
-			signal: y.signal,
-			onChunk: (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]);
-				}
-				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 () => {
-				j.flush(g), x(), B.current.has(h) ? B.current.delete(h) : v() || E?.notifyGenerationEnd?.(), await M();
-			},
-			onError: (e) => {
-				j.flush(g), A.current?.(e), e instanceof o && !v() ? V(e) : e instanceof r && m.resumeOnConflict && !v() ? U(e) : P(e);
-			}
-		}, 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: 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 === a ? {
-						...e,
-						streamErrorMessage: n
-					} : e)
-				};
-				return C.current = i, i;
-			});
-		}, he = async (n, r, i) => {
-			let a = () => H.current && !v() && z.current.get(g) === r;
-			try {
-				let o = await w.getConversation(e(t));
-				if (!a()) return;
-				if (r.messageIndex !== o.messages.length - 1 || !f(o)) {
-					F(n, r);
-					return;
-				}
-				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() && F(n, r);
-			}
-		}, 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], 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 = 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 = 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;
-				}
-				w.streamCompletion(g, n, s, N, ee, h, p, _, e);
-			} catch (e) {
-				N.onError(e instanceof Error ? e : Error(String(e)));
-			}
-		})();
-	}, [
-		ue,
-		de,
+		q,
 		J,
+		F,
+		P
+	]), le = d((e, t, n, r, i, a, o, s) => ee({
+		transport: v,
+		channel: x,
+		overlay: S,
+		startGeneration: y,
+		completeGeneration: b,
+		setConversation: g,
+		conversationRef: _,
+		generationConflictMessage: C,
+		generationPersistenceErrorMessage: w,
+		onStreamErrorRef: D,
+		batchChunksPerFrameRef: O,
+		mountedRef: A,
+		frameScheduler: k,
+		resumeGeneration: Y,
+		activeGenerationIdsRef: I,
+		latestGenerationIdsRef: V,
+		resumingPathsRef: L,
+		bufferedGenerationsRef: R,
+		stoppedGenerationIdsRef: z,
+		addStreamingPath: U,
+		removeStreamingPath: W,
+		setPathStoppable: G,
+		releaseActiveGeneration: K,
+		isPathDisplayed: H,
+		reportReloadError: F,
+		clearReloadError: P
+	}, e, t, n, r, i, a, o, s), [
+		y,
+		b,
+		U,
+		W,
+		G,
+		K,
+		H,
+		x?.channelId,
+		x?.ensureConnected,
+		x?.waitForChannel,
+		x?.notifyGenerationSettled,
+		S,
+		v,
 		Y,
-		X,
-		Ce,
-		q,
-		T?.channelId,
-		T?.ensureConnected,
-		T?.waitForChannel,
-		T?.notifyGenerationSettled,
-		E,
+		C,
 		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);
-			}
-		})();
+		k,
+		F,
+		P
+	]), ue = d((e, t) => {
+		let i = n(e), o = R.current.get(i);
+		return j.current.get(i)?.isCurrent() && o && t.messages.length - 1 >= o.messageIndex && t.messages[o.messageIndex].role === u.Assistant && !a(t) ? (P(i), R.current.delete(i), t) : o ? r(t, o) : t;
 	}, [
-		s,
-		C,
-		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;
+		R,
+		P,
+		j
+	]), de = c({
+		conversationId: h,
+		conversationRef: _,
+		transport: v,
+		overlay: S,
+		onStopError: re,
+		frameScheduler: k,
+		mountedRef: A,
+		activeGenerationIdsRef: I,
+		resumedGenerationIdsRef: ce,
+		bufferedGenerationsRef: R,
+		stoppedGenerationIdsRef: z,
+		stopRequestedIdsRef: B,
+		streamingPathsRef: oe
+	}), fe = d((e, t) => {
+		I.current.has(n(e)) || Y(e, t, { trackAttachGenerationId: !0 });
+	}, [I, Y]), X = h != null && ae.has(n(h)), Z = h == null ? null : n(h), pe = Z != null && (se.has(Z) || X && s(_.current) != null), Q = Z ? M.get(Z) : void 0, $ = Q?.isCurrent() ?? !1;
 	return {
-		hasConversationReloadError: Me,
-		isReloadingConversation: !!$ && Me && ye.has($),
-		retryConversationReload: xe,
-		startStream: Ee,
-		handleStop: Oe,
-		resumeIfAwaitingGeneration: ke,
-		restoreBufferedGeneration: De,
-		isStreaming: Ae,
-		canStopStreaming: je
+		hasConversationReloadError: $,
+		isReloadingConversation: !!Q && $ && N.has(Q),
+		retryConversationReload: ie,
+		startStream: le,
+		handleStop: de,
+		resumeIfAwaitingGeneration: fe,
+		restoreBufferedGeneration: ue,
+		isStreaming: X,
+		canStopStreaming: pe
 	};
 };
 //#endregion
-export { S as useConversationStream };
+export { h as useConversationStream };

package/conversation/useConversationStream/useStopGeneration.d.ts

@@ -0,0 +1,29 @@
+import { MutableRefObject } from 'react';
+import { ConversationStateAccessor, ConversationStreamOverlayNotifier, ConversationStreamTransport } from './conversation-stream.types';
+import { FrameScheduler } from './frame-scheduler';
+import { useStreamPathState } from './useStreamPathState';
+type PathState = ReturnType<typeof useStreamPathState>;
+interface UseStopGenerationParams {
+    conversationId: string | undefined;
+    conversationRef: ConversationStateAccessor['conversationRef'];
+    transport: ConversationStreamTransport;
+    overlay: ConversationStreamOverlayNotifier | undefined;
+    onStopError: ((error: Error) => void) | undefined;
+    frameScheduler: FrameScheduler;
+    mountedRef: MutableRefObject<boolean>;
+    activeGenerationIdsRef: PathState['activeGenerationIdsRef'];
+    resumedGenerationIdsRef: PathState['resumedGenerationIdsRef'];
+    bufferedGenerationsRef: PathState['bufferedGenerationsRef'];
+    stoppedGenerationIdsRef: PathState['stoppedGenerationIdsRef'];
+    stopRequestedIdsRef: PathState['stopRequestedIdsRef'];
+    streamingPathsRef: PathState['streamingPathsRef'];
+}
+/**
+ * Stop for the displayed conversation's generation: resolves its id (local,
+ * then a pending background generation, then a resumed one), flushes pending
+ * display writes, and asks the backend to stop it. A `404` within a short
+ * grace window is the generation ending on its own and is not reported.
+ */
+export declare const useStopGeneration: ({ conversationId, conversationRef, transport, overlay, onStopError, frameScheduler, mountedRef, activeGenerationIdsRef, resumedGenerationIdsRef, bufferedGenerationsRef, stoppedGenerationIdsRef, stopRequestedIdsRef, streamingPathsRef, }: UseStopGenerationParams) => () => void;
+export {};
+//# sourceMappingURL=useStopGeneration.d.ts.map

package/conversation/useConversationStream/useStopGeneration.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useStopGeneration.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useStopGeneration.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,gBAAgB,EAAe,MAAM,OAAO,CAAC;AAG3D,OAAO,KAAK,EACV,yBAAyB,EACzB,iCAAiC,EACjC,2BAA2B,EAC5B,MAAM,6BAA6B,CAAC;AACrC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAOxD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,KAAK,SAAS,GAAG,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAEvD,UAAU,uBAAuB;IAC/B,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,eAAe,EAAE,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;IAC9D,SAAS,EAAE,2BAA2B,CAAC;IACvC,OAAO,EAAE,iCAAiC,GAAG,SAAS,CAAC;IACvD,WAAW,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IAClD,cAAc,EAAE,cAAc,CAAC;IAE/B,UAAU,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACtC,sBAAsB,EAAE,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC5D,uBAAuB,EAAE,SAAS,CAAC,yBAAyB,CAAC,CAAC;IAC9D,sBAAsB,EAAE,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC5D,uBAAuB,EAAE,SAAS,CAAC,yBAAyB,CAAC,CAAC;IAC9D,mBAAmB,EAAE,SAAS,CAAC,qBAAqB,CAAC,CAAC;IACtD,iBAAiB,EAAE,SAAS,CAAC,mBAAmB,CAAC,CAAC;CACnD;AAED;;;;;GAKG;AACH,eAAO,MAAM,iBAAiB,GAAI,6OAc/B,uBAAuB,eAiFzB,CAAC"}

package/conversation/useConversationStream/useStopGeneration.js

@@ -0,0 +1,48 @@
+import { StopCompletionError as e } from "../create-chat-stream-api.js";
+import { getConversationPath as t } from "./conversation-path.js";
+import { STOP_NOT_FOUND_GRACE_MS as n, findResumedBackgroundGenerationId as r } from "./stream-errors.js";
+import { useCallback as i } from "react";
+//#region src/conversation/useConversationStream/useStopGeneration.ts
+var a = ({ conversationId: a, conversationRef: o, transport: s, overlay: c, onStopError: l, frameScheduler: u, mountedRef: d, activeGenerationIdsRef: f, resumedGenerationIdsRef: p, bufferedGenerationsRef: m, stoppedGenerationIdsRef: h, stopRequestedIdsRef: g, streamingPathsRef: _ }) => i(() => {
+	if (!a) return;
+	let i = t(a), v = f.current.get(i), y = v ?? r(o.current) ?? p.current.get(i);
+	if (!y || g.current.has(y)) return;
+	g.current.add(y), h.current.add(y), u.flush(i), c?.notifyStopGenerating?.();
+	let b = m.current.get(i)?.message.content, x = () => f.current.get(i) === y || v == null && _.current.has(i), S = async () => {
+		let e = n / 100;
+		for (let t = 0; t < e; t++) {
+			if (!x() || !d.current) return !0;
+			await new Promise((e) => setTimeout(e, 100));
+		}
+		return !x();
+	};
+	(async () => {
+		try {
+			await s.stopCompletion({
+				generationId: y,
+				path: i,
+				content: b
+			});
+		} catch (t) {
+			let n = t instanceof Error ? t : Error(String(t));
+			if (n instanceof e && n.status === 404 && await S()) return;
+			g.current.delete(y), l?.(n);
+		}
+	})();
+}, [
+	f,
+	m,
+	a,
+	o,
+	u,
+	d,
+	l,
+	c,
+	p,
+	g,
+	h,
+	_,
+	s
+]);
+//#endregion
+export { a as useStopGeneration };

package/conversation/useConversationStream/useStreamPathState.d.ts

@@ -0,0 +1,39 @@
+import { MutableRefObject } from 'react';
+import { BufferedGeneration } from './buffered-generation';
+import { FrameScheduler } from './frame-scheduler';
+interface UseStreamPathStateParams {
+    conversationId: string | undefined;
+    frameScheduler: FrameScheduler;
+    mountedRef: MutableRefObject<boolean>;
+    clearReloadError: (path: string) => void;
+}
+/**
+ * Per-path generation state of `useConversationStream`: which paths are
+ * streaming and which offer Stop, and the refs that own each path's local,
+ * latest, resumed and stopped generation ids, its live buffered message and
+ * its resume reservation, plus the displayed-conversation guard that keeps a
+ * stream from writing into a conversation that is no longer shown.
+ * `streamingPathsRef` is synced during render so Stop polling reads the
+ * latest set.
+ */
+export declare const useStreamPathState: ({ conversationId, frameScheduler, mountedRef, clearReloadError, }: UseStreamPathStateParams) => {
+    streamingPaths: Set<string>;
+    streamingPathsRef: import('react').RefObject<Set<string>>;
+    stoppablePaths: Set<string>;
+    activeGenerationIdsRef: import('react').RefObject<Map<string, string>>;
+    resumedGenerationIdsRef: import('react').RefObject<Map<string, string>>;
+    resumingPathsRef: import('react').RefObject<Set<string>>;
+    bufferedGenerationsRef: import('react').RefObject<Map<string, BufferedGeneration>>;
+    stoppedGenerationIdsRef: import('react').RefObject<Set<string>>;
+    stopRequestedIdsRef: import('react').RefObject<Set<string>>;
+    latestGenerationIdsRef: import('react').RefObject<Map<string, string>>;
+    isPathDisplayed: (path: string) => boolean;
+    addStreamingPath: (path: string) => void;
+    removeStreamingPath: (path: string) => void;
+    setPathStoppable: (path: string, isStoppable: boolean) => void;
+    releaseActiveGeneration: (path: string, genId: string) => void;
+    rememberResumedGenerationId: (path: string, generationId: string) => void;
+    forgetResumedGenerationId: (path: string, generationId: string) => void;
+};
+export {};
+//# sourceMappingURL=useStreamPathState.d.ts.map

package/conversation/useConversationStream/useStreamPathState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useStreamPathState.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useStreamPathState.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,gBAAgB,EAKtB,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,KAAK,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAEhE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExD,UAAU,wBAAwB;IAChC,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IAEnC,cAAc,EAAE,cAAc,CAAC;IAE/B,UAAU,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEtC,gBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;CAC1C;AAED;;;;;;;;GAQG;AACH,eAAO,MAAM,kBAAkB,GAAI,mEAKhC,wBAAwB;;;;;;;;;;;4BAwDhB,MAAM,KAAG,OAAO;6BAOhB,MAAM;gCAYgC,MAAM;6BAST,MAAM,eAAe,OAAO;oCAY/D,MAAM,SAAS,MAAM;wCAarB,MAAM,gBAAgB,MAAM;sCAU5B,MAAM,gBAAgB,MAAM;CA6BtC,CAAC"}