@gendive/chatllm 0.15.2 → 0.15.3

package/dist/react/index.js

@@ -2129,6 +2129,7 @@ ${currentContextSummary}` },
       const decoder = new TextDecoder();
       let buffer = "";
       let accumulatedContent = "";
+      let skillTagDetected = false;
       while (true) {
         const { done, value } = await reader.read();
         if (done) break;
@@ -2148,6 +2149,12 @@ ${currentContextSummary}` },
             const thinking = parsed.message?.thinking || "";
             if (content2 || thinking) {
               if (content2) accumulatedContent += content2;
+              if (!skipNextSkillParsingRef.current && accumulatedContent.includes("</skill_use>")) {
+                const endIdx = accumulatedContent.indexOf("</skill_use>");
+                accumulatedContent = accumulatedContent.substring(0, endIdx + "</skill_use>".length);
+                skillTagDetected = true;
+              }
+              const displayContent = skillTagDetected ? accumulatedContent : null;
               setSessions(
                 (prev) => prev.map((s) => {
                   if (s.id === capturedSessionId) {
@@ -2155,6 +2162,9 @@ ${currentContextSummary}` },
                       ...s,
                       messages: s.messages.map((m) => {
                         if (m.id !== assistantMessageId) return m;
+                        if (displayContent) {
+                          return { ...m, content: displayContent };
+                        }
                         let newContent = m.content;
                         if (thinking) {
                           if (!newContent.includes("<thinking>")) {
@@ -2176,10 +2186,12 @@ ${currentContextSummary}` },
                   return s;
                 })
               );
+              if (skillTagDetected) break;
             }
           } catch {
           }
         }
+        if (skillTagDetected) break;
       }
       if (buffer.trim()) {
         try {