@epam/ai-dial-chat-hooks 1.1.6 → 1.1.7

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

@@ -1,93 +1,129 @@
 //#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 = class extends Error {
+	constructor(e) {
+		super("The completion stream was interrupted", { cause: e }), this.name = "StreamInterruptedError";
+	}
+}, c = 45e3, l = (e, t) => {
+	let n = Date.now(), r, i = !1, a = () => {
+		if (i) return;
+		let o = Date.now() - n;
+		if (o >= e) {
+			t();
+			return;
+		}
+		clearTimeout(r), r = setTimeout(a, e - o);
+	}, o = () => {
+		document.visibilityState === "visible" && a();
+	}, s = typeof window < "u" && typeof document < "u";
+	return s && (document.addEventListener("visibilitychange", o), window.addEventListener("online", a), window.addEventListener("pageshow", a)), r = setTimeout(a, e), {
+		touch: () => {
+			n = Date.now();
+		},
+		stop: () => {
+			i || (i = !0, clearTimeout(r), s && (document.removeEventListener("visibilitychange", o), window.removeEventListener("online", a), window.removeEventListener("pageshow", a)));
+		}
+	};
+}, u = (e) => typeof e == "object" && !!e && e.name === "AbortError", d = (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);
+}, f = (n) => {
+	let { completionsBasePath: r } = n, i = (...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;
+		streamCompletion: (o, c, f, p, m, h, g, _, v) => {
+			let { onChunk: y, onComplete: b, onError: x, signal: S } = p;
 			(async () => {
-				let u;
+				let p;
 				try {
 					let t = n.getTimezone?.();
-					u = await o(`${a}/completions`, {
+					p = await i(`${r}/completions`, {
 						method: "POST",
 						credentials: "include",
-						signal: y,
+						signal: S,
 						headers: {
 							...e,
 							...n.getCsrfToken() == null ? {} : { "X-CSRF-Token": n.getCsrfToken() },
 							...t ? { "X-Timezone": t } : {}
 						},
 						body: JSON.stringify({
-							path: s,
+							path: o,
 							message: c ?? "",
-							model: l,
-							custom_content: d || {},
-							...f != null && { generationId: f },
-							...p != null && { mode: p },
-							...m != null && { messageIndex: m },
-							...h != null && { clientChannelId: h }
+							model: f,
+							custom_content: m || {},
+							...h != null && { generationId: h },
+							...g != null && { mode: g },
+							..._ != null && { messageIndex: _ },
+							...v != null && { clientChannelId: v }
 						})
 					});
 				} catch (e) {
-					if (e instanceof Error && e.name === "AbortError") return;
-					v(e instanceof Error ? e : Error(String(e)));
+					if (u(e)) return;
+					x(new s(e));
 					return;
 				}
-				let b = u.headers.get("x-csrf-token");
-				if (b && n.setCsrfToken(b), !u.ok) {
-					if (u.status === t) {
-						v(new r());
+				let C = p.headers.get("x-csrf-token");
+				if (C && n.setCsrfToken(C), !p.ok) {
+					if (p.status === t) {
+						x(new a());
 						return;
 					}
-					v(/* @__PURE__ */ Error(`Stream request failed with status ${u.status}`));
+					x(/* @__PURE__ */ Error(`Stream request failed with status ${p.status}`));
 					return;
 				}
-				if (!u.body) {
-					v(/* @__PURE__ */ Error("No response body"));
+				if (!p.body) {
+					x(/* @__PURE__ */ Error("No response body"));
 					return;
 				}
-				let x = u.body.getReader(), S = new TextDecoder(), C = "", w = !1, T = (e) => {
-					w = !0, v(e);
-				};
+				let w = p.body.getReader(), T = new TextDecoder(), E = "", D = !1, O = !1, k = (e) => {
+					D || (D = !0, A.stop(), x(e));
+				}, A = l(n.idleTimeoutMs ?? 45e3, () => {
+					O = !0, k(new s()), w.cancel().catch(() => void 0);
+				});
 				try {
 					for (;;) {
-						let { done: e, value: t } = await x.read();
+						let { done: e, value: t } = await w.read();
+						if (O) break;
 						if (e) {
-							C.trim() && i(C, g, T);
+							E.trim() && d(E, y, k);
 							break;
 						}
-						C += S.decode(t, { stream: !0 });
-						let n = C.split("\n");
-						C = n.pop() ?? "";
-						for (let e of n) i(e, g, T);
+						A.touch(), E += T.decode(t, { stream: !0 });
+						let n = E.split("\n");
+						E = n.pop() ?? "";
+						for (let e of n) d(e, y, k);
 					}
-					w || _();
+					D || (D = !0, b());
 				} catch (e) {
-					if (e instanceof Error && e.name === "AbortError") return;
-					v(e instanceof Error ? e : Error(String(e)));
+					if (u(e)) return;
+					k(new s(e));
 				} finally {
-					x.releaseLock();
+					A.stop(), w.releaseLock();
 				}
 			})();
 		},
 		stopCompletion: async (t) => {
-			let r = await o(`${a}/completions/stop`, {
+			let a = await i(`${r}/completions/stop`, {
 				method: "POST",
 				credentials: "include",
 				headers: {
@@ -95,10 +131,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, c as DEFAULT_STREAM_IDLE_TIMEOUT_MS, a as GenerationConflictError, r as GenerationPersistenceError, s as StreamInterruptedError, o as StreamUpstreamError, f as createChatStreamApi };

package/conversation/message-utils.js

@@ -9,6 +9,48 @@ var i = (t, n, r, i) => i && n === r - 1 && t.role === e.Assistant, a = (e) => {
 	return null;
 }, o = (t) => t.role === e.Assistant && (t.custom_content?.stages?.length ?? 0) > 0, s = (t) => {
 	for (let n = t.length - 1; n >= 0; n--) if (t[n].role === e.User) return t[n].custom_content?.configuration_value;
-}, c = (e) => (e ?? "").toLowerCase().replace(/[^a-z]/g, "") === "plaintext" ? t.PlainText : t.Markdown;
+}, c = (e) => {
+	if (typeof e.default == "boolean") return e.default;
+	let t = e.oneOf;
+	if (Array.isArray(t) && t.length === 1) {
+		let e = t[0];
+		if (typeof e == "object" && e && "const" in e && typeof e.const == "boolean") return e.const;
+	}
+}, l = (e) => {
+	let t = e?.properties;
+	if (t == null) return;
+	let n = Object.entries(t).flatMap(([e, t]) => {
+		let n = c(t);
+		return n === void 0 ? [] : [[e, n]];
+	});
+	return n.length > 0 ? Object.fromEntries(n) : void 0;
+}, u = (t) => {
+	let n = -1;
+	for (let r = t.length - 1; r >= 0; r--) if (t[r].role === e.User) {
+		n = r;
+		break;
+	}
+	let r = t[n]?.custom_content?.configuration_value;
+	for (let i = n + 1; i < t.length; i++) {
+		let n = t[i];
+		if (n.role !== e.Assistant) continue;
+		let a = l(n.custom_content?.form_schema);
+		a && (r = {
+			...r,
+			...a
+		});
+	}
+	return r;
+}, d = (t, n, r) => {
+	let i = n.length - 1, a = n[i];
+	if (a?.role !== e.Assistant) return;
+	let o = l(a.custom_content?.form_schema);
+	if (!o) return;
+	let s = Object.fromEntries(r.filter((e) => e in o).sort().map((e) => [e, o[e]]));
+	if (Object.keys(s).length !== 0) return {
+		key: `${t}|${i}|${JSON.stringify(s)}`,
+		values: s
+	};
+}, f = (e) => (e ?? "").toLowerCase().replace(/[^a-z]/g, "") === "plaintext" ? t.PlainText : t.Markdown;
 //#endregion
-export { a as getLastDeploymentId, s as getLastUserMessageToolConfiguration, i as isMessageStreaming, o as messageHasStages, c as normalizeResponseFormat };
+export { d as getFormSchemaToolSyncKey, a as getLastDeploymentId, s as getLastUserMessageToolConfiguration, u as getLatestToolConfiguration, l as getToolConfigurationFromFormSchema, i as isMessageStreaming, o as messageHasStages, f as normalizeResponseFormat };

package/conversation/useConversationStream/generation-resume.js

@@ -1,16 +1,39 @@
 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 = [
+	1e3,
+	2e3,
+	4e3,
+	8e3,
+	16e3
+], f = (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);
+}), p = async (e, t) => {
+	for (let n = 0; n <= d.length; n++) {
+		if (t()) return null;
+		try {
+			return await e();
+		} catch {
+			if (n === d.length) return null;
+			await f(d[n]);
+		}
+	}
+	return null;
+}, m = async (e, t) => {
 	let n = e.getReader(), r = new TextDecoder(), i = "";
 	try {
 		for (;;) {
@@ -39,88 +62,106 @@ 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 () => {
+}, h = ({ transport: o, setConversation: d, conversationRef: f, resumingPathsRef: p, bufferedGenerationsRef: h, addStreamingPath: g, removeStreamingPath: _, isPathDisplayed: v, generationPersistenceErrorMessage: y = t }) => (t, b, x = {}) => {
+	if (!c(b)) {
+		x.onSettled?.();
+		return;
+	}
+	let S = r(t);
+	if (!x.skipDedupe && p.current.has(S)) return;
+	p.current.add(S), g(S);
+	let C = b.messages.length - 1, w = {
+		generationId: u,
+		messageIndex: C,
+		message: x.seedMessage ?? b.messages[C]
+	};
+	h.current.set(S, w);
+	let T = () => h.current.get(S) === w, E = (e, t = !1) => {
+		if (!T()) return;
+		p.current.delete(S), _(S);
+		let n = e && e.messages.length - 1 === C && c(e) && s(w.message);
+		if (t || n) {
+			w.message = {
+				...w.message,
+				streamErrorMessage: y
+			}, v(S) && d((e) => {
+				if (!e || !T() || !v(S)) return e;
+				let t = a(e, w);
+				return f.current = t, t;
+			});
+			return;
+		}
+		h.current.delete(S), e && v(S) && (d(e), f.current = e);
+	}, D = async () => {
 		try {
-			b(await i.getConversation(e(g)));
+			E(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]);
+			E(void 0, s(w.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;
+	}, O = (e) => {
+		T() && (w.message = e, v(S) && d((e) => {
+			if (!e || !T() || !v(S)) return e;
+			let t = a(e, w);
+			return f.current = t, t;
+		}));
+	}, k = (e) => {
+		if (!T()) return;
+		let t = i([w.message], 0, e);
+		t && (w.message = t[0]), v(S) && d((e) => {
+			if (!e || !T() || !v(S)) return e;
+			let t = a(e, w);
+			return f.current = t, t;
 		});
-	}, w = async () => {
-		let t = new AbortController(), n;
+	}, A = async () => {
+		let n = new AbortController(), r;
 		try {
-			n = await i.watchConversation(v, t.signal);
+			r = await o.watchConversation(S, n.signal);
 		} catch {
-			await x();
+			await D();
 			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 m(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 E(n), i = !0, !0;
 				} catch {}
 				return !1;
 			});
 		} finally {
 			clearTimeout(a);
 		}
-		r || await x();
-	}, T = async () => {
+		i || await D();
+	}, j = async () => {
 		let e = new AbortController(), t;
 		try {
-			t = await i.attachToGeneration(v, e.signal);
+			t = await o.attachToGeneration(S, e.signal);
 		} catch {
 			return !1;
 		}
-		let n = !1;
-		return await l(t, (e) => {
+		let r = !1, i = !1;
+		return await m(t, (e) => {
+			if (!T()) return !0;
 			switch (e.type) {
-				case "snapshot": return S(e.message), !1;
-				case "chunk": return C(e.chunk), !1;
+				case "snapshot": return O(e.message), !1;
+				case "chunk": return k(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 ? E(void 0, !0) : await D(), !0) : !T();
 	};
 	(async () => {
-		await T() || await w();
+		try {
+			await j() || await A();
+		} finally {
+			x.onSettled?.();
+		}
 	})();
 };
 //#endregion
-export { u as createResumeIfAwaitingGeneration, o as isAwaitingGenerationResume };
+export { h as createResumeIfAwaitingGeneration, p as fetchConversationForRecovery, s as hasGeneratedPayload, c as isAwaitingGenerationResume };