@epam/ai-dial-chat-hooks 1.2.0-dev.99 → 1.3.0-dev.5

package/conversation/useConversationHandlers/useConversationHandlers.js

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