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

package/catalog/map-deployment-limits-to-input.d.ts

@@ -1,6 +1,7 @@
 import { CatalogItemLimits, UsageLimitProgressRow } from '@epam/ai-dial-catalog';
-import { DeploymentLimitsResponseDto, LimitStatsDto } from '@epam/ai-dial-chat-api-client';
+import { LimitStatsDto } from '@epam/ai-dial-chat-api-client';
 import { FormatResetTime } from '../usage/map-usage-data-to-dashboard';
+import { UsageLimitsSources } from '../usage/useDeploymentUsageLimits/useDeploymentUsageLimits';
 /** Labels and formatter callbacks for the conversation-input limits mapping utility. */
 export interface ConversationInputLimitsLabels {
     /** Heading of the group listing the selected deployment's token limits. */
@@ -23,6 +24,6 @@ export interface ConversationInputLimitsLabels {
     }) => string;
 }
 export declare const buildResetFields: (stats: LimitStatsDto, formatResetTime: FormatResetTime | undefined) => Pick<UsageLimitProgressRow, "resetLabel" | "resetIsoValue" | "resetAriaLabel">;
-/** Maps a deployment's day, week, and month token limits and the caller's cost budget to display-ready rows, or `undefined` when none qualify. */
-export declare const mapDeploymentLimitsToInput: (dto: DeploymentLimitsResponseDto | undefined, labels: ConversationInputLimitsLabels, formatResetTime?: FormatResetTime) => CatalogItemLimits | undefined;
+/** Maps the deployment's day, week, and month token limits (from `sources.deployment`) and the account cost budget (from `sources.account`) to display-ready rows, or `undefined` when none qualify. */
+export declare const mapDeploymentLimitsToInput: (sources: UsageLimitsSources, labels: ConversationInputLimitsLabels, formatResetTime?: FormatResetTime) => CatalogItemLimits | undefined;
 //# sourceMappingURL=map-deployment-limits-to-input.d.ts.map

