@gendive/chatllm 0.12.0 → 0.12.2

package/dist/react/index.js

@@ -512,7 +512,10 @@ function useInfoExtraction(options) {
               if (data === "[DONE]") continue;
               try {
                 const parsed = JSON.parse(data);
-                if (parsed.content) fullResponse += parsed.content;
+                {
+                  const chunk = parsed.content ?? parsed.text ?? "";
+                  if (chunk) fullResponse += chunk;
+                }
               } catch {
               }
             }
@@ -672,7 +675,7 @@ var callLLMStream = async (prompt, opts, onChunk) => {
         if (data === "[DONE]") continue;
         try {
           const parsed = JSON.parse(data);
-          const content = parsed.choices?.[0]?.delta?.content || parsed.delta?.content || parsed.message?.content || "";
+          const content = parsed.choices?.[0]?.delta?.content || parsed.delta?.content || parsed.message?.content || parsed.content || parsed.text || "";
           if (content) {
             fullContent += content;
             onChunk(content);
@@ -1207,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;
@@ -1278,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);
@@ -1489,7 +1527,10 @@ ${conversationText}
             if (data === "[DONE]") continue;
             try {
               const parsed = JSON.parse(data);
-              if (parsed.content) summary += parsed.content;
+              {
+                const chunk = parsed.content ?? parsed.text ?? "";
+                if (chunk) summary += chunk;
+              }
             } catch {
             }
           }
@@ -1545,7 +1586,10 @@ ${newConversation}
               if (data === "[DONE]") continue;
               try {
                 const parsed = JSON.parse(data);
-                if (parsed.content) summary += parsed.content;
+                {
+                  const chunk = parsed.content ?? parsed.text ?? "";
+                  if (chunk) summary += chunk;
+                }
               } catch {
               }
             }
@@ -1601,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 필드 사용
@@ -1611,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);
       }
@@ -1633,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) {
@@ -1945,7 +2026,7 @@ ${currentContextSummary}` },
           }
           try {
             const parsed = JSON.parse(data);
-            const content2 = parsed.message?.content || parsed.content || "";
+            const content2 = parsed.message?.content || parsed.content || parsed.text || "";
             const thinking = parsed.message?.thinking || "";
             if (content2 || thinking) {
               if (content2) accumulatedContent += content2;
@@ -1985,7 +2066,7 @@ ${currentContextSummary}` },
       if (buffer.trim()) {
         try {
           const parsed = JSON.parse(buffer);
-          const content2 = parsed.message?.content || parsed.content;
+          const content2 = parsed.message?.content || parsed.content || parsed.text;
           if (content2) {
             accumulatedContent += content2;
             setSessions(
@@ -2128,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;
@@ -2354,7 +2438,7 @@ ${formattedParts.join("\n")}
           }
           try {
             const parsed = JSON.parse(data);
-            const content = parsed.message?.content || parsed.content || "";
+            const content = parsed.message?.content || parsed.content || parsed.text || "";
             const thinking = parsed.message?.thinking || "";
             if (content || thinking) {
               setSessions(
@@ -2393,7 +2477,7 @@ ${formattedParts.join("\n")}
       if (buffer.trim()) {
         try {
           const parsed = JSON.parse(buffer);
-          const content = parsed.message?.content || parsed.content || "";
+          const content = parsed.message?.content || parsed.content || parsed.text || "";
           const thinking = parsed.message?.thinking || "";
           if (content || thinking) {
             setSessions(
@@ -2497,7 +2581,10 @@ ${currentSession.contextSummary}` },
                 if (data === "[DONE]") continue;
                 try {
                   const parsed = JSON.parse(data);
-                  if (parsed.content) responseContent += parsed.content;
+                  {
+                    const chunk = parsed.content ?? parsed.text ?? "";
+                    if (chunk) responseContent += chunk;
+                  }
                 } catch {
                 }
               }
@@ -2536,7 +2623,10 @@ ${currentSession.contextSummary}` },
               if (data === "[DONE]") continue;
               try {
                 const parsed = JSON.parse(data);
-                if (parsed.content) responseContent += parsed.content;
+                {
+                  const chunk = parsed.content ?? parsed.text ?? "";
+                  if (chunk) responseContent += chunk;
+                }
               } catch {
               }
             }