@agent-native/core 0.12.40 → 0.13.0

package/dist/client/AssistantChat.js

@@ -1,5 +1,5 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-import React, { useState, useRef, useEffect, useCallback, useMemo, forwardRef, useImperativeHandle, } from "react";
+import React, { useState, useRef, useEffect, useLayoutEffect, useCallback, useMemo, forwardRef, useImperativeHandle, } from "react";
 import { AssistantRuntimeProvider, useLocalRuntime, useThreadRuntime, useThread, useAui, useComposer, useMessageRuntime, ThreadPrimitive, ComposerPrimitive, MessagePrimitive, } from "@assistant-ui/react";
 import { SimpleImageAttachmentAdapter, CompositeAttachmentAdapter, } from "@assistant-ui/react";
 import { MarkdownTextPrimitive } from "@assistant-ui/react-markdown";
@@ -18,6 +18,7 @@ import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuLabel,
 import { IframeEmbed, parseEmbedBody } from "./IframeEmbed.js";
 import { useDevMode } from "./use-dev-mode.js";
 import { agentNativePath } from "./api-path.js";
+import { getThreadCacheKey } from "./use-chat-threads.js";
 import { BUILDER_SPACE_SETTINGS_URL } from "./error-format.js";
 import { ThumbsFeedback } from "./observability/ThumbsFeedback.js";
 import { TiptapComposer, } from "./composer/TiptapComposer.js";
@@ -1491,7 +1492,22 @@ const AssistantChatInner = forwardRef(function AssistantChatInner({ emptyStateTe
     }, [isRunning, tabId, threadId]);
     // ─── Chat persistence ──────────────────────────────────────────────
     const hasRestoredRef = useRef(false);
-    const [isRestoring, setIsRestoring] = useState(!!threadId && !isNewThread);
+    // Cached thread data from a prior session, read synchronously so existing
+    // chats can paint their messages on first commit instead of after the server
+    // round-trip. The server fetch still runs to refresh with the latest.
+    const [cachedThreadData] = useState(() => {
+        if (typeof window === "undefined")
+            return null;
+        if (!threadId || isNewThread)
+            return null;
+        try {
+            return localStorage.getItem(getThreadCacheKey(threadId));
+        }
+        catch {
+            return null;
+        }
+    });
+    const [isRestoring, setIsRestoring] = useState(!!threadId && !isNewThread && !cachedThreadData);
     const onSaveThreadRef = useRef(onSaveThread);
     onSaveThreadRef.current = onSaveThread;
     const onGenerateTitleRef = useRef(onGenerateTitle);
@@ -1692,6 +1708,23 @@ const AssistantChatInner = forwardRef(function AssistantChatInner({ emptyStateTe
             return false;
         }
     }, [apiUrl, refreshThreadFromServer, startReconnectToRun, threadId]);
+    // Hydrate from the localStorage cache synchronously so the message bubbles
+    // paint on the first commit instead of after the server round-trip. The
+    // server fetch below still runs to refresh with the latest content.
+    const hydratedFromCacheRef = useRef(false);
+    useLayoutEffect(() => {
+        if (hydratedFromCacheRef.current)
+            return;
+        if (!cachedThreadData)
+            return;
+        hydratedFromCacheRef.current = true;
+        try {
+            importThreadData(cachedThreadData, { markTitleGenerated: true });
+        }
+        catch {
+            // Corrupt cache entry — ignore and let the server fetch repopulate.
+        }
+    }, [cachedThreadData, importThreadData]);
     // Restore messages from server on mount (when threadId is set)
     useEffect(() => {
         if (hasRestoredRef.current)
@@ -1706,7 +1739,12 @@ const AssistantChatInner = forwardRef(function AssistantChatInner({ emptyStateTe
                         return;
                     const data = await res.json();
                     if (data.threadData) {
-                        importThreadData(data.threadData, { markTitleGenerated: true });
+                        // Skip the re-import if the server data matches what we already
+                        // hydrated from cache — avoids a second runtime.import that would
+                        // briefly flicker the message list.
+                        if (data.threadData !== cachedThreadData) {
+                            importThreadData(data.threadData, { markTitleGenerated: true });
+                        }
                     }
                     // Also skip title generation if thread already has a title
                     if (data.title) {
@@ -1746,6 +1784,7 @@ const AssistantChatInner = forwardRef(function AssistantChatInner({ emptyStateTe
         threadRuntime,
         importThreadData,
         reconnectActiveRunForThread,
+        cachedThreadData,
     ]);
     // If assistant-ui stops the local runtime while the background server run is
     // still alive, immediately switch into the same reconnect path used after a