@epam/ai-dial-chat-hooks 1.1.0-dev.423 → 1.1.0-dev.426

package/conversation/useConversationExport/useConversationExport.js

@@ -0,0 +1,285 @@
+import { ConversationExportMode as e, ConversationTransferWarningCode as t, ExportFileNameKind as n } from "../conversation-transfer/types.js";
+import { EXPORT_APP_NAME as r, buildExportEnvelope as i, buildExportFileName as a, serializeExportEnvelope as o, stripConversationAttachments as s } from "../conversation-transfer/export-conversation.js";
+import { collectAttachmentRefs as c } from "../conversation-transfer/attachment-refs.js";
+import { resolveDialFileBucketAndPath as l } from "../conversation-transfer/dial-file-resolve.js";
+import { runWithConcurrency as u } from "../conversation-transfer/async.js";
+import { ConversationTransferKind as d, ConversationTransferPhase as f, buildTransferProgress as p } from "../conversation-transfer/progress.js";
+import { useConversationTransferQueue as m } from "../conversation-transfer/queue.js";
+import { buildDialArchive as h } from "../conversation-transfer/zip-export.js";
+import { ConversationTransferErrorCode as g, ConversationTransferSubjectKind as _, ConversationTransferUnitKind as v, triggerBlobDownload as y } from "@epam/ai-dial-chat-shared";
+import { useCallback as b } from "react";
+import { ResponseError as x } from "@epam/ai-dial-chat-api-client";
+//#region src/conversation/useConversationExport/useConversationExport.ts
+var S = 5, C = 512 * 1024 * 1024, w = 429, T = 5, E = 2e3, D = (e) => e instanceof x && e.response.status === w, O = (e, t) => {
+	let n = Number(e.response.headers.get("Retry-After"));
+	return Number.isFinite(n) && n > 0 ? n * 1e3 : E * t;
+}, k = (e, t) => new Promise((n) => {
+	if (t.aborted) {
+		n();
+		return;
+	}
+	let r = setTimeout(n, e);
+	t.addEventListener("abort", () => {
+		clearTimeout(r), n();
+	}, { once: !0 });
+}), A = async (e, t, n) => {
+	for (let r = 1;; r += 1) try {
+		return await e.getConversation({ path: t }, { signal: n });
+	} catch (e) {
+		if (n.aborted || !D(e) || r >= T) throw e;
+		await k(O(e, r), n);
+	}
+}, j = (e) => !e.sharedWithMe && !e.publishedWithMe, M = (e) => e instanceof RangeError, N = (t) => t === e.WithoutAttachments ? d.ExportSingle : d.ExportSingleWithAttachments, P = ({ conversationsApi: x, filesApi: w, normalizeConversationPath: T, classifyTransferError: E = () => ({}), resolveErrorTraceId: D = async () => void 0, maxArchiveBytes: O = C, onSuccess: k, onWarning: P, onError: F }) => {
+	let I = m(), L = b(async (e, t, n) => {
+		let r = !1, i = !1, a = 0, o = () => {
+			a += 1, n(a, e.length);
+		};
+		return {
+			entries: await u(e, S, async (e) => {
+				if (t.aborted || i) return;
+				let n = l(e.fileId);
+				if (!n) {
+					r = !0, o();
+					return;
+				}
+				try {
+					let e = await w.downloadFileRaw({
+						bucket: n.bucket,
+						path: n.path
+					}, { signal: t }), r = new Uint8Array(await e.raw.arrayBuffer());
+					return o(), {
+						path: n.path,
+						data: r
+					};
+				} catch (e) {
+					if (t.aborted) return;
+					if (E(e).isUnauthorized) {
+						i = !0;
+						return;
+					}
+					r = !0, o();
+					return;
+				}
+			}),
+			anySkipped: !t.aborted && !i && r,
+			isUnauthorized: !t.aborted && i
+		};
+	}, [E, w]), R = b(async (n, r, a, l, u, d) => {
+		let m = N(l), _;
+		try {
+			_ = await A(x, T(r), d);
+		} catch (e) {
+			if (d.aborted) return;
+			if (E(e).isUnauthorized) F?.({
+				jobId: n,
+				code: g.Unauthorized
+			}), I.failJob(n, g.Unauthorized);
+			else {
+				let t = await D(e);
+				F?.({
+					jobId: n,
+					code: g.Unknown,
+					titles: [a],
+					traceId: t
+				}), I.failJob(n, g.Unknown);
+			}
+			console.error("Failed to fetch conversation for export", e);
+			return;
+		}
+		if (!d.aborted) {
+			I.setJobProgress(n, p({
+				kind: m,
+				phase: f.Prepare
+			}));
+			try {
+				if (l === e.WithoutAttachments) {
+					y(o(i([s(_)], [])), u), k?.({
+						jobId: n,
+						titles: [a]
+					}), I.succeedJob(n);
+					return;
+				}
+				let r = c(_), { entries: b, anySkipped: x, isUnauthorized: S } = await L(r, d, (e, t) => {
+					I.setJobProgress(n, p({
+						kind: m,
+						phase: f.Transfer,
+						completed: e,
+						total: t
+					}, {
+						completed: e,
+						total: t,
+						kind: v.Attachment
+					}));
+				});
+				if (d.aborted) return;
+				if (S) {
+					F?.({
+						jobId: n,
+						code: g.Unauthorized
+					}), I.failJob(n, g.Unauthorized);
+					return;
+				}
+				I.setJobProgress(n, p({
+					kind: m,
+					phase: f.Transfer,
+					completed: r.length,
+					total: r.length
+				}));
+				let C = b.reduce((e, t) => e + t.data.byteLength, 0);
+				if (C > O) {
+					F?.({
+						jobId: n,
+						code: g.FileTooLarge,
+						titles: [a]
+					}), I.failJob(n, g.FileTooLarge), console.error(`Refused to build an export archive of ${C} bytes: over the ${O}-byte limit`);
+					return;
+				}
+				let { blob: w, skippedPaths: T } = h(i([_], []), b), E = x || T.length > 0;
+				E && P?.({
+					jobId: n,
+					code: t.AttachmentSkipped
+				}), y(w, u), k?.({
+					jobId: n,
+					titles: [a]
+				}), E ? I.warnJob(n, t.AttachmentSkipped) : I.succeedJob(n);
+			} catch (e) {
+				if (d.aborted) return;
+				let t = M(e) ? g.FileTooLarge : g.Unknown, r = await D(e);
+				F?.({
+					jobId: n,
+					code: t,
+					titles: [a],
+					traceId: r
+				}), I.failJob(n, t), console.error("Failed to build conversation export archive", e);
+			}
+		}
+	}, [
+		E,
+		x,
+		L,
+		O,
+		T,
+		F,
+		k,
+		P,
+		I,
+		D
+	]), z = b((t, i, o) => {
+		let s = a(o === e.WithoutAttachments ? n.SingleConversation : n.SingleConversationWithAttachments, r), c = I.addJob({
+			kind: _.Single,
+			title: i
+		}, s);
+		return I.startJob(c, (e) => R(c, t, i, o, s, e));
+	}, [I, R]), B = b(async (e, t, n) => {
+		let r = d.ExportAll, a = [], s;
+		do {
+			if (n.aborted) return;
+			let t;
+			try {
+				t = await x.listConversations({ nextToken: s }, { signal: n });
+			} catch (t) {
+				if (n.aborted) return;
+				if (E(t).isUnauthorized) F?.({
+					jobId: e,
+					code: g.Unauthorized
+				}), I.failJob(e, g.Unauthorized);
+				else {
+					let n = await D(t);
+					F?.({
+						jobId: e,
+						code: g.Unknown,
+						traceId: n
+					}), I.failJob(e, g.Unknown);
+				}
+				console.error("Failed to list conversations for export", t);
+				return;
+			}
+			a.push(...t.items.filter(j).map((e) => ({
+				id: e.id,
+				title: e.title
+			}))), s = t.nextToken;
+		} while (s);
+		I.setJobProgress(e, p({
+			kind: r,
+			phase: f.Prepare
+		}));
+		let c = [];
+		for (let t of a) {
+			if (n.aborted) return;
+			try {
+				let e = await A(x, T(t.id), n);
+				c.push(e);
+			} catch (r) {
+				if (n.aborted) return;
+				let i = E(r);
+				if (i.isUnauthorized) {
+					F?.({
+						jobId: e,
+						code: g.Unauthorized
+					}), I.failJob(e, g.Unauthorized);
+					return;
+				}
+				if (i.isNotFound) {
+					let n = await D(r);
+					F?.({
+						jobId: e,
+						code: g.NotFound,
+						titles: [t.title],
+						traceId: n
+					}), console.error("Skipped a conversation during export-all: not found", r);
+					continue;
+				}
+				let a = await D(r);
+				F?.({
+					jobId: e,
+					code: g.Unknown,
+					traceId: a
+				}), I.failJob(e, g.Unknown), console.error("Aborted export-all: failed to fetch conversation", r);
+				return;
+			} finally {
+				I.setJobProgress(e, p({
+					kind: r,
+					phase: f.Transfer,
+					completed: c.length,
+					total: a.length
+				}, {
+					completed: c.length,
+					total: a.length,
+					kind: v.Conversation
+				}));
+			}
+		}
+		if (!n.aborted) try {
+			y(o(i(c, [])), t), k?.({ jobId: e }), I.succeedJob(e);
+		} catch (t) {
+			let n = M(t) ? g.FileTooLarge : g.Unknown, r = await D(t);
+			F?.({
+				jobId: e,
+				code: n,
+				traceId: r
+			}), I.failJob(e, n), console.error("Failed to build export-all archive", t);
+		}
+	}, [
+		E,
+		x,
+		T,
+		F,
+		k,
+		I,
+		D
+	]), V = b(() => {
+		let e = a(n.AllConversationsHistory, r), t = I.addJob({ kind: _.All }, e);
+		return I.startJob(t, (n) => B(t, e, n));
+	}, [I, B]);
+	return {
+		jobs: I.jobs,
+		exportSingle: z,
+		exportAll: V,
+		cancelJob: I.cancelJob,
+		dismissJob: I.dismissJob,
+		retryJob: I.retryJob,
+		dismissAll: I.dismissAll
+	};
+};
+//#endregion
+export { C as DEFAULT_MAX_ARCHIVE_BYTES, P as useConversationExport };

