@epam/ai-dial-chat-hooks 1.2.0-dev.99 → 1.3.0-dev.5

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