@epam/ai-dial-chat-hooks 1.3.0-dev.152 → 1.3.0-dev.157

package/conversation/useConversationStream/generation-resume.d.ts

@@ -42,6 +42,12 @@ export interface ResumeGenerationOptions {
     onSettled?: () => void;
     /** Resumes even though the path is already in `resumingPathsRef`, for a caller that reserved it itself. */
     skipDedupe?: boolean;
+    /**
+     * Takes the `generationId` the attach `snapshot` carries so Stop can reach a resumed
+     * non-background generation. Set only by the public entry point: stream recovery
+     * already knows its local id, and the conflict handover keeps Stop unavailable.
+     */
+    trackAttachGenerationId?: boolean;
 }
 /** Host-owned state {@link createResumeIfAwaitingGeneration} reads/writes through. */
 export interface ResumeIfAwaitingGenerationDeps {
@@ -66,6 +72,13 @@ export interface ResumeIfAwaitingGenerationDeps {
      * its id is removed when the resume settles.
      */
     stoppedGenerationIdsRef?: MutableRefObject<Set<string>>;
+    /** Records the id an attach `snapshot` supplied for `path`, so the host can offer Stop for it. */
+    rememberResumedGenerationId?: (path: string, generationId: string) => void;
+    /**
+     * Drops that id again once the resume no longer owns an attached generation. Compare-and-clear:
+     * the host removes the entry only while it still holds `generationId`.
+     */
+    forgetResumedGenerationId?: (path: string, generationId: string) => void;
 }
 /**
  * Builds `resumeIfAwaitingGeneration`: a hard refresh mid-generation loads a
@@ -81,5 +94,5 @@ export interface ResumeIfAwaitingGenerationDeps {
  * backend's finalize save resolves the placeholder whenever attach is
  * unavailable or ends without a terminal event.
  */
-export declare const createResumeIfAwaitingGeneration: ({ transport, setConversation, conversationRef, resumingPathsRef, bufferedGenerationsRef, addStreamingPath, removeStreamingPath, isPathDisplayed, generationPersistenceErrorMessage, frameScheduler, stoppedGenerationIdsRef, onReloadError, onReloadSuccess, }: ResumeIfAwaitingGenerationDeps) => (currentConversationId: string, conversation: Conversation, options?: ResumeGenerationOptions) => void;
+export declare const createResumeIfAwaitingGeneration: ({ transport, setConversation, conversationRef, resumingPathsRef, bufferedGenerationsRef, addStreamingPath, removeStreamingPath, isPathDisplayed, generationPersistenceErrorMessage, frameScheduler, stoppedGenerationIdsRef, rememberResumedGenerationId, forgetResumedGenerationId, onReloadError, onReloadSuccess, }: ResumeIfAwaitingGenerationDeps) => (currentConversationId: string, conversation: Conversation, options?: ResumeGenerationOptions) => void;
 //# sourceMappingURL=generation-resume.d.ts.map

package/conversation/useConversationStream/generation-resume.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generation-resume.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/generation-resume.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,YAAY,EAEjB,KAAK,OAAO,EAEb,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,OAAO,CAAC;AAOxE,OAAO,EACL,KAAK,kBAAkB,EAExB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,yBAAyB,CAAC;AAE3E;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,GAAI,SAAS,OAAO,KAAG,OAWtD,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,iCAAiC,GAC5C,cAAc,YAAY,KACzB,MAKA,CAAC;AAEJ;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B,GACrC,cAAc,YAAY,KACzB,OAUF,CAAC;AA6BF,eAAO,MAAM,0BAA0B,UAAkC,CAAC;AAe1E;;;;;GAKG;AACH,eAAO,MAAM,4BAA4B,GACvC,MAAM,MAAM,OAAO,CAAC,YAAY,CAAC,EACjC,YAAY,MAAM,OAAO,EACzB,YAAY,CAAC,YAAY,EAAE,YAAY,KAAK,OAAO,KAClD,OAAO,CAAC,YAAY,GAAG,IAAI,CAiB7B,CAAC;AAgEF,sGAAsG;AACtG,MAAM,WAAW,uBAAuB;IACtC,qIAAqI;IACrI,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,oGAAoG;IACpG,SAAS,CAAC,EAAE,MAAM,IAAI,CAAC;IACvB,2GAA2G;IAC3G,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,sFAAsF;AACtF,MAAM,WAAW,8BAA8B;IAC7C,SAAS,EAAE,2BAA2B,CAAC;IACvC,eAAe,EAAE,QAAQ,CAAC,cAAc,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC;IAC/D,eAAe,EAAE,gBAAgB,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC;IACvD,gBAAgB,EAAE,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAChD,sBAAsB,EAAE,gBAAgB,CAAC,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAC1E,gBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACzC,mBAAmB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC;IAC3C,iCAAiC,CAAC,EAAE,MAAM,CAAC;IAC3C,uEAAuE;IACvE,aAAa,CAAC,EAAE,CACd,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,EAC1B,SAAS,EAAE,MAAM,OAAO,KACrB,IAAI,CAAC;IACV,2DAA2D;IAC3D,eAAe,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACzC,iFAAiF;IACjF,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;;;OAIG;IACH,uBAAuB,CAAC,EAAE,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;CACzD;AAED;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,gCAAgC,GAAI,gQAc9C,8BAA8B,MAE7B,uBAAuB,MAAM,EAC7B,cAAc,YAAY,EAC1B,UAAS,uBAA4B,KACpC,IA8QJ,CAAC"}
+{"version":3,"file":"generation-resume.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/generation-resume.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,YAAY,EAEjB,KAAK,OAAO,EAEb,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,OAAO,CAAC;AAOxE,OAAO,EACL,KAAK,kBAAkB,EAExB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,yBAAyB,CAAC;AAE3E;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,GAAI,SAAS,OAAO,KAAG,OAWtD,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,iCAAiC,GAC5C,cAAc,YAAY,KACzB,MAKA,CAAC;AAEJ;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B,GACrC,cAAc,YAAY,KACzB,OAUF,CAAC;AA6BF,eAAO,MAAM,0BAA0B,UAAkC,CAAC;AAe1E;;;;;GAKG;AACH,eAAO,MAAM,4BAA4B,GACvC,MAAM,MAAM,OAAO,CAAC,YAAY,CAAC,EACjC,YAAY,MAAM,OAAO,EACzB,YAAY,CAAC,YAAY,EAAE,YAAY,KAAK,OAAO,KAClD,OAAO,CAAC,YAAY,GAAG,IAAI,CAiB7B,CAAC;AAgEF,sGAAsG;AACtG,MAAM,WAAW,uBAAuB;IACtC,qIAAqI;IACrI,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,oGAAoG;IACpG,SAAS,CAAC,EAAE,MAAM,IAAI,CAAC;IACvB,2GAA2G;IAC3G,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;;;OAIG;IACH,uBAAuB,CAAC,EAAE,OAAO,CAAC;CACnC;AAED,sFAAsF;AACtF,MAAM,WAAW,8BAA8B;IAC7C,SAAS,EAAE,2BAA2B,CAAC;IACvC,eAAe,EAAE,QAAQ,CAAC,cAAc,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC;IAC/D,eAAe,EAAE,gBAAgB,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC;IACvD,gBAAgB,EAAE,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAChD,sBAAsB,EAAE,gBAAgB,CAAC,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC,CAAC;IAC1E,gBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACzC,mBAAmB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC;IAC3C,iCAAiC,CAAC,EAAE,MAAM,CAAC;IAC3C,uEAAuE;IACvE,aAAa,CAAC,EAAE,CACd,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,EAC1B,SAAS,EAAE,MAAM,OAAO,KACrB,IAAI,CAAC;IACV,2DAA2D;IAC3D,eAAe,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACzC,iFAAiF;IACjF,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;;;OAIG;IACH,uBAAuB,CAAC,EAAE,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACxD,kGAAkG;IAClG,2BAA2B,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3E;;;OAGG;IACH,yBAAyB,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,IAAI,CAAC;CAC1E;AAED;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,gCAAgC,GAAI,wTAgB9C,8BAA8B,MAE7B,uBAAuB,MAAM,EAC7B,cAAc,YAAY,EAC1B,UAAS,uBAA4B,KACpC,IA6SJ,CAAC"}

package/conversation/useConversationStream/generation-resume.js

@@ -63,67 +63,76 @@ var c = (e) => {
 	} 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?.();
+}, _ = ({ transport: o, setConversation: s, conversationRef: p, resumingPathsRef: m, bufferedGenerationsRef: h, addStreamingPath: _, removeStreamingPath: v, isPathDisplayed: y, generationPersistenceErrorMessage: b = t, frameScheduler: x, stoppedGenerationIdsRef: S, rememberResumedGenerationId: C, forgetResumedGenerationId: w, onReloadError: T, onReloadSuccess: E }) => (t, D, O = {}) => {
+	if (!u(D)) {
+		O.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 = {
+	let k = r(t);
+	if (!O.skipDedupe && m.current.has(k)) return;
+	m.current.add(k), _(k);
+	let A = l(D), j = A !== -1, M = j ? A : D.messages.length - 1, N = {
 		generationId: f,
-		messageIndex: A,
-		message: E.seedMessage ?? T.messages[A]
+		messageIndex: M,
+		message: O.seedMessage ?? D.messages[M]
 	};
-	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);
-		t || n ? (j.message = {
-			...j.message,
+	h.current.set(k, N);
+	let P = () => h.current.get(k) === N, F = j ? D.messages[M].backgroundGeneration?.generationId : void 0, I, L = () => {
+		let e = F ?? I;
+		return e != null && (S?.current.has(e) ?? !1);
+	}, R = (e) => {
+		O.trackAttachGenerationId && !j && e != null && P() && (I = e, C?.(k, e));
+	}, z = () => {
+		if (I == null) return;
+		let e = I;
+		I = void 0, S?.current.delete(e), w?.(k, e);
+	}, B = (e, t = !1) => {
+		if (x?.flush(k), F != null && S?.current.delete(F), z(), !P()) return;
+		m.current.delete(k), v(k);
+		let n = !j && e && e.messages.length - 1 === M && u(e) && c(N.message);
+		t || n ? (N.message = {
+			...N.message,
 			streamErrorMessage: b
-		}, y(D) && s((e) => {
-			if (!e || !M() || !y(D)) return e;
-			let t = a(e, j);
+		}, y(k) && s((e) => {
+			if (!e || !P() || !y(k)) return e;
+			let t = a(e, N);
 			return p.current = t, t;
-		})) : (h.current.delete(D), e && y(D) && (s(e), p.current = e));
-	}, I = async () => {
-		if (M()) try {
+		})) : (h.current.delete(k), e && y(k) && (s(e), p.current = e));
+	}, V = async () => {
+		if (P()) try {
 			let n = await o.getConversation(e(t));
-			if (!M()) return;
-			w?.(D), F(n);
+			if (!P()) return;
+			E?.(k), B(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 (!P()) return;
+			x?.flush(k), m.current.delete(k), v(k), F != null && S?.current.delete(F), z(), T?.(k, V, P);
 		}
-	}, 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);
+	}, H = (e) => {
+		x?.cancel(k), P() && (N.message = e, y(k) && s((e) => {
+			if (!e || !P() || !y(k)) return e;
+			let t = a(e, N);
 			return p.current = t, t;
 		}));
-	}, R = () => s((e) => {
-		if (!e || !M() || !y(D)) return e;
-		let t = a(e, j);
+	}, U = () => s((e) => {
+		if (!e || !P() || !y(k)) return e;
+		let t = a(e, N);
 		return p.current = t, t;
-	}), z = (e) => {
-		if (!M() || P()) return;
-		let t = i([j.message], 0, e);
-		t && (j.message = t[0]), y(D) && (x ? x.schedule(D, R) : R());
-	}, B = async () => {
+	}), W = (e) => {
+		if (!P() || L()) return;
+		let t = i([N.message], 0, e);
+		t && (N.message = t[0]), y(k) && (x ? x.schedule(k, U) : U());
+	}, G = async () => {
 		let n = new AbortController(), r;
 		try {
-			r = await o.watchConversation(D, n.signal);
+			r = await o.watchConversation(k, n.signal);
 		} catch {
-			await I();
+			await V();
 			return;
 		}
 		try {
 			let i = await o.getConversation(e(t));
 			if (!u(i)) {
-				n.abort(), r.cancel().catch(() => void 0), F(i);
+				n.abort(), r.cancel().catch(() => void 0), B(i);
 				return;
 			}
 		} catch {}
@@ -135,38 +144,38 @@ var c = (e) => {
 				if (n?.action !== "UPDATE") return !1;
 				try {
 					let n = await o.getConversation(e(t));
-					if (!u(n)) return F(n), i = !0, !0;
+					if (!u(n)) return B(n), i = !0, !0;
 				} catch {}
 				return !1;
 			});
 		} finally {
 			clearTimeout(a);
 		}
-		i || await I();
-	}, V = async () => {
+		i || await V();
+	}, K = async () => {
 		let e = new AbortController(), t;
 		try {
-			t = await o.attachToGeneration(D, e.signal);
+			t = await o.attachToGeneration(k, e.signal);
 		} catch {
 			return !1;
 		}
 		let r = !1, i = !1;
 		return await g(t, (e) => {
-			if (!M()) return !0;
+			if (!P()) return !0;
 			switch (e.type) {
-				case "snapshot": return L(e.message), !1;
-				case "chunk": return z(e.chunk), !1;
+				case "snapshot": return R(e.generationId), H(e.message), !1;
+				case "chunk": return W(e.chunk), !1;
 				case "done":
 				case "error":
 				case "stopped": return i = e.type === "error" && e.errorType === n.type, r = !0, !0;
 			}
-		}), r ? (i ? F(void 0, !0) : await I(), !0) : !M();
+		}), r ? (i ? B(void 0, !0) : await V(), !0) : (z(), !P());
 	};
 	(async () => {
 		try {
-			await V() || await B();
+			await K() || await G();
 		} finally {
-			E.onSettled?.();
+			O.onSettled?.();
 		}
 	})();
 };

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

