@gendive/chatllm 0.17.3 → 0.17.4

package/dist/react/index.js

@@ -2426,6 +2426,39 @@ ${finalContent}`;
       ...isHidden && { hidden: true },
       ...userContentParts && { contentParts: userContentParts }
     };
+    const capturedSessionId = sessionId;
+    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;
+    const assistantMessageId = generateId3("msg");
+    const assistantMessage = {
+      id: assistantMessageId,
+      role: "assistant",
+      content: "",
+      model: selectedModel,
+      timestamp: Date.now()
+    };
+    setInput("");
+    setQuotedText(null);
+    setSelectedAction(null);
+    setAttachments([]);
+    setSessions(
+      (prev) => prev.map((s) => {
+        if (s.id === capturedSessionId) {
+          const newMessages = [...s.messages, userMessage, assistantMessage];
+          return {
+            ...s,
+            messages: newMessages,
+            title: s.messages.length === 0 ? generateTitle([userMessage]) : s.title,
+            updatedAt: Date.now()
+          };
+        }
+        return s;
+      })
+    );
     let attachmentResults = [];
     if (currentAttachments.length > 0) {
       const attachmentSkills = Object.entries(resolvedSkills).filter(
@@ -2444,10 +2477,29 @@ ${finalContent}`;
           });
         });
         if (matchedFiles.length === 0) continue;
+        setSessions(
+          (prev) => prev.map((s) => {
+            if (s.id !== capturedSessionId) return s;
+            return {
+              ...s,
+              messages: s.messages.map((m) => {
+                if (m.id !== assistantMessageId) return m;
+                return {
+                  ...m,
+                  contentParts: [...m.contentParts || [], {
+                    type: "tool_loading",
+                    toolName: skillName,
+                    label: skillConfig.label
+                  }]
+                };
+              })
+            };
+          })
+        );
         try {
           const filesToPass = skillConfig.autoConvertBase64 ? await convertAttachmentsToBase64(matchedFiles) : matchedFiles;
           const result = await skillConfig.execute({ files: filesToPass, userMessage: finalContent });
-          attachmentResults.push({
+          const toolResultPart = {
             type: "tool_result",
             toolName: skillName,
             label: skillConfig.label,
@@ -2458,9 +2510,44 @@ ${finalContent}`;
               metadata: result.metadata,
               sources: result.sources
             }
-          });
+          };
+          attachmentResults.push(toolResultPart);
+          setSessions(
+            (prev) => prev.map((s) => {
+              if (s.id !== capturedSessionId) return s;
+              return {
+                ...s,
+                messages: s.messages.map((m) => {
+                  if (m.id !== assistantMessageId) return m;
+                  return {
+                    ...m,
+                    contentParts: (m.contentParts || []).map(
+                      (p) => p.type === "tool_loading" && p.toolName === skillName ? toolResultPart : p
+                    )
+                  };
+                })
+              };
+            })
+          );
         } catch (error) {
           console.error(`[useChatUI] attachment skill ${skillName} failed:`, error);
+          setSessions(
+            (prev) => prev.map((s) => {
+              if (s.id !== capturedSessionId) return s;
+              return {
+                ...s,
+                messages: s.messages.map((m) => {
+                  if (m.id !== assistantMessageId) return m;
+                  return {
+                    ...m,
+                    contentParts: (m.contentParts || []).filter(
+                      (p) => !(p.type === "tool_loading" && p.toolName === skillName)
+                    )
+                  };
+                })
+              };
+            })
+          );
         }
       }
     }
@@ -2478,40 +2565,6 @@ ${finalContent}`;
         }
       }
     }
-    const assistantMessageId = generateId3("msg");
-    const assistantMessage = {
-      id: assistantMessageId,
-      role: "assistant",
-      content: "",
-      model: selectedModel,
-      timestamp: Date.now(),
-      ...attachmentResults.length > 0 && { contentParts: attachmentResults }
-    };
-    setInput("");
-    setQuotedText(null);
-    setSelectedAction(null);
-    setAttachments([]);
-    const capturedSessionId = sessionId;
-    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) {
-          const newMessages = [...s.messages, userMessage, assistantMessage];
-          return {
-            ...s,
-            messages: newMessages,
-            title: s.messages.length === 0 ? generateTitle([userMessage]) : s.title,
-            updatedAt: Date.now()
-          };
-        }
-        return s;
-      })
-    );
     if (isFirstMessage && generateTitleRef.current) {
       Promise.resolve(generateTitleRef.current(finalContent)).then((generatedTitle) => {
         if (generatedTitle && generatedTitle.trim()) {
@@ -8310,7 +8363,7 @@ var MessageBubble = ({
                     ] })
                   ] })
                 ] }),
-                message.contentParts?.length ? /* @__PURE__ */ (0, import_jsx_runtime15.jsx)("div", { style: { wordBreak: "break-word" }, children: /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
+                message.contentParts?.length && /* @__PURE__ */ (0, import_jsx_runtime15.jsx)("div", { style: { wordBreak: "break-word" }, children: /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
                   ContentPartRenderer,
                   {
                     parts: message.contentParts,
@@ -8318,7 +8371,8 @@ var MessageBubble = ({
                     showThinking,
                     thinkingDefaultOpen
                   }
-                ) }) : displayContent ? /* @__PURE__ */ (0, import_jsx_runtime15.jsx)("div", { style: { wordBreak: "break-word" }, children: /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
+                ) }),
+                displayContent ? /* @__PURE__ */ (0, import_jsx_runtime15.jsx)("div", { style: { wordBreak: "break-word" }, children: /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(
                   MarkdownRenderer,
                   {
                     content: displayContent,