@gendive/chatllm 0.10.1 → 0.10.2

package/dist/react/index.js

@@ -1030,7 +1030,12 @@ ${finalContent}`;
     setQuotedText(null);
     setSelectedAction(null);
     const capturedSessionId = sessionId;
-    const isFirstMessage = !sessions.find((s) => s.id === capturedSessionId)?.messages.length;
+    const currentSession2 = sessions.find((s) => s.id === capturedSessionId);
+    const existingMessages = currentSession2?.messages || [];
+    const isFirstMessage = !existingMessages.length;
+    const contextSummary = currentSession2?.compressionState?.contextSummary || currentSession2?.contextSummary;
+    const summaryAfterIndex = currentSession2?.compressionState?.summaryAfterIndex || currentSession2?.summaryAfterIndex || 0;
+    let compressionCount = currentSession2?.compressionState?.compressionCount || 0;
     setSessions(
       (prev) => prev.map((s) => {
         if (s.id === capturedSessionId) {
@@ -1061,29 +1066,25 @@ ${finalContent}`;
     setIsLoading(true);
     abortControllerRef.current = new AbortController();
     try {
-      const session = sessions.find((s) => s.id === capturedSessionId);
-      const existingMessages = session?.messages || [];
       let messagesToSend = [...existingMessages, userMessage];
       const recompressionThreshold = DEFAULT_RECOMPRESSION_THRESHOLD;
       const tokenLimit = DEFAULT_TOKEN_LIMIT;
-      let contextSummary = session?.compressionState?.contextSummary || session?.contextSummary;
-      const summaryAfterIndex = session?.compressionState?.summaryAfterIndex || session?.summaryAfterIndex || 0;
-      let compressionCount = session?.compressionState?.compressionCount || 0;
+      let currentContextSummary = contextSummary;
       const newMessageCount = messagesToSend.length - summaryAfterIndex;
       const estimatedTokensCount = estimateTokens(messagesToSend);
-      const needsInitialCompression = !contextSummary && messagesToSend.length > contextCompressionThreshold;
-      const needsRecompression = contextSummary && (newMessageCount > recompressionThreshold || estimatedTokensCount > tokenLimit * 0.7);
+      const needsInitialCompression = !currentContextSummary && messagesToSend.length > contextCompressionThreshold;
+      const needsRecompression = currentContextSummary && (newMessageCount > recompressionThreshold || estimatedTokensCount > tokenLimit * 0.7);
       if (needsInitialCompression || needsRecompression) {
         const toCompress = messagesToSend.slice(0, -keepRecentMessages);
         let summary;
-        if (needsRecompression && contextSummary) {
+        if (needsRecompression && currentContextSummary) {
           const newMessages = toCompress.slice(summaryAfterIndex);
-          summary = await incrementalCompressContext(contextSummary, newMessages, selectedModel);
+          summary = await incrementalCompressContext(currentContextSummary, newMessages, selectedModel);
         } else {
           summary = await compressContext(toCompress, selectedModel);
         }
         if (summary) {
-          contextSummary = summary;
+          currentContextSummary = summary;
           compressionCount += 1;
           setSessions(
             (prev) => prev.map(
@@ -1111,11 +1112,11 @@ ${finalContent}`;
         }
       }
       let chatMessages;
-      if (contextSummary) {
+      if (currentContextSummary) {
         const recentMessages = messagesToSend.slice(-keepRecentMessages);
         chatMessages = [
           { role: "system", content: `[\uC774\uC804 \uB300\uD654 \uC694\uC57D]
-${contextSummary}` },
+${currentContextSummary}` },
           ...recentMessages.map((m) => ({ role: m.role, content: m.content }))
         ];
       } else {