@@ -1 +1 @@
-{"version":3,"file":"useConversationStream.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useConversationStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EACL,KAAK,YAAY,EAEjB,KAAK,oBAAoB,EAEzB,KAAK,WAAW,EACjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,KAAK,QAAQ,EACb,KAAK,gBAAgB,EACrB,KAAK,cAAc,EAMpB,MAAM,OAAO,CAAC;AA6Cf,gFAAgF;AAChF,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;IACtC,UAAU,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IAChC,MAAM,EAAE,WAAW,CAAC;CACrB;AAED;;;;GAIG;AACH,MAAM,WAAW,2BAA2B;IAC1C,6GAA6G;IAC7G,gBAAgB,CACd,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,GAAG,SAAS,EAC3B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,uBAAuB,EAChC,aAAa,CAAC,EAAE,oBAAoB,EACpC,YAAY,CAAC,EAAE,MAAM,EACrB,IAAI,CAAC,EAAE,yBAAyB,EAChC,YAAY,CAAC,EAAE,MAAM,EACrB,eAAe,CAAC,EAAE,MAAM,GACvB,IAAI,CAAC;IACR;;;OAGG;IACH,cAAc,CAAC,MAAM,EAAE;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClB,uFAAuF;IACvF,iBAAiB,CACf,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IACvC;;;;;;;OAOG;IACH,kBAAkB,CAChB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IACvC,4EAA4E;IAC5E,eAAe,CACb,cAAc,EAAE,MAAM,EACtB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,YAAY,CAAC,CAAC;IACzB;;;;;OAKG;IACH,gBAAgB,CAAC,CACf,IAAI,EAAE,MAAM,EACZ,YAAY,EAAE,YAAY,GACzB,OAAO,CAAC,YAAY,CAAC,CAAC;CAC1B;AAED,6FAA6F;AAC7F,MAAM,WAAW,+BAA+B;IAC9C,iGAAiG;IACjG,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,eAAe,CAAC;IACzE,wGAAwG;IACxG,kBAAkB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,IAAI,CAAC;CAClE;AAED,yFAAyF;AACzF,MAAM,WAAW,yBAAyB;IACxC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,eAAe,EAAE,MAAM,IAAI,CAAC;IAC5B,oKAAoK;IACpK,cAAc,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC/D,+HAA+H;IAC/H,uBAAuB,CAAC,EAAE,MAAM,IAAI,CAAC;CACtC;AAED,6GAA6G;AAC7G,MAAM,WAAW,iCAAiC;IAChD,qBAAqB,CAAC,EAAE,MAAM,IAAI,CAAC;IACnC,mBAAmB,CAAC,EAAE,MAAM,IAAI,CAAC;IACjC,oBAAoB,CAAC,EAAE,MAAM,IAAI,CAAC;CACnC;AAED,+FAA+F;AAC/F,MAAM,WAAW,yBAAyB;IACxC,eAAe,EAAE,QAAQ,CAAC,cAAc,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC;IAC/D,eAAe,EAAE,gBAAgB,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC;CACxD;AAED,oDAAoD;AACpD,MAAM,WAAW,2BAA2B;IAC1C,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,KAAK,EAAE,yBAAyB,CAAC;IACjC,SAAS,EAAE,2BAA2B,CAAC;IACvC,UAAU,EAAE,+BAA+B,CAAC;IAC5C,OAAO,CAAC,EAAE,yBAAyB,CAAC;IACpC,OAAO,CAAC,EAAE,iCAAiC,CAAC;IAC5C,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACrC;;;;;OAKG;IACH,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,iFAAiF;IACjF,iCAAiC,CAAC,EAAE,MAAM,CAAC;IAC3C;;;;;OAKG;IACH,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACvC;;;;;OAKG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAqBD,4EAA4E;AAC5E,MAAM,WAAW,kBAAkB;IACjC;;;;;;OAMG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,qDAAqD;AACrD,MAAM,WAAW,2BAA2B;IAC1C,WAAW,EAAE,CACX,cAAc,EAAE,MAAM,EACtB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,EACb,aAAa,CAAC,EAAE,oBAAoB,EACpC,YAAY,CAAC,EAAE,MAAM,EACrB,IAAI,CAAC,EAAE,yBAAyB,EAChC,OAAO,CAAC,EAAE,kBAAkB,KACzB,IAAI,CAAC;IACV,UAAU,EAAE,MAAM,IAAI,CAAC;IACvB,0BAA0B,EAAE,CAC1B,qBAAqB,EAAE,MAAM,EAC7B,YAAY,EAAE,YAAY,KACvB,IAAI,CAAC;IACV,+GAA+G;IAC/G,yBAAyB,EAAE,CACzB,qBAAqB,EAAE,MAAM,EAC7B,YAAY,EAAE,YAAY,KACvB,YAAY,CAAC;IAClB,WAAW,EAAE,OAAO,CAAC;IACrB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,iEAAiE;IACjE,0BAA0B,EAAE,OAAO,CAAC;IACpC,iDAAiD;IACjD,uBAAuB,EAAE,OAAO,CAAC;IACjC,yEAAyE;IACzE,uBAAuB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9C;AAkBD;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,GAAI,mQAYnC,2BAA2B,KAAG,2BAm6BhC,CAAC"}
+{"version":3,"file":"useConversationStream.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useConversationStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC1E,OAAO,EACL,KAAK,YAAY,EAEjB,KAAK,oBAAoB,EAEzB,KAAK,WAAW,EACjB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,KAAK,QAAQ,EACb,KAAK,gBAAgB,EACrB,KAAK,cAAc,EAMpB,MAAM,OAAO,CAAC;AA6Cf,gFAAgF;AAChF,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;IACtC,UAAU,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IAChC,MAAM,EAAE,WAAW,CAAC;CACrB;AAED;;;;GAIG;AACH,MAAM,WAAW,2BAA2B;IAC1C,6GAA6G;IAC7G,gBAAgB,CACd,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,GAAG,SAAS,EAC3B,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,uBAAuB,EAChC,aAAa,CAAC,EAAE,oBAAoB,EACpC,YAAY,CAAC,EAAE,MAAM,EACrB,IAAI,CAAC,EAAE,yBAAyB,EAChC,YAAY,CAAC,EAAE,MAAM,EACrB,eAAe,CAAC,EAAE,MAAM,GACvB,IAAI,CAAC;IACR;;;OAGG;IACH,cAAc,CAAC,MAAM,EAAE;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClB,uFAAuF;IACvF,iBAAiB,CACf,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IACvC;;;;;;;OAOG;IACH,kBAAkB,CAChB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC;IACvC,4EAA4E;IAC5E,eAAe,CACb,cAAc,EAAE,MAAM,EACtB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,YAAY,CAAC,CAAC;IACzB;;;;;OAKG;IACH,gBAAgB,CAAC,CACf,IAAI,EAAE,MAAM,EACZ,YAAY,EAAE,YAAY,GACzB,OAAO,CAAC,YAAY,CAAC,CAAC;CAC1B;AAED,6FAA6F;AAC7F,MAAM,WAAW,+BAA+B;IAC9C,iGAAiG;IACjG,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,eAAe,CAAC;IACzE,wGAAwG;IACxG,kBAAkB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,IAAI,CAAC;CAClE;AAED,yFAAyF;AACzF,MAAM,WAAW,yBAAyB;IACxC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,eAAe,EAAE,MAAM,IAAI,CAAC;IAC5B,oKAAoK;IACpK,cAAc,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC/D,+HAA+H;IAC/H,uBAAuB,CAAC,EAAE,MAAM,IAAI,CAAC;CACtC;AAED,6GAA6G;AAC7G,MAAM,WAAW,iCAAiC;IAChD,qBAAqB,CAAC,EAAE,MAAM,IAAI,CAAC;IACnC,mBAAmB,CAAC,EAAE,MAAM,IAAI,CAAC;IACjC,oBAAoB,CAAC,EAAE,MAAM,IAAI,CAAC;CACnC;AAED,+FAA+F;AAC/F,MAAM,WAAW,yBAAyB;IACxC,eAAe,EAAE,QAAQ,CAAC,cAAc,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC;IAC/D,eAAe,EAAE,gBAAgB,CAAC,YAAY,GAAG,IAAI,CAAC,CAAC;CACxD;AAED,oDAAoD;AACpD,MAAM,WAAW,2BAA2B;IAC1C,cAAc,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,KAAK,EAAE,yBAAyB,CAAC;IACjC,SAAS,EAAE,2BAA2B,CAAC;IACvC,UAAU,EAAE,+BAA+B,CAAC;IAC5C,OAAO,CAAC,EAAE,yBAAyB,CAAC;IACpC,OAAO,CAAC,EAAE,iCAAiC,CAAC;IAC5C,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACrC;;;;;OAKG;IACH,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,iFAAiF;IACjF,iCAAiC,CAAC,EAAE,MAAM,CAAC;IAC3C;;;;;OAKG;IACH,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACvC;;;;;OAKG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAqBD,4EAA4E;AAC5E,MAAM,WAAW,kBAAkB;IACjC;;;;;;OAMG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,qDAAqD;AACrD,MAAM,WAAW,2BAA2B;IAC1C,WAAW,EAAE,CACX,cAAc,EAAE,MAAM,EACtB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,EACb,aAAa,CAAC,EAAE,oBAAoB,EACpC,YAAY,CAAC,EAAE,MAAM,EACrB,IAAI,CAAC,EAAE,yBAAyB,EAChC,OAAO,CAAC,EAAE,kBAAkB,KACzB,IAAI,CAAC;IACV,UAAU,EAAE,MAAM,IAAI,CAAC;IACvB,0BAA0B,EAAE,CAC1B,qBAAqB,EAAE,MAAM,EAC7B,YAAY,EAAE,YAAY,KACvB,IAAI,CAAC;IACV,+GAA+G;IAC/G,yBAAyB,EAAE,CACzB,qBAAqB,EAAE,MAAM,EAC7B,YAAY,EAAE,YAAY,KACvB,YAAY,CAAC;IAClB,WAAW,EAAE,OAAO,CAAC;IACrB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,iEAAiE;IACjE,0BAA0B,EAAE,OAAO,CAAC;IACpC,iDAAiD;IACjD,uBAAuB,EAAE,OAAO,CAAC;IACjC,yEAAyE;IACzE,uBAAuB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9C;AAkBD;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,GAAI,mQAYnC,2BAA2B,KAAG,2BA88BhC,CAAC"}

