@epam/ai-dial-chat-hooks 1.2.0-dev.28 → 1.2.0-dev.284

package/conversation/useConversationScroll/useConversationScroll.js

@@ -1,121 +1,125 @@
 import { useCallback as e, useEffect as t, useLayoutEffect as n, useRef as r, useState as i } from "react";
 //#region src/conversation/useConversationScroll/useConversationScroll.ts
 var a = 80, o = 1, s = ({ messages: s, isAssistantTyping: c, conversationId: l }) => {
-	let [u, d] = i(!1), f = r(null), p = r(null), m = r(null), h = r(!1), g = r(!1), _ = r(null), v = r(/* @__PURE__ */ new Map()), y = e((e, t) => {
-		t ? v.current.set(e, t) : v.current.delete(e);
-	}, []), b = r(null), x = e((e) => {
-		b.current = e;
-	}, []), S = e((e, t) => {
+	let [u, d] = i(!1), f = r(null), p = r(null), m = r(null), h = r(!1), g = r(!1), _ = r(null), v = r(!0), y = r(/* @__PURE__ */ new Map()), b = e((e, t) => {
+		t ? y.current.set(e, t) : y.current.delete(e);
+	}, []), x = r(null), S = e((e) => {
+		x.current = e;
+	}, []), C = e((e, t) => {
 		let n = f.current;
 		n && n.scrollTo({
 			top: e,
 			behavior: t ? "instant" : "smooth"
 		});
-	}, []), C = e(() => {
-		m.current && (m.current.style.height = "0px"), h.current = !1, g.current = !1, _.current = null;
 	}, []), w = e(() => {
+		m.current && (m.current.style.height = "0px"), h.current = !1, g.current = !1, _.current = null;
+	}, []), T = e(() => {
 		let e = f.current, t = p.current;
 		if (!e || !t) return null;
 		let n = e.getBoundingClientRect(), r = t.getBoundingClientRect();
 		return e.scrollTop + (r.bottom - n.top) - e.clientHeight;
-	}, []), T = e((e = !1) => {
-		let t = w();
-		t != null && S(Math.max(t, 0), e);
-	}, [S, w]), E = e(() => {
+	}, []), E = e((e = !1) => {
+		let t = T();
+		t != null && C(Math.max(t, 0), e);
+	}, [C, T]), D = e(() => {
 		let e = f.current, t = p.current;
 		if (!e || !t) return null;
 		let n = e.getBoundingClientRect();
 		return t.getBoundingClientRect().bottom - n.top - e.clientHeight;
-	}, []), D = e(() => {
+	}, []), O = e(() => {
 		if (!g.current) return;
-		let e = E();
-		e == null || e < -1 || C();
-	}, [C, E]), O = e((e) => {
-		let t = f.current, n = v.current.get(e);
+		let e = D();
+		e == null || e < -1 || w();
+	}, [w, D]), k = e((e) => {
+		let t = f.current, n = y.current.get(e);
 		if (!t || !n) return null;
 		let r = t.getBoundingClientRect(), i = n.getBoundingClientRect(), a = t.scrollTop + (i.top - r.top);
 		return Math.max(a, 0);
-	}, []), k = e((e, t = !1) => {
-		let n = O(e);
-		n != null && S(n, t);
-	}, [O, S]), A = e(() => {
-		let e = w();
+	}, []), A = e((e, t = !1) => {
+		let n = k(e);
+		n != null && C(n, t);
+	}, [k, C]), j = e(() => {
+		let e = T();
 		return e == null ? null : Math.max(e, 0);
-	}, [w]), j = e(() => {
+	}, [T]), M = e(() => {
 		if (!h.current) return null;
-		let e = _.current, t = A();
+		let e = _.current, t = j();
 		return e == null || t == null ? null : Math.max(e, t);
-	}, [A]), M = e(() => {
-		let e = f.current, t = j();
+	}, [j]), N = e(() => {
+		let e = f.current, t = M();
 		!e || t == null || e.scrollTop > t + o && (e.scrollTop = t);
-	}, [j]), N = e((e) => {
-		let t = m.current, n = A();
+	}, [M]), P = e((e) => {
+		let t = m.current, n = j();
 		if (!t || n == null) return;
 		let r = Math.max(e - n, 0);
 		t.style.height = `${r}px`, h.current = r > 0, g.current = !1, _.current = r > 0 ? e : null;
-	}, [A]), P = e((e) => {
-		let t = O(e);
-		t != null && (N(t), k(e, !1));
+	}, [j]), F = e((e) => {
+		let t = k(e);
+		t != null && (P(t), A(e, !1));
 	}, [
-		N,
-		O,
-		k
-	]), F = e(() => {
-		let e = E();
-		e != null && d(e >= a);
-	}, [E]), I = r(s.length), L = r(null);
+		P,
+		k,
+		A
+	]), I = e(() => {
+		let e = D();
+		if (e == null) return;
+		let t = e < a;
+		v.current = t, d(!t);
+	}, [D]), L = r(s.length), R = r(null);
 	n(() => {
-		let e = s.length !== I.current, t = l !== L.current, n = () => {
-			I.current = s.length, L.current = l;
+		let e = s.length !== L.current, t = l !== R.current, n = () => {
+			L.current = s.length, R.current = l;
 		};
-		t && C();
-		let r = b.current;
-		if (r != null && v.current.has(r)) {
-			b.current = null, P(r), n();
+		t && w();
+		let r = x.current;
+		if (r != null && y.current.has(r)) {
+			x.current = null, F(r), n();
 			return;
 		}
-		c || ((e || t) && T(!1), n());
+		c || ((e || t) && E(!1), n());
 	}, [
 		s,
 		c,
 		l,
-		T,
-		P,
-		C
+		E,
+		F,
+		w
 	]), n(() => {
-		!c && h.current && (g.current = !0, D());
-	}, [c, D]);
-	let R = e(() => {
-		M(), D(), F();
+		!c && h.current && (g.current = !0, O());
+	}, [c, O]);
+	let z = e(() => {
+		N(), O(), I();
 	}, [
-		M,
-		D,
-		F
+		N,
+		O,
+		I
 	]);
 	return t(() => {
 		let e = f.current;
-		if (e) return e.addEventListener("scroll", R, { passive: !0 }), () => e.removeEventListener("scroll", R);
-	}, [R]), t(() => {
+		if (e) return e.addEventListener("scroll", z, { passive: !0 }), () => e.removeEventListener("scroll", z);
+	}, [z]), t(() => {
 		let e = p.current;
 		if (!e) return;
 		let t = new ResizeObserver(() => {
-			M(), D(), F();
+			let e = v.current && !h.current;
+			N(), e && E(!0), O(), I();
 		});
 		return t.observe(e), () => t.disconnect();
 	}, [
-		M,
-		D,
-		F
+		N,
+		O,
+		I,
+		E
 	]), {
 		containerRef: f,
 		contentRef: p,
 		spacerRef: m,
-		setMessageRef: y,
+		setMessageRef: b,
 		isScrollButtonVisible: u,
 		scrollToBottom: e(() => {
-			T(!1);
-		}, [T]),
-		armAnchor: x
+			E(!1);
+		}, [E]),
+		armAnchor: S
 	};
 };
 //#endregion

package/conversation/useConversationStream/apply-chunk.js

@@ -75,4 +75,4 @@ var t = (e) => {
 	});
 };
 //#endregion
-export { o as applyChunkToMessages };
+export { o as applyChunkToMessages, a as mergeStages };

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

@@ -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 };