@gendive/chatllm 0.12.1 → 0.12.2

package/dist/react/index.mjs

@@ -1154,6 +1154,37 @@ var formatPollResponse = (question, selectedOptions, otherText) => {
   return `${selectedLabels.join(", ")}\uC744(\uB97C) \uC120\uD0DD\uD588\uC2B5\uB2C8\uB2E4. \uC774 \uC120\uD0DD\uC744 \uBC14\uD0D5\uC73C\uB85C \uC9C4\uD589\uD574\uC8FC\uC138\uC694.`;
 };
 
+// src/react/utils/sessionCache.ts
+var buildCacheKey = (storageKey, sessionId) => `${storageKey}_cache_${sessionId}`;
+var writeSessionCache = (storageKey, session) => {
+  if (typeof window === "undefined") return;
+  try {
+    const key = buildCacheKey(storageKey, session.id);
+    localStorage.setItem(key, JSON.stringify(session));
+  } catch (error) {
+    console.warn("[sessionCache] Failed to write cache:", error);
+  }
+};
+var readSessionCache = (storageKey, sessionId) => {
+  if (typeof window === "undefined") return null;
+  try {
+    const key = buildCacheKey(storageKey, sessionId);
+    const data = localStorage.getItem(key);
+    if (!data) return null;
+    return JSON.parse(data);
+  } catch {
+    return null;
+  }
+};
+var removeSessionCache = (storageKey, sessionId) => {
+  if (typeof window === "undefined") return;
+  try {
+    const key = buildCacheKey(storageKey, sessionId);
+    localStorage.removeItem(key);
+  } catch {
+  }
+};
+
 // src/react/hooks/useChatUI.ts
 var DEFAULT_STORAGE_KEY = "chatllm_sessions";
 var DEFAULT_COMPRESSION_THRESHOLD = 20;
@@ -1225,6 +1256,10 @@ var useChatUI = (options) => {
   const [deepResearchProgress, setDeepResearchProgress] = useState4(
     null
   );
+  const sessionsRef = useRef3(sessions);
+  useEffect2(() => {
+    sessionsRef.current = sessions;
+  }, [sessions]);
   const abortControllerRef = useRef3(null);
   const skipNextPollParsingRef = useRef3(false);
   const skipNextSkillParsingRef = useRef3(false);
@@ -1554,7 +1589,7 @@ ${newConversation}
       setIsSessionLoading(true);
       try {
         const sessionDetail = await onLoadSession(id);
-        const loadedMessages = sessionDetail.messages.map((m, idx) => ({
+        let loadedMessages = sessionDetail.messages.map((m, idx) => ({
           id: m.id || generateId2("msg"),
           role: typeof m.role === "string" ? m.role.toLowerCase() : m.role,
           // API는 message 필드, 내부는 content 필드 사용
@@ -1564,18 +1599,54 @@ ${newConversation}
           alternatives: m.alternatives,
           sources: m.sources
         }));
+        let resolvedTitle = sessionDetail.title;
+        if (loadedMessages.length === 0) {
+          const cached = readSessionCache(storageKey, id);
+          if (cached && cached.messages.length > 0) {
+            console.warn("[useChatUI] Server returned empty messages, using localStorage cache");
+            loadedMessages = cached.messages;
+            if (!resolvedTitle && cached.title) {
+              resolvedTitle = cached.title;
+            }
+          }
+        }
         setSessions(
           (prev) => prev.map(
-            (s) => s.id === id ? { ...s, title: sessionDetail.title, messages: loadedMessages, updatedAt: Date.now() } : s
+            (s) => s.id === id ? { ...s, title: resolvedTitle, messages: loadedMessages, updatedAt: Date.now() } : s
           )
         );
         setCurrentSessionId(id);
+        if (loadedMessages.length > 0) {
+          const existingSession2 = sessions.find((s) => s.id === id);
+          writeSessionCache(storageKey, {
+            id,
+            title: resolvedTitle,
+            messages: loadedMessages,
+            model: existingSession2?.model || initialModel || models[0]?.id || "",
+            createdAt: existingSession2?.createdAt || Date.now(),
+            updatedAt: Date.now()
+          });
+        }
         const existingSession = sessions.find((s) => s.id === id);
         if (existingSession) {
           setSelectedModel(existingSession.model);
         }
       } catch (error) {
         onError?.(error instanceof Error ? error : new Error("Failed to load session"));
+        const cached = readSessionCache(storageKey, id);
+        if (cached && cached.messages.length > 0) {
+          console.warn("[useChatUI] onLoadSession failed, using localStorage cache");
+          setSessions(
+            (prev) => prev.map(
+              (s) => s.id === id ? { ...s, title: cached.title || s.title, messages: cached.messages, updatedAt: Date.now() } : s
+            )
+          );
+          setCurrentSessionId(id);
+          const existingSession = sessions.find((s) => s.id === id);
+          if (existingSession) {
+            setSelectedModel(existingSession.model);
+          }
+        }
       } finally {
         setIsSessionLoading(false);
       }
@@ -1586,11 +1657,12 @@ ${newConversation}
       setCurrentSessionId(id);
       setSelectedModel(session.model);
     }
-  }, [sessions, useExternalStorage, onLoadSession, onError]);
+  }, [sessions, useExternalStorage, onLoadSession, onError, storageKey, initialModel, models]);
   const deleteSession = useCallback4(async (id) => {
     if (useExternalStorage && onDeleteSessionCallback) {
       try {
         await onDeleteSessionCallback(id);
+        removeSessionCache(storageKey, id);
         setSessions((prev) => {
           const filtered = prev.filter((s) => s.id !== id);
           if (currentSessionId === id) {
@@ -2081,20 +2153,23 @@ ${result.content}
           };
         })
       );
-      if (useExternalStorage && onSaveMessages && capturedSessionId) {
-        const updatedSession = sessions.find((s) => s.id === capturedSessionId);
-        const assistantContent = updatedSession?.messages.find(
-          (m) => m.id === assistantMessageId
-        )?.content || "";
-        if (assistantContent) {
+      if (useExternalStorage && capturedSessionId) {
+        const assistantContentForSave = accumulatedContent;
+        if (assistantContentForSave && onSaveMessages) {
           const messagesToSave = [
             { role: "USER", message: finalContent },
-            { role: "ASSISTANT", message: assistantContent }
+            { role: "ASSISTANT", message: assistantContentForSave }
           ];
           onSaveMessages(capturedSessionId, messagesToSave).catch((saveError) => {
             console.error("[useChatUI] Failed to save messages:", saveError);
           });
         }
+        queueMicrotask(() => {
+          const sessionToCache = sessionsRef.current.find((s) => s.id === capturedSessionId);
+          if (sessionToCache && sessionToCache.messages.length > 0) {
+            writeSessionCache(storageKey, sessionToCache);
+          }
+        });
       }
       if (enableAutoExtraction && globalMemory) {
         const currentMsgCount = existingMessages.length + 2;