package/conversation/useConversationStream/useConversationStream.js

@@ -5,166 +5,174 @@ import { applyChunkToMessages as l } from "./apply-chunk.js";
 import { restoreBufferedMessage as u } from "./buffered-generation.js";
 import { createResumeIfAwaitingGeneration as ee, fetchConversationForRecovery as te, findPendingBackgroundMessageIndex as d, hasGeneratedPayload as ne, isAwaitingGenerationResume as f } from "./generation-resume.js";
 import { createFrameScheduler as re } from "./frame-scheduler.js";
-import { MessageRole as p, generateUUID as ie } from "@epam/ai-dial-chat-shared";
-import { useCallback as m, useEffect as h, useMemo as g, useRef as _, useState as v } from "react";
-import { SendCompletionDtoModeEnum as y } from "@epam/ai-dial-chat-api-client";
+import { MessageRole as ie, generateUUID as ae } from "@epam/ai-dial-chat-shared";
+import { useCallback as p, useEffect as m, useMemo as h, useRef as g, useState as _ } from "react";
+import { SendCompletionDtoModeEnum as oe } from "@epam/ai-dial-chat-api-client";
 //#region src/conversation/useConversationStream/useConversationStream.ts
-var ae = 2e4, oe = 401, b = 404, x = 2e3, S = 100, se = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof s ? e.message : "", C = (e) => {
+var se = 2e4, ce = 401, v = 404, y = 2e3, b = 100, le = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof s ? e.message : "", x = (e) => {
 	if (!e) return;
 	let t = d(e);
 	return t === -1 ? void 0 : e.messages[t].backgroundGeneration?.generationId;
-}, w = ({ conversationId: s, state: { setConversation: w, conversationRef: T }, transport: E, generation: { startGeneration: ce, completeGeneration: le }, channel: D, overlay: O, onStopError: ue, generationConflictMessage: de = t, generationPersistenceErrorMessage: k = n, onStreamError: A, batchChunksPerFrame: j = !1 }) => {
-	let fe = _(A);
-	h(() => {
-		fe.current = A;
-	}, [A]);
-	let pe = _(j);
-	h(() => {
-		pe.current = j;
-	}, [j]);
-	let [M] = v(re), [N, P] = v(() => /* @__PURE__ */ new Set()), F = _(N);
-	F.current = N;
-	let [I, me] = v(() => /* @__PURE__ */ new Set()), L = _(/* @__PURE__ */ new Map()), R = _(/* @__PURE__ */ new Set()), z = _(/* @__PURE__ */ new Map()), B = _(/* @__PURE__ */ new Set()), V = _(/* @__PURE__ */ new Set()), he = _(/* @__PURE__ */ new Map()), H = _(!0), U = _(/* @__PURE__ */ new Map()), [ge, W] = v(U.current), G = _(/* @__PURE__ */ new Set()), [_e, ve] = v(G.current);
-	h(() => (H.current = !0, () => {
-		H.current = !1;
+}, S = ({ conversationId: s, state: { setConversation: S, conversationRef: C }, transport: w, generation: { startGeneration: ue, completeGeneration: de }, channel: T, overlay: E, onStopError: fe, generationConflictMessage: pe = t, generationPersistenceErrorMessage: D = n, onStreamError: O, batchChunksPerFrame: k = !1 }) => {
+	let me = g(O);
+	m(() => {
+		me.current = O;
+	}, [O]);
+	let he = g(k);
+	m(() => {
+		he.current = k;
+	}, [k]);
+	let [A] = _(re), [j, M] = _(() => /* @__PURE__ */ new Set()), N = g(j);
+	N.current = j;
+	let [P, ge] = _(() => /* @__PURE__ */ new Set()), F = g(/* @__PURE__ */ new Map()), I = g(/* @__PURE__ */ new Map()), L = g(/* @__PURE__ */ new Set()), R = g(/* @__PURE__ */ new Map()), z = g(/* @__PURE__ */ new Set()), B = g(/* @__PURE__ */ new Set()), _e = g(/* @__PURE__ */ new Map()), V = g(!0), H = g(/* @__PURE__ */ new Map()), [U, ve] = _(H.current), W = g(/* @__PURE__ */ new Set()), [ye, be] = _(W.current);
+	m(() => (V.current = !0, () => {
+		V.current = !1;
 	}), []);
-	let K = m((e) => {
-		if (!U.current.has(e)) return;
-		let t = new Map(U.current);
-		t.delete(e), U.current = t, H.current && W(t);
-	}, []), q = m((e, t, n) => {
-		if (!H.current || !n()) return;
-		let r = new Map(U.current);
+	let G = p((e) => {
+		if (!H.current.has(e)) return;
+		let t = new Map(H.current);
+		t.delete(e), H.current = t, V.current && ve(t);
+	}, []), K = p((e, t, n) => {
+		if (!V.current || !n()) return;
+		let r = new Map(H.current);
 		r.set(e, {
 			retry: t,
 			isCurrent: n
-		}), U.current = r, W(r);
-	}, []), ye = m(async () => {
+		}), H.current = r, ve(r);
+	}, []), xe = p(async () => {
 		if (!s) return;
-		let e = U.current.get(c(s));
-		if (e && e.isCurrent() && !G.current.has(e)) {
-			G.current.add(e), ve(new Set(G.current));
+		let e = H.current.get(c(s));
+		if (e && e.isCurrent() && !W.current.has(e)) {
+			W.current.add(e), be(new Set(W.current));
 			try {
 				await e.retry();
 			} finally {
-				G.current.delete(e), H.current && ve(new Set(G.current));
+				W.current.delete(e), V.current && be(new Set(W.current));
 			}
 		}
-	}, [s]), be = _(s);
-	h(() => (be.current = s, () => M.cancelAll()), [s, M]);
-	let J = m((e) => H.current && c(be.current ?? "") === e, []), Y = m((e) => {
-		K(e), P((t) => {
+	}, [s]), Se = g(s);
+	m(() => (Se.current = s, () => A.cancelAll()), [s, A]);
+	let q = p((e) => V.current && c(Se.current ?? "") === e, []), J = p((e) => {
+		G(e), M((t) => {
 			if (t.has(e)) return t;
 			let n = new Set(t);
 			return n.add(e), n;
 		});
-	}, [K]), X = m((e) => {
-		P((t) => {
+	}, [G]), Y = p((e) => {
+		M((t) => {
 			if (!t.has(e)) return t;
 			let n = new Set(t);
 			return n.delete(e), n;
 		});
-	}, []), Z = m((e, t) => {
-		me((n) => {
+	}, []), X = p((e, t) => {
+		ge((n) => {
 			if (n.has(e) === t) return n;
 			let r = new Set(n);
 			return t ? r.add(e) : r.delete(e), r;
 		});
-	}, []), xe = m((e, t) => {
-		L.current.get(e) === t && (L.current.delete(e), Z(e, !1));
-	}, [Z]), Q = g(() => ee({
-		transport: E,
-		setConversation: w,
-		conversationRef: T,
-		resumingPathsRef: R,
-		bufferedGenerationsRef: z,
-		addStreamingPath: Y,
-		removeStreamingPath: X,
-		isPathDisplayed: J,
-		generationPersistenceErrorMessage: k,
-		frameScheduler: j ? M : void 0,
-		stoppedGenerationIdsRef: B,
-		onReloadError: q,
-		onReloadSuccess: K
+	}, []), Ce = p((e, t) => {
+		F.current.get(e) === t && (F.current.delete(e), X(e, !1));
+	}, [X]), we = p((e, t) => {
+		F.current.has(e) || (I.current.set(e, t), X(e, !0));
+	}, [X]), Te = p((e, t) => {
+		I.current.get(e) === t && (I.current.delete(e), F.current.has(e) || X(e, !1));
+	}, [X]), Z = h(() => ee({
+		transport: w,
+		setConversation: S,
+		conversationRef: C,
+		resumingPathsRef: L,
+		bufferedGenerationsRef: R,
+		addStreamingPath: J,
+		removeStreamingPath: Y,
+		isPathDisplayed: q,
+		generationPersistenceErrorMessage: D,
+		frameScheduler: k ? A : void 0,
+		stoppedGenerationIdsRef: z,
+		rememberResumedGenerationId: we,
+		forgetResumedGenerationId: Te,
+		onReloadError: K,
+		onReloadSuccess: G
 	}), [
-		Y,
-		X,
 		J,
-		E,
-		k,
-		j,
-		M,
+		Y,
 		q,
-		K
-	]), Se = m((t, n, a, s, ee, re, m = y.Append, h = {}) => {
-		let g = re ?? ie(), _ = c(t);
-		M.flush(_), R.current.delete(_), L.current.set(_, g), he.current.set(_, g), Z(_, !0);
-		let v;
-		m === y.Regenerate ? v = a : m === y.Edit && (v = a - 1);
-		let b = () => he.current.get(_) !== g, x = ce(_, g), S = T.current?.messages[a];
-		S ? z.current.set(_, {
-			generationId: g,
+		w,
+		D,
+		k,
+		A,
+		we,
+		Te,
+		K,
+		G
+	]), Ee = p((t, n, a, s, ee, re, p = oe.Append, m = {}) => {
+		let h = re ?? ae(), g = c(t);
+		A.flush(g), L.current.delete(g), F.current.set(g, h), _e.current.set(g, h), X(g, !0);
+		let _;
+		p === oe.Regenerate ? _ = a : p === oe.Edit && (_ = a - 1);
+		let v = () => _e.current.get(g) !== h, y = ue(g, h), b = C.current?.messages[a];
+		b ? R.current.set(g, {
+			generationId: h,
 			messageIndex: a,
-			message: S
-		}) : z.current.delete(_), Y(_), O?.notifyGenerationStart?.(), D?.ensureConnected();
-		let C = () => {
-			b() || X(_), xe(_, g), le(_, g), D?.notifyGenerationSettled?.();
-		}, ue = () => {
-			if (b()) return;
-			let e = z.current.get(_);
-			e && e.generationId === g && (e.message = {
+			message: b
+		}) : R.current.delete(g), J(g), E?.notifyGenerationStart?.(), T?.ensureConnected();
+		let x = () => {
+			v() || Y(g), Ce(g, h), de(g, h), T?.notifyGenerationSettled?.();
+		}, fe = () => {
+			if (v()) return;
+			let e = R.current.get(g);
+			e && e.generationId === h && (e.message = {
 				...e.message,
-				streamErrorMessage: k
-			}, fe.current?.(new i()), J(_) && w((t) => {
-				if (!t || b() || !J(_)) return t;
+				streamErrorMessage: D
+			}, me.current?.(new i()), q(g) && S((t) => {
+				if (!t || v() || !q(g)) return t;
 				let n = u(t, e);
-				return T.current = n, n;
+				return C.current = n, n;
 			}));
-		}, A = z.current.get(_), j = () => H.current && !b() && z.current.get(_) === A, N = async () => {
-			if (j()) {
-				if (!J(_)) A && z.current.delete(_);
+		}, O = R.current.get(g), k = () => V.current && !v() && R.current.get(g) === O, j = async () => {
+			if (k()) {
+				if (!q(g)) O && R.current.delete(g);
 				else try {
-					let n = await E.getConversation(e(t));
-					if (!j()) return;
-					if (K(_), !J(_)) {
-						A && z.current.delete(_);
+					let n = await w.getConversation(e(t));
+					if (!k()) return;
+					if (G(g), !q(g)) {
+						O && R.current.delete(g);
 						return;
 					}
 					if (d(n) !== -1) {
-						Q(t, n, { seedMessage: A?.message });
+						Z(t, n, { seedMessage: O?.message });
 						return;
 					}
-					if (A && A.messageIndex === n.messages.length - 1 && f(n) && ne(A.message)) {
-						ue();
+					if (O && O.messageIndex === n.messages.length - 1 && f(n) && ne(O.message)) {
+						fe();
 						return;
 					}
-					A && z.current.delete(_), w(n), T.current = n;
+					O && R.current.delete(g), S(n), C.current = n;
 				} catch {
-					j() && q(_, N, j);
+					k() && K(g, j, k);
 				}
 			}
-		}, P = {
-			signal: x.signal,
+		}, M = {
+			signal: y.signal,
 			onChunk: (e) => {
-				if (b()) return;
-				let t = z.current.get(_);
-				if (t?.generationId === g) {
+				if (v()) return;
+				let t = R.current.get(g);
+				if (t?.generationId === h) {
 					let n = l([t.message], 0, e);
 					n && (t.message = n[0]);
 				}
-				J(_) && (pe.current && t?.generationId === g ? M.schedule(_, () => {
-					let e = z.current.get(_);
-					if (b() || !J(_) || e?.generationId !== g) return;
+				q(g) && (he.current && t?.generationId === h ? A.schedule(g, () => {
+					let e = R.current.get(g);
+					if (v() || !q(g) || e?.generationId !== h) return;
 					let t = { ...e };
-					w((e) => {
+					S((e) => {
 						if (!e) return e;
 						let n = u(e, t);
-						return T.current = n, n;
+						return C.current = n, n;
 					});
-				}) : w((t) => {
+				}) : S((t) => {
 					if (!t) return t;
-					let n = z.current.get(_), r;
-					if (n?.generationId === g) r = u(t, n);
+					let n = R.current.get(g), r;
+					if (n?.generationId === h) r = u(t, n);
 					else {
 						let n = l(t.messages, a, e);
 						if (!n) return t;
@@ -173,157 +181,157 @@ var ae = 2e4, oe = 401, b = 404, x = 2e3, S = 100, se = (e, t, n) => e instanceo
 							messages: n
 						};
 					}
-					return T.current = r, r;
+					return C.current = r, r;
 				}));
 			},
 			onComplete: async () => {
-				M.flush(_), C(), B.current.has(g) ? B.current.delete(g) : b() || O?.notifyGenerationEnd?.(), await N();
+				A.flush(g), x(), z.current.has(h) ? z.current.delete(h) : v() || E?.notifyGenerationEnd?.(), await j();
 			},
 			onError: (e) => {
-				M.flush(_), fe.current?.(e), e instanceof o && !b() ? U(e) : e instanceof r && h.resumeOnConflict && !b() ? ge(e) : F(e);
+				A.flush(g), me.current?.(e), e instanceof o && !v() ? B(e) : e instanceof r && m.resumeOnConflict && !v() ? H(e) : N(e);
 			}
-		}, F = (e) => {
-			let t = z.current.get(_), n = t?.generationId === g ? t : void 0;
-			n && !(e instanceof i) && z.current.delete(_), C(), n && E.saveConversation && e instanceof i && e.status === oe ? me(e, n, E.saveConversation.bind(E)) : I(e, n);
-		}, I = (e, t) => {
+		}, N = (e) => {
+			let t = R.current.get(g), n = t?.generationId === h ? t : void 0;
+			n && !(e instanceof i) && R.current.delete(g), x(), n && w.saveConversation && e instanceof i && e.status === ce ? ge(e, n, w.saveConversation.bind(w)) : P(e, n);
+		}, P = (e, t) => {
 			if (t && e instanceof i && (t.message = {
 				...t.message,
-				streamErrorMessage: k
-			}), b() || !J(_)) return;
-			let n = se(e, de, k);
-			w((e) => {
-				if (!e || b() || !J(_)) return e;
-				let r = t?.generationId === g ? u(e, t) : e, i = {
+				streamErrorMessage: D
+			}), v() || !q(g)) return;
+			let n = le(e, pe, D);
+			S((e) => {
+				if (!e || v() || !q(g)) return e;
+				let r = t?.generationId === h ? u(e, t) : e, i = {
 					...r,
 					messages: r.messages.map((e, t) => t === a ? {
 						...e,
 						streamErrorMessage: n
 					} : e)
 				};
-				return T.current = i, i;
+				return C.current = i, i;
 			});
-		}, me = async (n, r, i) => {
-			let a = () => H.current && !b() && z.current.get(_) === r;
+		}, ge = async (n, r, i) => {
+			let a = () => V.current && !v() && R.current.get(g) === r;
 			try {
-				let o = await E.getConversation(e(t));
+				let o = await w.getConversation(e(t));
 				if (!a()) return;
 				if (r.messageIndex !== o.messages.length - 1 || !f(o)) {
-					I(n, r);
+					P(n, r);
 					return;
 				}
-				let s = await i(_, u(o, r));
-				if (!a() || (z.current.delete(_), !J(_))) return;
-				w(s), T.current = s;
+				let s = await i(g, u(o, r));
+				if (!a() || (R.current.delete(g), !q(g))) return;
+				S(s), C.current = s;
 			} catch {
-				a() && I(n, r);
+				a() && P(n, r);
 			}
-		}, V = () => {
-			C(), B.current.has(g) ? B.current.delete(g) : b() || O?.notifyGenerationEnd?.();
-		}, U = async (n) => {
-			R.current.add(_);
-			let r = await te(() => E.getConversation(e(t)), b);
-			W(r, n, !1);
-		}, ge = async (n) => {
-			L.current.get(_) === g && Z(_, !1), R.current.add(_);
-			let r = await te(() => E.getConversation(e(t)), b, (e) => e.messages.length === a && e.messages.at(-1)?.role === p.User);
-			W(r, n, !0);
-		}, W = (e, n, r) => {
-			if (b()) {
-				C();
+		}, I = () => {
+			x(), z.current.has(h) ? z.current.delete(h) : v() || E?.notifyGenerationEnd?.();
+		}, B = async (n) => {
+			L.current.add(g);
+			let r = await te(() => w.getConversation(e(t)), v);
+			U(r, n, !1);
+		}, H = async (n) => {
+			F.current.get(g) === h && X(g, !1), L.current.add(g);
+			let r = await te(() => w.getConversation(e(t)), v, (e) => e.messages.length === a && e.messages.at(-1)?.role === ie.User);
+			U(r, n, !0);
+		}, U = (e, n, r) => {
+			if (v()) {
+				x();
 				return;
 			}
-			let i = e?.messages[e.messages.length - 1], o = e != null && e.messages.length - 1 === a && i?.role === p.Assistant, s = r && e != null && d(e) !== -1;
+			let i = e?.messages[e.messages.length - 1], o = e != null && e.messages.length - 1 === a && i?.role === ie.Assistant, s = r && e != null && d(e) !== -1;
 			if (!e || !o && !s) {
-				R.current.delete(_), F(n);
+				L.current.delete(g), N(n);
 				return;
 			}
-			let c = z.current.get(_), l = c?.generationId === g ? c : void 0;
-			f(e) ? Q(t, e, {
+			let c = R.current.get(g), l = c?.generationId === h ? c : void 0;
+			f(e) ? Z(t, e, {
 				seedMessage: o ? l?.message : void 0,
 				skipDedupe: !0,
-				onSettled: V
-			}) : (R.current.delete(_), l && z.current.delete(_), J(_) && (w(e), T.current = e), V());
+				onSettled: I
+			}) : (L.current.delete(g), l && R.current.delete(g), q(g) && (S(e), C.current = e), I());
 		};
 		(async () => {
 			try {
-				let e = D?.channelId ?? await D?.waitForChannel(ae) ?? void 0, t = B.current.has(g);
-				if (x.signal.aborted || b() || t) {
-					z.current.get(_)?.generationId === g && z.current.delete(_), b() || X(_), xe(_, g), le(_, g), D?.notifyGenerationSettled?.(), t ? B.current.delete(g) : b() || O?.notifyGenerationEnd?.();
+				let e = T?.channelId ?? await T?.waitForChannel(se) ?? void 0, t = z.current.has(h);
+				if (y.signal.aborted || v() || t) {
+					R.current.get(g)?.generationId === h && R.current.delete(g), v() || Y(g), Ce(g, h), de(g, h), T?.notifyGenerationSettled?.(), t ? z.current.delete(h) : v() || E?.notifyGenerationEnd?.();
 					return;
 				}
-				E.streamCompletion(_, n, s, P, ee, g, m, v, e);
+				w.streamCompletion(g, n, s, M, ee, h, p, _, e);
 			} catch (e) {
-				P.onError(e instanceof Error ? e : Error(String(e)));
+				M.onError(e instanceof Error ? e : Error(String(e)));
 			}
 		})();
 	}, [
-		ce,
-		le,
+		ue,
+		de,
+		J,
 		Y,
 		X,
-		Z,
-		xe,
-		J,
-		D?.channelId,
-		D?.ensureConnected,
-		D?.waitForChannel,
-		D?.notifyGenerationSettled,
-		O,
-		E,
-		Q,
-		de,
-		k,
-		M,
+		Ce,
 		q,
-		K
-	]), Ce = m((e, t) => {
-		let n = c(e), r = z.current.get(n);
-		return U.current.get(n)?.isCurrent() && r && t.messages.length - 1 >= r.messageIndex && t.messages[r.messageIndex].role === p.Assistant && !f(t) ? (K(n), z.current.delete(n), t) : r ? u(t, r) : t;
-	}, [K]), we = m(() => {
+		T?.channelId,
+		T?.ensureConnected,
+		T?.waitForChannel,
+		T?.notifyGenerationSettled,
+		E,
+		w,
+		Z,
+		pe,
+		D,
+		A,
+		K,
+		G
+	]), De = p((e, t) => {
+		let n = c(e), r = R.current.get(n);
+		return H.current.get(n)?.isCurrent() && r && t.messages.length - 1 >= r.messageIndex && t.messages[r.messageIndex].role === ie.Assistant && !f(t) ? (G(n), R.current.delete(n), t) : r ? u(t, r) : t;
+	}, [G]), Oe = p(() => {
 		if (!s) return;
-		let e = c(s), t = L.current.get(e), n = t ?? C(T.current);
-		if (!n || V.current.has(n)) return;
-		V.current.add(n), B.current.add(n), M.flush(e), O?.notifyStopGenerating?.();
-		let r = z.current.get(e)?.message.content, i = () => L.current.get(e) === n || t == null && F.current.has(e), o = async () => {
-			let e = x / S;
+		let e = c(s), t = F.current.get(e), n = t ?? x(C.current) ?? I.current.get(e);
+		if (!n || B.current.has(n)) return;
+		B.current.add(n), z.current.add(n), A.flush(e), E?.notifyStopGenerating?.();
+		let r = R.current.get(e)?.message.content, i = () => F.current.get(e) === n || t == null && N.current.has(e), o = async () => {
+			let e = y / b;
 			for (let t = 0; t < e; t++) {
-				if (!i() || !H.current) return !0;
-				await new Promise((e) => setTimeout(e, S));
+				if (!i() || !V.current) return !0;
+				await new Promise((e) => setTimeout(e, b));
 			}
 			return !i();
 		};
 		(async () => {
 			try {
-				await E.stopCompletion({
+				await w.stopCompletion({
 					generationId: n,
 					path: e,
 					content: r
 				});
 			} catch (e) {
 				let t = e instanceof Error ? e : Error(String(e));
-				if (t instanceof a && t.status === b && await o()) return;
-				V.current.delete(n), ue?.(t);
+				if (t instanceof a && t.status === v && await o()) return;
+				B.current.delete(n), fe?.(t);
 			}
 		})();
 	}, [
 		s,
-		T,
-		M,
-		ue,
-		O,
-		E
-	]), Te = m((e, t) => Q(e, t), [Q]), Ee = s != null && N.has(c(s)), $ = s == null ? null : c(s), De = $ != null && (I.has($) || Ee && C(T.current) != null), Oe = $ ? ge.get($) : void 0, ke = Oe?.isCurrent() ?? !1;
+		C,
+		A,
+		fe,
+		E,
+		w
+	]), ke = p((e, t) => Z(e, t, { trackAttachGenerationId: !0 }), [Z]), Ae = s != null && j.has(c(s)), Q = s == null ? null : c(s), je = Q != null && (P.has(Q) || Ae && x(C.current) != null), $ = Q ? U.get(Q) : void 0, Me = $?.isCurrent() ?? !1;
 	return {
-		hasConversationReloadError: ke,
-		isReloadingConversation: !!Oe && ke && _e.has(Oe),
-		retryConversationReload: ye,
-		startStream: Se,
-		handleStop: we,
-		resumeIfAwaitingGeneration: Te,
-		restoreBufferedGeneration: Ce,
-		isStreaming: Ee,
-		canStopStreaming: De
+		hasConversationReloadError: Me,
+		isReloadingConversation: !!$ && Me && ye.has($),
+		retryConversationReload: xe,
+		startStream: Ee,
+		handleStop: Oe,
+		resumeIfAwaitingGeneration: ke,
+		restoreBufferedGeneration: De,
+		isStreaming: Ae,
+		canStopStreaming: je
 	};
 };
 //#endregion
-export { w as useConversationStream };
+export { S as useConversationStream };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@epam/ai-dial-chat-hooks",
   "description": "Framework-level React hooks extracted from AI DIAL Chat for building custom chat interfaces",
-  "version": "1.3.0-dev.152",
+  "version": "1.3.0-dev.157",
   "license": "Apache-2.0",
   "type": "module",
   "main": "./index.js",
@@ -121,7 +121,7 @@
     }
   },
   "dependencies": {
-    "@epam/ai-dial-chat-api-client": "1.3.0-dev.152",
+    "@epam/ai-dial-chat-api-client": "1.3.0-dev.157",
     "dompurify": "3.4.16",
     "fflate": "^0.8.3",
     "lru-cache": "^10.4.3",
@@ -130,22 +130,22 @@
   },
   "peerDependencies": {
     "react": "^19.2.8",
-    "@epam/ai-dial-attachment-canvas": "1.3.0-dev.152",
-    "@epam/ai-dial-attachment-input": "1.3.0-dev.152",
-    "@epam/ai-dial-catalog": "1.3.0-dev.152",
-    "@epam/ai-dial-chat-overlay": "1.3.0-dev.152",
-    "@epam/ai-dial-chat-shared": "1.3.0-dev.152",
-    "@epam/ai-dial-conversation-input": "1.3.0-dev.152",
-    "@epam/ai-dial-mcp-apps": "1.3.0-dev.152",
-    "@epam/ai-dial-publish-panel": "1.3.0-dev.152",
-    "@epam/ai-dial-quotations": "1.3.0-dev.152",
+    "@epam/ai-dial-attachment-canvas": "1.3.0-dev.157",
+    "@epam/ai-dial-attachment-input": "1.3.0-dev.157",
+    "@epam/ai-dial-catalog": "1.3.0-dev.157",
+    "@epam/ai-dial-chat-overlay": "1.3.0-dev.157",
+    "@epam/ai-dial-chat-shared": "1.3.0-dev.157",
+    "@epam/ai-dial-conversation-input": "1.3.0-dev.157",
+    "@epam/ai-dial-mcp-apps": "1.3.0-dev.157",
+    "@epam/ai-dial-publish-panel": "1.3.0-dev.157",
+    "@epam/ai-dial-quotations": "1.3.0-dev.157",
     "@epam/ai-dial-react-file-manager": "^0.4.0-dev.11",
-    "@epam/ai-dial-scheduled-tasks": "1.3.0-dev.152",
-    "@epam/ai-dial-share": "1.3.0-dev.152",
-    "@epam/ai-dial-skill-editor": "1.3.0-dev.152",
-    "@epam/ai-dial-source-panel": "1.3.0-dev.152",
+    "@epam/ai-dial-scheduled-tasks": "1.3.0-dev.157",
+    "@epam/ai-dial-share": "1.3.0-dev.157",
+    "@epam/ai-dial-skill-editor": "1.3.0-dev.157",
+    "@epam/ai-dial-source-panel": "1.3.0-dev.157",
     "@epam/ai-dial-ui-kit": "^0.16.0-dev.22",
-    "@epam/ai-dial-usage-dashboard": "1.3.0-dev.152",
+    "@epam/ai-dial-usage-dashboard": "1.3.0-dev.157",
     "@modelcontextprotocol/sdk": "^1.32.1",
     "@epam/pdf-highlighter-kit": "^0.0.19",
     "@mcp-ui/client": "^7.1.1"