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

package/conversation/useConversationStream/useConversationStream.d.ts

@@ -1,138 +1,5 @@
-import { SendCompletionDtoModeEnum } from '@epam/ai-dial-chat-api-client';
-import { Conversation, MessageCustomContent, StreamChunk } from '@epam/ai-dial-chat-shared';
-import { Dispatch, MutableRefObject, SetStateAction } from 'react';
-/** Options accepted by {@link ConversationStreamTransport.streamCompletion}. */
-export interface StreamCompletionOptions {
-    onChunk: (chunk: StreamChunk) => void;
-    onComplete: () => void | Promise<void>;
-    onError: (error: Error) => void;
-    signal: AbortSignal;
-}
-/**
- * Host-owned transport for the DIAL Core completion protocol. Implemented at
- * the app edge — never hardcodes an `/api` path, CSRF handling, or a
- * `server-api` import; the library only depends on this interface's shape.
- */
-export interface ConversationStreamTransport {
-    /** Starts a completion; delivers chunks/completion/error through `options`, aborted via `options.signal`. */
-    streamCompletion(path: string, message: string | undefined, model: string, options: StreamCompletionOptions, customContent?: MessageCustomContent, generationId?: string, mode?: SendCompletionDtoModeEnum, messageIndex?: number, clientChannelId?: string): void;
-    /**
-     * Requests the backend stop an active generation. `content` is the answer text shown
-     * so far; the backend saves it for a generation whose text it does not hold.
-     */
-    stopCompletion(params: {
-        generationId: string;
-        path: string;
-        content?: string;
-    }): Promise<void>;
-    /** Opens a stream of resource-update events for `path`, until aborted via `signal`. */
-    watchConversation(path: string, signal: AbortSignal): Promise<ReadableStream<Uint8Array>>;
-    /**
-     * Attaches to an active generation's live replay: a `snapshot` event
-     * carrying the assistant message as assembled so far, then a `chunk` event
-     * per subsequent delta, then exactly one terminal event (`done`/`error`/
-     * `stopped`), until aborted via `signal`. Rejects when no active
-     * generation exists for `path` — including one that already finished —
-     * so the caller can fall back to `watchConversation`.
-     */
-    attachToGeneration(path: string, signal: AbortSignal): Promise<ReadableStream<Uint8Array>>;
-    /** Reloads the persisted conversation by its full (bucket-qualified) id. */
-    getConversation(conversationId: string, signal?: AbortSignal): Promise<Conversation>;
-    /**
-     * Saves `conversation` at the bucket-stripped `path`, resolving with the saved
-     * conversation. Used only for the single recovery save after the backend's
-     * terminal save was rejected with `401`; without it that failure shows the
-     * persistence warning immediately.
-     */
-    saveConversation?(path: string, conversation: Conversation): Promise<Conversation>;
-}
-/** Host-owned generation lifecycle, backing cross-navigation `AbortController` ownership. */
-export interface ConversationGenerationLifecycle {
-    /** Starts (or replaces) tracking for a generation on `path`, returning its `AbortController`. */
-    startGeneration: (path: string, generationId: string) => AbortController;
-    /** Marks the generation identified by `path`/`generationId` complete, if it is still the active one. */
-    completeGeneration: (path: string, generationId: string) => void;
-}
-/** Optional host-owned client-channel connection, used to nudge tool-signin delivery. */
-export interface ConversationStreamChannel {
-    channelId: string | null;
-    ensureConnected: () => void;
-    /** Resolves with a channel id, waiting for an in-flight subscribe if one isn't established yet, so a completion sent immediately after mount can still carry it. */
-    waitForChannel: (timeoutMs?: number) => Promise<string | null>;
-    /** Notifies the host that a generation just ended (completed or errored), so it can consider disconnecting an idle channel. */
-    notifyGenerationSettled?: () => void;
-}
-/** Optional host-owned overlay generation-lifecycle notifications. Each method is independently optional. */
-export interface ConversationStreamOverlayNotifier {
-    notifyGenerationStart?: () => void;
-    notifyGenerationEnd?: () => void;
-    notifyStopGenerating?: () => void;
-}
-/** Shared mutable channel through which the displayed conversation's state is read/written. */
-export interface ConversationStateAccessor {
-    setConversation: Dispatch<SetStateAction<Conversation | null>>;
-    conversationRef: MutableRefObject<Conversation | null>;
-}
-/** Parameters for {@link useConversationStream}. */
-export interface UseConversationStreamParams {
-    conversationId: string | undefined;
-    state: ConversationStateAccessor;
-    transport: ConversationStreamTransport;
-    generation: ConversationGenerationLifecycle;
-    channel?: ConversationStreamChannel;
-    overlay?: ConversationStreamOverlayNotifier;
-    onStopError?: (error: Error) => void;
-    /**
-     * Message shown on the message bubble when the backend rejects a completion
-     * because this conversation is already generating — the case a second
-     * browser tab of the same session hits. Defaults to
-     * {@link DEFAULT_GENERATION_CONFLICT_MESSAGE}.
-     */
-    generationConflictMessage?: string;
-    /** Host-translated warning shown when the received answer could not be saved. */
-    generationPersistenceErrorMessage?: string;
-    /**
-     * Receives the original error of every failed stream. The hook shows only
-     * host-supplied or upstream-supplied text on the message bubble and hides
-     * transport detail (e.g. `Failed to fetch`), so this is where the host can
-     * still log or report the raw error.
-     */
-    onStreamError?: (error: Error) => void;
-    /**
-     * Coalesces the displayed chunk updates into one `setConversation` per
-     * animation frame. Chunks still reach the per-path buffer immediately, and
-     * any pending update is flushed before completion, error, stop, or a
-     * superseding generation. Defaults to `false` (one update per chunk).
-     */
-    batchChunksPerFrame?: boolean;
-}
-/** Per-start options of {@link UseConversationStreamResult.startStream}. */
-export interface StartStreamOptions {
-    /**
-     * Treats a `GenerationConflictError` as "this turn is already being
-     * generated" — e.g. a page reloaded before that generation saved its start
-     * state — and joins the running generation instead of showing the conflict
-     * message, which remains the fallback. Only for a start that sends no new
-     * user text; a send the user typed must keep reporting the conflict.
-     */
-    resumeOnConflict?: boolean;
-}
-/** Return value of {@link useConversationStream}. */
-export interface UseConversationStreamResult {
-    startStream: (conversationId: string, userContent: string, messageIndex: number, model: string, customContent?: MessageCustomContent, generationId?: string, mode?: SendCompletionDtoModeEnum, options?: StartStreamOptions) => void;
-    handleStop: () => void;
-    resumeIfAwaitingGeneration: (currentConversationId: string, conversation: Conversation) => void;
-    /** Restores the accumulated live assistant message when its conversation is loaded again before completion. */
-    restoreBufferedGeneration: (currentConversationId: string, conversation: Conversation) => Conversation;
-    isStreaming: boolean;
-    canStopStreaming: boolean;
-    /** Whether the displayed conversation's terminal read failed. */
-    hasConversationReloadError: boolean;
-    /** Whether a retry of that read is in flight. */
-    isReloadingConversation: boolean;
-    /** Retries the failed read without saving or regenerating the answer. */
-    retryConversationReload: () => Promise<void>;
-}
+import { UseConversationStreamParams, UseConversationStreamResult } from './conversation-stream.types';
+export type { ConversationGenerationLifecycle, ConversationStateAccessor, ConversationStreamChannel, ConversationStreamOverlayNotifier, ConversationStreamTransport, StartStreamOptions, StreamCompletionOptions, UseConversationStreamParams, UseConversationStreamResult, } from './conversation-stream.types';
 /**
  * Owns completion-streaming state: per-path streaming/stoppable tracking,
  * stale-chunk rejection, cross-navigation live-message buffering,
@@ -141,5 +8,5 @@ export interface UseConversationStreamResult {
  * `transport`/`generation`/`channel`/`overlay` capabilities rather than any
  * app context or `server-api` import.
  */
