@epam/ai-dial-chat-hooks 1.3.0-dev.37 → 1.3.0-dev.371

package/conversation/useConversationStream/generation-resume.js

@@ -1,184 +1,173 @@
 import { safeDecodeURI as e } from "../../shared/string-utils.js";
-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 { BackgroundGenerationStatus as o, MessageRole as s } from "@epam/ai-dial-chat-shared";
+import { DEFAULT_GENERATION_INTERRUPTED_ERROR_MESSAGE as t, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as n, GenerationPersistenceError as r } from "../create-chat-stream-api.js";
+import { readSseEvents as i } from "../../shared/read-sse-events.js";
+import { getConversationPath as a } from "./conversation-path.js";
+import { applyChunkToMessages as o } from "./apply-chunk.js";
+import { restoreBufferedMessage as s } from "./buffered-generation.js";
+import { BackgroundGenerationStatus as c, MessageRole as l } from "@epam/ai-dial-chat-shared";
 //#region src/conversation/useConversationStream/generation-resume.ts
-var c = (e) => {
+var u = (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;
-}, l = (e) => e.messages.findIndex((e) => e.backgroundGeneration?.status === o.Pending), u = (e) => {
-	if (l(e) !== -1) return !0;
+}, d = (e) => e.messages.findIndex((e) => e.backgroundGeneration?.status === c.Pending), f = (e) => {
+	if (d(e) !== -1) return !0;
 	let t = e.messages[e.messages.length - 1];
-	return !!t && t.role === s.Assistant && !c(t) && t.streamErrorMessage == null && !t.wasStoppedByUser;
-}, d = 300 * 1e3, f = "awaiting-resume", p = [
+	return !!t && t.role === l.Assistant && !u(t) && t.streamErrorMessage == null && !t.wasStoppedByUser;
+}, p = 3e5, m = "awaiting-resume", h = [
 	1e3,
 	2e3,
 	4e3,
 	8e3,
 	16e3
-], m = (e) => new Promise((t) => {
+], g = (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);
-}), h = async (e, t, n) => {
-	for (let r = 0; r <= p.length; r++) {
+}), _ = async (e, t, n) => {
+	for (let r = 0; r <= h.length; r++) {
 		if (t()) return null;
 		try {
 			let t = await e();
 			if (!n?.(t)) return t;
 		} catch {}
-		if (r === p.length) return null;
-		await m(p[r]);
+		if (r === h.length) return null;
+		await g(h[r]);
 	}
 	return null;
-}, g = async (e, t) => {
-	let n = e.getReader(), r = new TextDecoder(), i = "";
-	try {
-		for (;;) {
-			let { done: e, value: a } = await n.read();
-			if (e) break;
-			i += r.decode(a, { stream: !0 });
-			let o = i.split("\n");
-			i = o.pop() ?? "";
-			let s = !1;
-			for (let e of o) {
-				let n = e.trim();
-				if (!n.startsWith("data:")) continue;
-				let r = n.slice(5).trim(), i;
-				try {
-					i = JSON.parse(r);
-				} catch {
-					continue;
-				}
-				if (await t(i)) {
-					s = !0;
-					break;
-				}
-			}
-			if (s) break;
-		}
-	} catch {} finally {
-		n.releaseLock();
-	}
-}, _ = ({ transport: o, setConversation: s, conversationRef: p, resumingPathsRef: m, bufferedGenerationsRef: h, addStreamingPath: _, removeStreamingPath: v, isPathDisplayed: y, generationPersistenceErrorMessage: b = t, frameScheduler: x, stoppedGenerationIdsRef: S, onReloadError: C, onReloadSuccess: w }) => (t, T, E = {}) => {
-	if (!u(T)) {
-		E.onSettled?.();
+}, v = ({ transport: c, setConversation: l, conversationRef: h, resumingPathsRef: g, bufferedGenerationsRef: _, addStreamingPath: v, removeStreamingPath: y, isPathDisplayed: b, generationPersistenceErrorMessage: x = n, generationInterruptedMessage: S = t, frameScheduler: C, stoppedGenerationIdsRef: w, rememberResumedGenerationId: T, forgetResumedGenerationId: E, onReloadError: D, onReloadSuccess: O }) => (t, n, k = {}) => {
+	if (!f(n)) {
+		k.onSettled?.();
 		return;
 	}
-	let D = r(t);
-	if (!E.skipDedupe && m.current.has(D)) return;
-	m.current.add(D), _(D);
-	let O = l(T), k = O !== -1, A = k ? O : T.messages.length - 1, j = {
-		generationId: f,
-		messageIndex: A,
-		message: E.seedMessage ?? T.messages[A]
+	let A = a(t);
+	if (!k.skipDedupe && g.current.has(A)) return;
+	g.current.add(A), v(A);
+	let j = d(n), M = j !== -1, N = M ? j : n.messages.length - 1, P = {
+		generationId: m,
+		messageIndex: N,
+		message: k.seedMessage ?? n.messages[N]
 	};
-	h.current.set(D, j);
-	let M = () => h.current.get(D) === j, N = k ? T.messages[A].backgroundGeneration?.generationId : void 0, P = () => N != null && (S?.current.has(N) ?? !1), F = (e, t = !1) => {
-		if (x?.flush(D), N != null && S?.current.delete(N), !M()) return;
-		m.current.delete(D), v(D);
-		let n = !k && e && e.messages.length - 1 === A && u(e) && c(j.message);
-		if (t || n) {
-			j.message = {
-				...j.message,
-				streamErrorMessage: b
-			}, y(D) && s((e) => {
-				if (!e || !M() || !y(D)) return e;
-				let t = a(e, j);
-				return p.current = t, t;
-			});
-			return;
+	_.current.set(A, P);
+	let F = () => _.current.get(A) === P, I = M ? n.messages[N].backgroundGeneration?.generationId : void 0, L, R = () => {
+		let e = I ?? L;
+		return e != null && (w?.current.has(e) ?? !1);
+	}, z = (e) => {
+		k.trackAttachGenerationId && !M && e != null && F() && (L = e, T?.(A, e));
+	}, B = () => {
+		if (L == null) return;
+		let e = L;
+		L = void 0, w?.current.delete(e), E?.(A, e);
+	}, V = (e) => !!k.shouldMarkOrphanedAsInterrupted && !M && !!e && e.messages.length - 1 === N && d(e) === -1 && f(e) && !u(P.message), H = () => {
+		P.message = {
+			...P.message,
+			streamErrorMessage: S
+		};
+	}, U = () => {
+		F() && (H(), g.current.delete(A), y(A), B(), b(A) && l((e) => {
+			if (!e || !F() || !b(A)) return e;
+			let t = s(e, P);
+			return h.current = t, t;
+		}));
+	}, W = (e, t = !1) => {
+		if (C?.flush(A), I != null && w?.current.delete(I), B(), !F()) return;
+		g.current.delete(A), y(A);
+		let n = !M && e && e.messages.length - 1 === N && f(e) && u(P.message);
+		if (t || n) P.message = {
+			...P.message,
+			streamErrorMessage: x
+		}, b(A) && l((e) => {
+			if (!e || !F() || !b(A)) return e;
+			let t = s(e, P);
+			return h.current = t, t;
+		});
+		else if (_.current.delete(A), e && b(A)) {
+			let t = V(e);
+			t && H();
+			let n = t ? s(e, P) : e;
+			l(n), h.current = n;
 		}
-		h.current.delete(D), e && y(D) && (s(e), p.current = e);
-	}, I = async () => {
-		if (M()) try {
-			let n = await o.getConversation(e(t));
-			if (!M()) return;
-			w?.(D), F(n);
+	}, G = async () => {
+		if (F()) try {
+			let n = await c.getConversation(e(t));
+			if (!F()) return;
+			O?.(A), W(n);
 		} catch {
-			if (!M()) return;
-			x?.flush(D), m.current.delete(D), v(D), N != null && S?.current.delete(N), C?.(D, I, M);
+			if (!F()) return;
+			C?.flush(A), g.current.delete(A), y(A), I != null && w?.current.delete(I), B(), D?.(A, G, F);
 		}
-	}, L = (e) => {
-		x?.cancel(D), M() && (j.message = e, y(D) && s((e) => {
-			if (!e || !M() || !y(D)) return e;
-			let t = a(e, j);
-			return p.current = t, t;
+	}, K = (e) => {
+		C?.cancel(A), F() && (P.message = e, b(A) && l((e) => {
+			if (!e || !F() || !b(A)) return e;
+			let t = s(e, P);
+			return h.current = t, t;
 		}));
-	}, R = () => s((e) => {
-		if (!e || !M() || !y(D)) return e;
-		let t = a(e, j);
-		return p.current = t, t;
-	}), z = (e) => {
-		if (!M() || P()) return;
-		let t = i([j.message], 0, e);
-		if (t && (j.message = t[0]), y(D)) {
-			if (x) {
-				x.schedule(D, R);
-				return;
-			}
-			R();
-		}
-	}, B = async () => {
+	}, q = () => l((e) => {
+		if (!e || !F() || !b(A)) return e;
+		let t = s(e, P);
+		return h.current = t, t;
+	}), J = (e) => {
+		if (!F() || R()) return;
+		let t = o([P.message], 0, e);
+		t && (P.message = t[0]), b(A) && (C ? C.schedule(A, q) : q());
+	}, Y = async () => {
 		let n = new AbortController(), r;
 		try {
-			r = await o.watchConversation(D, n.signal);
+			r = await c.watchConversation(A, n.signal);
 		} catch {
-			await I();
+			await G();
 			return;
 		}
 		try {
-			let i = await o.getConversation(e(t));
-			if (!u(i)) {
-				n.abort(), r.cancel().catch(() => void 0), F(i);
+			let i = await c.getConversation(e(t));
+			if (!f(i)) {
+				n.abort(), r.cancel().catch(() => void 0), W(i);
 				return;
 			}
+			V(i) && U();
 		} catch {}
-		let i = !1, a = window.setTimeout(() => {
+		let a = !1, o = window.setTimeout(() => {
 			n.abort();
-		}, d);
+		}, p);
 		try {
-			await g(r, async (n) => {
+			await i(r, async (n) => {
 				if (n?.action !== "UPDATE") return !1;
 				try {
-					let n = await o.getConversation(e(t));
-					if (!u(n)) return F(n), i = !0, !0;
+					let n = await c.getConversation(e(t));
+					if (!f(n)) return W(n), a = !0, !0;
 				} catch {}
 				return !1;
 			});
 		} finally {
-			clearTimeout(a);
+			clearTimeout(o);
 		}
-		i || await I();
-	}, V = async () => {
+		a || await G();
+	}, X = async () => {
 		let e = new AbortController(), t;
 		try {
-			t = await o.attachToGeneration(D, e.signal);
+			t = await c.attachToGeneration(A, e.signal);
 		} catch {
 			return !1;
 		}
-		let r = !1, i = !1;
-		return await g(t, (e) => {
-			if (!M()) return !0;
+		let n = !1, a = !1;
+		return await i(t, (e) => {
+			if (!F()) return !0;
 			switch (e.type) {
-				case "snapshot": return L(e.message), !1;
-				case "chunk": return z(e.chunk), !1;
+				case "snapshot": return z(e.generationId), K(e.message), !1;
+				case "chunk": return J(e.chunk), !1;
 				case "done":
 				case "error":
-				case "stopped": return i = e.type === "error" && e.errorType === n.type, r = !0, !0;
+				case "stopped": return a = e.type === "error" && e.errorType === r.type, n = !0, !0;
 			}
-		}), r ? (i ? F(void 0, !0) : await I(), !0) : !M();
+		}), n ? (a ? W(void 0, !0) : await G(), !0) : (B(), !F());
 	};
 	(async () => {
 		try {
-			await V() || await B();
+			await X() || await Y();
 		} finally {
-			E.onSettled?.();
+			k.onSettled?.();
 		}
 	})();
 };
 //#endregion
-export { _ as createResumeIfAwaitingGeneration, h as fetchConversationForRecovery, l as findPendingBackgroundMessageIndex, c as hasGeneratedPayload, u as isAwaitingGenerationResume };
+export { h as RECOVERY_REFETCH_DELAYS_MS, v as createResumeIfAwaitingGeneration, _ as fetchConversationForRecovery, d as findPendingBackgroundMessageIndex, u as hasGeneratedPayload, f as isAwaitingGenerationResume };

package/conversation/useConversationStream/stream-errors.d.ts

@@ -0,0 +1,10 @@
+import { Conversation } from '@epam/ai-dial-chat-shared';
+export declare const CHANNEL_WAIT_TIMEOUT_MS = 20000;
+export declare const UNAUTHORIZED_STATUS = 401;
+export declare const NOT_FOUND_STATUS = 404;
+export declare const STOP_NOT_FOUND_GRACE_MS = 2000;
+export declare const STOP_NOT_FOUND_POLL_MS = 100;
+export declare const resolveStreamErrorMessage: (error: Error, generationConflictMessage: string, generationPersistenceErrorMessage: string) => string;
+/** Generation id of the displayed conversation's pending background message, if any. */
+export declare const findResumedBackgroundGenerationId: (conversation: Conversation | null) => string | undefined;
+//# sourceMappingURL=stream-errors.d.ts.map

package/conversation/useConversationStream/stream-errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-errors.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/stream-errors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAoB9D,eAAO,MAAM,uBAAuB,QAAQ,CAAC;AAG7C,eAAO,MAAM,mBAAmB,MAAM,CAAC;AAEvC,eAAO,MAAM,gBAAgB,MAAM,CAAC;AAKpC,eAAO,MAAM,uBAAuB,OAAO,CAAC;AAC5C,eAAO,MAAM,sBAAsB,MAAM,CAAC;AAQ1C,eAAO,MAAM,yBAAyB,GACpC,OAAO,KAAK,EACZ,2BAA2B,MAAM,EACjC,mCAAmC,MAAM,KACxC,MAOF,CAAC;AAEF,wFAAwF;AACxF,eAAO,MAAM,iCAAiC,GAC5C,cAAc,YAAY,GAAG,IAAI,KAChC,MAAM,GAAG,SAMX,CAAC"}

package/conversation/useConversationStream/stream-errors.js

@@ -0,0 +1,10 @@
+import { GenerationConflictError as e, GenerationPersistenceError as t, StreamUpstreamError as n } from "../create-chat-stream-api.js";
+import { findPendingBackgroundMessageIndex as r } from "./generation-resume.js";
+//#region src/conversation/useConversationStream/stream-errors.ts
+var i = 2e4, a = 2e3, o = (r, i, a) => r instanceof t ? a : r instanceof e ? i : r instanceof n ? r.message : "", s = (e) => {
+	if (!e) return;
+	let t = r(e);
+	return t === -1 ? void 0 : e.messages[t].backgroundGeneration?.generationId;
+};
+//#endregion
+export { i as CHANNEL_WAIT_TIMEOUT_MS, a as STOP_NOT_FOUND_GRACE_MS, s as findResumedBackgroundGenerationId, o as resolveStreamErrorMessage };

package/conversation/useConversationStream/stream-session.d.ts

@@ -0,0 +1,51 @@
+import { SendCompletionDtoModeEnum } from '@epam/ai-dial-chat-api-client';
+import { MessageCustomContent } from '@epam/ai-dial-chat-shared';
+import { MutableRefObject } from 'react';
+import { ConversationGenerationLifecycle, ConversationStateAccessor, ConversationStreamChannel, ConversationStreamOverlayNotifier, ConversationStreamTransport, StartStreamOptions } from './conversation-stream.types';
+import { FrameScheduler } from './frame-scheduler';
+import { createResumeIfAwaitingGeneration } from './generation-resume';
+import { useConversationReloadErrors } from './useConversationReloadErrors';
+import { useStreamPathState } from './useStreamPathState';
+type PathState = ReturnType<typeof useStreamPathState>;
+type ReloadErrors = ReturnType<typeof useConversationReloadErrors>;
+/** Everything a stream session reads from `useConversationStream`: its capabilities, messages, refs and mutators. */
+export interface StreamSessionContext {
+    transport: ConversationStreamTransport;
+    channel: ConversationStreamChannel | undefined;
+    overlay: ConversationStreamOverlayNotifier | undefined;
+    startGeneration: ConversationGenerationLifecycle['startGeneration'];
+    completeGeneration: ConversationGenerationLifecycle['completeGeneration'];
+    setConversation: ConversationStateAccessor['setConversation'];
+    conversationRef: ConversationStateAccessor['conversationRef'];
+    generationConflictMessage: string;
+    generationPersistenceErrorMessage: string;
+    onStreamErrorRef: MutableRefObject<((error: Error) => void) | undefined>;
+    batchChunksPerFrameRef: MutableRefObject<boolean>;
+    mountedRef: MutableRefObject<boolean>;
+    frameScheduler: FrameScheduler;
+    resumeGeneration: ReturnType<typeof createResumeIfAwaitingGeneration>;
+    activeGenerationIdsRef: PathState['activeGenerationIdsRef'];
+    latestGenerationIdsRef: PathState['latestGenerationIdsRef'];
+    resumingPathsRef: PathState['resumingPathsRef'];
+    bufferedGenerationsRef: PathState['bufferedGenerationsRef'];
+    stoppedGenerationIdsRef: PathState['stoppedGenerationIdsRef'];
+    addStreamingPath: PathState['addStreamingPath'];
+    removeStreamingPath: PathState['removeStreamingPath'];
+    setPathStoppable: PathState['setPathStoppable'];
+    releaseActiveGeneration: PathState['releaseActiveGeneration'];
+    isPathDisplayed: PathState['isPathDisplayed'];
+    reportReloadError: ReloadErrors['reportReloadError'];
+    clearReloadError: ReloadErrors['clearReloadError'];
+}
+/**
+ * One generation of `useConversationStream.startStream`: registers the
+ * generation for its path, sends the completion, applies chunks (batched per
+ * frame when asked), reloads the persisted conversation on completion, shows
+ * a failure on the message, recovers an interrupted or conflicted start, and
+ * saves an answer whose terminal save was rejected with `401`. Each generation
+ * keeps its own closures over its id, path and abort controller; the hook
+ * passes a fresh context built from the same values its callback closes over.
+ */
+export declare const runStreamSession: ({ transport, channel, overlay, startGeneration, completeGeneration, setConversation, conversationRef, generationConflictMessage, generationPersistenceErrorMessage, onStreamErrorRef, batchChunksPerFrameRef, mountedRef, frameScheduler, resumeGeneration, activeGenerationIdsRef, latestGenerationIdsRef, resumingPathsRef, bufferedGenerationsRef, stoppedGenerationIdsRef, addStreamingPath, removeStreamingPath, setPathStoppable, releaseActiveGeneration, isPathDisplayed, reportReloadError, clearReloadError, }: StreamSessionContext, currentConversationId: string, userContent: string, messageIndex: number, model: string, customContent?: MessageCustomContent, generationId?: string, mode?: SendCompletionDtoModeEnum, options?: StartStreamOptions) => void;
+export {};
+//# sourceMappingURL=stream-session.d.ts.map

package/conversation/useConversationStream/stream-session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-session.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/stream-session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EAGL,KAAK,oBAAoB,EAE1B,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,OAAO,CAAC;AAa9C,OAAO,KAAK,EACV,+BAA+B,EAC/B,yBAAyB,EACzB,yBAAyB,EACzB,iCAAiC,EACjC,2BAA2B,EAC3B,kBAAkB,EAEnB,MAAM,6BAA6B,CAAC;AACrC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,gCAAgC,EAKjC,MAAM,qBAAqB,CAAC;AAM7B,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,+BAA+B,CAAC;AACjF,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,KAAK,SAAS,GAAG,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAEvD,KAAK,YAAY,GAAG,UAAU,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAEnE,qHAAqH;AACrH,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,2BAA2B,CAAC;IACvC,OAAO,EAAE,yBAAyB,GAAG,SAAS,CAAC;IAC/C,OAAO,EAAE,iCAAiC,GAAG,SAAS,CAAC;IACvD,eAAe,EAAE,+BAA+B,CAAC,iBAAiB,CAAC,CAAC;IACpE,kBAAkB,EAAE,+BAA+B,CAAC,oBAAoB,CAAC,CAAC;IAC1E,eAAe,EAAE,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;IAC9D,eAAe,EAAE,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;IAC9D,yBAAyB,EAAE,MAAM,CAAC;IAClC,iCAAiC,EAAE,MAAM,CAAC;IAC1C,gBAAgB,EAAE,gBAAgB,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;IACzE,sBAAsB,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAClD,UAAU,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACtC,cAAc,EAAE,cAAc,CAAC;IAC/B,gBAAgB,EAAE,UAAU,CAAC,OAAO,gCAAgC,CAAC,CAAC;IACtE,sBAAsB,EAAE,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC5D,sBAAsB,EAAE,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC5D,gBAAgB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAChD,sBAAsB,EAAE,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC5D,uBAAuB,EAAE,SAAS,CAAC,yBAAyB,CAAC,CAAC;IAC9D,gBAAgB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAChD,mBAAmB,EAAE,SAAS,CAAC,qBAAqB,CAAC,CAAC;IACtD,gBAAgB,EAAE,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAChD,uBAAuB,EAAE,SAAS,CAAC,yBAAyB,CAAC,CAAC;IAC9D,eAAe,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC9C,iBAAiB,EAAE,YAAY,CAAC,mBAAmB,CAAC,CAAC;IACrD,gBAAgB,EAAE,YAAY,CAAC,kBAAkB,CAAC,CAAC;CACpD;AAED;;;;;;;;GAQG;AACH,eAAO,MAAM,gBAAgB,GAC3B,0fA2BG,oBAAoB,EACvB,uBAAuB,MAAM,EAC7B,aAAa,MAAM,EACnB,cAAc,MAAM,EACpB,OAAO,MAAM,EACb,gBAAgB,oBAAoB,EACpC,eAAe,MAAM,EACrB,OAAM,yBAA4D,EAClE,UAAS,kBAAuB,KAC/B,IA2iBF,CAAC"}

package/conversation/useConversationStream/stream-session.js

@@ -0,0 +1,179 @@
+import { safeDecodeURI as e } from "../../shared/string-utils.js";
+import { GenerationConflictError as t, GenerationPersistenceError as n, StreamInterruptedError 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 { fetchConversationForRecovery as s, findPendingBackgroundMessageIndex as c, hasGeneratedPayload as l, isAwaitingGenerationResume as u } from "./generation-resume.js";
+import { resolveStreamErrorMessage as ee } from "./stream-errors.js";
+import { SendCompletionDtoModeEnum as d } from "@epam/ai-dial-chat-api-client";
+import { MessageRole as f, generateUUID as p } from "@epam/ai-dial-chat-shared";
+//#region src/conversation/useConversationStream/stream-session.ts
+var m = ({ transport: m, channel: h, overlay: g, startGeneration: te, completeGeneration: _, setConversation: v, conversationRef: y, generationConflictMessage: ne, generationPersistenceErrorMessage: b, onStreamErrorRef: x, batchChunksPerFrameRef: re, mountedRef: S, frameScheduler: C, resumeGeneration: w, activeGenerationIdsRef: T, latestGenerationIdsRef: E, resumingPathsRef: D, bufferedGenerationsRef: O, stoppedGenerationIdsRef: k, addStreamingPath: ie, removeStreamingPath: A, setPathStoppable: j, releaseActiveGeneration: M, isPathDisplayed: N, reportReloadError: ae, clearReloadError: P }, F, I, L, oe, se, ce, R = d.Append, le = {}) => {
+	let z = ce ?? p(), B = i(F);
+	C.flush(B), D.current.delete(B), T.current.set(B, z), E.current.set(B, z), j(B, !0);
+	let V;
+	R === d.Regenerate ? V = L : R === d.Edit && (V = L - 1);
+	let H = () => E.current.get(B) !== z, U = te(B, z), W = y.current?.messages[L];
+	W ? O.current.set(B, {
+		generationId: z,
+		messageIndex: L,
+		message: W
+	}) : O.current.delete(B), ie(B), g?.notifyGenerationStart?.(), h?.ensureConnected();
+	let G = () => {
+		H() || A(B), M(B, z), _(B, z), h?.notifyGenerationSettled?.();
+	}, ue = () => {
+		if (H()) return;
+		let e = O.current.get(B);
+		e && e.generationId === z && (e.message = {
+			...e.message,
+			streamErrorMessage: b
+		}, x.current?.(new n()), N(B) && v((t) => {
+			if (!t || H() || !N(B)) return t;
+			let n = o(t, e);
+			return y.current = n, n;
+		}));
+	}, K = O.current.get(B), q = () => S.current && !H() && O.current.get(B) === K, J = async () => {
+		if (q()) {
+			if (!N(B)) K && O.current.delete(B);
+			else try {
+				let t = await m.getConversation(e(F));
+				if (!q()) return;
+				if (P(B), !N(B)) {
+					K && O.current.delete(B);
+					return;
+				}
+				if (c(t) !== -1) {
+					w(F, t, { seedMessage: K?.message });
+					return;
+				}
+				if (K && K.messageIndex === t.messages.length - 1 && u(t) && l(K.message)) {
+					ue();
+					return;
+				}
+				K && O.current.delete(B), v(t), y.current = t;
+			} catch {
+				q() && ae(B, J, q);
+			}
+		}
+	}, Y = {
+		signal: U.signal,
+		onChunk: (e) => {
+			if (H()) return;
+			let t = O.current.get(B);
+			if (t?.generationId === z) {
+				let n = a([t.message], 0, e);
+				n && (t.message = n[0]);
+			}
+			N(B) && (re.current && t?.generationId === z ? C.schedule(B, () => {
+				let e = O.current.get(B);
+				if (H() || !N(B) || e?.generationId !== z) return;
+				let t = { ...e };
+				v((e) => {
+					if (!e) return e;
+					let n = o(e, t);
+					return y.current = n, n;
+				});
+			}) : v((t) => {
+				if (!t) return t;
+				let n = O.current.get(B), r;
+				if (n?.generationId === z) r = o(t, n);
+				else {
+					let n = a(t.messages, L, e);
+					if (!n) return t;
+					r = {
+						...t,
+						messages: n
+					};
+				}
+				return y.current = r, r;
+			}));
+		},
+		onComplete: async () => {
+			C.flush(B), G(), k.current.has(z) ? k.current.delete(z) : H() || g?.notifyGenerationEnd?.(), await J();
+		},
+		onError: (e) => {
+			C.flush(B), x.current?.(e), e instanceof r && !H() ? fe(e) : e instanceof t && le.resumeOnConflict && !H() ? pe(e) : X(e);
+		}
+	}, X = (e) => {
+		let t = O.current.get(B), r = t?.generationId === z ? t : void 0;
+		r && !(e instanceof n) && O.current.delete(B), G(), r && m.saveConversation && e instanceof n && e.status === 401 ? de(e, r, m.saveConversation.bind(m)) : Z(e, r);
+	}, Z = (e, t) => {
+		if (t && e instanceof n && (t.message = {
+			...t.message,
+			streamErrorMessage: b
+		}), H() || !N(B)) return;
+		let r = ee(e, ne, b);
+		v((e) => {
+			if (!e || H() || !N(B)) return e;
+			let n = t?.generationId === z ? o(e, t) : e, i = {
+				...n,
+				messages: n.messages.map((e, t) => t === L ? {
+					...e,
+					streamErrorMessage: r
+				} : e)
+			};
+			return y.current = i, i;
+		});
+	}, de = async (t, n, r) => {
+		let i = () => S.current && !H() && O.current.get(B) === n;
+		try {
+			let a = await m.getConversation(e(F));
+			if (!i()) return;
+			if (n.messageIndex !== a.messages.length - 1 || !u(a)) {
+				Z(t, n);
+				return;
+			}
+			let s = t.terminalMarker ? {
+				...n,
+				message: {
+					...n.message,
+					...t.terminalMarker
+				}
+			} : n, c = await r(B, o(a, s));
+			if (!i() || (O.current.delete(B), !N(B))) return;
+			v(c), y.current = c;
+		} catch {
+			i() && Z(t, n);
+		}
+	}, Q = () => {
+		G(), k.current.has(z) ? k.current.delete(z) : H() || g?.notifyGenerationEnd?.();
+	}, fe = async (t) => {
+		D.current.add(B);
+		let n = await s(() => m.getConversation(e(F)), H);
+		$(n, t, !1);
+	}, pe = async (t) => {
+		T.current.get(B) === z && j(B, !1), D.current.add(B);
+		let n = await s(() => m.getConversation(e(F)), H, (e) => e.messages.length === L && e.messages.at(-1)?.role === f.User);
+		$(n, t, !0);
+	}, $ = (e, t, n) => {
+		if (H()) {
+			G();
+			return;
+		}
+		let r = e?.messages[e.messages.length - 1], i = e != null && e.messages.length - 1 === L && r?.role === f.Assistant, a = n && e != null && c(e) !== -1;
+		if (!e || !i && !a) {
+			D.current.delete(B), X(t);
+			return;
+		}
+		let o = O.current.get(B), s = o?.generationId === z ? o : void 0;
+		u(e) ? w(F, e, {
+			seedMessage: i ? s?.message : void 0,
+			skipDedupe: !0,
+			onSettled: Q
+		}) : (D.current.delete(B), s && O.current.delete(B), N(B) && (v(e), y.current = e), Q());
+	};
+	(async () => {
+		try {
+			let e = h?.channelId ?? await h?.waitForChannel(2e4) ?? void 0, t = k.current.has(z);
+			if (U.signal.aborted || H() || t) {
+				O.current.get(B)?.generationId === z && O.current.delete(B), H() || A(B), M(B, z), _(B, z), h?.notifyGenerationSettled?.(), t ? k.current.delete(z) : H() || g?.notifyGenerationEnd?.();
+				return;
+			}
+			m.streamCompletion(B, I, oe, Y, se, z, R, V, e);
+		} catch (e) {
+			Y.onError(e instanceof Error ? e : Error(String(e)));
+		}
+	})();
+};
+//#endregion
+export { m as runStreamSession };

package/conversation/useConversationStream/useConversationReloadErrors.d.ts

@@ -0,0 +1,25 @@
+import { MutableRefObject } from 'react';
+interface ConversationReloadFailure {
+    retry: () => Promise<void>;
+    isCurrent: () => boolean;
+}
+interface UseConversationReloadErrorsParams {
+    conversationId: string | undefined;
+    mountedRef: MutableRefObject<boolean>;
+}
+/**
+ * Terminal-read failures of `useConversationStream`, per conversation path:
+ * records a failed read with its retry, clears it on success or a new
+ * stream, and retries the displayed conversation's read once at a time. The
+ * ref is the source of truth; the state copies re-render the host.
+ */
+export declare const useConversationReloadErrors: ({ conversationId, mountedRef, }: UseConversationReloadErrorsParams) => {
+    reloadFailuresRef: import('react').RefObject<Map<string, ConversationReloadFailure>>;
+    reloadFailures: Map<string, ConversationReloadFailure>;
+    retrying: Set<ConversationReloadFailure>;
+    clearReloadError: (path: string) => void;
+    reportReloadError: (path: string, retry: () => Promise<void>, isCurrent: () => boolean) => void;
+    retryConversationReload: () => Promise<void>;
+};
+export {};
+//# sourceMappingURL=useConversationReloadErrors.d.ts.map

package/conversation/useConversationStream/useConversationReloadErrors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useConversationReloadErrors.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useConversationReloadErrors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,gBAAgB,EAAiC,MAAM,OAAO,CAAC;AAG7E,UAAU,yBAAyB;IACjC,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3B,SAAS,EAAE,MAAM,OAAO,CAAC;CAC1B;AAED,UAAU,iCAAiC;IACzC,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IAEnC,UAAU,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;CACvC;AAED;;;;;GAKG;AACH,eAAO,MAAM,2BAA2B,GAAI,iCAGzC,iCAAiC;;;;6BAWzB,MAAM;8BAWN,MAAM,SAAS,MAAM,OAAO,CAAC,IAAI,CAAC,aAAa,MAAM,OAAO;;CAmCtE,CAAC"}

package/conversation/useConversationStream/useConversationReloadErrors.js

@@ -0,0 +1,38 @@
+import { getConversationPath as e } from "./conversation-path.js";
+import { useCallback as t, useRef as n, useState as r } from "react";
+//#region src/conversation/useConversationStream/useConversationReloadErrors.ts
+var i = ({ conversationId: i, mountedRef: a }) => {
+	let o = n(/* @__PURE__ */ new Map()), [s, c] = r(o.current), l = n(/* @__PURE__ */ new Set()), [u, d] = r(l.current);
+	return {
+		reloadFailuresRef: o,
+		reloadFailures: s,
+		retrying: u,
+		clearReloadError: t((e) => {
+			if (!o.current.has(e)) return;
+			let t = new Map(o.current);
+			t.delete(e), o.current = t, a.current && c(t);
+		}, [a]),
+		reportReloadError: t((e, t, n) => {
+			if (!a.current || !n()) return;
+			let r = new Map(o.current);
+			r.set(e, {
+				retry: t,
+				isCurrent: n
+			}), o.current = r, c(r);
+		}, [a]),
+		retryConversationReload: t(async () => {
+			if (!i) return;
+			let t = o.current.get(e(i));
+			if (t && t.isCurrent() && !l.current.has(t)) {
+				l.current.add(t), d(new Set(l.current));
+				try {
+					await t.retry();
+				} finally {
+					l.current.delete(t), a.current && d(new Set(l.current));
+				}
+			}
+		}, [i, a])
+	};
+};
+//#endregion
+export { i as useConversationReloadErrors };