package/conversation/useConversationHandlers/attachment-to-dto.js

@@ -0,0 +1,13 @@
+//#region src/conversation/useConversationHandlers/attachment-to-dto.ts
+var e = (e) => {
+	if (!e.url) throw Error(`Attachment "${e.name}" has not been uploaded`);
+	return {
+		type: e.contentType,
+		title: e.name,
+		url: e.url
+	};
+}, t = (t) => {
+	if (t.length) return t.map(e);
+};
+//#endregion
+export { e as attachmentToDto, t as attachmentsToDtos };

package/conversation/useConversationHandlers/message-factory.js

@@ -0,0 +1,21 @@
+import { MessageRole as e } from "@epam/ai-dial-chat-shared";
+//#region src/conversation/useConversationHandlers/message-factory.ts
+var t = (t, n, r) => {
+	let i = new Date(Date.now()).toISOString();
+	return {
+		userMessage: {
+			role: e.User,
+			content: t,
+			timestamp: i,
+			...n && Object.keys(n).length ? { custom_content: n } : {}
+		},
+		assistantMessage: {
+			role: e.Assistant,
+			content: "",
+			timestamp: i,
+			...r ? { deploymentId: r } : {}
+		}
+	};
+};
+//#endregion
+export { t as createMessagePair };

package/conversation/useConversationHandlers/message-utils.js