-export declare const useConversationStream: ({ conversationId, state: { setConversation, conversationRef }, transport, generation: { startGeneration, completeGeneration }, channel, overlay, onStopError, generationConflictMessage, generationPersistenceErrorMessage, onStreamError, batchChunksPerFrame, }: UseConversationStreamParams) => UseConversationStreamResult;
+export declare const useConversationStream: ({ conversationId, state: { setConversation, conversationRef }, transport, generation: { startGeneration, completeGeneration }, channel, overlay, onStopError, generationConflictMessage, generationPersistenceErrorMessage, generationInterruptedMessage, onStreamError, batchChunksPerFrame, }: UseConversationStreamParams) => UseConversationStreamResult;
 //# sourceMappingURL=useConversationStream.d.ts.map

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;AAoCf,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,2BAg4BhC,CAAC"}
+{"version":3,"file":"useConversationStream.d.ts","sourceRoot":"","sources":["../../../src/conversation/useConversationStream/useConversationStream.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAEV,2BAA2B,EAC3B,2BAA2B,EAC5B,MAAM,6BAA6B,CAAC;AAYrC,YAAY,EACV,+BAA+B,EAC/B,yBAAyB,EACzB,yBAAyB,EACzB,iCAAiC,EACjC,2BAA2B,EAC3B,kBAAkB,EAClB,uBAAuB,EACvB,2BAA2B,EAC3B,2BAA2B,GAC5B,MAAM,6BAA6B,CAAC;AAErC;;;;;;;GAOG;AACH,eAAO,MAAM,qBAAqB,GAAI,iSAanC,2BAA2B,KAAG,2BAwRhC,CAAC"}

package/conversation/useConversationStream/useConversationStream.js

@@ -1,336 +1,152 @@
-import { safeDecodeURI as e } from "../../shared/string-utils.js";
-import { DEFAULT_GENERATION_CONFLICT_MESSAGE as t, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as n, GenerationConflictError as r, GenerationPersistenceError as i, StreamInterruptedError as a, StreamUpstreamError as o } from "../create-chat-stream-api.js";
-import { getConversationPath as s } from "./conversation-path.js";
-import { applyChunkToMessages as c } from "./apply-chunk.js";
-import { restoreBufferedMessage as l } from "./buffered-generation.js";
-import { createResumeIfAwaitingGeneration as ee, fetchConversationForRecovery as te, findPendingBackgroundMessageIndex as u, hasGeneratedPayload as ne, isAwaitingGenerationResume as d } from "./generation-resume.js";
-import { createFrameScheduler as re } from "./frame-scheduler.js";
-import { MessageRole as ie, generateUUID as ae } from "@epam/ai-dial-chat-shared";
-import { useCallback as f, useEffect as p, useMemo as oe, useRef as m, useState as h } from "react";
-import { SendCompletionDtoModeEnum as g } from "@epam/ai-dial-chat-api-client";
+import { DEFAULT_GENERATION_CONFLICT_MESSAGE as e, DEFAULT_GENERATION_INTERRUPTED_ERROR_MESSAGE as t, DEFAULT_GENERATION_PERSISTENCE_ERROR_MESSAGE as n } from "../create-chat-stream-api.js";
+import { getConversationPath as r } from "./conversation-path.js";
+import { restoreBufferedMessage as i } from "./buffered-generation.js";
+import { createResumeIfAwaitingGeneration as a, isAwaitingGenerationResume as o } from "./generation-resume.js";
+import { createFrameScheduler as s } from "./frame-scheduler.js";
+import { findResumedBackgroundGenerationId as ee } from "./stream-errors.js";
+import { runStreamSession as c } from "./stream-session.js";
+import { useConversationReloadErrors as l } from "./useConversationReloadErrors.js";
+import { useStopGeneration as te } from "./useStopGeneration.js";
+import { useStreamPathState as ne } from "./useStreamPathState.js";
+import { MessageRole as re } from "@epam/ai-dial-chat-shared";
+import { useCallback as u, useEffect as d, useMemo as ie, useRef as f, useState as ae } from "react";
 //#region src/conversation/useConversationStream/useConversationStream.ts
-var se = 2e4, ce = 401, le = (e, t, n) => e instanceof i ? n : e instanceof r ? t : e instanceof o ? e.message : "", _ = (e) => {
-	if (!e) return;
-	let t = u(e);
-	return t === -1 ? void 0 : e.messages[t].backgroundGeneration?.generationId;
-}, v = ({ conversationId: o, state: { setConversation: v, conversationRef: y }, transport: b, generation: { startGeneration: ue, completeGeneration: x }, channel: S, overlay: C, onStopError: w, generationConflictMessage: de = t, generationPersistenceErrorMessage: T = n, onStreamError: E, batchChunksPerFrame: D = !1 }) => {
-	let O = m(E);
-	p(() => {
-		O.current = E;
-	}, [E]);
-	let fe = m(D);
-	p(() => {
-		fe.current = D;
-	}, [D]);
-	let [k] = h(re), [A, j] = h(() => /* @__PURE__ */ new Set()), [M, N] = h(() => /* @__PURE__ */ new Set()), P = m(/* @__PURE__ */ new Map()), F = m(/* @__PURE__ */ new Set()), I = m(/* @__PURE__ */ new Map()), L = m(/* @__PURE__ */ new Set()), pe = m(/* @__PURE__ */ new Map()), R = m(!0), z = m(/* @__PURE__ */ new Map()), [B, V] = h(z.current), H = m(/* @__PURE__ */ new Set()), [me, U] = h(H.current);
-	p(() => (R.current = !0, () => {
-		R.current = !1;
+var p = ({ conversationId: p, state: { setConversation: m, conversationRef: h }, transport: g, generation: { startGeneration: _, completeGeneration: v }, channel: y, overlay: b, onStopError: oe, generationConflictMessage: x = e, generationPersistenceErrorMessage: S = n, generationInterruptedMessage: C = t, onStreamError: w, batchChunksPerFrame: T = !1 }) => {
+	let E = f(w);
+	d(() => {
+		E.current = w;
+	}, [w]);
+	let D = f(T);
+	d(() => {
+		D.current = T;
+	}, [T]);
+	let [O] = ae(s), k = f(!0);
+	d(() => (k.current = !0, () => {
+		k.current = !1;
 	}), []);
-	let W = f((e) => {
-		if (!z.current.has(e)) return;
-		let t = new Map(z.current);
-		t.delete(e), z.current = t, R.current && V(t);
-	}, []), G = f((e, t, n) => {
-		if (!R.current || !n()) return;
-		let r = new Map(z.current);
-		r.set(e, {
-			retry: t,
-			isCurrent: n
-		}), z.current = r, V(r);
-	}, []), he = f(async () => {
-		if (!o) return;
-		let e = z.current.get(s(o));
-		if (!(!e || !e.isCurrent() || H.current.has(e))) {
-			H.current.add(e), U(new Set(H.current));
-			try {
-				await e.retry();
-			} finally {
-				H.current.delete(e), R.current && U(new Set(H.current));
-			}
-		}
-	}, [o]), K = m(o);
-	p(() => (K.current = o, () => k.cancelAll()), [o, k]);
-	let q = f((e) => R.current && s(K.current ?? "") === e, []), J = f((e) => {
-		W(e), j((t) => {
-			if (t.has(e)) return t;
-			let n = new Set(t);
-			return n.add(e), n;
-		});
-	}, [W]), Y = f((e) => {
-		j((t) => {
-			if (!t.has(e)) return t;
-			let n = new Set(t);
-			return n.delete(e), n;
-		});
-	}, []), X = f((e, t) => {
-		N((n) => {
-			if (n.has(e) === t) return n;
-			let r = new Set(n);
-			return t ? r.add(e) : r.delete(e), r;
-		});
-	}, []), Z = f((e, t) => {
-		P.current.get(e) === t && (P.current.delete(e), X(e, !1));
-	}, [X]), Q = oe(() => ee({
-		transport: b,
-		setConversation: v,
-		conversationRef: y,
-		resumingPathsRef: F,
-		bufferedGenerationsRef: I,
-		addStreamingPath: J,
-		removeStreamingPath: Y,
-		isPathDisplayed: q,
-		generationPersistenceErrorMessage: T,
-		frameScheduler: D ? k : void 0,
-		stoppedGenerationIdsRef: L,
-		onReloadError: G,
-		onReloadSuccess: W
+	let { reloadFailuresRef: A, reloadFailures: j, retrying: M, clearReloadError: N, reportReloadError: P, retryConversationReload: se } = l({
+		conversationId: p,
+		mountedRef: k
+	}), { streamingPaths: ce, streamingPathsRef: le, stoppablePaths: ue, activeGenerationIdsRef: F, resumedGenerationIdsRef: I, resumingPathsRef: L, bufferedGenerationsRef: R, stoppedGenerationIdsRef: z, stopRequestedIdsRef: B, latestGenerationIdsRef: V, isPathDisplayed: H, addStreamingPath: U, removeStreamingPath: W, setPathStoppable: G, releaseActiveGeneration: K, rememberResumedGenerationId: q, forgetResumedGenerationId: J } = ne({
+		conversationId: p,
+		frameScheduler: O,
+		mountedRef: k,
+		clearReloadError: N
+	}), Y = ie(() => a({
+		transport: g,
+		setConversation: m,
+		conversationRef: h,
+		resumingPathsRef: L,
+		bufferedGenerationsRef: R,
+		addStreamingPath: U,
+		removeStreamingPath: W,
+		isPathDisplayed: H,
+		generationPersistenceErrorMessage: S,
+		generationInterruptedMessage: C,
+		frameScheduler: T ? O : void 0,
+		stoppedGenerationIdsRef: z,
+		rememberResumedGenerationId: q,
+		forgetResumedGenerationId: J,
+		onReloadError: P,
+		onReloadSuccess: N
 	}), [
-		J,
-		Y,
-		q,
-		b,
-		T,
-		D,
-		k,
-		G,
-		W
-	]), ge = f((t, n, o, ee, re, f, p = g.Append, oe = {}) => {
-		let m = f ?? ae(), h = s(t);
-		k.flush(h), F.current.delete(h), P.current.set(h, m), pe.current.set(h, m), X(h, !0);
-		let _;
-		p === g.Regenerate ? _ = o : p === g.Edit && (_ = o - 1);
-		let w = () => pe.current.get(h) !== m, E = ue(h, m), D = y.current?.messages[o];
-		D ? I.current.set(h, {
-			generationId: m,
-			messageIndex: o,
-			message: D
-		}) : I.current.delete(h), J(h), C?.notifyGenerationStart?.(), S?.ensureConnected();
-		let A = () => {
-			w() || Y(h), Z(h, m), x(h, m), S?.notifyGenerationSettled?.();
-		}, j = () => {
-			if (w()) return;
-			let e = I.current.get(h);
-			!e || e.generationId !== m || (e.message = {
-				...e.message,
-				streamErrorMessage: T
-			}, O.current?.(new i()), q(h) && v((t) => {
-				if (!t || w() || !q(h)) return t;
-				let n = l(t, e);
-				return y.current = n, n;
-			}));
-		}, M = I.current.get(h), N = () => R.current && !w() && I.current.get(h) === M, z = async () => {
-			if (N()) {
-				if (!q(h)) {
-					M && I.current.delete(h);
-					return;
-				}
-				try {
-					let n = await b.getConversation(e(t));
-					if (!N()) return;
-					if (W(h), !q(h)) {
-						M && I.current.delete(h);
-						return;
-					}
-					if (u(n) !== -1) {
-						Q(t, n, { seedMessage: M?.message });
-						return;
-					}
-					if (M && M.messageIndex === n.messages.length - 1 && d(n) && ne(M.message)) {
-						j();
-						return;
-					}
-					M && I.current.delete(h), v(n), y.current = n;
-				} catch {
-					N() && G(h, z, N);
-				}
-			}
-		}, B = {
-			signal: E.signal,
-			onChunk: (e) => {
-				if (w()) return;
-				let t = I.current.get(h);
-				if (t?.generationId === m) {
-					let n = c([t.message], 0, e);
-					n && (t.message = n[0]);
-				}
-				if (q(h)) {
-					if (fe.current && t?.generationId === m) {
-						k.schedule(h, () => {
-							let e = I.current.get(h);
-							if (w() || !q(h) || e?.generationId !== m) return;
-							let t = { ...e };
-							v((e) => {
-								if (!e) return e;
-								let n = l(e, t);
-								return y.current = n, n;
-							});
-						});
-						return;
-					}
-					v((t) => {
-						if (!t) return t;
-						let n = I.current.get(h), r;
-						if (n?.generationId === m) r = l(t, n);
-						else {
-							let n = c(t.messages, o, e);
-							if (!n) return t;
-							r = {
-								...t,
-								messages: n
-							};
-						}
-						return y.current = r, r;
-					});
-				}
-			},
-			onComplete: async () => {
-				k.flush(h), A(), L.current.has(m) ? L.current.delete(m) : w() || C?.notifyGenerationEnd?.(), await z();
-			},
-			onError: (e) => {
-				if (k.flush(h), O.current?.(e), e instanceof a && !w()) {
-					he(e);
-					return;
-				}
-				if (e instanceof r && oe.resumeOnConflict && !w()) {
-					K(e);
-					return;
-				}
-				V(e);
-			}
-		}, V = (e) => {
-			let t = I.current.get(h), n = t?.generationId === m ? t : void 0;
-			if (n && !(e instanceof i) && I.current.delete(h), A(), n && b.saveConversation && e instanceof i && e.status === ce) {
-				me(e, n, b.saveConversation.bind(b));
-				return;
-			}
-			H(e, n);
-		}, H = (e, t) => {
-			if (t && e instanceof i && (t.message = {
-				...t.message,
-				streamErrorMessage: T
-			}), w() || !q(h)) return;
-			let n = le(e, de, T);
-			v((e) => {
-				if (!e || w() || !q(h)) return e;
-				let r = t?.generationId === m ? l(e, t) : e, i = {
-					...r,
-					messages: r.messages.map((e, t) => t === o ? {
-						...e,
-						streamErrorMessage: n
-					} : e)
-				};
-				return y.current = i, i;
-			});
-		}, me = async (n, r, i) => {
-			let a = () => R.current && !w() && I.current.get(h) === r;
-			try {
-				let o = await b.getConversation(e(t));
-				if (!a()) return;
-				if (r.messageIndex !== o.messages.length - 1 || !d(o)) {
-					H(n, r);
-					return;
-				}
-				let s = await i(h, l(o, r));
-				if (!a() || (I.current.delete(h), !q(h))) return;
-				v(s), y.current = s;
-			} catch {
-				a() && H(n, r);
-			}
-		}, U = () => {
-			A(), L.current.has(m) ? L.current.delete(m) : w() || C?.notifyGenerationEnd?.();
-		}, he = async (n) => {
-			F.current.add(h), ge(await te(() => b.getConversation(e(t)), w), n, !1);
-		}, K = async (n) => {
-			P.current.get(h) === m && X(h, !1), F.current.add(h), ge(await te(() => b.getConversation(e(t)), w, (e) => e.messages.length === o && e.messages.at(-1)?.role === ie.User), n, !0);
-		}, ge = (e, n, r) => {
-			if (w()) {
-				A();
-				return;
-			}
-			let i = e?.messages[e.messages.length - 1], a = e != null && e.messages.length - 1 === o && i?.role === ie.Assistant, s = r && e != null && u(e) !== -1;
-			if (!e || !a && !s) {
-				F.current.delete(h), V(n);
-				return;
-			}
-			let c = I.current.get(h), l = c?.generationId === m ? c : void 0;
-			if (d(e)) {
-				Q(t, e, {
-					seedMessage: a ? l?.message : void 0,
-					skipDedupe: !0,
-					onSettled: U
-				});
-				return;
-			}
-			F.current.delete(h), l && I.current.delete(h), q(h) && (v(e), y.current = e), U();
-		};
-		(async () => {
-			try {
-				let e = S?.channelId ?? await S?.waitForChannel(se) ?? void 0, t = L.current.has(m);
-				if (E.signal.aborted || w() || t) {
-					I.current.get(h)?.generationId === m && I.current.delete(h), w() || Y(h), Z(h, m), x(h, m), S?.notifyGenerationSettled?.(), t ? L.current.delete(m) : w() || C?.notifyGenerationEnd?.();
-					return;
-				}
-				b.streamCompletion(h, n, ee, B, re, m, p, _, e);
-			} catch (e) {
-				B.onError(e instanceof Error ? e : Error(String(e)));
-			}
-		})();
-	}, [
-		ue,
-		x,
-		J,
-		Y,
-		X,
-		Z,
-		q,
-		S?.channelId,
-		S?.ensureConnected,
-		S?.waitForChannel,
-		S?.notifyGenerationSettled,
+		U,
+		W,
+		H,
+		g,
+		S,
 		C,
-		b,
-		Q,
-		de,
 		T,
-		k,
+		O,
+		q,
+		J,
+		P,
+		N
+	]), de = u((e, t, n, r, i, a, o, s) => c({
+		transport: g,
+		channel: y,
+		overlay: b,
+		startGeneration: _,
+		completeGeneration: v,
+		setConversation: m,
+		conversationRef: h,
+		generationConflictMessage: x,
+		generationPersistenceErrorMessage: S,
+		onStreamErrorRef: E,
+		batchChunksPerFrameRef: D,
+		mountedRef: k,
+		frameScheduler: O,
+		resumeGeneration: Y,
+		activeGenerationIdsRef: F,
+		latestGenerationIdsRef: V,
+		resumingPathsRef: L,
+		bufferedGenerationsRef: R,
+		stoppedGenerationIdsRef: z,
+		addStreamingPath: U,
+		removeStreamingPath: W,
+		setPathStoppable: G,
+		releaseActiveGeneration: K,
+		isPathDisplayed: H,
+		reportReloadError: P,
+		clearReloadError: N
+	}, e, t, n, r, i, a, o, s), [
+		_,
+		v,
+		U,
+		W,
 		G,
-		W
-	]), _e = f((e, t) => {
-		let n = s(e), r = I.current.get(n);
-		return z.current.get(n)?.isCurrent() && r && t.messages.length - 1 >= r.messageIndex && t.messages[r.messageIndex].role === ie.Assistant && !d(t) ? (W(n), I.current.delete(n), t) : r ? l(t, r) : t;
-	}, [W]), ve = f(() => {
-		if (!o) return;
-		let e = s(o), t = P.current.get(e) ?? _(y.current);
-		if (!t) return;
-		L.current.add(t), k.flush(e), C?.notifyStopGenerating?.();
-		let n = I.current.get(e)?.message.content;
-		b.stopCompletion({
-			generationId: t,
-			path: e,
-			content: n
-		}).catch((e) => {
-			let t = e instanceof Error ? e : Error(String(e));
-			w?.(t);
-		});
+		K,
+		H,
+		y?.channelId,
+		y?.ensureConnected,
+		y?.waitForChannel,
+		y?.notifyGenerationSettled,
+		b,
+		g,
+		Y,
+		x,
+		S,
+		O,
+		P,
+		N
+	]), fe = u((e, t) => {
+		let n = r(e), a = R.current.get(n);
+		return A.current.get(n)?.isCurrent() && a && t.messages.length - 1 >= a.messageIndex && t.messages[a.messageIndex].role === re.Assistant && !o(t) ? (N(n), R.current.delete(n), t) : a ? i(t, a) : t;
 	}, [
-		o,
-		y,
-		k,
-		w,
-		C,
-		b
-	]), ye = f((e, t) => Q(e, t), [Q]), be = o != null && A.has(s(o)), $ = o == null ? null : s(o), xe = $ != null && (M.has($) || be && _(y.current) != null), Se = $ ? B.get($) : void 0, Ce = Se?.isCurrent() ?? !1;
+		R,
+		N,
+		A
+	]), pe = te({
+		conversationId: p,
+		conversationRef: h,
+		transport: g,
+		overlay: b,
+		onStopError: oe,
+		frameScheduler: O,
+		mountedRef: k,
+		activeGenerationIdsRef: F,
+		resumedGenerationIdsRef: I,
+		bufferedGenerationsRef: R,
+		stoppedGenerationIdsRef: z,
+		stopRequestedIdsRef: B,
+		streamingPathsRef: le
+	}), me = u((e, t) => {
+		F.current.has(r(e)) || Y(e, t, {
+			trackAttachGenerationId: !0,
+			shouldMarkOrphanedAsInterrupted: !0
+		});
+	}, [F, Y]), X = p != null && ce.has(r(p)), Z = p == null ? null : r(p), he = Z != null && (ue.has(Z) || X && ee(h.current) != null), Q = Z ? j.get(Z) : void 0, $ = Q?.isCurrent() ?? !1;
 	return {
-		hasConversationReloadError: Ce,
-		isReloadingConversation: !!Se && Ce && me.has(Se),
-		retryConversationReload: he,
-		startStream: ge,
-		handleStop: ve,
-		resumeIfAwaitingGeneration: ye,
-		restoreBufferedGeneration: _e,
-		isStreaming: be,
-		canStopStreaming: xe
+		hasConversationReloadError: $,
+		isReloadingConversation: !!Q && $ && M.has(Q),
+		retryConversationReload: se,
+		startStream: de,
+		handleStop: pe,
+		resumeIfAwaitingGeneration: me,
+		restoreBufferedGeneration: fe,
+		isStreaming: X,
+		canStopStreaming: he
 	};
 };
 //#endregion
-export { v as useConversationStream };
+export { p as useConversationStream };

package/conversation/useConversationStream/useStopGeneration.d.ts

@@ -0,0 +1,29 @@
+import { MutableRefObject } from 'react';
+import { ConversationStateAccessor, ConversationStreamOverlayNotifier, ConversationStreamTransport } from './conversation-stream.types';
+import { FrameScheduler } from './frame-scheduler';
+import { useStreamPathState } from './useStreamPathState';
+type PathState = ReturnType<typeof useStreamPathState>;
+interface UseStopGenerationParams {
+    conversationId: string | undefined;
+    conversationRef: ConversationStateAccessor['conversationRef'];
+    transport: ConversationStreamTransport;
+    overlay: ConversationStreamOverlayNotifier | undefined;
+    onStopError: ((error: Error) => void) | undefined;
+    frameScheduler: FrameScheduler;
+    mountedRef: MutableRefObject<boolean>;
+    activeGenerationIdsRef: PathState['activeGenerationIdsRef'];
+    resumedGenerationIdsRef: PathState['resumedGenerationIdsRef'];
+    bufferedGenerationsRef: PathState['bufferedGenerationsRef'];
+    stoppedGenerationIdsRef: PathState['stoppedGenerationIdsRef'];
+    stopRequestedIdsRef: PathState['stopRequestedIdsRef'];
+    streamingPathsRef: PathState['streamingPathsRef'];
+}
+/**
+ * Stop for the displayed conversation's generation: resolves its id (local,
+ * then a pending background generation, then a resumed one), flushes pending
+ * display writes, and asks the backend to stop it. A `404` within a short
+ * grace window is the generation ending on its own and is not reported.
+ */
+export declare const useStopGeneration: ({ conversationId, conversationRef, transport, overlay, onStopError, frameScheduler, mountedRef, activeGenerationIdsRef, resumedGenerationIdsRef, bufferedGenerationsRef, stoppedGenerationIdsRef, stopRequestedIdsRef, streamingPathsRef, }: UseStopGenerationParams) => () => void;
+export {};
+//# sourceMappingURL=useStopGeneration.d.ts.map