@epam/ai-dial-chat-hooks 1.2.0-dev.219 → 1.2.0-dev.229

package/conversation/stage.js

@@ -14,10 +14,14 @@ var t = (t) => t === e.Completed ? e.Completed : t === e.Failed ? e.Failed : nul
 	status: t(e.status),
 	...e.content != null && { content: e.content },
 	...e.tag != null && { tag: e.tag },
-	...e.attachments?.length && { attachments: e.attachments.map(n) }
+	...e.attachments?.length && { attachments: e.attachments.map(n) },
+	...e.parent_stage_index != null && { parent_stage_index: e.parent_stage_index }
 }), i = (e) => {
 	let t = Array.isArray(e) ? e : e?.custom_content?.stages ?? e?.customContent?.stages;
-	if (t?.length) return t.map(r);
+	if (t?.length) return t.map((e, t) => r(e.index == null ? {
+		...e,
+		index: t
+	} : e));
 };
 //#endregion
 export { i as mapStages, r as toStage };

package/conversation/useConversationHandlers/useConversationHandlers.js

@@ -41,34 +41,34 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 		_,
 		E
 	]), F = u((e) => {
-		if (g || !m || !p || e === -1 || p.messages[e]?.role !== c.Assistant) return;
-		let t = p.messages[e - 1];
-		if (!t || t.role !== c.User) return;
-		let n = C(), i = r(E) ? {
-			...t.custom_content,
+		let t = y.current;
+		if (g || !m || !t || e === -1 || t.messages[e]?.role !== c.Assistant) return;
+		let n = t.messages[e - 1];
+		if (!n || n.role !== c.User) return;
+		let i = C(), a = r(E) ? {
+			...n.custom_content,
 			configuration_value: E
-		} : t.custom_content, a = {
-			...p.messages[e],
+		} : n.custom_content, o = {
+			...t.messages[e],
 			content: "",
 			custom_content: void 0,
 			wasStoppedByUser: void 0,
 			stoppedWithoutContent: void 0,
 			streamErrorMessage: void 0,
-			deploymentId: n
-		}, o = {
-			...p,
+			deploymentId: i
+		}, s = {
+			...t,
 			messages: [
-				...p.messages.slice(0, e - 1),
+				...t.messages.slice(0, e - 1),
 				{
-					...t,
-					custom_content: i
+					...n,
+					custom_content: a
 				},
-				a
+				o
 			]
 		};
-		y.current = o, v(o), A((t) => new Set([...t].filter((t) => t < e))), _(m, t.content, e, n, i, l(), f.Regenerate);
+		y.current = s, v(s), A((t) => new Set([...t].filter((t) => t < e))), _(m, n.content, e, i, a, l(), f.Regenerate);
 	}, [
-		p,
 		m,
 		y,
 		g,
@@ -105,54 +105,59 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 		D,
 		v
 	]), R = u(async (e, t, n) => {
-		if (!m || !p) return !1;
-		let r = p.messages[e];
-		if (!r) return !1;
-		let i = r.rating, a = {
-			...p,
-			messages: p.messages.map((n, r) => r === e ? {
+		let r = y.current;
+		if (!m || !r) return !1;
+		let i = r.messages[e];
+		if (!i) return !1;
+		let a = i.rating, o = {
+			...r,
+			messages: r.messages.map((n, r) => r === e ? {
 				...n,
 				rating: t ?? void 0
 			} : n)
 		};
-		v(() => (y.current = a, a));
-		let o = s(m), c = () => {
-			v((t) => t && {
-				...t,
-				messages: t.messages.map((t, n) => n === e ? {
+		v(() => (y.current = o, o));
+		let c = s(m), l = () => {
+			v((t) => {
+				if (!t) return t;
+				let n = {
 					...t,
-					rating: i
-				} : t)
+					messages: t.messages.map((t, n) => n === e ? {
+						...t,
+						rating: a
+					} : t)
+				};
+				return y.current = n, n;
 			});
-		}, l = () => x.saveConversation({
-			path: o,
-			saveConversationBodyDto: { conversation: a }
-		}), u = r.responseId;
-		if (!u) return c(), !1;
+		}, u = () => x.saveConversation({
+			path: c,
+			saveConversationBodyDto: { conversation: o }
+		}), d = i.responseId;
+		if (!d) return l(), !1;
 		try {
 			return await S.rateMessage({ rateMessageDto: {
-				conversationId: p.id,
-				responseId: u,
-				modelId: p.model.id,
+				conversationId: r.id,
+				responseId: d,
+				modelId: r.model.id,
 				rate: t,
 				...n ? { comment: n } : {}
-			} }), await l(), !0;
+			} }), await u(), !0;
 		} catch {
-			return c(), !1;
+			return l(), !1;
 		}
 	}, [
-		p,
 		m,
 		y,
 		x,
 		S,
 		v
 	]), z = u((e, t, i) => {
-		if (!m || !p) return;
-		let s = a(e, i), c = o(e, i), u = t ? { [t]: e.const } : void 0, d = r(E), h = u || d ? {
-			...u ? { form_value: u } : {},
-			...d ? { configuration_value: E } : {}
-		} : void 0, g = C(), { userMessage: b, assistantMessage: x } = n(s, h, g);
+		let s = y.current;
+		if (!m || !s) return;
+		let c = a(e, i), u = o(e, i), d = t ? { [t]: e.const } : void 0, p = r(E), h = d || p ? {
+			...d ? { form_value: d } : {},
+			...p ? { configuration_value: E } : {}
+		} : void 0, g = C(), { userMessage: b, assistantMessage: x } = n(c, h, g);
 		v((e) => {
 			if (!e) return e;
 			let t = {
@@ -164,9 +169,8 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 				]
 			};
 			return y.current = t, t;
-		}), _(m, c, p.messages.length + 1, g, h, l(), f.Append);
+		}), _(m, u, s.messages.length + 1, g, h, l(), f.Append);
 	}, [
-		p,
 		m,
 		y,
 		C,
@@ -182,14 +186,15 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 		handleConfirmDelete: L,
 		handleRateMessage: R,
 		handleButtonSelect: u((e, t, n) => {
-			!m || !p || g || (e["dial:widgetOptions"].confirmationMessage ? M({
+			let r = y.current;
+			!m || !r || g || (e["dial:widgetOptions"].confirmationMessage ? M({
 				starter: e,
 				propertyKey: t,
 				description: n
 			}) : z(e, t, n));
 		}, [
-			p,
 			m,
+			y,
 			g,
 			z
 		]),
@@ -208,28 +213,29 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 			});
 		}, []),
 		handleEditMessage: u(async (e, n, r, a, o) => {
-			if (g || !m || !p) return;
-			let s = e;
-			if (s === -1 || p.messages[s].role !== c.User) return;
-			let u = p.messages[s];
-			if (!i(p.messages, s, n, r, a, o)) {
+			let s = y.current;
+			if (g || !m || !s) return;
+			let u = e;
+			if (u === -1 || s.messages[u].role !== c.User) return;
+			let d = s.messages[u];
+			if (!i(s.messages, u, n, r, a, o)) {
 				A((e) => {
 					let t = new Set(e);
-					return t.delete(s), t;
+					return t.delete(u), t;
 				});
 				return;
 			}
-			let d = t(a), h = new Set(r.map((e) => e.id)), b = [...(u.custom_content?.attachments ?? []).filter((e) => {
+			let p = t(a), h = new Set(r.map((e) => e.id)), b = [...(d.custom_content?.attachments ?? []).filter((e) => {
 				let t = e.url ?? e.data ?? e.title;
 				return h.has(t);
-			}), ...d ?? []], { attachments: x, skills: S, ...w } = u.custom_content ?? {}, T = o ?? S, E;
+			}), ...p ?? []], { attachments: x, skills: S, ...w } = d.custom_content ?? {}, T = o ?? S, E;
 			E = b.length > 0 || T?.length ? {
 				...w,
 				...b.length > 0 ? { attachments: b } : {},
 				...T?.length ? { skills: T } : {}
 			} : Object.keys(w).length > 0 ? w : void 0;
 			let D = {
-				...u,
+				...d,
 				content: n,
 				custom_content: E
 			}, O = Date.now(), k = {
@@ -237,18 +243,17 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 				content: "",
 				timestamp: new Date(O).toISOString()
 			}, j = [
-				...p.messages.slice(0, s),
+				...s.messages.slice(0, u),
 				D,
 				k
 			], M = {
-				...p,
+				...s,
 				messages: j
 			};
 			y.current = M, v(M);
 			let N = C();
 			_(m, n, j.length - 1, N, E, l(), f.Edit), A(/* @__PURE__ */ new Set());
 		}, [
-			p,
 			m,
 			y,
 			g,

package/conversation/useConversationStream/frame-scheduler.js

@@ -0,0 +1,39 @@
+//#region src/conversation/useConversationStream/frame-scheduler.ts
+var e = 16, t = (t) => {
+	if (typeof requestAnimationFrame == "function") {
+		let e = requestAnimationFrame(t);
+		return () => cancelAnimationFrame(e);
+	}
+	let n = setTimeout(t, e);
+	return () => clearTimeout(n);
+}, n = () => {
+	let e = /* @__PURE__ */ new Map(), n = (t) => {
+		let n = e.get(t);
+		n && (e.delete(t), n.task());
+	};
+	return {
+		schedule: (r, i) => {
+			let a = e.get(r);
+			if (a) {
+				a.task = i;
+				return;
+			}
+			let o = {
+				task: i,
+				cancelFrame: () => void 0
+			};
+			e.set(r, o), o.cancelFrame = t(() => n(r));
+		},
+		flush: (t) => {
+			e.get(t)?.cancelFrame(), n(t);
+		},
+		cancel: (t) => {
+			e.get(t)?.cancelFrame(), e.delete(t);
+		},
+		cancelAll: () => {
+			e.forEach((e) => e.cancelFrame()), e.clear();
+		}
+	};
+};
+//#endregion
+export { n as createFrameScheduler };

package/conversation/useConversationStream/generation-resume.js

@@ -62,62 +62,68 @@ var s = (e) => {
 	} catch {} finally {
 		n.releaseLock();
 	}
-}, h = ({ transport: o, setConversation: d, conversationRef: f, resumingPathsRef: p, bufferedGenerationsRef: h, addStreamingPath: g, removeStreamingPath: _, isPathDisplayed: v, generationPersistenceErrorMessage: y = t }) => (t, b, x = {}) => {
-	if (!c(b)) {
-		x.onSettled?.();
+}, h = ({ transport: o, setConversation: d, conversationRef: f, resumingPathsRef: p, bufferedGenerationsRef: h, addStreamingPath: g, removeStreamingPath: _, isPathDisplayed: v, generationPersistenceErrorMessage: y = t, frameScheduler: b }) => (t, x, S = {}) => {
+	if (!c(x)) {
+		S.onSettled?.();
 		return;
 	}
-	let S = r(t);
-	if (!x.skipDedupe && p.current.has(S)) return;
-	p.current.add(S), g(S);
-	let C = b.messages.length - 1, w = {
+	let C = r(t);
+	if (!S.skipDedupe && p.current.has(C)) return;
+	p.current.add(C), g(C);
+	let w = x.messages.length - 1, T = {
 		generationId: u,
-		messageIndex: C,
-		message: x.seedMessage ?? b.messages[C]
+		messageIndex: w,
+		message: S.seedMessage ?? x.messages[w]
 	};
-	h.current.set(S, w);
-	let T = () => h.current.get(S) === w, E = (e, t = !1) => {
-		if (!T()) return;
-		p.current.delete(S), _(S);
-		let n = e && e.messages.length - 1 === C && c(e) && s(w.message);
+	h.current.set(C, T);
+	let E = () => h.current.get(C) === T, D = (e, t = !1) => {
+		if (b?.flush(C), !E()) return;
+		p.current.delete(C), _(C);
+		let n = e && e.messages.length - 1 === w && c(e) && s(T.message);
 		if (t || n) {
-			w.message = {
-				...w.message,
+			T.message = {
+				...T.message,
 				streamErrorMessage: y
-			}, v(S) && d((e) => {
-				if (!e || !T() || !v(S)) return e;
-				let t = a(e, w);
+			}, v(C) && d((e) => {
+				if (!e || !E() || !v(C)) return e;
+				let t = a(e, T);
 				return f.current = t, t;
 			});
 			return;
 		}
-		h.current.delete(S), e && v(S) && (d(e), f.current = e);
-	}, D = async () => {
+		h.current.delete(C), e && v(C) && (d(e), f.current = e);
+	}, O = async () => {
 		try {
-			E(await o.getConversation(e(t)));
+			D(await o.getConversation(e(t)));
 		} catch {
-			E(void 0, s(w.message));
+			D(void 0, s(T.message));
 		}
-	}, O = (e) => {
-		T() && (w.message = e, v(S) && d((e) => {
-			if (!e || !T() || !v(S)) return e;
-			let t = a(e, w);
-			return f.current = t, t;
-		}));
 	}, k = (e) => {
-		if (!T()) return;
-		let t = i([w.message], 0, e);
-		t && (w.message = t[0]), v(S) && d((e) => {
-			if (!e || !T() || !v(S)) return e;
-			let t = a(e, w);
+		b?.cancel(C), E() && (T.message = e, v(C) && d((e) => {
+			if (!e || !E() || !v(C)) return e;
+			let t = a(e, T);
 			return f.current = t, t;
-		});
-	}, A = async () => {
+		}));
+	}, A = () => d((e) => {
+		if (!e || !E() || !v(C)) return e;
+		let t = a(e, T);
+		return f.current = t, t;
+	}), j = (e) => {
+		if (!E()) return;
+		let t = i([T.message], 0, e);
+		if (t && (T.message = t[0]), v(C)) {
+			if (b) {
+				b.schedule(C, A);
+				return;
+			}
+			A();
+		}
+	}, M = async () => {
 		let n = new AbortController(), r;
 		try {
-			r = await o.watchConversation(S, n.signal);
+			r = await o.watchConversation(C, n.signal);
 		} catch {
-			await D();
+			await O();
 			return;
 		}
 		let i = !1, a = window.setTimeout(() => {
@@ -128,38 +134,38 @@ var s = (e) => {
 				if (n?.action !== "UPDATE") return !1;
 				try {
 					let n = await o.getConversation(e(t));
-					if (!c(n)) return E(n), i = !0, !0;
+					if (!c(n)) return D(n), i = !0, !0;
 				} catch {}
 				return !1;
 			});
 		} finally {
 			clearTimeout(a);
 		}
-		i || await D();
-	}, j = async () => {
+		i || await O();
+	}, N = async () => {
 		let e = new AbortController(), t;
 		try {
-			t = await o.attachToGeneration(S, e.signal);
+			t = await o.attachToGeneration(C, e.signal);
 		} catch {
 			return !1;
 		}
 		let r = !1, i = !1;
 		return await m(t, (e) => {
-			if (!T()) return !0;
+			if (!E()) return !0;
 			switch (e.type) {
-				case "snapshot": return O(e.message), !1;
-				case "chunk": return k(e.chunk), !1;
+				case "snapshot": return k(e.message), !1;
+				case "chunk": return j(e.chunk), !1;
 				case "done":
 				case "error":
 				case "stopped": return i = e.type === "error" && e.errorType === n.type, r = !0, !0;
 			}
-		}), r ? (i ? E(void 0, !0) : await D(), !0) : !T();
+		}), r ? (i ? D(void 0, !0) : await O(), !0) : !E();
 	};
 	(async () => {
 		try {
-			await j() || await A();
+			await N() || await M();
 		} finally {
-			x.onSettled?.();
+			S.onSettled?.();
 		}
 	})();
 };

package/conversation/useConversationStream/useConversationStream.js

@@ -4,224 +4,248 @@ 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 u, fetchConversationForRecovery as d, hasGeneratedPayload as f, isAwaitingGenerationResume as p } from "./generation-resume.js";
-import { MessageRole as m, generateUUID as h } from "@epam/ai-dial-chat-shared";
+import { createFrameScheduler as m } from "./frame-scheduler.js";
+import { MessageRole as ee, generateUUID as h } from "@epam/ai-dial-chat-shared";
 import { useCallback as g, useEffect as _, useMemo as v, useRef as y, useState as b } from "react";
 import { SendCompletionDtoModeEnum as x } from "@epam/ai-dial-chat-api-client";
 //#region src/conversation/useConversationStream/useConversationStream.ts
-var S = 2e4, ee = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof o ? e.message : "", C = ({ conversationId: r, state: { setConversation: o, conversationRef: C }, transport: w, generation: { startGeneration: T, completeGeneration: E }, channel: D, overlay: O, onStopError: k, generationConflictMessage: A = t, generationPersistenceErrorMessage: j = n, onStreamError: M }) => {
-	let N = y(M);
+var te = 2e4, ne = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof o ? e.message : "", S = ({ conversationId: r, state: { setConversation: o, conversationRef: S }, transport: C, generation: { startGeneration: w, completeGeneration: T }, channel: E, overlay: D, onStopError: O, generationConflictMessage: k = t, generationPersistenceErrorMessage: A = n, onStreamError: j, batchChunksPerFrame: M = !1 }) => {
+	let N = y(j);
 	_(() => {
-		N.current = M;
-	}, [M]);
-	let [P, F] = b(() => /* @__PURE__ */ new Set()), [I, L] = b(null), R = y(null), z = y(null), B = y(/* @__PURE__ */ new Set()), V = y(/* @__PURE__ */ new Map()), H = y(/* @__PURE__ */ new Set()), U = y(/* @__PURE__ */ new Map()), W = y(r);
+		N.current = j;
+	}, [j]);
+	let P = y(M);
 	_(() => {
-		W.current = r;
-	}, [r]);
-	let G = g((e) => s(W.current ?? "") === e, []), K = g((e) => {
-		F((t) => {
+		P.current = M;
+	}, [M]);
+	let [F] = b(m), [I, L] = b(() => /* @__PURE__ */ new Set()), [R, z] = b(null), B = y(null), V = y(null), H = y(/* @__PURE__ */ new Set()), U = y(/* @__PURE__ */ new Map()), W = y(/* @__PURE__ */ new Set()), G = y(/* @__PURE__ */ new Map()), K = y(r);
+	_(() => (K.current = r, () => F.cancelAll()), [r, F]);
+	let q = g((e) => s(K.current ?? "") === e, []), J = g((e) => {
+		L((t) => {
 			if (t.has(e)) return t;
 			let n = new Set(t);
 			return n.add(e), n;
 		});
-	}, []), q = g((e) => {
-		F((t) => {
+	}, []), Y = g((e) => {
+		L((t) => {
 			if (!t.has(e)) return t;
 			let n = new Set(t);
 			return n.delete(e), n;
 		});
-	}, []), J = v(() => u({
-		transport: w,
+	}, []), X = v(() => u({
+		transport: C,
 		setConversation: o,
-		conversationRef: C,
-		resumingPathsRef: B,
-		bufferedGenerationsRef: V,
-		addStreamingPath: K,
-		removeStreamingPath: q,
-		isPathDisplayed: G,
-		generationPersistenceErrorMessage: j
+		conversationRef: S,
+		resumingPathsRef: H,
+		bufferedGenerationsRef: U,
+		addStreamingPath: J,
+		removeStreamingPath: Y,
+		isPathDisplayed: q,
+		generationPersistenceErrorMessage: A,
+		frameScheduler: M ? F : void 0
 	}), [
-		K,
+		J,
+		Y,
 		q,
-		G,
-		w,
-		j
-	]), Y = g((t, n, r, u, g, _, v = x.Append) => {
-		let y = _ ?? h(), b = s(t);
-		B.current.delete(b), R.current = y, z.current = b, U.current.set(b, y), L(b);
-		let k;
-		v === x.Regenerate ? k = r : v === x.Edit && (k = r - 1);
-		let M = () => U.current.get(b) !== y, P = T(b, y), F = C.current?.messages[r];
-		F ? V.current.set(b, {
-			generationId: y,
+		C,
+		A,
+		M,
+		F
+	]), Z = g((t, n, r, u, m, g, _ = x.Append) => {
+		let v = g ?? h(), y = s(t);
+		F.flush(y), H.current.delete(y), B.current = v, V.current = y, G.current.set(y, v), z(y);
+		let b;
+		_ === x.Regenerate ? b = r : _ === x.Edit && (b = r - 1);
+		let O = () => G.current.get(y) !== v, j = w(y, v), M = S.current?.messages[r];
+		M ? U.current.set(y, {
+			generationId: v,
 			messageIndex: r,
-			message: F
-		}) : V.current.delete(b), K(b), O?.notifyGenerationStart?.(), D?.ensureConnected();
+			message: M
+		}) : U.current.delete(y), J(y), D?.notifyGenerationStart?.(), E?.ensureConnected();
 		let I = () => {
-			M() || q(b), R.current === y && (R.current = null, z.current = null, L(null)), E(b, y), D?.notifyGenerationSettled?.();
-		}, W = () => {
-			if (M()) return;
-			let e = V.current.get(b);
-			!e || e.generationId !== y || (e.message = {
+			O() || Y(y), B.current === v && (B.current = null, V.current = null, z(null)), T(y, v), E?.notifyGenerationSettled?.();
+		}, L = () => {
+			if (O()) return;
+			let e = U.current.get(y);
+			!e || e.generationId !== v || (e.message = {
 				...e.message,
-				streamErrorMessage: j
-			}, N.current?.(new i()), G(b) && o((t) => {
-				if (!t || M() || !G(b)) return t;
+				streamErrorMessage: A
+			}, N.current?.(new i()), q(y) && o((t) => {
+				if (!t || O() || !q(y)) return t;
 				let n = l(t, e);
-				return C.current = n, n;
+				return S.current = n, n;
 			}));
-		}, Y = {
-			signal: P.signal,
+		}, R = {
+			signal: j.signal,
 			onChunk: (e) => {
-				if (R.current !== y) return;
-				let t = V.current.get(b);
-				if (t?.generationId === y) {
+				if (B.current !== v) return;
+				let t = U.current.get(y);
+				if (t?.generationId === v) {
 					let n = c([t.message], 0, e);
 					n && (t.message = n[0]);
 				}
-				G(b) && o((t) => {
-					if (!t) return t;
-					let n = V.current.get(b), i;
-					if (n?.generationId === y) i = l(t, n);
-					else {
-						let n = c(t.messages, r, e);
-						if (!n) return t;
-						i = {
-							...t,
-							messages: n
-						};
+				if (q(y)) {
+					if (P.current && t?.generationId === v) {
+						F.schedule(y, () => {
+							let e = U.current.get(y);
+							if (B.current !== v || !q(y) || e?.generationId !== v) return;
+							let t = { ...e };
+							o((e) => {
+								if (!e) return e;
+								let n = l(e, t);
+								return S.current = n, n;
+							});
+						});
+						return;
 					}
-					return C.current = i, i;
-				});
+					o((t) => {
+						if (!t) return t;
+						let n = U.current.get(y), i;
+						if (n?.generationId === v) i = l(t, n);
+						else {
+							let n = c(t.messages, r, e);
+							if (!n) return t;
+							i = {
+								...t,
+								messages: n
+							};
+						}
+						return S.current = i, i;
+					});
+				}
 			},
 			onComplete: async () => {
-				let n = V.current.get(b), r = n?.generationId === y ? n : void 0;
-				if (I(), H.current.has(y) ? H.current.delete(y) : M() || O?.notifyGenerationEnd?.(), !M()) {
-					if (!G(b)) {
-						r && V.current.delete(b);
+				F.flush(y);
+				let n = U.current.get(y), r = n?.generationId === v ? n : void 0;
+				if (I(), W.current.has(v) ? W.current.delete(v) : O() || D?.notifyGenerationEnd?.(), !O()) {
+					if (!q(y)) {
+						r && U.current.delete(y);
 						return;
 					}
 					try {
-						let n = await w.getConversation(e(t));
-						if (M()) return;
-						if (!G(b)) {
-							r && V.current.delete(b);
+						let n = await C.getConversation(e(t));
+						if (O()) return;
+						if (!q(y)) {
+							r && U.current.delete(y);
 							return;
 						}
 						if (r && r.messageIndex === n.messages.length - 1 && p(n) && f(r.message)) {
-							W();
+							L();
 							return;
 						}
-						r && V.current.delete(b), o(n), C.current = n;
+						r && U.current.delete(y), o(n), S.current = n;
 					} catch {
-						M() || W();
+						O() || L();
 					}
 				}
 			},
 			onError: (e) => {
-				if (N.current?.(e), e instanceof a && !M()) {
+				if (F.flush(y), N.current?.(e), e instanceof a && !O()) {
 					Q(e);
 					return;
 				}
-				X(e);
+				K(e);
 			}
-		}, X = (e) => {
-			let t = V.current.get(b), n = t?.generationId === y ? t : void 0;
+		}, K = (e) => {
+			let t = U.current.get(y), n = t?.generationId === v ? t : void 0;
 			if (n && (e instanceof i ? n.message = {
 				...n.message,
-				streamErrorMessage: j
-			} : V.current.delete(b)), I(), M() || !G(b)) return;
-			let a = ee(e, A, j);
+				streamErrorMessage: A
+			} : U.current.delete(y)), I(), O() || !q(y)) return;
+			let a = ne(e, k, A);
 			o((e) => {
-				if (!e || M() || !G(b)) return e;
-				let t = n?.generationId === y ? l(e, n) : e, i = {
+				if (!e || O() || !q(y)) return e;
+				let t = n?.generationId === v ? l(e, n) : e, i = {
 					...t,
 					messages: t.messages.map((e, t) => t === r ? {
 						...e,
 						streamErrorMessage: a
 					} : e)
 				};
-				return C.current = i, i;
+				return S.current = i, i;
 			});
 		}, Z = () => {
-			I(), H.current.has(y) ? H.current.delete(y) : M() || O?.notifyGenerationEnd?.();
+			I(), W.current.has(v) ? W.current.delete(v) : O() || D?.notifyGenerationEnd?.();
 		}, Q = async (n) => {
-			B.current.add(b);
-			let i = await d(() => w.getConversation(e(t)), M);
-			if (M()) {
+			H.current.add(y);
+			let i = await d(() => C.getConversation(e(t)), O);
+			if (O()) {
 				I();
 				return;
 			}
-			let a = i?.messages[i.messages.length - 1], s = i != null && i.messages.length - 1 === r && a?.role === m.Assistant;
+			let a = i?.messages[i.messages.length - 1], s = i != null && i.messages.length - 1 === r && a?.role === ee.Assistant;
 			if (!i || !s) {
-				B.current.delete(b), X(n);
+				H.current.delete(y), K(n);
 				return;
 			}
-			let c = V.current.get(b), l = c?.generationId === y ? c : void 0;
+			let c = U.current.get(y), l = c?.generationId === v ? c : void 0;
 			if (p(i)) {
-				J(t, i, {
+				X(t, i, {
 					seedMessage: l?.message,
 					skipDedupe: !0,
 					onSettled: Z
 				});
 				return;
 			}
-			B.current.delete(b), l && V.current.delete(b), G(b) && (o(i), C.current = i), Z();
+			H.current.delete(y), l && U.current.delete(y), q(y) && (o(i), S.current = i), Z();
 		};
 		(async () => {
 			try {
-				let e = D?.channelId ?? await D?.waitForChannel(S) ?? void 0, t = H.current.has(y);
-				if (P.signal.aborted || M() || t) {
-					V.current.get(b)?.generationId === y && V.current.delete(b), M() || q(b), R.current === y && (R.current = null, z.current = null, L(null)), E(b, y), D?.notifyGenerationSettled?.(), t ? H.current.delete(y) : M() || O?.notifyGenerationEnd?.();
+				let e = E?.channelId ?? await E?.waitForChannel(te) ?? void 0, t = W.current.has(v);
+				if (j.signal.aborted || O() || t) {
+					U.current.get(y)?.generationId === v && U.current.delete(y), O() || Y(y), B.current === v && (B.current = null, V.current = null, z(null)), T(y, v), E?.notifyGenerationSettled?.(), t ? W.current.delete(v) : O() || D?.notifyGenerationEnd?.();
 					return;
 				}
-				w.streamCompletion(b, n, u, Y, g, y, v, k, e);
+				C.streamCompletion(y, n, u, R, m, v, _, b, e);
 			} catch (e) {
-				Y.onError(e instanceof Error ? e : Error(String(e)));
+				R.onError(e instanceof Error ? e : Error(String(e)));
 			}
 		})();
 	}, [
-		T,
-		E,
-		K,
-		q,
-		G,
-		D?.channelId,
-		D?.ensureConnected,
-		D?.waitForChannel,
-		D?.notifyGenerationSettled,
-		O,
 		w,
+		T,
 		J,
+		Y,
+		q,
+		E?.channelId,
+		E?.ensureConnected,
+		E?.waitForChannel,
+		E?.notifyGenerationSettled,
+		D,
+		C,
+		X,
+		k,
 		A,
-		j
-	]), X = g((e, t) => {
-		let n = s(e), r = V.current.get(n);
+		F
+	]), Q = g((e, t) => {
+		let n = s(e), r = U.current.get(n);
 		return r ? l(t, r) : t;
-	}, []), Z = g(() => {
-		let e = R.current;
+	}, []), re = g(() => {
+		let e = B.current;
 		if (!e || !r) return;
 		let t = s(r);
-		z.current === t && (H.current.add(e), O?.notifyStopGenerating?.(), w.stopCompletion({
+		V.current === t && (W.current.add(e), F.flush(t), D?.notifyStopGenerating?.(), C.stopCompletion({
 			generationId: e,
 			path: t
 		}).catch((e) => {
 			let t = e instanceof Error ? e : Error(String(e));
-			k?.(t);
+			O?.(t);
 		}));
 	}, [
 		r,
-		k,
+		F,
 		O,
-		w
-	]), Q = g((e, t) => J(e, t), [J]), te = r != null && P.has(s(r)), $ = r == null ? null : s(r);
+		D,
+		C
+	]), ie = g((e, t) => X(e, t), [X]), ae = r != null && I.has(s(r)), $ = r == null ? null : s(r);
 	return {
-		startStream: Y,
-		handleStop: Z,
-		resumeIfAwaitingGeneration: Q,
-		restoreBufferedGeneration: X,
-		isStreaming: te,
-		canStopStreaming: $ != null && I === $
+		startStream: Z,
+		handleStop: re,
+		resumeIfAwaitingGeneration: ie,
+		restoreBufferedGeneration: Q,
+		isStreaming: ae,
+		canStopStreaming: $ != null && R === $
 	};
 };
 //#endregion
-export { C as useConversationStream };
+export { S as useConversationStream };

package/conversation.d.ts

@@ -643,9 +643,12 @@ declare interface LocalModeValues {
 }
 
 /**
- * Returns the normalized stages of a raw stage array or a message-like
- * payload (reading `custom_content.stages`, falling back to
- * `customContent.stages`), or `undefined` when there are none.
+ * Returns the normalized stages of a complete raw stage array or a
+ * message-like payload (reading `custom_content.stages`, falling back to
+ * `customContent.stages`), or `undefined` when there are none. An entry
+ * without an `index` takes its array position — the identity a complete
+ * non-streaming array's `parent_stage_index` refers to — while explicit
+ * (possibly sparse) indexes are kept. Not for partial stream deltas.
  */
 export declare const mapStages: (source: RawStage[] | RawStageSource | null | undefined) => Stage[] | undefined;
 
@@ -731,6 +734,11 @@ export declare interface RawStage {
     tag?: string | null;
     /** Files produced or referenced by this stage. */
     attachments?: RawStageAttachment[] | null;
+    /**
+     * `index` of the parent stage (its array position in a complete unindexed
+     * array). Nullish means top level; `0` is a valid parent.
+     */
+    parent_stage_index?: number | null;
 }
 
 /**
@@ -851,7 +859,8 @@ export declare class StreamUpstreamError extends Error {
 /**
  * Returns a renderable `Stage` for one raw payload: `index` defaults to `0`,
  * `name` to `''`, an unrecognized `status` to `null` (still running), and
- * `content`/`tag`/`attachments` are passed through when present.
+ * `content`/`tag`/`attachments`/`parent_stage_index` are passed through when
+ * present (a parent of `0` included).
  */
 export declare const toStage: (stage: RawStage) => Stage;
 
@@ -1133,7 +1142,7 @@ export declare interface UseConversationPanelItemsResolvers {
  * `transport`/`generation`/`channel`/`overlay` capabilities rather than any
  * app context or `server-api` import.
  */
-export declare const useConversationStream: ({ conversationId, state: { setConversation, conversationRef }, transport, generation: { startGeneration, completeGeneration }, channel, overlay, onStopError, generationConflictMessage, generationPersistenceErrorMessage, onStreamError, }: UseConversationStreamParams) => UseConversationStreamResult;
+export declare const useConversationStream: ({ conversationId, state: { setConversation, conversationRef }, transport, generation: { startGeneration, completeGeneration }, channel, overlay, onStopError, generationConflictMessage, generationPersistenceErrorMessage, onStreamError, batchChunksPerFrame, }: UseConversationStreamParams) => UseConversationStreamResult;
 
 /** Parameters for {@link useConversationStream}. */
 export declare interface UseConversationStreamParams {
@@ -1160,6 +1169,13 @@ export declare interface UseConversationStreamParams {
      * still log or report the raw error.
      */
     onStreamError?: (error: Error) => void;
+    /**
+     * Coalesces the displayed chunk updates into one `setConversation` per
+     * animation frame. Chunks still reach the per-path buffer immediately, and
+     * any pending update is flushed before completion, error, stop, or a
+     * superseding generation. Defaults to `false` (one update per chunk).
+     */
+    batchChunksPerFrame?: boolean;
 }
 
 /** Return value of {@link useConversationStream}. */

package/file-manager-canvas.d.ts

@@ -21,6 +21,9 @@ export declare const annotationToOoxmlCanvasContent: (annotation: Annotation, an
 /**
  * Builds a `PdfCanvasContent` for a PDF citation annotation, including highlights
  * for the clicked annotation's document within its citation group.
+ * When the clicked annotation has a page but no selector with valid geometry,
+ * an invisible page-anchor highlight is added and selected so the viewer
+ * still opens that page.
  * Returns `null` if the annotation has no PDF source attachment.
  */
 export declare const annotationToPdfCanvasContent: (annotation: Annotation, groups: AnnotationGroup[], resolvers: AttachmentCanvasUrlResolvers) => PdfCanvasContent | null;

package/files/attachment-canvas.js

@@ -147,20 +147,43 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 		type: t.Html,
 		srcdoc: i
 	} : i;
-}, R = (n, r, i) => {
+}, R = (e, t) => ({
+	id: e,
+	bboxes: [{
+		page: t,
+		x1: 0,
+		y1: 0,
+		x2: 0,
+		y2: 0
+	}],
+	style: {
+		backgroundColor: "transparent",
+		opacity: 0
+	}
+}), z = (n, r, i) => {
 	let a = n.body?.source?.attachment;
 	if (a?.type !== o.PDF) return null;
 	let s = e(a.url) ? i.resolveDialFileDownloadUrl(a.url) : a.url;
 	if (s == null) return null;
-	let c = (r.find((e) => e.annotations.includes(n))?.annotations ?? [n]).filter((e) => e.body?.source?.attachment?.url === a.url), u = c.indexOf(n), f = d(c), m = l(n, u);
+	let c = (r.find((e) => e.annotations.includes(n))?.annotations ?? [n]).filter((e) => e.body?.source?.attachment?.url === a.url), u = c.indexOf(n), f = d(c), m = l(n, u), h = p(n), g = f.some((e) => e.id === m);
+	if (!g && h != null) {
+		let e = `page-anchor-${m}`;
+		return {
+			type: t.Pdf,
+			url: s,
+			highlights: [...f, R(e, h)],
+			selectedHighlightId: e,
+			page: h
+		};
+	}
 	return {
 		type: t.Pdf,
 		url: s,
 		highlights: f,
-		selectedHighlightId: f.some((e) => e.id === m) ? m : void 0,
-		page: p(n)
+		selectedHighlightId: g ? m : void 0,
+		page: h
 	};
-}, z = (e) => {
+}, B = (e) => {
 	switch (e.type) {
 		case "docx_text_anchor": return {
 			kind: i.DocxTableRow,
@@ -196,7 +219,7 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 			end: e.end
 		};
 	}
-}, B = (n, i, o) => {
+}, V = (n, i, o) => {
 	let s = n.body?.source?.attachment;
 	if (s?.url == null) return null;
 	let c = a(s.title ?? s.url, s.type);
@@ -210,7 +233,7 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 		let i = l(e, t);
 		m.push({
 			id: i,
-			locations: r.map(z)
+			locations: r.map(B)
 		}), e === n && (h = i);
 	}), {
 		type: t.Ooxml,
@@ -221,7 +244,7 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 			selectedHighlightId: h
 		} : {}
 	};
-}, V = (n, r) => {
+}, H = (n, r) => {
 	let i = m(n.url);
 	if (i == null) return null;
 	let a = e(i.baseUrl) ? r.resolveDialFileDownloadUrl(i.baseUrl) : i.baseUrl;
@@ -234,55 +257,42 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 	return {
 		type: t.Pdf,
 		url: a,
-		highlights: [{
-			id: o,
-			bboxes: [{
-				page: i.page,
-				x1: 0,
-				y1: 0,
-				x2: 0,
-				y2: 0
-			}],
-			style: {
-				backgroundColor: "transparent",
-				opacity: 0
-			}
-		}],
+		highlights: [R(o, i.page)],
 		selectedHighlightId: o,
 		page: i.page
 	};
-}, H = (e) => {
+}, U = (e) => {
 	try {
 		let t = new URL(e).protocol;
 		return t === "http:" || t === "https:" || t === "blob:";
 	} catch {
 		return !1;
 	}
-}, U = (t) => t.url != null && !e(t.url) && H(t.url) ? t.url : void 0, W = async (e, n) => {
+}, W = (t) => t.url != null && !e(t.url) && U(t.url) ? t.url : void 0, G = async (e, n) => {
 	let r = await O(e, n);
 	if (r != null) return typeof r == "string" ? {
 		type: t.Pdf,
 		url: r
 	} : r;
-	let i = U(e);
+	let i = W(e);
 	return i == null ? null : {
 		type: t.Pdf,
 		url: i
 	};
-}, G = async (e, n, r) => {
+}, K = async (e, n, r) => {
 	let i = await O(e, n);
 	if (i != null) return typeof i == "string" ? {
 		type: t.Ooxml,
 		url: i,
 		format: r
 	} : i;
-	let a = U(e);
+	let a = W(e);
 	return a == null ? null : {
 		type: t.Ooxml,
 		url: a,
 		format: r
 	};
-}, K = async (e, n) => {
+}, q = async (e, n) => {
 	let r = await k(e, n);
 	if (r == null) return null;
 	if (typeof r != "string") return r;
@@ -298,7 +308,7 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 			text: r
 		};
 	}
-}, q = async (e, n, r, i) => {
+}, J = async (e, n, r, i) => {
 	let a = await k(e, n);
 	if (a == null || typeof a != "string") return null;
 	let o = {};
@@ -320,7 +330,7 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 		visualizerName: r.title,
 		requestTimeout: r.requestTimeout
 	};
-}, J = (e, n, r, i) => {
+}, Y = (e, n, r, i) => {
 	let a = {
 		width: r.width,
 		height: r.height,
@@ -350,4 +360,4 @@ var g = (e, t) => URL.createObjectURL(s(e, t)), _ = (e) => {
 	};
 };
 //#endregion
-export { B as annotationToOoxmlCanvasContent, R as annotationToPdfCanvasContent, S as clearAttachmentCache, A as hasAttachmentTextSource, V as referenceAttachmentToPdfCanvasContent, F as resolveCodeCanvasContent, J as resolveGroupedVisualizerCanvasContent, L as resolveHtmlCanvasContent, j as resolveImageCanvasContent, K as resolveJsonCanvasContent, N as resolveMarkdownCanvasContent, G as resolveOoxmlCanvasContent, W as resolvePdfCanvasContent, M as resolveTextCanvasContent, q as resolveVisualizerCanvasContent };
+export { V as annotationToOoxmlCanvasContent, z as annotationToPdfCanvasContent, S as clearAttachmentCache, A as hasAttachmentTextSource, H as referenceAttachmentToPdfCanvasContent, F as resolveCodeCanvasContent, Y as resolveGroupedVisualizerCanvasContent, L as resolveHtmlCanvasContent, j as resolveImageCanvasContent, q as resolveJsonCanvasContent, N as resolveMarkdownCanvasContent, K as resolveOoxmlCanvasContent, G as resolvePdfCanvasContent, M as resolveTextCanvasContent, J as resolveVisualizerCanvasContent };

package/index.d.ts

@@ -225,6 +225,9 @@ export declare const annotationToOoxmlCanvasContent: (annotation: Annotation, an
 /**
  * Builds a `PdfCanvasContent` for a PDF citation annotation, including highlights
  * for the clicked annotation's document within its citation group.
+ * When the clicked annotation has a page but no selector with valid geometry,
+ * an invisible page-anchor highlight is added and selected so the viewer
+ * still opens that page.
  * Returns `null` if the annotation has no PDF source attachment.
  */
 export declare const annotationToPdfCanvasContent: (annotation: Annotation, groups: AnnotationGroup[], resolvers: AttachmentCanvasUrlResolvers) => PdfCanvasContent | null;
@@ -2400,9 +2403,12 @@ export declare interface MapSkillToCatalogItemOptions {
 }
 
 /**
- * Returns the normalized stages of a raw stage array or a message-like
- * payload (reading `custom_content.stages`, falling back to
- * `customContent.stages`), or `undefined` when there are none.
+ * Returns the normalized stages of a complete raw stage array or a
+ * message-like payload (reading `custom_content.stages`, falling back to
+ * `customContent.stages`), or `undefined` when there are none. An entry
+ * without an `index` takes its array position — the identity a complete
+ * non-streaming array's `parent_stage_index` refers to — while explicit
+ * (possibly sparse) indexes are kept. Not for partial stream deltas.
  */
 export declare const mapStages: (source: RawStage[] | RawStageSource | null | undefined) => Stage[] | undefined;
 
@@ -3221,6 +3227,11 @@ export declare interface RawStage {
     tag?: string | null;
     /** Files produced or referenced by this stage. */
     attachments?: RawStageAttachment[] | null;
+    /**
+     * `index` of the parent stage (its array position in a complete unindexed
+     * array). Nullish means top level; `0` is a valid parent.
+     */
+    parent_stage_index?: number | null;
 }
 
 /**
@@ -4435,7 +4446,8 @@ export declare const toPublishRuleDto: (rule: PublicationRule) => PublishRuleDto
 /**
  * Returns a renderable `Stage` for one raw payload: `index` defaults to `0`,
  * `name` to `''`, an unrecognized `status` to `null` (still running), and
- * `content`/`tag`/`attachments` are passed through when present.
+ * `content`/`tag`/`attachments`/`parent_stage_index` are passed through when
+ * present (a parent of `0` included).
  */
 export declare const toStage: (stage: RawStage) => Stage;
 
@@ -5225,7 +5237,7 @@ export declare interface UseConversationSourcesResult {
  * `transport`/`generation`/`channel`/`overlay` capabilities rather than any
  * app context or `server-api` import.
  */
-export declare const useConversationStream: ({ conversationId, state: { setConversation, conversationRef }, transport, generation: { startGeneration, completeGeneration }, channel, overlay, onStopError, generationConflictMessage, generationPersistenceErrorMessage, onStreamError, }: UseConversationStreamParams) => UseConversationStreamResult;
+export declare const useConversationStream: ({ conversationId, state: { setConversation, conversationRef }, transport, generation: { startGeneration, completeGeneration }, channel, overlay, onStopError, generationConflictMessage, generationPersistenceErrorMessage, onStreamError, batchChunksPerFrame, }: UseConversationStreamParams) => UseConversationStreamResult;
 
 /** Parameters for {@link useConversationStream}. */
 export declare interface UseConversationStreamParams {
@@ -5252,6 +5264,13 @@ export declare interface UseConversationStreamParams {
      * still log or report the raw error.
      */
     onStreamError?: (error: Error) => void;
+    /**
+     * Coalesces the displayed chunk updates into one `setConversation` per
+     * animation frame. Chunks still reach the per-path buffer immediately, and
+     * any pending update is flushed before completion, error, stop, or a
+     * superseding generation. Defaults to `false` (one update per chunk).
+     */
+    batchChunksPerFrame?: boolean;
 }
 
 /** Return value of {@link useConversationStream}. */

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@epam/ai-dial-chat-hooks",
   "description": "Framework-level React hooks extracted from AI DIAL Chat for building custom chat interfaces",
-  "version": "1.2.0-dev.219",
+  "version": "1.2.0-dev.229",
   "license": "Apache-2.0",
   "type": "module",
   "main": "./index.js",
@@ -111,7 +111,7 @@
     }
   },
   "dependencies": {
-    "@epam/ai-dial-chat-api-client": "1.2.0-dev.219",
+    "@epam/ai-dial-chat-api-client": "1.2.0-dev.229",
     "dompurify": "3.4.15",
     "fflate": "^0.8.3",
     "lru-cache": "^10.4.3",
@@ -120,21 +120,21 @@
   },
   "peerDependencies": {
     "react": "^19.2.8",
-    "@epam/ai-dial-attachment-canvas": "1.2.0-dev.219",
-    "@epam/ai-dial-attachment-input": "1.2.0-dev.219",
-    "@epam/ai-dial-catalog": "1.2.0-dev.219",
-    "@epam/ai-dial-chat-overlay": "1.2.0-dev.219",
-    "@epam/ai-dial-chat-shared": "1.2.0-dev.219",
-    "@epam/ai-dial-mcp-apps": "1.2.0-dev.219",
-    "@epam/ai-dial-publish-panel": "1.2.0-dev.219",
-    "@epam/ai-dial-quotations": "1.2.0-dev.219",
-    "@epam/ai-dial-react-file-manager": "^0.3.0-dev.19",
-    "@epam/ai-dial-scheduled-tasks": "1.2.0-dev.219",
-    "@epam/ai-dial-share": "1.2.0-dev.219",
-    "@epam/ai-dial-skill-editor": "1.2.0-dev.219",
-    "@epam/ai-dial-source-panel": "1.2.0-dev.219",
-    "@epam/ai-dial-ui-kit": "^0.15.0-dev.27",
-    "@epam/ai-dial-usage-dashboard": "1.2.0-dev.219",
+    "@epam/ai-dial-attachment-canvas": "1.2.0-dev.229",
+    "@epam/ai-dial-attachment-input": "1.2.0-dev.229",
+    "@epam/ai-dial-catalog": "1.2.0-dev.229",
+    "@epam/ai-dial-chat-overlay": "1.2.0-dev.229",
+    "@epam/ai-dial-chat-shared": "1.2.0-dev.229",
+    "@epam/ai-dial-mcp-apps": "1.2.0-dev.229",
+    "@epam/ai-dial-publish-panel": "1.2.0-dev.229",
+    "@epam/ai-dial-quotations": "1.2.0-dev.229",
+    "@epam/ai-dial-react-file-manager": "^0.3.0-dev.20",
+    "@epam/ai-dial-scheduled-tasks": "1.2.0-dev.229",
+    "@epam/ai-dial-share": "1.2.0-dev.229",
+    "@epam/ai-dial-skill-editor": "1.2.0-dev.229",
+    "@epam/ai-dial-source-panel": "1.2.0-dev.229",
+    "@epam/ai-dial-ui-kit": "^0.15.0-dev.30",
+    "@epam/ai-dial-usage-dashboard": "1.2.0-dev.229",
     "@modelcontextprotocol/sdk": "^1.29.0",
     "@epam/pdf-highlighter-kit": "^0.0.19",
     "@mcp-ui/client": "^7.1.1"