@epam/ai-dial-chat-hooks 1.2.0-dev.15 → 1.2.0-dev.152

package/conversation/useConversationStream/generation-resume.js

@@ -1,16 +1,17 @@
 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 { MessageRole as o } from "@epam/ai-dial-chat-shared";
 //#region src/conversation/useConversationStream/generation-resume.ts
-var a = (e) => {
+var s = (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) => {
+}, c = (e) => {
 	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 === o.Assistant && !s(t) && t.streamErrorMessage == null && !t.wasStoppedByUser;
+}, l = 300 * 1e3, u = "awaiting-resume", d = async (e, t) => {
 	let n = e.getReader(), r = new TextDecoder(), i = "";
 	try {
 		for (;;) {
@@ -39,88 +40,99 @@ 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 () => {
+}, f = ({ transport: o, setConversation: f, conversationRef: p, resumingPathsRef: m, bufferedGenerationsRef: h, addStreamingPath: g, removeStreamingPath: _, isPathDisplayed: v, generationPersistenceErrorMessage: y = t }) => (t, b) => {
+	if (!c(b)) return;
+	let x = r(t);
+	if (m.current.has(x)) return;
+	m.current.add(x), g(x);
+	let S = b.messages.length - 1, C = {
+		generationId: u,
+		messageIndex: S,
+		message: b.messages[S]
+	};
+	h.current.set(x, C);
+	let w = () => h.current.get(x) === C, T = (e, t = !1) => {
+		if (!w()) return;
+		m.current.delete(x), _(x);
+		let n = e && e.messages.length - 1 === S && c(e) && s(C.message);
+		if (t || n) {
+			C.message = {
+				...C.message,
+				streamErrorMessage: y
+			}, v(x) && f((e) => {
+				if (!e || !w() || !v(x)) return e;
+				let t = a(e, C);
+				return p.current = t, t;
+			});
+			return;
+		}
+		h.current.delete(x), e && v(x) && (f(e), p.current = e);
+	}, E = async () => {
 		try {
-			b(await i.getConversation(e(g)));
+			T(await o.getConversation(e(t)));
 		} catch {
-			b();
-		}
-	}, 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]);
+			T(void 0, s(C.message));
 		}
-		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;
+	}, D = (e) => {
+		w() && (C.message = e, v(x) && f((e) => {
+			if (!e || !w() || !v(x)) return e;
+			let t = a(e, C);
+			return p.current = t, t;
+		}));
+	}, O = (e) => {
+		if (!w()) return;
+		let t = i([C.message], 0, e);
+		t && (C.message = t[0]), v(x) && f((e) => {
+			if (!e || !w() || !v(x)) return e;
+			let t = a(e, C);
+			return p.current = t, t;
 		});
-	}, w = async () => {
-		let t = new AbortController(), n;
+	}, k = async () => {
+		let n = new AbortController(), r;
 		try {
-			n = await i.watchConversation(v, t.signal);
+			r = await o.watchConversation(x, n.signal);
 		} catch {
-			await x();
+			await E();
 			return;
 		}
-		let r = !1, a = window.setTimeout(() => {
-			t.abort();
-		}, s);
+		let i = !1, a = window.setTimeout(() => {
+			n.abort();
+		}, l);
 		try {
-			await l(n, async (t) => {
-				if (t?.action !== "UPDATE") return !1;
+			await d(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 (!c(n)) return T(n), i = !0, !0;
 				} catch {}
 				return !1;
 			});
 		} finally {
 			clearTimeout(a);
 		}
-		r || await x();
-	}, T = async () => {
+		i || await E();
+	}, A = async () => {
 		let e = new AbortController(), t;
 		try {
-			t = await i.attachToGeneration(v, e.signal);
+			t = await o.attachToGeneration(x, e.signal);
 		} catch {
 			return !1;
 		}
-		let n = !1;
-		return await l(t, (e) => {
+		let r = !1, i = !1;
+		return await d(t, (e) => {
+			if (!w()) return !0;
 			switch (e.type) {
-				case "snapshot": return S(e.message), !1;
-				case "chunk": return C(e.chunk), !1;
+				case "snapshot": return D(e.message), !1;
+				case "chunk": return O(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 ? T(void 0, !0) : await E(), !0) : !w();
 	};
 	(async () => {
-		await T() || await w();
+		await A() || await k();
 	})();
 };
 //#endregion
-export { u as createResumeIfAwaitingGeneration, o as isAwaitingGenerationResume };
+export { f as createResumeIfAwaitingGeneration, s as hasGeneratedPayload, c as isAwaitingGenerationResume };

package/conversation/useConversationStream/useConversationStream.js

@@ -1,152 +1,195 @@
 import { safeDecodeURI as e } from "../../shared/string-utils.js";
-import { DEFAULT_GENERATION_CONFLICT_MESSAGE as t, GenerationConflictError as n } from "../create-chat-stream-api.js";
-import { getConversationPath as r } from "./conversation-path.js";
-import { applyChunkToMessages as i } from "./apply-chunk.js";
-import { restoreBufferedMessage as a } from "./buffered-generation.js";
-import { createResumeIfAwaitingGeneration as o } from "./generation-resume.js";
-import { generateUUID as s } from "@epam/ai-dial-chat-shared";
-import { useCallback as c, useEffect as l, useMemo as u, useRef as d, useState as f } from "react";
-import { SendCompletionDtoModeEnum as p } from "@epam/ai-dial-chat-api-client";
+import { DEFAULT_GENERATION_CONFLICT_MESSAGE as t, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as n, GenerationConflictError as r, GenerationPersistenceError as i, StreamUpstreamError as a } from "../create-chat-stream-api.js";
+import { getConversationPath as o } from "./conversation-path.js";
+import { applyChunkToMessages as s } from "./apply-chunk.js";
+import { restoreBufferedMessage as c } from "./buffered-generation.js";
+import { createResumeIfAwaitingGeneration as l, hasGeneratedPayload as u, isAwaitingGenerationResume as d } from "./generation-resume.js";
+import { generateUUID as f } from "@epam/ai-dial-chat-shared";
+import { useCallback as p, useEffect as m, useMemo as h, useRef as g, useState as _ } from "react";
+import { SendCompletionDtoModeEnum as v } from "@epam/ai-dial-chat-api-client";
 //#region src/conversation/useConversationStream/useConversationStream.ts
-var m = 2e4, h = ({ conversationId: h, state: { setConversation: g, conversationRef: _ }, transport: v, generation: { startGeneration: y, completeGeneration: b }, channel: x, overlay: S, onStopError: C, generationConflictMessage: w = t }) => {
-	let [T, E] = f(() => /* @__PURE__ */ new Set()), [D, O] = f(null), k = d(null), A = d(null), j = d(/* @__PURE__ */ new Set()), M = d(/* @__PURE__ */ new Map()), N = d(/* @__PURE__ */ new Set()), P = d(/* @__PURE__ */ new Map()), F = d(h);
-	l(() => {
-		F.current = h;
-	}, [h]);
-	let I = c((e) => r(F.current ?? "") === e, []), L = c((e) => {
-		E((t) => {
+var y = 2e4, b = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof a ? e.message : "", x = ({ conversationId: r, state: { setConversation: a, conversationRef: x }, transport: S, generation: { startGeneration: C, completeGeneration: w }, channel: T, overlay: E, onStopError: D, generationConflictMessage: O = t, generationPersistenceErrorMessage: k = n, onStreamError: A }) => {
+	let j = g(A);
+	m(() => {
+		j.current = A;
+	}, [A]);
+	let [M, N] = _(() => /* @__PURE__ */ new Set()), [P, F] = _(null), I = g(null), L = g(null), R = g(/* @__PURE__ */ new Set()), z = g(/* @__PURE__ */ new Map()), B = g(/* @__PURE__ */ new Set()), V = g(/* @__PURE__ */ new Map()), H = g(r);
+	m(() => {
+		H.current = r;
+	}, [r]);
+	let U = p((e) => o(H.current ?? "") === e, []), W = p((e) => {
+		N((t) => {
 			if (t.has(e)) return t;
 			let n = new Set(t);
 			return n.add(e), n;
 		});
-	}, []), R = c((e) => {
-		E((t) => {
+	}, []), G = p((e) => {
+		N((t) => {
 			if (!t.has(e)) return t;
 			let n = new Set(t);
 			return n.delete(e), n;
 		});
-	}, []), z = c((t, o, c, l, u, d, f = p.Append) => {
-		let h = d ?? s(), C = r(t);
-		k.current = h, A.current = C, P.current.set(C, h), O(C);
-		let T;
-		f === p.Regenerate ? T = c : f === p.Edit && (T = c - 1);
-		let E = () => P.current.get(C) !== h, D = y(C, h), j = _.current?.messages[c];
-		j ? M.current.set(C, {
-			generationId: h,
-			messageIndex: c,
-			message: j
-		}) : M.current.delete(C), L(C), S?.notifyGenerationStart?.(), x?.ensureConnected();
-		let F = {
-			signal: D.signal,
+	}, []), K = p((t, n, r, l, p, m, h = v.Append) => {
+		let g = m ?? f(), _ = o(t);
+		R.current.delete(_), I.current = g, L.current = _, V.current.set(_, g), F(_);
+		let D;
+		h === v.Regenerate ? D = r : h === v.Edit && (D = r - 1);
+		let A = () => V.current.get(_) !== g, M = C(_, g), N = x.current?.messages[r];
+		N ? z.current.set(_, {
+			generationId: g,
+			messageIndex: r,
+			message: N
+		}) : z.current.delete(_), W(_), E?.notifyGenerationStart?.(), T?.ensureConnected();
+		let P = () => {
+			if (A()) return;
+			let e = z.current.get(_);
+			!e || e.generationId !== g || (e.message = {
+				...e.message,
+				streamErrorMessage: k
+			}, j.current?.(new i()), U(_) && a((t) => {
+				if (!t || A() || !U(_)) return t;
+				let n = c(t, e);
+				return x.current = n, n;
+			}));
+		}, H = {
+			signal: M.signal,
 			onChunk: (e) => {
-				if (k.current !== h) return;
-				let t = M.current.get(C);
-				if (t?.generationId === h) {
-					let n = i([t.message], 0, e);
+				if (I.current !== g) return;
+				let t = z.current.get(_);
+				if (t?.generationId === g) {
+					let n = s([t.message], 0, e);
 					n && (t.message = n[0]);
 				}
-				I(C) && g((t) => {
+				U(_) && a((t) => {
 					if (!t) return t;
-					let n = M.current.get(C), r;
-					if (n?.generationId === h) r = a(t, n);
+					let n = z.current.get(_), i;
+					if (n?.generationId === g) i = c(t, n);
 					else {
-						let n = i(t.messages, c, e);
+						let n = s(t.messages, r, e);
 						if (!n) return t;
-						r = {
+						i = {
 							...t,
 							messages: n
 						};
 					}
-					return _.current = r, r;
+					return x.current = i, i;
 				});
 			},
 			onComplete: async () => {
-				if (M.current.get(C)?.generationId === h && M.current.delete(C), E() || R(C), k.current === h && (k.current = null, A.current = null, O(null)), b(C, h), x?.notifyGenerationSettled?.(), N.current.has(h) ? N.current.delete(h) : E() || S?.notifyGenerationEnd?.(), !E() && I(C)) try {
-					let n = await v.getConversation(e(t));
-					if (E() || !I(C)) return;
-					g(n), _.current = n;
-				} catch {}
+				let n = z.current.get(_), r = n?.generationId === g ? n : void 0;
+				if (A() || G(_), I.current === g && (I.current = null, L.current = null, F(null)), w(_, g), T?.notifyGenerationSettled?.(), B.current.has(g) ? B.current.delete(g) : A() || E?.notifyGenerationEnd?.(), !A()) {
+					if (!U(_)) {
+						r && z.current.delete(_);
+						return;
+					}
+					try {
+						let n = await S.getConversation(e(t));
+						if (A()) return;
+						if (!U(_)) {
+							r && z.current.delete(_);
+							return;
+						}
+						if (r && r.messageIndex === n.messages.length - 1 && d(n) && u(r.message)) {
+							P();
+							return;
+						}
+						r && z.current.delete(_), a(n), x.current = n;
+					} catch {
+						A() || P();
+					}
+				}
 			},
 			onError: (e) => {
-				let t = M.current.get(C), r = t?.generationId === h ? t : void 0;
-				if (r && M.current.delete(C), E() || R(C), k.current === h && (k.current = null, A.current = null, O(null)), b(C, h), x?.notifyGenerationSettled?.(), E() || !I(C)) return;
-				let i = e instanceof n ? w : e.message;
-				g((e) => {
-					if (!e) return e;
-					let t = r?.generationId === h ? a(e, r) : e, n = {
+				j.current?.(e);
+				let t = z.current.get(_), n = t?.generationId === g ? t : void 0;
+				if (n && (e instanceof i ? n.message = {
+					...n.message,
+					streamErrorMessage: k
+				} : z.current.delete(_)), A() || G(_), I.current === g && (I.current = null, L.current = null, F(null)), w(_, g), T?.notifyGenerationSettled?.(), A() || !U(_)) return;
+				let o = b(e, O, k);
+				a((e) => {
+					if (!e || A() || !U(_)) return e;
+					let t = n?.generationId === g ? c(e, n) : e, i = {
 						...t,
-						messages: t.messages.map((e, t) => t === c ? {
+						messages: t.messages.map((e, t) => t === r ? {
 							...e,
-							streamErrorMessage: i
+							streamErrorMessage: o
 						} : e)
 					};
-					return _.current = n, n;
+					return x.current = i, i;
 				});
 			}
 		};
 		(async () => {
 			try {
-				let e = x?.channelId ?? await x?.waitForChannel(m) ?? void 0;
-				v.streamCompletion(C, o, l, F, u, h, f, T, e);
+				let e = T?.channelId ?? await T?.waitForChannel(y) ?? void 0, t = B.current.has(g);
+				if (M.signal.aborted || A() || t) {
+					z.current.get(_)?.generationId === g && z.current.delete(_), A() || G(_), I.current === g && (I.current = null, L.current = null, F(null)), w(_, g), T?.notifyGenerationSettled?.(), t ? B.current.delete(g) : A() || E?.notifyGenerationEnd?.();
+					return;
+				}
+				S.streamCompletion(_, n, l, H, p, g, h, D, e);
 			} catch (e) {
-				F.onError(e instanceof Error ? e : Error(String(e)));
+				H.onError(e instanceof Error ? e : Error(String(e)));
 			}
 		})();
 	}, [
-		y,
-		b,
-		L,
-		R,
-		I,
-		x?.channelId,
-		x?.ensureConnected,
-		x?.waitForChannel,
-		x?.notifyGenerationSettled,
+		C,
+		w,
+		W,
+		G,
+		U,
+		T?.channelId,
+		T?.ensureConnected,
+		T?.waitForChannel,
+		T?.notifyGenerationSettled,
+		E,
 		S,
-		v,
-		w
-	]), B = c((e, t) => {
-		let n = r(e), i = M.current.get(n);
-		return i ? a(t, i) : t;
-	}, []), V = c(() => {
-		let e = k.current;
-		if (!e || !h) return;
-		let t = r(h);
-		A.current === t && (N.current.add(e), S?.notifyStopGenerating?.(), v.stopCompletion({
+		O,
+		k
+	]), q = p((e, t) => {
+		let n = o(e), r = z.current.get(n);
+		return r ? c(t, r) : t;
+	}, []), J = p(() => {
+		let e = I.current;
+		if (!e || !r) return;
+		let t = o(r);
+		L.current === t && (B.current.add(e), E?.notifyStopGenerating?.(), S.stopCompletion({
 			generationId: e,
 			path: t
 		}).catch((e) => {
 			let t = e instanceof Error ? e : Error(String(e));
-			C?.(t);
+			D?.(t);
 		}));
 	}, [
-		h,
-		C,
-		S,
-		v
-	]), H = u(() => o({
-		transport: v,
-		setConversation: g,
-		conversationRef: _,
-		resumingPathsRef: j,
-		bufferedGenerationsRef: M,
-		addStreamingPath: L,
-		removeStreamingPath: R,
-		isPathDisplayed: I
+		r,
+		D,
+		E,
+		S
+	]), Y = h(() => l({
+		transport: S,
+		setConversation: a,
+		conversationRef: x,
+		resumingPathsRef: R,
+		bufferedGenerationsRef: z,
+		addStreamingPath: W,
+		removeStreamingPath: G,
+		isPathDisplayed: U,
+		generationPersistenceErrorMessage: k
 	}), [
-		L,
-		R,
-		I,
-		v
-	]), U = h != null && T.has(r(h)), W = h == null ? null : r(h);
+		W,
+		G,
+		U,
+		S,
+		k
+	]), X = r != null && M.has(o(r)), Z = r == null ? null : o(r);
 	return {
-		startStream: z,
-		handleStop: V,
-		resumeIfAwaitingGeneration: H,
-		restoreBufferedGeneration: B,
-		isStreaming: U,
-		canStopStreaming: W != null && D === W
+		startStream: K,
+		handleStop: J,
+		resumeIfAwaitingGeneration: Y,
+		restoreBufferedGeneration: q,
+		isStreaming: X,
+		canStopStreaming: Z != null && P === Z
 	};
 };
 //#endregion
-export { h as useConversationStream };
+export { x as useConversationStream };

package/conversation-overlay.d.ts

@@ -0,0 +1,7 @@
+import { Message } from '@epam/ai-dial-chat-shared';
+import { OverlayChatMessage } from '@epam/ai-dial-chat-overlay';
+
+/** Maps chat messages to the DIAL Chat Overlay protocol's message shape. */
+export declare const toOverlayMessages: (messages: Message[]) => OverlayChatMessage[];
+
+export { }

package/conversation-overlay.js

@@ -0,0 +1,2 @@
+import { toOverlayMessages as e } from "./conversation/overlay-messages.js";
+export { e as toOverlayMessages };

package/conversation-sources/useConversationSources/useConversationSources.js

@@ -1,35 +1,42 @@
 import { MessageRole as e, messageAttachmentToDisplayAttachment as t } from "@epam/ai-dial-chat-shared";
-import { isReferenceOnlyAttachment as n, resolveMessageAnnotations as r } from "@epam/ai-dial-quotations";
-import { useMemo as i } from "react";
+import { getAnnotationPdfPage as n, isReferenceOnlyAttachment as r, parsePdfPageReference as i, resolveMessageAnnotations as a } from "@epam/ai-dial-quotations";
+import { useMemo as o } from "react";
 //#region src/conversation-sources/useConversationSources/useConversationSources.ts
-var a = {}, o = (o, s = a) => i(() => {
-	let i = [], a = [], c = [], l = /* @__PURE__ */ new Set(), u = /* @__PURE__ */ new Set(), d = /* @__PURE__ */ new Set();
-	for (let f of o) {
-		let o = f.custom_content?.attachments;
-		if (f.role === e.User) for (let e of o ?? []) {
-			let n = t(e, s);
-			u.has(n.id) || (u.add(n.id), i.push(n));
+var s = {}, c = (e, t) => {
+	let r = n(e);
+	if (r == null) return t;
+	let a = i(t);
+	return a == null || a.page != null ? t : `${t}#page=${r}`;
+}, l = (n, i = s) => o(() => {
+	let o = [], s = [], l = [], u = /* @__PURE__ */ new Set(), d = /* @__PURE__ */ new Set(), f = /* @__PURE__ */ new Set();
+	for (let p of n) {
+		let n = p.custom_content?.attachments;
+		if (p.role === e.User) for (let e of n ?? []) {
+			let n = t(e, i);
+			d.has(n.id) || (d.add(n.id), o.push(n));
 		}
-		else if (f.role === e.Assistant) {
-			let e = o?.filter((e) => !n(e));
+		else if (p.role === e.Assistant) {
+			let e = n?.filter((e) => !r(e));
 			for (let n of e ?? []) {
-				let e = t(n, s);
-				d.has(e.id) || (d.add(e.id), a.push(e));
+				let e = t(n, i);
+				f.has(e.id) || (f.add(e.id), s.push(e));
 			}
-			for (let e of o ?? []) {
-				if (!n(e)) continue;
+			for (let e of n ?? []) {
+				if (!r(e)) continue;
 				let t = e.reference_url;
-				l.has(t) || (l.add(t), c.push({
+				u.has(t) || (u.add(t), l.push({
 					url: t,
 					title: e.title ?? t,
 					contentType: e.reference_type ?? e.type ?? "",
 					quote: e.data
 				}));
 			}
-			for (let e of r(f)) {
+			for (let e of a(p)) {
 				let t = e?.body?.source?.attachment, n = t?.url;
-				!n || l.has(n) || (l.add(n), c.push({
-					url: n,
+				if (!n) continue;
+				let r = c(e, n);
+				u.has(r) || (u.add(r), l.push({
+					url: r,
 					title: t?.title ?? n.split("/").pop() ?? n,
 					contentType: t?.type ?? "",
 					quote: e.body?.quote
@@ -38,10 +45,10 @@ var a = {}, o = (o, s = a) => i(() => {
 		}
 	}
 	return {
-		uploaded: i,
-		generated: a,
-		sources: c
+		uploaded: o,
+		generated: s,
+		sources: l
 	};
-}, [o, s]);
+}, [n, i]);
 //#endregion
-export { o as useConversationSources };
+export { l as useConversationSources };

package/conversation-sources.d.ts

@@ -18,7 +18,7 @@ export declare interface UseConversationSourcesResult {
     uploaded: DisplayAttachment[];
     /** Deduplicated attachments the assistant generated across the conversation. */
     generated: DisplayAttachment[];
-    /** Quotation sources referenced by assistant messages, deduplicated by URL. */
+    /** Quotation sources referenced by assistant messages, deduplicated by URL; PDF sources are qualified with `#page=N`. */
     sources: QuotationSource[];
 }