@epam/ai-dial-chat-hooks 1.2.0-dev.14 → 1.2.0-dev.142

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

@@ -1,30 +1,39 @@
 //#region src/conversation/create-chat-stream-api.ts
-var e = { "Content-Type": "application/json" }, t = 409, n = "A response is already being generated in this conversation. Wait for it to finish or stop it before sending another message.", r = class extends Error {
-	constructor(e = n) {
+var e = { "Content-Type": "application/json" }, t = 409, n = "The response could not be saved. It is still shown here, but may be lost if you reload or leave this page. Copy it before continuing.", r = class extends Error {
+	static type = "conversation_save_failed";
+	constructor() {
+		super(n), this.name = "GenerationPersistenceError";
+	}
+}, i = "A response is already being generated in this conversation. Wait for it to finish or stop it before sending another message.", a = class extends Error {
+	constructor(e = i) {
 		super(e), this.name = "GenerationConflictError";
 	}
-}, i = (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);
+}, o = class extends Error {
+	constructor(e) {
+		super(e), this.name = "StreamUpstreamError";
+	}
+}, s = (e, t, n) => {
+	let i = e.trim();
+	if (!i || i.startsWith(":") || !i.startsWith("data: ")) return;
+	let a = i.slice(6);
+	if (a !== "[DONE]") try {
+		let e = JSON.parse(a);
 		if (e.error) {
-			n(Error(e.error.message));
+			n(e.error.type === r.type ? new r() : new o(e.error.message));
 			return;
 		}
 		t(e);
 	} catch {}
-}, a = (n) => {
-	let { completionsBasePath: a } = n, o = (...e) => (n.fetchImpl ?? fetch)(...e);
+}, c = (n) => {
+	let { completionsBasePath: r } = n, i = (...e) => (n.fetchImpl ?? fetch)(...e);
 	return {
-		streamCompletion: (s, c, l, u, d, f, p, m, h) => {
+		streamCompletion: (o, c, l, u, d, f, p, m, h) => {
 			let { onChunk: g, onComplete: _, onError: v, signal: y } = u;
 			(async () => {
 				let u;
 				try {
 					let t = n.getTimezone?.();
-					u = await o(`${a}/completions`, {
+					u = await i(`${r}/completions`, {
 						method: "POST",
 						credentials: "include",
 						signal: y,
@@ -34,7 +43,7 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 							...t ? { "X-Timezone": t } : {}
 						},
 						body: JSON.stringify({
-							path: s,
+							path: o,
 							message: c ?? "",
 							model: l,
 							custom_content: d || {},
@@ -52,7 +61,7 @@ var e = { "Content-Type": "application/json" }, t = 409, n = "A response is alre
 				let b = u.headers.get("x-csrf-token");
 				if (b && n.setCsrfToken(b), !u.ok) {
 					if (u.status === t) {
-						v(new r());
+						v(new a());
 						return;
 					}
 					v(/* @__PURE__ */ Error(`Stream request failed with status ${u.status}`));
@@ -69,13 +78,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() && s(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) s(e, g, T);
 					}
 					w || _();
 				} catch (e) {
@@ -87,7 +96,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 a = await i(`${r}/completions/stop`, {
 				method: "POST",
 				credentials: "include",
 				headers: {
@@ -95,10 +104,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}`);
+			}), o = a.headers.get("x-csrf-token");
+			if (o && n.setCsrfToken(o), !a.ok) throw Error(`stopCompletion failed: ${a.status}`);
 		}
 	};
 };
 //#endregion
-export { n as DEFAULT_GENERATION_CONFLICT_MESSAGE, r as GenerationConflictError, a as createChatStreamApi };
+export { i as DEFAULT_GENERATION_CONFLICT_MESSAGE, n as DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE, a as GenerationConflictError, r as GenerationPersistenceError, o as StreamUpstreamError, c 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/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/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 };