@epam/ai-dial-chat-hooks 1.2.0-dev.12 → 1.2.0-dev.120

package/conversation/create-chat-stream-api.js

@@ -3,20 +3,24 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 	constructor(e = n) {
 		super(e), this.name = "GenerationConflictError";
 	}
-}, i = (e, t, n) => {
+}, i = class extends Error {
+	constructor(e) {
+		super(e), this.name = "StreamUpstreamError";
+	}
+}, a = (e, t, n) => {
 	let r = e.trim();
 	if (!r || r.startsWith(":") || !r.startsWith("data: ")) return;
-	let i = r.slice(6);
-	if (i !== "[DONE]") try {
-		let e = JSON.parse(i);
+	let a = r.slice(6);
+	if (a !== "[DONE]") try {
+		let e = JSON.parse(a);
 		if (e.error) {
-			n(Error(e.error.message));
+			n(new i(e.error.message));
 			return;
 		}
 		t(e);
 	} catch {}
-}, a = (n) => {
-	let { completionsBasePath: a } = n, o = (...e) => (n.fetchImpl ?? fetch)(...e);
+}, o = (n) => {
+	let { completionsBasePath: i } = n, o = (...e) => (n.fetchImpl ?? fetch)(...e);
 	return {
 		streamCompletion: (s, c, l, u, d, f, p, m, h) => {
 			let { onChunk: g, onComplete: _, onError: v, signal: y } = u;
@@ -24,7 +28,7 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 				let u;
 				try {
 					let t = n.getTimezone?.();
-					u = await o(`${a}/completions`, {
+					u = await o(`${i}/completions`, {
 						method: "POST",
 						credentials: "include",
 						signal: y,
@@ -69,13 +73,13 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 					for (;;) {
 						let { done: e, value: t } = await x.read();
 						if (e) {
-							C.trim() && i(C, g, T);
+							C.trim() && a(C, g, T);
 							break;
 						}
 						C += S.decode(t, { stream: !0 });
 						let n = C.split("\n");
 						C = n.pop() ?? "";
-						for (let e of n) i(e, g, T);
+						for (let e of n) a(e, g, T);
 					}
 					w || _();
 				} catch (e) {
@@ -87,7 +91,7 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 			})();
 		},
 		stopCompletion: async (t) => {
-			let r = await o(`${a}/completions/stop`, {
+			let r = await o(`${i}/completions/stop`, {
 				method: "POST",
 				credentials: "include",
 				headers: {
@@ -95,10 +99,10 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 					...n.getCsrfToken() == null ? {} : { "X-CSRF-Token": n.getCsrfToken() }
 				},
 				body: JSON.stringify(t)
-			}), i = r.headers.get("x-csrf-token");
-			if (i && n.setCsrfToken(i), !r.ok) throw Error(`stopCompletion failed: ${r.status}`);
+			}), a = r.headers.get("x-csrf-token");
+			if (a && n.setCsrfToken(a), !r.ok) throw Error(`stopCompletion failed: ${r.status}`);
 		}
 	};
 };
 //#endregion
-export { n as DEFAULT_GENERATION_CONFLICT_MESSAGE, r as GenerationConflictError, a as createChatStreamApi };
+export { n as DEFAULT_GENERATION_CONFLICT_MESSAGE, r as GenerationConflictError, i as StreamUpstreamError, o as createChatStreamApi };

package/conversation/stage.js

@@ -0,0 +1,23 @@
+import { StageStatus as e } from "@epam/ai-dial-chat-shared";
+//#region src/conversation/stage.ts
+var t = (t) => t === e.Completed ? e.Completed : t === e.Failed ? e.Failed : null, n = (e) => ({
+	title: e.title ?? "",
+	...e.index != null && { index: e.index },
+	...e.type != null && { type: e.type },
+	...e.data != null && { data: e.data },
+	...e.url != null && { url: e.url },
+	...e.reference_type != null && { reference_type: e.reference_type },
+	...e.reference_url != null && { reference_url: e.reference_url }
+}), r = (e) => ({
+	index: e.index ?? 0,
+	name: e.name ?? "",
+	status: t(e.status),
+	...e.content != null && { content: e.content },
+	...e.tag != null && { tag: e.tag },
+	...e.attachments?.length && { attachments: e.attachments.map(n) }
+}), i = (e) => {
+	let t = Array.isArray(e) ? e : e?.custom_content?.stages ?? e?.customContent?.stages;
+	if (t?.length) return t.map(r);
+};
+//#endregion
+export { i as mapStages, r as toStage };

package/conversation/useAttachmentUpload/useAttachmentUpload.js

@@ -26,9 +26,16 @@ var o = 700, s = 5, c = (e) => t(e.split("/").filter(Boolean).pop() ?? "file").r
 				})).url,
 				name: a.fileName
 			};
