@constructor-io/constructorio-ui-asa 1.7.2 → 1.8.0

package/lib/cjs/components/Chat/AiMessage.js

@@ -12,9 +12,8 @@ function AiMessage({ message, componentOverrides, translations }) {
     const hasGroups = !!((_a = message.groups) === null || _a === void 0 ? void 0 : _a.length);
     // On error, fall back to the (translatable) error message when the agent
     // returned no partial text of its own.
-    const text = isError
-        ? message.text || (0, translate_1.default)('CioAsa.error.message', translations)
-        : message.text || '';
+    const errorKey = message.interrupted ? 'CioAsa.error.interrupted' : 'CioAsa.error.message';
+    const text = isError ? message.text || (0, translate_1.default)(errorKey, translations) : message.text || '';
     const hasText = text.length > 0;
     const loaderRenderProps = { translations };
     const textRenderProps = { text };

package/lib/cjs/components/Chat/Chat.js

@@ -3,7 +3,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const tslib_1 = require("tslib");
 const react_1 = tslib_1.__importStar(require("react"));
 const useAsaResults_1 = tslib_1.__importDefault(require("../../hooks/useAsaResults"));
+const useCioAsaContext_1 = require("../../hooks/useCioAsaContext");
 const useFocusTrap_1 = tslib_1.__importDefault(require("../../hooks/useFocusTrap"));
+const useLatest_1 = tslib_1.__importDefault(require("../../hooks/useLatest"));
+const productNormalizer_1 = require("../../utils/productNormalizer");
 const translate_1 = tslib_1.__importDefault(require("../../utils/translate"));
 const ChatHeader_1 = tslib_1.__importDefault(require("./ChatHeader"));
 const WelcomeScreen_1 = tslib_1.__importDefault(require("./WelcomeScreen"));
@@ -26,20 +29,43 @@ function getAnnouncement(messages, translations) {
     }
     return '';
 }