@@ -0,0 +1,8 @@
+//#region src/conversation/useConversationHandlers/message-utils.ts
+var e = (e) => e != null && Object.keys(e).length > 0, t = (e, t, n, r) => {
+	if (t !== e.content || r.length > 0) return !0;
+	let i = e.custom_content?.attachments?.length ?? 0;
+	return n.length !== i;
+};
+//#endregion
+export { e as hasActiveToolConfig, t as isMessageChanged };

package/conversation/useConversationHandlers/starter-option.js

@@ -0,0 +1,11 @@
+import { getStarterPopulateText as e } from "../starter-option.js";
+//#region src/conversation/useConversationHandlers/starter-option.ts
+var t = (t, n) => {
+	let { populateText: r } = t["dial:widgetOptions"];
+	return r == null && n ? n : e(t);
+}, n = (t, n) => {
+	let { populateText: r, submit: i } = t["dial:widgetOptions"];
+	return i && r == null ? "" : n ?? e(t);
+};
+//#endregion
+export { t as getStarterConversationText, n as getStarterSubmitText };

package/conversation/useConversationHandlers/useConversationHandlers.js

@@ -0,0 +1,263 @@
+import { useAttachmentUpload as e } from "../useAttachmentUpload/useAttachmentUpload.js";
+import { attachmentsToDtos as t } from "./attachment-to-dto.js";
+import { createMessagePair as n } from "./message-factory.js";
+import { hasActiveToolConfig as r, isMessageChanged as i } from "./message-utils.js";
+import { getStarterSubmitText as a } from "./starter-option.js";
+import { getConversationPath as o } from "../useConversationStream/conversation-path.js";
+import { MessageRole as s, generateUUID as c } from "@epam/ai-dial-chat-shared";
+import { useCallback as l, useState as u } from "react";
+import { SendCompletionDtoModeEnum as d } from "@epam/ai-dial-chat-api-client";
+//#region src/conversation/useConversationHandlers/useConversationHandlers.ts
+var f = ({ conversation: f, conversationId: p, bucket: m, isStreaming: h, startStream: g, state: { setConversation: _, conversationRef: v }, filesApi: y, conversationsApi: b, rateApi: x, resolveModelId: S, onConversationDeleted: C, showNetworkError: w, toolConfigurationValue: T }) => {
+	let [E, D] = u(null), [O, k] = u(/* @__PURE__ */ new Set()), [A, j] = u(null), { handleUploadAttachment: M } = e({
+		filesApi: y,
+		bucket: m,
+		onNetworkError: w
+	}), N = l(async (e, i) => {
+		if (!p || !f) return;
+		let a = t(i), o = r(T), s = a?.length || o ? {
+			...a?.length ? { attachments: a } : {},
+			...o ? { configuration_value: T } : {}
+		} : void 0, l = S(), { userMessage: u, assistantMessage: m } = n(e, s, l);
+		_((e) => {
+			if (!e) return e;
+			let t = {
+				...e,
+				messages: [
+					...e.messages,
+					u,
+					m
+				]
+			};
+			return v.current = t, t;
+		}), g(p, e, f.messages.length + 1, l, s, c(), d.Append);
+	}, [
+		f,
+		p,
+		v,
+		S,
+		_,
+		g,
+		T
+	]), P = l((e) => {
+		if (h || !p || !f || e === -1 || f.messages[e]?.role !== s.Assistant) return;
+		let t = f.messages[e - 1];
+		if (!t || t.role !== s.User) return;
+		let n = S();
+		_((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 v.current = i, i;
+		}), k((t) => new Set([...t].filter((t) => t < e))), g(p, t.content, e, n, t.custom_content, c(), d.Regenerate);
+	}, [
+		f,
+		p,
+		v,
+		h,
+		S,
+		_,
+		g
+	]), F = l((e) => {
+		h || D(e);
+	}, [h]), I = l(() => {
+		if (!p || E == null) return;
+		let e = E;
+		D(null);
+		let t = v.current;
+		if (!t || e === -1) return;
+		let n = o(p), r = t.messages[e + 1]?.role === s.Assistant ? t.messages.filter((t, n) => n !== e && n !== e + 1) : t.messages.filter((t, n) => n !== e);
+		if (r.length === 0 || r.length === 1 && r[0].role === s.Status) {
+			b.deleteConversation({ path: n }), C?.();
+			return;
+		}
+		let i = {
+			...t,
+			messages: r
+		};
+		v.current = i, _(i), b.saveConversation({
+			path: n,
+			saveConversationBodyDto: { conversation: i }
+		});
+	}, [
+		p,
+		v,
+		b,
+		C,
+		E,
+		_
+	]), L = l(async (e, t, n) => {
+		if (!p || !f) return !1;
+		let r = f.messages[e];
+		if (!r) return !1;
+		let i = r.rating, a = {
+			...f,
+			messages: f.messages.map((n, r) => r === e ? {
+				...n,
+				rating: t ?? void 0
+			} : n)
+		};
+		_(() => (v.current = a, a));
+		let s = o(p), c = () => {
+			_((t) => t && {
+				...t,
+				messages: t.messages.map((t, n) => n === e ? {
+					...t,
+					rating: i
+				} : t)
+			});
+		}, l = () => b.saveConversation({
+			path: s,
+			saveConversationBodyDto: { conversation: a }
+		});
+		if (t != null) {
+			let e = r.responseId;
+			if (!e) return c(), !1;
+			try {
+				return await x.rateMessage({ rateMessageDto: {
+					conversationId: f.id,
+					responseId: e,
+					modelId: f.model.id,
+					rate: t,
+					...n ? { comment: n } : {}
+				} }), await l(), !0;
+			} catch {
+				return c(), !1;
+			}
+		} else try {
+			return await l(), !0;
+		} catch {
+			return c(), !1;
+		}
+	}, [
+		f,
+		p,
+		v,
+		b,
+		x,
+		_
+	]), R = l((e, t, i) => {
+		if (!p || !f) return;
+		let o = i ?? e.title, s = a(e, i), l = t ? { [t]: e.const } : void 0, u = r(T), m = l || u ? {
+			...l ? { form_value: l } : {},
+			...u ? { configuration_value: T } : {}
+		} : void 0, h = S(), { userMessage: y, assistantMessage: b } = n(o, m, h);
+		_((e) => {
+			if (!e) return e;
+			let t = {
+				...e,
+				messages: [
+					...e.messages,
+					y,
+					b
+				]
+			};
+			return v.current = t, t;
+		}), g(p, s, f.messages.length + 1, h, m, c(), d.Append);
+	}, [
+		f,
+		p,
+		v,
+		S,
+		_,
+		g,
+		T
+	]);
+	return {
+		handleSend: N,
+		handleUploadAttachment: M,
+		handleRegenerateMessage: P,
+		handleDeleteMessage: F,
+		handleConfirmDelete: I,
+		handleRateMessage: L,
+		handleButtonSelect: l((e, t, n) => {
+			!p || !f || h || (e["dial:widgetOptions"].confirmationMessage ? j({
+				starter: e,
+				propertyKey: t,
+				description: n
+			}) : R(e, t, n));
+		}, [
+			f,
+			p,
+			h,
+			R
+		]),
+		handleConfirmStarter: l(() => {
+			if (!A) return;
+			let { starter: e, propertyKey: t, description: n } = A;
+			j(null), R(e, t, n);
+		}, [A, R]),
+		handleStartEdit: l((e) => {
+			k((t) => new Set([...t, e]));
+		}, []),
+		handleCancelEdit: l((e) => {
+			k((t) => {
+				let n = new Set(t);
+				return n.delete(e), n;
+			});
+		}, []),
+		handleEditMessage: l(async (e, n, r, a) => {
+			if (h || !p || !f) return;
+			let o = e;
+			if (o === -1 || f.messages[o].role !== s.User) return;
+			let l = f.messages[o];
+			if (!i(l, n, r, a)) {
+				k((e) => {
+					let t = new Set(e);
+					return t.delete(o), t;
+				});
+				return;
+			}
+			let u = t(a), m = new Set(r.map((e) => e.id)), y = [...(l.custom_content?.attachments ?? []).filter((e) => {
+				let t = e.url ?? e.data ?? e.title;
+				return m.has(t);
+			}), ...u ?? []], { attachments: b, ...x } = l.custom_content ?? {}, C = y.length > 0 ? {
+				...x,
+				attachments: y
+			} : Object.keys(x).length > 0 ? x : void 0, w = {
+				...l,
+				content: n,
+				custom_content: C
+			}, T = Date.now(), E = {
+				role: s.Assistant,
+				content: "",
+				timestamp: new Date(T).toISOString()
+			}, D = [
+				...f.messages.slice(0, o),
+				w,
+				E
+			], O = {
+				...f,
+				messages: D
+			};
+			_(() => (v.current = O, O));
+			let A = S();
+			g(p, n, D.length - 1, A, C, c(), d.Edit), k(/* @__PURE__ */ new Set());
+		}, [
+			f,
+			p,
+			v,
+			h,
+			S,
+			_,
+			g
+		]),
+		editingMessageIndexes: O,
+		pendingDeleteIndex: E,
+		setPendingDeleteIndex: D,
+		pendingStarterContext: A,
+		setPendingStarterContext: j
+	};
+};
+//#endregion
+export { f as useConversationHandlers };