@meetsmore-oss/use-ai-client 1.9.4 → 1.10.0

package/dist/bundled.js

@@ -1553,6 +1553,54 @@ function getTextFromContent(content3) {
   return content3.filter((part) => part.type === "text").map((part) => part.text).join("\n");
 }
 
+// src/utils/mergeAssistantMessages.ts
+function mergeAssistantMessagesForDisplay(messages) {
+  const result = [];
+  let pendingTexts = [];
+  let pendingIds = [];
+  for (const msg of messages) {
+    if (msg.role === "tool") {
+      continue;
+    }
+    if (msg.role === "assistant") {
+      const text5 = getTextFromContent(msg.content);
+      if (msg.toolCalls && msg.toolCalls.length > 0) {
+        pendingIds.push(msg.id);
+        if (text5) {
+          pendingTexts.push(text5);
+        }
+      } else {
+        const allTexts = text5 ? [...pendingTexts, text5] : pendingTexts;
+        const combined = allTexts.join("\n\n");
+        result.push({ ...msg, content: combined || "" });
+        pendingTexts = [];
+        pendingIds = [];
+      }
+    } else {
+      if (pendingTexts.length > 0) {
+        result.push({
+          id: `merged-${pendingIds.join("-")}`,
+          role: "assistant",
+          content: pendingTexts.join("\n\n"),
+          createdAt: /* @__PURE__ */ new Date()
+        });
+        pendingTexts = [];
+        pendingIds = [];
+      }
+      result.push(msg);
+    }
+  }
+  if (pendingTexts.length > 0) {
+    result.push({
+      id: `merged-${pendingIds.join("-")}`,
+      role: "assistant",
+      content: pendingTexts.join("\n\n"),
+      createdAt: /* @__PURE__ */ new Date()
+    });
+  }
+  return result;
+}
+
 // ../../node_modules/.bun/react-markdown@8.0.7+32264e8fb3466d46/node_modules/react-markdown/lib/uri-transformer.js
 var protocols = ["http", "https", "mailto", "tel"];
 function uriTransformer(uri) {
@@ -14451,11 +14499,7 @@ function UseAIChatPanel({
 }) {
   const strings = useStrings();
   const theme = useTheme();
-  const displayMessages = messages.filter((m) => {
-    if (m.role === "tool") return false;
-    if (m.role === "assistant" && m.toolCalls && m.toolCalls.length > 0 && !getTextFromContent(m.content)) return false;
-    return true;
-  });
+  const displayMessages = mergeAssistantMessagesForDisplay(messages);
   const [input, setInput] = useState5("");
   const chatHistoryDropdown = useDropdownState();
   const agentDropdown = useDropdownState();
@@ -23589,6 +23633,20 @@ var UseAIClient = class {
           toolCallId: e.toolCallId
         });
       }
