@gendive/chatllm 0.17.23 → 0.17.24

package/dist/react/index.mjs

@@ -2735,6 +2735,26 @@ ${attachmentContext}
         } catch {
         }
       }
+      const saveMessagesOnEarlyReturn = () => {
+        if (!useExternalStorage || !capturedSessionId) return;
+        queueMicrotask(() => {
+          const latestSession = sessionsRef.current.find((s) => s.id === capturedSessionId);
+          if (!latestSession) return;
+          const latestMessages = latestSession.messages;
+          const assistantMsg = [...latestMessages].reverse().find((m) => m.role === "assistant");
+          const userMsg = latestMessages.find((m) => m.role === "user" && m.content === finalContent);
+          const assistantContent = assistantMsg?.content || "";
+          if (assistantContent && onSaveMessagesRef.current) {
+            onSaveMessagesRef.current(capturedSessionId, [
+              { role: "user", message: finalContent, ...userMsg?.contentParts && { contentParts: userMsg.contentParts } },
+              { role: "assistant", message: assistantContent, ...assistantMsg?.contentParts && { contentParts: assistantMsg.contentParts } }
+            ]).catch((e) => console.error("[useChatUI] Failed to save messages:", e));
+          }
+          if (latestSession.messages.length > 0) {
+            writeSessionCache(storageKey, latestSession);
+          }
+        });
+      };
       if (!shouldSkipSkillParsing) {
         const assistantContent = accumulatedContent;
         const { skillCall: detectedSkill, cleanContent: skillCleanContent } = parseSkillCallFromContent(assistantContent);
@@ -2838,6 +2858,7 @@ ${attachmentContext}
                 })
               );
               if (resultType === "image" || resultType === "file") {
+                saveMessagesOnEarlyReturn();
                 setIsLoading(false);
                 abortControllerRef.current = null;
                 return;
@@ -2848,11 +2869,13 @@ ${attachmentContext}
                 shouldContinue = decision === "continue";
               }
               if (!shouldContinue) {
+                saveMessagesOnEarlyReturn();
                 setIsLoading(false);
                 abortControllerRef.current = null;
                 return;
               }
               skipNextSkillParsingRef.current = true;
+              saveMessagesOnEarlyReturn();
               const feedbackPrompt = resultType === "error" ? `\uB3C4\uAD6C "${toolName}" \uC2E4\uD589 \uC911 \uC624\uB958 \uBC1C\uC0DD: ${result.content}
 
 \uC0AC\uC6A9\uC790\uC5D0\uAC8C \uC624\uB958\uB97C \uC548\uB0B4\uD574\uC8FC\uC138\uC694. skill_use \uD0DC\uADF8\uB294 \uC0AC\uC6A9\uD558\uC9C0 \uB9C8\uC138\uC694.` : `\uB3C4\uAD6C "${toolName}" \uACB0\uACFC:
@@ -2888,6 +2911,7 @@ ${result.content}
               })
             );
             if (streamedReport) {
+              saveMessagesOnEarlyReturn();
               setIsLoading(false);
               abortControllerRef.current = null;
               return;
@@ -2898,11 +2922,13 @@ ${result.content}
               shouldContinueSkill = decision === "continue";
             }
             if (!shouldContinueSkill) {
+              saveMessagesOnEarlyReturn();
               setIsLoading(false);
               abortControllerRef.current = null;
               return;
             }
             skipNextSkillParsingRef.current = true;
+            saveMessagesOnEarlyReturn();
             const resultPrompt = `\uC2A4\uD0AC "${detectedSkill.name}" \uC2E4\uD589 \uACB0\uACFC:
 
 ${result.content}
@@ -6439,6 +6465,7 @@ var MarkdownRenderer = ({
     if (hasUnfinishedPoll) {
       processedContent += "\n\xA7POLL_LOADING\xA7";
     }
+    processedContent = processedContent.replace(/<skill_use[^>]*>[\s\S]*?<\/skill_use>/gi, "");
     processedContent = processedContent.replace(UNCLOSED_SKILL_TAG_REGEX, "");
     const codeBlocks = [];
     processedContent = processedContent.replace(CODE_BLOCK_REGEX, (match, lang, code) => {