package/catalog/map-deployment-limits-to-input.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"map-deployment-limits-to-input.d.ts","sourceRoot":"","sources":["../../src/catalog/map-deployment-limits-to-input.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,iBAAiB,EAEtB,KAAK,qBAAqB,EAC3B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EACV,2BAA2B,EAC3B,aAAa,EACd,MAAM,+BAA+B,CAAC;AAEvC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,sCAAsC,CAAC;AAE5E,wFAAwF;AACxF,MAAM,WAAW,6BAA6B;IAC5C,2EAA2E;IAC3E,UAAU,EAAE,MAAM,CAAC;IACnB,kFAAkF;IAClF,SAAS,EAAE,MAAM,CAAC;IAClB,8DAA8D;IAC9D,SAAS,EAAE,MAAM,CAAC;IAClB,+DAA+D;IAC/D,UAAU,EAAE,MAAM,CAAC;IACnB,gEAAgE;IAChE,WAAW,EAAE,MAAM,CAAC;IACpB,sEAAsE;IACtE,gBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,MAAM,CAAC;IAC1D,wDAAwD;IACxD,uBAAuB,EAAE,CAAC,MAAM,EAAE;QAChC,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;KACf,KAAK,MAAM,CAAC;CACd;AA+FD,eAAO,MAAM,gBAAgB,GAC3B,OAAO,aAAa,EACpB,iBAAiB,eAAe,GAAG,SAAS,KAC3C,IAAI,CACL,qBAAqB,EACrB,YAAY,GAAG,eAAe,GAAG,gBAAgB,CAYlD,CAAC;AAuGF,kJAAkJ;AAClJ,eAAO,MAAM,0BAA0B,GACrC,KAAK,2BAA2B,GAAG,SAAS,EAC5C,QAAQ,6BAA6B,EACrC,kBAAkB,eAAe,KAChC,iBAAiB,GAAG,SAyCtB,CAAC"}
+{"version":3,"file":"map-deployment-limits-to-input.d.ts","sourceRoot":"","sources":["../../src/catalog/map-deployment-limits-to-input.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,iBAAiB,EAEtB,KAAK,qBAAqB,EAC3B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EAEV,aAAa,EACd,MAAM,+BAA+B,CAAC;AAEvC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,sCAAsC,CAAC;AAC5E,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,4DAA4D,CAAC;AAErG,wFAAwF;AACxF,MAAM,WAAW,6BAA6B;IAC5C,2EAA2E;IAC3E,UAAU,EAAE,MAAM,CAAC;IACnB,kFAAkF;IAClF,SAAS,EAAE,MAAM,CAAC;IAClB,8DAA8D;IAC9D,SAAS,EAAE,MAAM,CAAC;IAClB,+DAA+D;IAC/D,UAAU,EAAE,MAAM,CAAC;IACnB,gEAAgE;IAChE,WAAW,EAAE,MAAM,CAAC;IACpB,sEAAsE;IACtE,gBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,MAAM,CAAC;IAC1D,wDAAwD;IACxD,uBAAuB,EAAE,CAAC,MAAM,EAAE;QAChC,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;KACf,KAAK,MAAM,CAAC;CACd;AAgGD,eAAO,MAAM,gBAAgB,GAC3B,OAAO,aAAa,EACpB,iBAAiB,eAAe,GAAG,SAAS,KAC3C,IAAI,CACL,qBAAqB,EACrB,YAAY,GAAG,eAAe,GAAG,gBAAgB,CAYlD,CAAC;AA4GF,wMAAwM;AACxM,eAAO,MAAM,0BAA0B,GACrC,SAAS,kBAAkB,EAC3B,QAAQ,6BAA6B,EACrC,kBAAkB,eAAe,KAChC,iBAAiB,GAAG,SAqCtB,CAAC"}

package/catalog/map-deployment-limits-to-input.js

@@ -61,7 +61,7 @@ var n = [
 	return r >= 1 ? t.LimitReached : r >= a && e !== t.LimitReached ? t.RunningLow : e;
 }, void 0), _ = (e, t, n, r, i, a, o) => {
 	let s = t.flatMap((t) => {
-		let n = e[t.key];
+		let n = e?.[t.key];
 		return !u(n) || !h(n) ? [] : (o.push(n), [p(n, r[t.labelField], r, i, a)]);
 	});
 	return s.length > 0 ? {
@@ -69,11 +69,10 @@ var n = [
 		rows: s
 	} : void 0;
 }, v = (t, i, a) => {
-	if (t == null) return;
-	let o = [], u = [_(t, n, i.tokenGroup, i, {
+	let o = [], u = [_(t.deployment, n, i.tokenGroup, i, {
 		format: (e) => s.format(l(e)),
 		formatFull: (e) => c.format(e)
-	}, a, o), _(t, r, i.costGroup, i, {
+	}, a, o), _(t.account, r, i.costGroup, i, {
 		format: e,
 		formatFull: e
 	}, a, o)].filter((e) => e != null);

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.149",
+  "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.149",
+    "@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.149",
-    "@epam/ai-dial-attachment-input": "1.3.0-dev.149",
-    "@epam/ai-dial-catalog": "1.3.0-dev.149",
-    "@epam/ai-dial-chat-overlay": "1.3.0-dev.149",
-    "@epam/ai-dial-chat-shared": "1.3.0-dev.149",
-    "@epam/ai-dial-conversation-input": "1.3.0-dev.149",
-    "@epam/ai-dial-mcp-apps": "1.3.0-dev.149",
-    "@epam/ai-dial-publish-panel": "1.3.0-dev.149",
-    "@epam/ai-dial-quotations": "1.3.0-dev.149",
+    "@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.149",
-    "@epam/ai-dial-share": "1.3.0-dev.149",
-    "@epam/ai-dial-skill-editor": "1.3.0-dev.149",
-    "@epam/ai-dial-source-panel": "1.3.0-dev.149",
+    "@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.149",
+    "@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"

package/usage/usage-banner-source.d.ts

@@ -1,6 +1,6 @@
-import { DeploymentLimitsResponseDto } from '@epam/ai-dial-chat-api-client';
 import { UsageLimitBannerSeverity } from '@epam/ai-dial-conversation-input';
 import { UsagePeriod } from './usage-period';
+import { UsageLimitsSources } from './useDeploymentUsageLimits/useDeploymentUsageLimits';
 /** The day/week/month usage stat the composer banner reports on, and how close it is to its limit. */
 export interface UsageBannerSource {
     /** Calendar period of the stat. */
@@ -15,9 +15,10 @@ export interface UsageBannerSource {
 /** `true` when `total` is a real cap: finite, positive, and below DIAL Core's unlimited sentinel. */
 export declare const isCappedStat: (total: number) => boolean;
 /**
- * Returns the capped day/week/month token or cost stat with the highest
- * used/total ratio, classified against `thresholdPercent`, or `undefined` when
- * no stat is capped or the worst one is below the threshold.
+ * Returns the capped day/week/month token stat of `sources.deployment` or cost
+ * stat of `sources.account` with the highest used/total ratio, classified
+ * against `thresholdPercent`, or `undefined` when no stat is capped or the
+ * worst one is below the threshold.
  */
-export declare const selectUsageBannerSource: (dto: DeploymentLimitsResponseDto | undefined, thresholdPercent: number) => UsageBannerSource | undefined;
+export declare const selectUsageBannerSource: (sources: UsageLimitsSources, thresholdPercent: number) => UsageBannerSource | undefined;
 //# sourceMappingURL=usage-banner-source.d.ts.map

package/usage/usage-banner-source.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usage-banner-source.d.ts","sourceRoot":"","sources":["../../src/usage/usage-banner-source.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,2BAA2B,EAE5B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,sGAAsG;AACtG,MAAM,WAAW,iBAAiB;IAChC,mCAAmC;IACnC,MAAM,EAAE,WAAW,CAAC;IACpB,uDAAuD;IACvD,QAAQ,EAAE,wBAAwB,CAAC;IACnC,wEAAwE;IACxE,WAAW,EAAE,MAAM,CAAC;IACpB,6DAA6D;IAC7D,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAQD,qGAAqG;AACrG,eAAO,MAAM,YAAY,GAAI,OAAO,MAAM,KAAG,OAC4B,CAAC;AA8B1E;;;;GAIG;AACH,eAAO,MAAM,uBAAuB,GAClC,KAAK,2BAA2B,GAAG,SAAS,EAC5C,kBAAkB,MAAM,KACvB,iBAAiB,GAAG,SAoCtB,CAAC"}
+{"version":3,"file":"usage-banner-source.d.ts","sourceRoot":"","sources":["../../src/usage/usage-banner-source.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,qDAAqD,CAAC;AAE9F,sGAAsG;AACtG,MAAM,WAAW,iBAAiB;IAChC,mCAAmC;IACnC,MAAM,EAAE,WAAW,CAAC;IACpB,uDAAuD;IACvD,QAAQ,EAAE,wBAAwB,CAAC;IACnC,wEAAwE;IACxE,WAAW,EAAE,MAAM,CAAC;IACpB,6DAA6D;IAC7D,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAQD,qGAAqG;AACrG,eAAO,MAAM,YAAY,GAAI,OAAO,MAAM,KAAG,OAC4B,CAAC;AAsC1E;;;;;GAKG;AACH,eAAO,MAAM,uBAAuB,GAClC,SAAS,kBAAkB,EAC3B,kBAAkB,MAAM,KACvB,iBAAiB,GAAG,SAsCtB,CAAC"}

package/usage/usage-banner-source.js

@@ -2,22 +2,46 @@ import { UsagePeriod as e } from "./usage-period.js";
 import { UsageLimitBannerSeverity as t } from "@epam/ai-dial-conversation-input";
 //#region src/usage/usage-banner-source.ts
 var n = 2 ** 53 - 1, r = (e) => Number.isFinite(e) && e > 0 && e < n, i = [
-	["monthTokenStats", e.Month],
-	["monthCostStats", e.Month],
-	["weekTokenStats", e.Week],
-	["weekCostStats", e.Week],
-	["dayTokenStats", e.Day],
-	["dayCostStats", e.Day]
+	[
+		"deployment",
+		"monthTokenStats",
+		e.Month
+	],
+	[
+		"account",
+		"monthCostStats",
+		e.Month
+	],
+	[
+		"deployment",
+		"weekTokenStats",
+		e.Week
+	],
+	[
+		"account",
+		"weekCostStats",
+		e.Week
+	],
+	[
+		"deployment",
+		"dayTokenStats",
+		e.Day
+	],
+	[
+		"account",
+		"dayCostStats",
+		e.Day
+	]
 ], a = (e, n) => {
 	let a;
-	for (let [t, n] of i) {
-		let i = e?.[t];
+	for (let [t, n, o] of i) {
+		let i = e[t]?.[n];
 		if (i == null || !r(i.total)) continue;
-		let o = Math.max(i.used, 0) / i.total;
-		(a == null || o > a.ratio) && (a = {
+		let s = Math.max(i.used, 0) / i.total;
+		(a == null || s > a.ratio) && (a = {
 			stat: i,
-			period: n,
-			ratio: o
+			period: o,
+			ratio: s
 		});
 	}
 	if (a == null) return;

package/usage/useDeploymentUsageLimits/useDeploymentUsageLimits.d.ts

@@ -1,32 +1,56 @@
-import { DeploymentLimitsResponseDto } from '@epam/ai-dial-chat-api-client';
-/** Host-configured call that fetches the limits of one deployment. */
-export type GetDeploymentLimits = (deploymentId: string) => Promise<DeploymentLimitsResponseDto>;
+import { DeploymentLimitsResponseDto, UserLimitStatsResponseDto } from '@epam/ai-dial-chat-api-client';
+/** The caller's account-wide cost budget and spend: the top level of a user-usage response. */
+export type AccountUsage = Pick<UserLimitStatsResponseDto, 'dayCostStats' | 'weekCostStats' | 'monthCostStats'>;
+/**
+ * What the usage dial and the usage banner are mapped from. Each field has a
+ * single scope, so a deployment's figures are never read as the account's or
+ * the other way round.
+ */
+export interface UsageLimitsSources {
+    /** The selected deployment's own stats (its deployment-usage response). Only its token stats are read. */
+    deployment: DeploymentLimitsResponseDto | undefined;
+    /** The caller's account budget and spend. Only its cost stats are read. */
+    account: AccountUsage | undefined;
+}
+/** Host-configured calls that fetch the two usage sources. */
+export interface UsageLimitsApi {
+    /** Fetches one deployment's own usage, every stat scoped to that deployment. */
+    getDeploymentUsage: (deploymentId: string) => Promise<DeploymentLimitsResponseDto>;
+    /** Fetches the caller's account usage; only its top-level cost stats are read. */
+    getAccountUsage: () => Promise<AccountUsage>;
+}
 /** Return shape of {@link useDeploymentUsageLimits}. */
 export interface UseDeploymentUsageLimitsResult {
-    /** Raw limits response for the selected deployment, or `undefined` before the first success. */
-    limitsDto: DeploymentLimitsResponseDto | undefined;
-    /** `true` while a fetch is in flight. */
+    /** The last successfully fetched value of each source. */
+    sources: UsageLimitsSources;
+    /** `true` until both calls of the current fetch have settled. */
     isLoading: boolean;
-    /** `true` after the most recent fetch rejected. */
+    /** `true` after either call of the most recent fetch rejected. */
     hasError: boolean;
-    /** Re-fetches the current deployment when no request is in flight. */
+    /** Re-fetches both sources when no request is in flight. */
     refresh: () => void;
 }
 /**
- * Fetches the selected deployment's usage limits via the provided
- * `getDeploymentLimits` function. The hook accepts the fetch function as a
- * parameter so the caller supplies an already-configured API client — the hook
- * never constructs or imports one itself. The function's identity is not a
- * re-fetch trigger; only `deploymentId` and `refresh()` are.
+ * Fetches the selected deployment's own usage and the caller's account usage
+ * through the provided `api`. The hook accepts the calls as a parameter so the
+ * caller supplies an already-configured API client; the hook never constructs
+ * or imports one itself. The identity of `api` is not a re-fetch trigger; only
+ * `deploymentId` and `refresh()` are.
+ *
+ * Two sources rather than one deployment-limits response: there, the cost
+ * stats add an application's aggregated callee cost to the account spend that
+ * already contains it, so an agent's spend would be counted twice.
  *
- * A monotonic request ID prevents a slower response for a previous deployment
- * from replacing the current value. Refresh failures preserve the last known
- * response so the open popover can surface a non-blocking error.
+ * Both calls share one monotonic request ID, so a slower response for a
+ * previous deployment never replaces the current value. Each source keeps its
+ * last known value when its own call fails, so the open popover can surface a
+ * non-blocking error while the other source still renders. The account source
+ * survives a deployment change, since it does not depend on the selection.
  *
- * The raw response is returned rather than a mapped display model: mapping
+ * The raw responses are returned rather than a mapped display model: mapping
  * needs the caller's translated labels and locale-bound reset formatter, and
  * holding the mapped value in state here would leave it stale after a language
  * change until the next fetch.
  */
-export declare const useDeploymentUsageLimits: (deploymentId: string | undefined, getDeploymentLimits: GetDeploymentLimits) => UseDeploymentUsageLimitsResult;
+export declare const useDeploymentUsageLimits: (deploymentId: string | undefined, api: UsageLimitsApi) => UseDeploymentUsageLimitsResult;
 //# sourceMappingURL=useDeploymentUsageLimits.d.ts.map

package/usage/useDeploymentUsageLimits/useDeploymentUsageLimits.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useDeploymentUsageLimits.d.ts","sourceRoot":"","sources":["../../../src/usage/useDeploymentUsageLimits/useDeploymentUsageLimits.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,+BAA+B,CAAC;AAGjF,sEAAsE;AACtE,MAAM,MAAM,mBAAmB,GAAG,CAChC,YAAY,EAAE,MAAM,KACjB,OAAO,CAAC,2BAA2B,CAAC,CAAC;AAE1C,wDAAwD;AACxD,MAAM,WAAW,8BAA8B;IAC7C,gGAAgG;IAChG,SAAS,EAAE,2BAA2B,GAAG,SAAS,CAAC;IACnD,yCAAyC;IACzC,SAAS,EAAE,OAAO,CAAC;IACnB,mDAAmD;IACnD,QAAQ,EAAE,OAAO,CAAC;IAClB,sEAAsE;IACtE,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB;AAED;;;;;;;;;;;;;;;GAeG;AACH,eAAO,MAAM,wBAAwB,GACnC,cAAc,MAAM,GAAG,SAAS,EAChC,qBAAqB,mBAAmB,KACvC,8BAgEF,CAAC"}
+{"version":3,"file":"useDeploymentUsageLimits.d.ts","sourceRoot":"","sources":["../../../src/usage/useDeploymentUsageLimits/useDeploymentUsageLimits.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,2BAA2B,EAC3B,yBAAyB,EAC1B,MAAM,+BAA+B,CAAC;AAGvC,+FAA+F;AAC/F,MAAM,MAAM,YAAY,GAAG,IAAI,CAC7B,yBAAyB,EACzB,cAAc,GAAG,eAAe,GAAG,gBAAgB,CACpD,CAAC;AAEF;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IACjC,0GAA0G;IAC1G,UAAU,EAAE,2BAA2B,GAAG,SAAS,CAAC;IACpD,2EAA2E;IAC3E,OAAO,EAAE,YAAY,GAAG,SAAS,CAAC;CACnC;AAED,8DAA8D;AAC9D,MAAM,WAAW,cAAc;IAC7B,gFAAgF;IAChF,kBAAkB,EAAE,CAClB,YAAY,EAAE,MAAM,KACjB,OAAO,CAAC,2BAA2B,CAAC,CAAC;IAC1C,kFAAkF;IAClF,eAAe,EAAE,MAAM,OAAO,CAAC,YAAY,CAAC,CAAC;CAC9C;AAED,wDAAwD;AACxD,MAAM,WAAW,8BAA8B;IAC7C,0DAA0D;IAC1D,OAAO,EAAE,kBAAkB,CAAC;IAC5B,iEAAiE;IACjE,SAAS,EAAE,OAAO,CAAC;IACnB,kEAAkE;IAClE,QAAQ,EAAE,OAAO,CAAC;IAClB,4DAA4D;IAC5D,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,eAAO,MAAM,wBAAwB,GACnC,cAAc,MAAM,GAAG,SAAS,EAChC,KAAK,cAAc,KAClB,8BA6EF,CAAC"}

package/usage/useDeploymentUsageLimits/useDeploymentUsageLimits.js

@@ -1,39 +1,34 @@
-import { useCallback as e, useEffect as t, useRef as n, useState as r } from "react";
+import { useCallback as e, useEffect as t, useMemo as n, useRef as r, useState as i } from "react";
 //#region src/usage/useDeploymentUsageLimits/useDeploymentUsageLimits.ts
-var i = (i, a) => {
-	let [o, s] = r(), [c, l] = r(!1), [u, d] = r(!1), [f, p] = r(0), m = n(0), h = n(!1), g = n(void 0), _ = n(a);
-	_.current = a;
-	let v = e(() => {
-		h.current || p((e) => e + 1);
+var a = (a, o) => {
+	let [s, c] = i(), [l, u] = i(), [d, f] = i(!1), [p, m] = i(!1), [h, g] = i(0), _ = r(0), v = r(!1), y = r(void 0), b = r(o);
+	b.current = o;
+	let x = e(() => {
+		v.current || g((e) => e + 1);
 	}, []);
 	return t(() => {
-		let e = g.current !== i;
-		if (g.current = i, !i) {
-			m.current += 1, s(void 0), l(!1), d(!1), h.current = !1;
+		let e = y.current !== a;
+		if (y.current = a, !a) {
+			_.current += 1, c(void 0), f(!1), m(!1), v.current = !1;
 			return;
 		}
-		e && s(void 0);
-		let t = ++m.current, n = !1;
-		return l(!0), d(!1), h.current = !0, (async () => {
-			try {
-				let e = await _.current(i);
-				if (n || t !== m.current) return;
-				s(e);
-			} catch {
-				if (n || t !== m.current) return;
-				d(!0);
-			} finally {
-				!n && t === m.current && (l(!1), h.current = !1);
-			}
+		e && c(void 0);
+		let t = ++_.current, n = !1, r = () => !n && t === _.current;
+		return f(!0), m(!1), v.current = !0, (async () => {
+			let [e, t] = await Promise.allSettled([b.current.getDeploymentUsage(a), b.current.getAccountUsage()]);
+			r() && (e.status === "fulfilled" && c(e.value), t.status === "fulfilled" && u(t.value), m(e.status === "rejected" || t.status === "rejected"), f(!1), v.current = !1);
 		})(), () => {
 			n = !0;
 		};
-	}, [i, f]), {
-		limitsDto: o,
-		isLoading: c,
-		hasError: u,
-		refresh: v
+	}, [a, h]), {
+		sources: n(() => ({
+			deployment: s,
+			account: l
+		}), [s, l]),
+		isLoading: d,
+		hasError: p,
+		refresh: x
 	};
 };
 //#endregion
-export { i as useDeploymentUsageLimits };
+export { a as useDeploymentUsageLimits };

package/usage/useUsageLimitBanner/useUsageLimitBanner.d.ts

@@ -1,6 +1,6 @@
 import { UsageLimitBannerViewProps } from '@epam/ai-dial-conversation-input';
 import { TranslateFn } from '../usage-reset-time';
-import { GetDeploymentLimits } from '../useDeploymentUsageLimits/useDeploymentUsageLimits';
+import { UsageLimitsApi } from '../useDeploymentUsageLimits/useDeploymentUsageLimits';
 /**
  * i18n key strings the consuming app's translation bundle must define for
  * `useUsageLimitBanner` to produce translated strings, in addition to
@@ -29,8 +29,10 @@ export interface UseUsageLimitBannerOptions {
     deploymentId: string | undefined;
     /** Whether the selected deployment is currently generating a response. Defaults to `false`. */
     isGenerationInProgress?: boolean;
-    /** Host-configured call that fetches one deployment's limits. */
-    getDeploymentLimits: GetDeploymentLimits;
+    /** Host-configured call that fetches one deployment's own usage. */
+    getDeploymentUsage: UsageLimitsApi['getDeploymentUsage'];
+    /** Host-configured call that fetches the caller's account usage. */
+    getAccountUsage: UsageLimitsApi['getAccountUsage'];
     /** Used percentage at or above which a not-yet-exhausted limit raises a warning. */
     warningThresholdPercent: number;
     /** Active UI locale, used to format the reset time. */
@@ -47,10 +49,10 @@ export interface UseUsageLimitBannerResult {
 }
 /**
  * Everything the composer usage banner needs except its rendering: fetches the
- * selected deployment's limits, refreshes them after each generation, picks
- * the limit worth reporting, formats its sentences, and tracks tab-lifetime
- * dismissal. Spread `banner` onto `UsageLimitBannerView` from
- * `@epam/ai-dial-conversation-input`.
+ * selected deployment's own usage and the account usage, refreshes both after
+ * each generation, picks the limit worth reporting, formats its sentences, and
+ * tracks tab-lifetime dismissal. Spread `banner` onto `UsageLimitBannerView`
+ * from `@epam/ai-dial-conversation-input`.
  */
-export declare const useUsageLimitBanner: ({ deploymentId, isGenerationInProgress, getDeploymentLimits, warningThresholdPercent, activeLocale, t, }: UseUsageLimitBannerOptions) => UseUsageLimitBannerResult;
+export declare const useUsageLimitBanner: ({ deploymentId, isGenerationInProgress, getDeploymentUsage, getAccountUsage, warningThresholdPercent, activeLocale, t, }: UseUsageLimitBannerOptions) => UseUsageLimitBannerResult;
 //# sourceMappingURL=useUsageLimitBanner.d.ts.map

package/usage/useUsageLimitBanner/useUsageLimitBanner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useUsageLimitBanner.d.ts","sourceRoot":"","sources":["../../../src/usage/useUsageLimitBanner/useUsageLimitBanner.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,yBAAyB,EAC/B,MAAM,kCAAkC,CAAC;AAI1C,OAAO,EAAE,KAAK,WAAW,EAAyB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EACL,KAAK,mBAAmB,EAEzB,MAAM,sDAAsD,CAAC;AAI9D;;;;;GAKG;AACH,eAAO,MAAM,sBAAsB;IACjC,0EAA0E;;IAE1E,2EAA2E;;IAE3E,4EAA4E;;IAE5E,0EAA0E;;IAE1E,2EAA2E;;IAE3E,4EAA4E;;IAE5E,2CAA2C;;CAEnC,CAAC;AAkBX,6CAA6C;AAC7C,MAAM,WAAW,0BAA0B;IACzC,+CAA+C;IAC/C,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,+FAA+F;IAC/F,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,iEAAiE;IACjE,mBAAmB,EAAE,mBAAmB,CAAC;IACzC,oFAAoF;IACpF,uBAAuB,EAAE,MAAM,CAAC;IAChC,uDAAuD;IACvD,YAAY,EAAE,MAAM,CAAC;IACrB,gEAAgE;IAChE,CAAC,EAAE,WAAW,CAAC;CAChB;AAED,gEAAgE;AAChE,MAAM,MAAM,qBAAqB,GAAG,IAAI,CACtC,yBAAyB,EACzB,UAAU,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,GAAG,gBAAgB,CAClE,CAAC;AAEF,mDAAmD;AACnD,MAAM,WAAW,yBAAyB;IACxC,wIAAwI;IACxI,MAAM,EAAE,qBAAqB,GAAG,SAAS,CAAC;CAC3C;AAED;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,GAAI,0GAOjC,0BAA0B,KAAG,yBAiE/B,CAAC"}
+{"version":3,"file":"useUsageLimitBanner.d.ts","sourceRoot":"","sources":["../../../src/usage/useUsageLimitBanner/useUsageLimitBanner.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,yBAAyB,EAC/B,MAAM,kCAAkC,CAAC;AAI1C,OAAO,EAAE,KAAK,WAAW,EAAyB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EACL,KAAK,cAAc,EAEpB,MAAM,sDAAsD,CAAC;AAI9D;;;;;GAKG;AACH,eAAO,MAAM,sBAAsB;IACjC,0EAA0E;;IAE1E,2EAA2E;;IAE3E,4EAA4E;;IAE5E,0EAA0E;;IAE1E,2EAA2E;;IAE3E,4EAA4E;;IAE5E,2CAA2C;;CAEnC,CAAC;AAkBX,6CAA6C;AAC7C,MAAM,WAAW,0BAA0B;IACzC,+CAA+C;IAC/C,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,+FAA+F;IAC/F,sBAAsB,CAAC,EAAE,OAAO,CAAC;IACjC,oEAAoE;IACpE,kBAAkB,EAAE,cAAc,CAAC,oBAAoB,CAAC,CAAC;IACzD,oEAAoE;IACpE,eAAe,EAAE,cAAc,CAAC,iBAAiB,CAAC,CAAC;IACnD,oFAAoF;IACpF,uBAAuB,EAAE,MAAM,CAAC;IAChC,uDAAuD;IACvD,YAAY,EAAE,MAAM,CAAC;IACrB,gEAAgE;IAChE,CAAC,EAAE,WAAW,CAAC;CAChB;AAED,gEAAgE;AAChE,MAAM,MAAM,qBAAqB,GAAG,IAAI,CACtC,yBAAyB,EACzB,UAAU,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,GAAG,gBAAgB,CAClE,CAAC;AAEF,mDAAmD;AACnD,MAAM,WAAW,yBAAyB;IACxC,wIAAwI;IACxI,MAAM,EAAE,qBAAqB,GAAG,SAAS,CAAC;CAC3C;AAED;;;;;;GAMG;AACH,eAAO,MAAM,mBAAmB,GAAI,0HAQjC,0BAA0B,KAAG,yBAqE/B,CAAC"}

package/usage/useUsageLimitBanner/useUsageLimitBanner.js

@@ -26,41 +26,44 @@ var l = {
 		[e.Week]: l.reachedWeek,
 		[e.Month]: l.reachedMonth
 	}
-}, d = ({ deploymentId: e, isGenerationInProgress: c = !1, getDeploymentLimits: d, warningThresholdPercent: f, activeLocale: p, t: m }) => {
-	let { limitsDto: h, isLoading: g, refresh: _ } = r(e, d), { isDismissed: v, dismiss: y } = a();
+}, d = ({ deploymentId: e, isGenerationInProgress: c = !1, getDeploymentUsage: d, getAccountUsage: f, warningThresholdPercent: p, activeLocale: m, t: h }) => {
+	let g = s(() => ({
+		getDeploymentUsage: d,
+		getAccountUsage: f
+	}), [d, f]), { sources: _, isLoading: v, refresh: y } = r(e, g), { isDismissed: b, dismiss: x } = a();
 	i({
 		isGenerationInProgress: c,
-		isLoading: g,
-		refresh: _
+		isLoading: v,
+		refresh: y
 	});
-	let b = s(() => t(h, f), [h, f]), x = s(() => {
-		if (b != null && e) return {
-			dismissalKey: `${e}:${b.period}:${b.severity}`,
-			limitText: m(u[b.severity][b.period], { percent: b.usedPercent }),
-			reset: n(b.resetsAt, b.period, p, m)
+	let S = s(() => t(_, p), [_, p]), C = s(() => {
+		if (S != null && e) return {
+			dismissalKey: `${e}:${S.period}:${S.severity}`,
+			limitText: h(u[S.severity][S.period], { percent: S.usedPercent }),
+			reset: n(S.resetsAt, S.period, m, h)
 		};
 	}, [
-		b,
+		S,
 		e,
-		p,
-		m
-	]), S = x?.dismissalKey, C = o(() => {
-		S != null && y(S);
-	}, [y, S]), w = m(l.closeLabel);
+		m,
+		h
+	]), w = C?.dismissalKey, T = o(() => {
+		w != null && x(w);
+	}, [x, w]), E = h(l.closeLabel);
 	return { banner: s(() => {
-		if (!(b == null || x == null || v(x.dismissalKey))) return {
-			severity: b.severity,
-			limitText: x.limitText,
-			reset: x.reset,
-			onClose: C,
-			closeAriaLabel: w
+		if (!(S == null || C == null || b(C.dismissalKey))) return {
+			severity: S.severity,
+			limitText: C.limitText,
+			reset: C.reset,
+			onClose: T,
+			closeAriaLabel: E
 		};
 	}, [
-		b,
-		x,
-		v,
+		S,
 		C,
-		w
+		b,
+		T,
+		E
 	]) };
 };
 //#endregion