@gendive/chatllm 0.12.1 → 0.12.3

package/dist/react/index.js

@@ -1210,6 +1210,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;
@@ -1281,6 +1312,10 @@ var useChatUI = (options) => {
   const [deepResearchProgress, setDeepResearchProgress] = (0, import_react4.useState)(
     null
   );
+  const sessionsRef = (0, import_react4.useRef)(sessions);
+  (0, import_react4.useEffect)(() => {
+    sessionsRef.current = sessions;
+  }, [sessions]);
   const abortControllerRef = (0, import_react4.useRef)(null);
   const skipNextPollParsingRef = (0, import_react4.useRef)(false);
   const skipNextSkillParsingRef = (0, import_react4.useRef)(false);
@@ -1610,7 +1645,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 필드 사용
@@ -1620,18 +1655,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);
       }
@@ -1642,11 +1713,12 @@ ${newConversation}
       setCurrentSessionId(id);
       setSelectedModel(session.model);
     }
-  }, [sessions, useExternalStorage, onLoadSession, onError]);
+  }, [sessions, useExternalStorage, onLoadSession, onError, storageKey, initialModel, models]);
   const deleteSession = (0, import_react4.useCallback)(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) {
@@ -2137,20 +2209,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;
@@ -6086,7 +6161,7 @@ var MessageBubble = ({
           display: "flex",
           flexDirection: "column",
           alignItems: "flex-end",
-          padding: "8px 24px"
+          padding: "4px 24px"
         },
         onMouseEnter: () => setShowActions(true),
         onMouseLeave: () => setShowActions(false),
@@ -6152,7 +6227,7 @@ var MessageBubble = ({
         display: "flex",
         flexDirection: "column",
         alignItems: "flex-start",
-        padding: "8px 24px"
+        padding: "4px 24px"
       },
       onMouseEnter: () => setShowActions(true),
       onMouseLeave: () => setShowActions(false),
@@ -6164,25 +6239,25 @@ var MessageBubble = ({
             className: "chatllm-sheet",
             style: {
               width: "100%",
-              padding: "24px",
+              padding: "16px",
               display: "flex",
-              gap: "16px"
+              gap: "12px"
             },
             children: [
               /* @__PURE__ */ (0, import_jsx_runtime10.jsx)("div", { style: { flexShrink: 0 }, children: /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(
                 "div",
                 {
                   style: {
-                    width: "40px",
-                    height: "40px",
-                    borderRadius: "12px",
+                    width: "32px",
+                    height: "32px",
+                    borderRadius: "10px",
                     backgroundColor: "var(--chatllm-primary-light)",
                     display: "flex",
                     alignItems: "center",
                     justifyContent: "center",
                     color: "var(--chatllm-primary)"
                   },
-                  children: /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(IconSvg, { name: "magic-line", size: 24 })
+                  children: /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(IconSvg, { name: "magic-line", size: 20 })
                 }
               ) }),
               /* @__PURE__ */ (0, import_jsx_runtime10.jsxs)("div", { style: { flex: 1, minWidth: 0 }, children: [
@@ -6193,7 +6268,7 @@ var MessageBubble = ({
                       display: "flex",
                       alignItems: "center",
                       gap: "8px",
-                      marginBottom: "16px"
+                      marginBottom: "8px"
                     },
                     children: /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(
                       "span",
@@ -6443,7 +6518,7 @@ var MessageBubble = ({
               display: "flex",
               gap: "2px",
               marginTop: "4px",
-              marginLeft: "56px",
+              marginLeft: "44px",
               height: "24px",
               opacity: showActions ? 1 : 0,
               transition: "opacity 0.15s ease"