-		} catch (t) {
-			if (l(t)) {
+		} catch (c) {
+			if (l(c)) {
 				if (i.markTaken(a.fileName), o < s) {
+					if (t.listFiles) try {
+						let e = await t.listFiles({
+							bucket: u,
+							path: a.path.slice(0, a.path.lastIndexOf("/"))
+						});
+						for (let t of e.items) i.markTaken(t.name);
+					} catch {}
 					a = i.allocate(n);
 					continue;
 				}
@@ -38,10 +45,10 @@ var o = 700, s = 5, c = (e) => t(e.split("/").filter(Boolean).pop() ?? "file").r
 					let e = p.current.splice(0);
 					d?.(e), m.current = null;
 				}, f);
-				let n = t instanceof Error ? t : /* @__PURE__ */ Error("Network upload failed");
-				throw n.errorReason = r.Network, n;
+				let t = c instanceof Error ? c : /* @__PURE__ */ Error("Network upload failed");
+				throw t.errorReason = r.Network, t;
 			}
-			throw t;
+			throw c;
 		}
 	}, [
 		u,

package/conversation/useConversationHandlers/useConversationHandlers.js

@@ -44,23 +44,19 @@ var p = ({ conversation: p, conversationId: m, bucket: h, isStreaming: g, startS
 		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();
-		v((t) => {
-			if (!t) return t;
-			let r = {
-				...t.messages[e],
-				content: "",
-				custom_content: void 0,
-				wasStoppedByUser: void 0,
-				stoppedWithoutContent: void 0,
-				streamErrorMessage: void 0,
-				deploymentId: n
-			}, i = {
-				...t,
-				messages: [...t.messages.slice(0, e), r]
-			};
-			return y.current = i, i;
-		}), A((t) => new Set([...t].filter((t) => t < e))), _(m, t.content, e, n, t.custom_content, l(), f.Regenerate);
+		let n = C(), r = {
+			...p.messages[e],
+			content: "",
+			custom_content: void 0,
+			wasStoppedByUser: void 0,
+			stoppedWithoutContent: void 0,
+			streamErrorMessage: void 0,
+			deploymentId: n
+		}, i = {
+			...p,
+			messages: [...p.messages.slice(0, e), r]
+		};
+		y.current = i, v(i), A((t) => new Set([...t].filter((t) => t < e))), _(m, t.content, e, n, t.custom_content, l(), f.Regenerate);
 	}, [
 		p,
 		m,

package/conversation/useConversationImport/useConversationImport.js

@@ -191,7 +191,7 @@ var b = 5, x = 5, S = (e) => /\.(dial|zip)$/i.test(e.name), C = (e) => new Promi
 				jobId: n,
 				code: t.AttachmentSkipped,
 				names: [...j]
-			}), A.length > 0 ? D.failJob(n, h.Unknown) : j.size > 0 ? D.warnJob(n, t.AttachmentSkipped) : D.succeedJob(n);
+			}), A.length > 0 ? D.failJob(n, h.Unknown) : j.size > 0 ? D.warnJob(n, t.AttachmentSkipped, [...j]) : D.succeedJob(n);
 		}
 	}, [
 		l,

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/useConversationStream.js

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