+    } else if (event.type === export_EventType.STEP_FINISHED) {
+      if (this._currentAssistantToolCalls.length > 0 && this._currentAssistantMessage) {
+        const assistantMsg = {
+          id: this._currentAssistantMessage.id || v4_default(),
+          role: "assistant",
+          content: this._currentAssistantMessage.content || "",
+          toolCalls: [...this._currentAssistantToolCalls]
+        };
+        this._messages.push(assistantMsg);
+        this._messages.push(...this._pendingToolResults);
+        this._currentAssistantMessage = { id: v4_default(), role: "assistant", content: "" };
+        this._currentAssistantToolCalls = [];
+        this._pendingToolResults = [];
+      }
     } else if (event.type === export_EventType.RUN_FINISHED) {
       if (this._currentAssistantMessage) {
         if (this._currentAssistantToolCalls.length > 0) {
@@ -38015,13 +38073,8 @@ var LocalStorageChatRepository = class {
 
 // src/hooks/useChatManagement.ts
 import { useState as useState6, useCallback as useCallback4, useRef as useRef5, useEffect as useEffect5 } from "react";
-var CHAT_TITLE_MAX_LENGTH = 50;
-function deepEquals(a, b) {
-  return JSON.stringify(a) === JSON.stringify(b);
-}
-function generateChatTitle(message) {
-  return message.length > CHAT_TITLE_MAX_LENGTH ? message.substring(0, CHAT_TITLE_MAX_LENGTH) + "..." : message;
-}
+
+// src/utils/messageConversion.ts
 function transformMessagesToClientFormat(persistedMessages) {
   return persistedMessages.map((msg) => {
     const textContent = getTextFromContent(msg.content);
@@ -38049,6 +38102,39 @@ function transformMessagesToClientFormat(persistedMessages) {
     }
   });
 }
+function extractTurnMessages(messages, startIndex) {
+  const turnSlice = messages.slice(startIndex);
+  const result = [];
+  for (const msg of turnSlice) {
+    if (msg.role === "assistant" && "toolCalls" in msg && msg.toolCalls) {
+      result.push({
+        id: msg.id,
+        role: "assistant",
+        content: typeof msg.content === "string" ? msg.content : "",
+        createdAt: /* @__PURE__ */ new Date(),
+        toolCalls: msg.toolCalls
+      });
+    } else if (msg.role === "tool") {
+      result.push({
+        id: msg.id,
+        role: "tool",
+        content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content),
+        createdAt: /* @__PURE__ */ new Date(),
+        toolCallId: "toolCallId" in msg && msg.toolCallId ? msg.toolCallId : void 0
+      });
+    }
+  }
+  return result;
+}
+
+// src/hooks/useChatManagement.ts
+var CHAT_TITLE_MAX_LENGTH = 50;
+function deepEquals(a, b) {
+  return JSON.stringify(a) === JSON.stringify(b);
+}
+function generateChatTitle(message) {
+  return message.length > CHAT_TITLE_MAX_LENGTH ? message.substring(0, CHAT_TITLE_MAX_LENGTH) + "..." : message;
+}
 function useChatManagement({
   repository,
   clientRef,
@@ -38934,30 +39020,6 @@ function useFeedback({
 
 // src/hooks/useServerEvents.ts
 import { useState as useState12, useCallback as useCallback10, useRef as useRef10 } from "react";
-function extractTurnMessages(messages, startIndex) {
-  const turnSlice = messages.slice(startIndex);
-  const result = [];
-  for (const msg of turnSlice) {
-    if (msg.role === "assistant" && "toolCalls" in msg && msg.toolCalls) {
-      result.push({
-        id: msg.id,
-        role: "assistant",
-        content: "",
-        createdAt: /* @__PURE__ */ new Date(),
-        toolCalls: msg.toolCalls
-      });
-    } else if (msg.role === "tool") {
-      result.push({
-        id: msg.id,
-        role: "tool",
-        content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content),
-        createdAt: /* @__PURE__ */ new Date(),
-        toolCallId: "toolCallId" in msg && msg.toolCallId ? msg.toolCallId : void 0
-      });
-    }
-  }
-  return result;
-}
 function useServerEvents({
   toolSystem,
   saveAIResponse,
@@ -38967,6 +39029,7 @@ function useServerEvents({
   const [streamingText, setStreamingText] = useState12("");
   const streamingChatIdRef = useRef10(null);
   const messageCountAtRunStartRef = useRef10(0);
+  const hasTextFromPriorStepRef = useRef10(false);
   const [executingToolRaw, setExecutingTool] = useState12(null);
   const executingToolFallbackRef = useRef10(null);
   const clearStreamingText = useCallback10(() => {
@@ -38983,6 +39046,11 @@ function useServerEvents({
     const strs = stringsRef.current;
     if (event.type === export_EventType.RUN_STARTED) {
       messageCountAtRunStartRef.current = client.messages.length;
+      hasTextFromPriorStepRef.current = false;
+    } else if (event.type === export_EventType.TEXT_MESSAGE_START) {
+      if (hasTextFromPriorStepRef.current) {
+        setStreamingText((prev) => prev + "\n\n");
+      }
     } else if (event.type === export_EventType.TOOL_CALL_START) {
       const e = event;
       const tool = ts.aggregatedToolsRef.current[e.toolCallName];
@@ -39019,10 +39087,9 @@ function useServerEvents({
       ts.handleApprovalRequest(e);
     } else if (event.type === export_EventType.TEXT_MESSAGE_CONTENT) {
       const contentEvent = event;
+      hasTextFromPriorStepRef.current = true;
       setStreamingText((prev) => prev + contentEvent.delta);
     } else if (event.type === export_EventType.TEXT_MESSAGE_END) {
-      setStreamingText("");
-      streamingChatIdRef.current = null;
     } else if (event.type === export_EventType.RUN_FINISHED) {
       const content3 = client.currentMessageContent;
       if (content3) {
@@ -39031,15 +39098,19 @@ function useServerEvents({
         const turnMessages = extractTurnMessages(client.messages, messageCountAtRunStartRef.current);
         saveAIResponseRef.current(content3, void 0, traceId, turnMessages);
       }
+      setStreamingText("");
+      streamingChatIdRef.current = null;
+      setExecutingTool(null);
       setLoading(false);
     } else if (event.type === export_EventType.RUN_ERROR) {
       const errorEvent = event;
       const errorCode = errorEvent.message;
       console.error("[ServerEvents] Run error:", errorCode);
-      const userMessage = strs.errors[errorCode] || strs.errors[ErrorCode.UNKNOWN_ERROR];
+      const userMessage = strs.errors[errorCode] || errorEvent.message || strs.errors[ErrorCode.UNKNOWN_ERROR];
       saveAIResponseRef.current(userMessage, "error");
       setStreamingText("");
       streamingChatIdRef.current = null;
+      setExecutingTool(null);
       setLoading(false);
     }
   }, []);