-const Chat = (0, react_1.forwardRef)(({ onClose, className, initialSuggestions, termsText, onProductClick, onViewMore, onAddToCart, aspectRatio, currency, normalizeItem, componentOverrides, translations, initialThreadId, showStopButton = false, }, ref) => {
-    const { messages, sendMessage, isStreaming, abort, clearHistory } = (0, useAsaResults_1.default)({
+const Chat = (0, react_1.forwardRef)(({ onClose, className, initialSuggestions, termsText, onProductClick, onViewMore, onAddToCart, aspectRatio, currency, normalizeItem, componentOverrides, translations, initialThreadId, onThreadsChange, showStopButton = false, }, ref) => {
+    var _a;
+    const { messages, sendMessage, isStreaming, canAbort, abort, clearHistory, isHydrating, threads, activeThreadId, newThread, switchThread, } = (0, useAsaResults_1.default)({
         initialThreadId,
+        normalizeItem: (item) => (normalizeItem !== null && normalizeItem !== void 0 ? normalizeItem : productNormalizer_1.normalizeItemToProduct)(item, {
+            saleBadgeText: (0, translate_1.default)('CioAsa.results.saleBadge', translations),
+        }),
     });
     const chatViewRef = (0, react_1.useRef)(null);
     const containerRef = (0, react_1.useRef)(null);
     const isWelcome = messages.length === 0;
+    // While another thread loads, keep the chat frame (header, close button) instead of a blank body.
+    const lastViewRef = (0, react_1.useRef)(null);
+    if (!isHydrating)
+        lastViewRef.current = isWelcome ? 'welcome' : 'chat';
+    const showWelcome = !isHydrating && isWelcome;
+    const showChat = !isHydrating && !isWelcome;
+    const showChatFrame = showChat || (isHydrating && lastViewRef.current === 'chat');
     const isModal = typeof onClose === 'function';
     const announcement = getAnnouncement(messages, translations);
-    (0, react_1.useImperativeHandle)(ref, () => ({
+    (0, react_1.useImperativeHandle)(ref, () => ({ abort, clearHistory, newThread, switchThread }), [
         abort,
         clearHistory,
-    }));
+        newThread,
+        switchThread,
+    ]);
+    const hasPersistence = Boolean((_a = (0, useCioAsaContext_1.useCioAsaContext)()) === null || _a === void 0 ? void 0 : _a.persistence);
+    const onThreadsChangeRef = (0, useLatest_1.default)(onThreadsChange);
     (0, react_1.useEffect)(() => {
+        var _a;
+        if (!hasPersistence || isHydrating)
+            return;
+        (_a = onThreadsChangeRef.current) === null || _a === void 0 ? void 0 : _a.call(onThreadsChangeRef, threads, activeThreadId);
+    }, [threads, activeThreadId, isHydrating, hasPersistence, onThreadsChangeRef]);
+    (0, react_1.useEffect)(() => {
+        if (isHydrating)
+            return;
         const root = isWelcome ? containerRef.current : chatViewRef.current;
         const input = root === null || root === void 0 ? void 0 : root.querySelector('input');
         if (input) {
@@ -52,15 +78,17 @@ const Chat = (0, react_1.forwardRef)(({ onClose, className, initialSuggestions,
                 input.focus();
             }
         }
-    }, [isWelcome]);
+    }, [isWelcome, isHydrating]);
     (0, useFocusTrap_1.default)(containerRef, { onEscape: onClose, trapFocus: isModal });
     return (react_1.default.createElement("div", { className: ['cio-asa', 'cio-asa-chat', className].filter(Boolean).join(' '), ref: containerRef, role: 'dialog', "aria-modal": isModal ? 'true' : undefined, "aria-label": (0, translate_1.default)(isWelcome ? 'CioAsa.welcome.title' : 'CioAsa.header.title', translations) },
         react_1.default.createElement("div", { className: 'cio-sr-only', role: 'status' }, announcement),
-        react_1.default.createElement("div", { className: 'cio-asa-chat-body' }, isWelcome ? (react_1.default.createElement("div", { className: 'cio-asa-chat-view cio-asa-chat-view--welcome' },
-            react_1.default.createElement(WelcomeScreen_1.default, { suggestions: initialSuggestions, onSend: sendMessage, onClose: onClose, termsText: termsText, translations: translations, componentOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.welcomeScreen }))) : (react_1.default.createElement("div", { className: 'cio-asa-chat-view cio-asa-chat-view--chat', ref: chatViewRef },
-            react_1.default.createElement(ChatHeader_1.default, { onClose: onClose, translations: translations, componentOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.header }),
-            react_1.default.createElement(ChatMessageList_1.default, { messages: messages, onProductClick: onProductClick, onViewMore: onViewMore, onAddToCart: onAddToCart, aspectRatio: aspectRatio, currency: currency, normalizeItem: normalizeItem, addToCartText: (0, translate_1.default)('CioAsa.results.addToCart', translations), viewMoreText: (0, translate_1.default)('CioAsa.results.viewMore', translations), aiMessageOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.aiMessage, userMessageOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.userMessage, resultsBlockOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.resultsBlock, translations: translations, onSend: sendMessage, isStreaming: isStreaming }),
-            react_1.default.createElement(ChatInput_1.default, { onSubmit: sendMessage, isDisabled: isStreaming, isStreaming: isStreaming, onAbort: abort, showStopButton: showStopButton, translations: translations, componentOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.input }))))));
+        react_1.default.createElement("div", { className: 'cio-asa-chat-body' },
+            showWelcome && (react_1.default.createElement("div", { className: 'cio-asa-chat-view cio-asa-chat-view--welcome' },
+                react_1.default.createElement(WelcomeScreen_1.default, { suggestions: initialSuggestions, onSend: sendMessage, onClose: onClose, termsText: termsText, translations: translations, componentOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.welcomeScreen }))),
+            showChatFrame && (react_1.default.createElement("div", { className: 'cio-asa-chat-view cio-asa-chat-view--chat', ref: chatViewRef },
+                react_1.default.createElement(ChatHeader_1.default, { onClose: onClose, translations: translations, componentOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.header }),
+                react_1.default.createElement(ChatMessageList_1.default, { messages: messages, onProductClick: onProductClick, onViewMore: onViewMore, onAddToCart: onAddToCart, aspectRatio: aspectRatio, currency: currency, normalizeItem: normalizeItem, addToCartText: (0, translate_1.default)('CioAsa.results.addToCart', translations), viewMoreText: (0, translate_1.default)('CioAsa.results.viewMore', translations), aiMessageOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.aiMessage, userMessageOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.userMessage, resultsBlockOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.resultsBlock, translations: translations, onSend: sendMessage, isStreaming: isStreaming }),
+                react_1.default.createElement(ChatInput_1.default, { onSubmit: sendMessage, isDisabled: isStreaming || isHydrating, isStreaming: isStreaming, onAbort: canAbort ? abort : undefined, showStopButton: showStopButton, translations: translations, componentOverrides: componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.input }))))));
 });
 Chat.displayName = 'Chat';
 exports.default = Chat;

package/lib/cjs/components/CioAsa/CioAsa.js

@@ -15,8 +15,12 @@ const Chat_1 = tslib_1.__importDefault(require("../Chat/Chat"));
 const CioAsa = react_1.default.forwardRef((props, ref) => {
     const { 
     // Provider configuration — everything else belongs to Chat.
-    apiKey, cioClient, testCells, staticRequestConfigs, formatters, urlHelpers, callbacks, section } = props, chatProps = tslib_1.__rest(props, ["apiKey", "cioClient", "testCells", "staticRequestConfigs", "formatters", "urlHelpers", "callbacks", "section"]);
-    return (react_1.default.createElement(CioAsaProvider_1.default, { apiKey: apiKey, cioClient: cioClient, testCells: testCells, staticRequestConfigs: staticRequestConfigs, formatters: formatters, urlHelpers: urlHelpers, callbacks: callbacks, section: section },
+    apiKey, cioClient, testCells, staticRequestConfigs, formatters, urlHelpers, callbacks, section, persistConversation, userId = null } = props, chatProps = tslib_1.__rest(props, ["apiKey", "cioClient", "testCells", "staticRequestConfigs", "formatters", "urlHelpers", "callbacks", "section", "persistConversation", "userId"]);
+    // Split so TypeScript can match the pair against the provider's props.
+    const persistenceProps = persistConversation
+        ? { persistConversation, userId }
+        : { userId };
+    return (react_1.default.createElement(CioAsaProvider_1.default, Object.assign({ apiKey: apiKey, cioClient: cioClient, testCells: testCells, staticRequestConfigs: staticRequestConfigs, formatters: formatters, urlHelpers: urlHelpers, callbacks: callbacks, section: section }, persistenceProps),
         react_1.default.createElement(Chat_1.default, Object.assign({}, chatProps, { ref: ref }))));
 });
 CioAsa.displayName = 'CioAsa';

package/lib/cjs/components/CioAsaProvider/CioAsaProvider.js

@@ -6,15 +6,42 @@ const useCioClient_1 = tslib_1.__importDefault(require("../../hooks/useCioClient
 const useCioAsaContext_1 = require("../../hooks/useCioAsaContext");
 const defaultFormatters = tslib_1.__importStar(require("../../utils/formatters"));
 const defaultUrlHelpers = tslib_1.__importStar(require("../../utils/urlHelpers"));
+const clientOptions_1 = require("../../utils/clientOptions");
+const localStoragePersistence_1 = require("../../utils/localStoragePersistence");
 function CioAsaProvider(props) {
-    const { apiKey, formatters, urlHelpers, staticRequestConfigs = { domain: 'chatbot' }, cioClient: customCioClient, testCells, callbacks, section = 'Products', children, } = props;
+    var _a;
+    const { apiKey, formatters, urlHelpers, staticRequestConfigs = { domain: 'chatbot' }, cioClient: customCioClient, testCells, callbacks, section = 'Products', persistConversation: persistOptions, userId: userIdProp, children, } = props;
+    // Read field by field: an inline options object is a new reference on every render.
+    const persistenceEnabled = Boolean(persistOptions === null || persistOptions === void 0 ? void 0 : persistOptions.enabled);
     const [cioClientOptions, setCioClientOptions] = (0, react_1.useState)({});
+    const initialUserId = (0, react_1.useRef)(userIdProp).current;
+    const clientInit = (0, react_1.useMemo)(() => initialUserId === undefined
+        ? cioClientOptions
+        : Object.assign(Object.assign({}, cioClientOptions), { userId: (0, localStoragePersistence_1.shopperId)(initialUserId) }), [cioClientOptions, initialUserId]);
     const cioClient = (0, useCioClient_1.default)({
         apiKey,
         cioClient: customCioClient,
-        cioClientOptions,
+        cioClientOptions: clientInit,
         testCells,
     });
+    const userId = (0, localStoragePersistence_1.shopperId)(userIdProp);
+    const syncUserId = persistenceEnabled || userIdProp !== undefined;
+    (0, react_1.useEffect)(() => {
+        if (customCioClient || !cioClient || !syncUserId)
+            return;
+        cioClient.setClientOptions({ userId });
+    }, [cioClient, customCioClient, syncUserId, userId]);
+    const resolvedApiKey = apiKey !== null && apiKey !== void 0 ? apiKey : (_a = (0, clientOptions_1.readClientOptions)(cioClient)) === null || _a === void 0 ? void 0 : _a.apiKey;
+    const { domain } = staticRequestConfigs;
+    const persistence = (0, react_1.useMemo)(() => {
+        if (!persistenceEnabled || resolvedApiKey === undefined)
+            return undefined;
+        return (0, localStoragePersistence_1.sharedLocalStoragePersistence)((0, localStoragePersistence_1.persistenceNamespace)({ apiKey: resolvedApiKey, domain, userId }), (0, localStoragePersistence_1.storageAreaFor)(userId));
+    }, [persistenceEnabled, resolvedApiKey, domain, userId]);
+    const persistenceScope = persistence && (userId === undefined ? 'guest' : 'user');
+    const persistenceIndex = persistence && resolvedApiKey !== undefined
+        ? (0, localStoragePersistence_1.persistenceNamespace)({ apiKey: resolvedApiKey, domain })
+        : undefined;
     const contextValue = (0, react_1.useMemo)(() => ({
         cioClient,
         cioClientOptions,
@@ -24,7 +51,21 @@ function CioAsaProvider(props) {
         urlHelpers: Object.assign(Object.assign({}, defaultUrlHelpers), urlHelpers),
         callbacks,
         section,
-    }), [cioClient, cioClientOptions, formatters, urlHelpers, staticRequestConfigs, callbacks, section]);
+        persistence,
+        persistenceScope,
+        persistenceIndex,
+    }), [
+        cioClient,
+        cioClientOptions,
+        formatters,
+        urlHelpers,
+        staticRequestConfigs,
+        callbacks,
+        section,
+        persistence,
+        persistenceScope,
+        persistenceIndex,
+    ]);
     return (react_1.default.createElement(useCioAsaContext_1.AsaContext.Provider, { value: contextValue }, typeof children === 'function' ? children(contextValue) : children));
 }
 exports.default = CioAsaProvider;

package/lib/cjs/components/ResultsBlock/ResultsGroup.js

@@ -20,9 +20,9 @@ function NextButton() {
  * `searchResultId` are known.
  */
 function ResultsGroup({ groupData, label, slidesToShow, gap, showTitle, viewMoreText, addToCartText, saleBadgeText, currency, intent, intentResultId, normalizeItem, onProductClick, onAddToCart, onViewMore, componentOverrides, tracking, callbacks, }) {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e;
     const { searchResultId } = groupData;
-    const products = groupData.searchResults.map((item) => normalizeItem(item, { saleBadgeText }));
+    const products = (_a = groupData.products) !== null && _a !== void 0 ? _a : groupData.searchResults.map((item) => normalizeItem(item, { saleBadgeText }));
     const canTrack = !!intent && !!searchResultId;
     const { ref: viewRef } = (0, useViewportTracking_1.default)({
         enabled: canTrack,
@@ -83,7 +83,7 @@ function ResultsGroup({ groupData, label, slidesToShow, gap, showTitle, viewMore
         addToCartText,
         onProductClick: handleProductClick,
         onAddToCart,
-        componentOverrides: (_b = (_a = componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.carousel) === null || _a === void 0 ? void 0 : _a.item) === null || _b === void 0 ? void 0 : _b.productCard,
+        componentOverrides: (_c = (_b = componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.carousel) === null || _b === void 0 ? void 0 : _b.item) === null || _c === void 0 ? void 0 : _c.productCard,
     });
     const titleRenderProps = { label };
     const viewMoreRenderProps = {
@@ -95,12 +95,12 @@ function ResultsGroup({ groupData, label, slidesToShow, gap, showTitle, viewMore
                 reactNode: renderProductCard,
             } }) });
     return (react_1.default.createElement("div", { className: 'cio-asa-results-group', ref: viewRef },
-        showTitle && label && (react_1.default.createElement(constructorio_ui_components_1.RenderPropsWrapper, { override: (_c = componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.groupTitle) === null || _c === void 0 ? void 0 : _c.reactNode, props: titleRenderProps },
+        showTitle && label && (react_1.default.createElement(constructorio_ui_components_1.RenderPropsWrapper, { override: (_d = componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.groupTitle) === null || _d === void 0 ? void 0 : _d.reactNode, props: titleRenderProps },
             react_1.default.createElement("h3", { className: 'cio-asa-results-group__label' }, label))),
         react_1.default.createElement(constructorio_ui_components_1.Carousel, { items: products, loop: false, responsive: {
                 0: { gap, slidesToShow },
             }, componentOverrides: carouselOverrides }),
-        onViewMore && (react_1.default.createElement(constructorio_ui_components_1.RenderPropsWrapper, { override: (_d = componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.viewMore) === null || _d === void 0 ? void 0 : _d.reactNode, props: viewMoreRenderProps },
+        onViewMore && (react_1.default.createElement(constructorio_ui_components_1.RenderPropsWrapper, { override: (_e = componentOverrides === null || componentOverrides === void 0 ? void 0 : componentOverrides.viewMore) === null || _e === void 0 ? void 0 : _e.reactNode, props: viewMoreRenderProps },
             react_1.default.createElement("button", { type: 'button', className: 'cio-asa-results-group__view-more', onClick: handleViewMore },
                 react_1.default.createElement("span", { className: 'cio-asa-results-group__view-more-text' }, viewMoreText),
                 react_1.default.createElement(icons_1.ArrowRightIcon, null))))));

package/lib/cjs/hooks/agentStream.js

@@ -0,0 +1,113 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.startTurn = exports.readAgentStream = void 0;
+const tslib_1 = require("tslib");
+const asaStreamHandlers_1 = require("./asaStreamHandlers");
+/** Drains an agent stream into the assistant message identified by `assistantId`. */
+function readAgentStream(stream, assistantId, setMessages, callbacks) {
+    const reader = stream.getReader();
+    let cancelled = false;
+    let pendingGroup = null;
+    let intentResultId;
+    let loadStartFired = false;
+    let groupCount = 0;
+    const patchAssistant = (patch) => {
+        (0, asaStreamHandlers_1.updateMessageById)(setMessages, assistantId, (m) => (Object.assign(Object.assign({}, m), patch)));
+    };
+    const fireLoadStart = () => {
+        if (loadStartFired)
+            return;
+        loadStartFired = true;
+        callbacks.onLoadStart(intentResultId);
+    };
+    const done = (() => tslib_1.__awaiter(this, void 0, void 0, function* () {
+        var _a, _b, _c, _d;
+        try {
+            while (!cancelled) {
+                // eslint-disable-next-line no-await-in-loop
+                const res = yield reader.read();
+                if (cancelled)
+                    break;
+                if (res.done) {
+                    (0, asaStreamHandlers_1.handleStreamEnd)(assistantId, setMessages);
+                    callbacks.onFinish({ searchResultCount: groupCount, intentResultId });
+                    break;
+                }
+                const { type, data } = res.value;
+                if (!intentResultId && (data === null || data === void 0 ? void 0 : data.intent_result_id)) {
+                    intentResultId = data.intent_result_id;
+                    patchAssistant({ intentResultId });
+                }
+                if (type === 'start') {
+                    if (data === null || data === void 0 ? void 0 : data.thread_id)
+                        patchAssistant({ threadId: data.thread_id });
+                    callbacks.onStart(data === null || data === void 0 ? void 0 : data.thread_id);
+                    fireLoadStart();
+                }
+                else if (type === 'group') {
+                    pendingGroup = {
+                        display_name: (_b = (_a = data === null || data === void 0 ? void 0 : data.display_name) !== null && _a !== void 0 ? _a : data === null || data === void 0 ? void 0 : data.group) !== null && _b !== void 0 ? _b : '',
+                        value: (_d = (_c = data === null || data === void 0 ? void 0 : data.value) !== null && _c !== void 0 ? _c : data === null || data === void 0 ? void 0 : data.group) !== null && _d !== void 0 ? _d : '',
+                    };
+                }
+                else if (type === 'search_result') {
+                    fireLoadStart();
+                    groupCount += 1;
+                    pendingGroup = (0, asaStreamHandlers_1.handleSearchResult)(data, pendingGroup, assistantId, setMessages);
+                }
+                else if (type === 'message') {
+                    fireLoadStart();
+                    (0, asaStreamHandlers_1.handleMessage)(data, assistantId, setMessages);
+                }
+                else if (type === 'follow_up_refinement') {
+                    fireLoadStart();
+                    (0, asaStreamHandlers_1.handleFollowUpRefinement)(data, assistantId, setMessages);
+                }
+                else if (type === 'server_error') {
+                    (0, asaStreamHandlers_1.handleServerError)(assistantId, setMessages);
+                    break;
+                }
+            }
+        }
+        catch (_e) {
+            if (!cancelled)
+                (0, asaStreamHandlers_1.handleStreamError)(assistantId, setMessages);
+        }
+        finally {
+            // Cancelling an errored stream rejects with the stored error; nothing here needs it.
+            if (!cancelled)
+                reader.cancel().catch(() => { });
+        }
+    }))();
+    return {
+        assistantId,
+        done,
+        cancel: () => {
+            cancelled = true;
+            reader.cancel().catch(() => { });
+        },
+    };
+}
+exports.readAgentStream = readAgentStream;
+/** Streams into `assistant`, reporting to the owner `createOwner` returns for the new turn. */
+function startTurn(stream, assistant, createOwner, callbacks) {
+    const turn = { assistant };
+    turn.owner = createOwner(turn);
+    const handle = readAgentStream(stream, assistant.id, (action) => {
+        if (typeof action === 'function')
+            [turn.assistant] = action([turn.assistant]);
+        turn.owner.setMessages(action);
+    }, {
+        onLoadStart: callbacks.onLoadStart,
+        onFinish: (result) => callbacks.onFinish(Object.assign(Object.assign({}, result), { threadId: turn.serverThreadId })),
+        onStart: (threadId) => {
+            if (threadId)
+                turn.serverThreadId = threadId;
+            turn.owner.onStart(threadId);
+        },
+    });
+    Object.assign(turn, handle);
+    handle.done.then(() => turn.owner.onDone());
+    return turn;
+}
+exports.startTurn = startTurn;

package/lib/cjs/hooks/asaStreamHandlers.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.handleStreamError = exports.handleStreamEnd = exports.handleServerError = exports.handleFollowUpRefinement = exports.handleMessage = exports.handleSearchResult = void 0;
+exports.handleStreamError = exports.handleStreamEnd = exports.handleServerError = exports.handleFollowUpRefinement = exports.handleMessage = exports.handleSearchResult = exports.updateMessageById = void 0;
 /**
  * Internal A/B configuration echoed back on the request. Stripped before the request reaches
  * consumers: it is unstable (the backend omits it on some pods within a single response) and
@@ -14,6 +14,7 @@ function omitInternalRequestFields(request) {
 function updateMessageById(setMessages, id, patch) {
     setMessages((prev) => prev.map((msg) => (msg.id === id ? patch(msg) : msg)));
 }
+exports.updateMessageById = updateMessageById;
 function handleSearchResult(data, pendingGroup, assistantId, setMessages) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
     const baseGroup = pendingGroup !== null && pendingGroup !== void 0 ? pendingGroup : {

package/lib/cjs/hooks/persistence/backgroundTurns.js

@@ -0,0 +1,121 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.subscribeBackgroundTurns = exports.moveBackgroundTurns = exports.resumeBackgroundTurn = exports.continueInBackground = exports.BackgroundTurn = void 0;
+const chatSession_1 = require("../../utils/chatSession");
+const chatThreads_1 = require("../../utils/chatThreads");
+const turns = new Set();
+const listeners = new Set();
+const notify = () => listeners.forEach((listener) => listener());
+/** An answer that went on after its conversation left the screen, until it is stored or shown again. */
+class BackgroundTurn {
+    constructor(turn, params) {
+        this.streaming = true;
+        /** Whether the latest snapshot is known to be stored. */
+        this.saved = false;
+        /** Every id the conversation was stored under, so a list read before a re-key still finds it. */
+        this.ids = new Set();
+        this.turn = turn;
+        this.store = params.store;
+        this.session = params.session;
+        this.messages = params.messages;
+        this.writes = params.after;
+        if (this.session.storageThreadId)
+            this.ids.add(this.session.storageThreadId);
+        this.turn.owner = {
+            setMessages: (action) => {
+                this.messages = typeof action === 'function' ? action(this.messages) : action;
+            },
+            onStart: (threadId) => {
+                if (threadId)
+                    this.session.serverThreadId = threadId;
+                this.save().then(notify);
+            },
+            onDone: () => {
+                this.streaming = false;
+                this.messages = (0, chatThreads_1.normalizeHydratedMessages)(this.messages);
+                this.save().then(() => {
+                    if (this.saved)
+                        turns.delete(this);
+                    notify();
+                });
+            },
+        };
+    }
+    save() {
+        const { snapshot, staleIds } = (0, chatSession_1.prepareSnapshot)(this.session, this.messages);
+        const { store } = this;
+        this.ids.add(snapshot.threadId);
+        this.saved = false;
+        this.writes = this.writes
+            .then(() => (0, chatThreads_1.saveThreadAndRetireStale)(store, snapshot, staleIds))
+            .then(({ saved, leftover }) => {
+            if (leftover.length > 0) {
+                this.session.orphanIds = Array.from(new Set([...this.session.orphanIds, ...leftover]));
+            }
+            if (this.session.lastSyncedAt === snapshot.updatedAt)
+                this.saved = saved;
+        })
+            .catch(() => { });
+        return this.writes;
+    }
+    /** Later writes go into `store`. */
+    moveTo(store) {
+        this.store = store;
+    }
+    /** Stores the conversation settled, without awaiting: the page is going away. */
+    flush() {
+        if (!this.streaming && this.saved)
+            return;
+        const { snapshot, staleIds } = (0, chatSession_1.prepareSnapshot)(this.session, (0, chatThreads_1.normalizeHydratedMessages)(this.messages));
+        if (this.store.saveThreadSync)
+            this.store.saveThreadSync(snapshot, staleIds);
+        else
+            this.store.saveThread(snapshot).catch(() => { });
+    }
+}
+exports.BackgroundTurn = BackgroundTurn;
+let listeningForPagehide = false;
+/**
+ * Keeps `turn` streaming with no chat showing it. It is saved into `store` at once, on `start`
+ * and when done, each write queued behind `after`, and is dropped from here once the last one landed.
+ */
+function continueInBackground(turn, params) {
+    const bg = new BackgroundTurn(turn, params);
+    turns.add(bg);
+    if (!listeningForPagehide && typeof window !== 'undefined') {
+        listeningForPagehide = true;
+        window.addEventListener('pagehide', () => turns.forEach((t) => t.flush()));
+    }
+    bg.save().then(notify);
+}
+exports.continueInBackground = continueInBackground;
+/**
+ * Takes back the background turn stored in `store` under `threadId` for a chat to show, handing an
+ * unfinished one to `createOwner`; `null` if there is none.
+ */
+function resumeBackgroundTurn(store, threadId, createOwner) {
+    const bg = Array.from(turns).find((t) => t.store === store && t.ids.has(threadId));
+    if (!bg)
+        return null;
+    turns.delete(bg);
+    if (bg.streaming)
+        bg.turn.owner = createOwner(bg.turn);
+    return bg;
+}
+exports.resumeBackgroundTurn = resumeBackgroundTurn;
+/** Points the background turns of `from` at `to`; resolves once their writes into `from` settled. */
+function moveBackgroundTurns(from, to) {
+    const moving = Array.from(turns).filter((bg) => bg.store === from);
+    const settled = Promise.all(moving.map((bg) => bg.writes)).then(() => { });
+    moving.forEach((bg) => bg.moveTo(to));
+    return settled;
+}
+exports.moveBackgroundTurns = moveBackgroundTurns;
+/** Called whenever a background turn is stored or finishes. */
+function subscribeBackgroundTurns(listener) {
+    listeners.add(listener);
+    return () => {
+        listeners.delete(listener);
+    };
+}
+exports.subscribeBackgroundTurns = subscribeBackgroundTurns;