@axiom-lattice/react-sdk 2.1.57 → 2.1.58

package/dist/index.mjs

@@ -1668,6 +1668,7 @@ function AgentThreadProvider({
     }));
   }, [clientAssistantId, tenantId, threadId]);
   const stopStreamingRef = useRef5(null);
+  const sendMessageStreamsRef = useRef5(/* @__PURE__ */ new Map());
   const chunkMessageMerger = useRef5(createSimpleMessageMerger2());
   const lastAgentStateCreatedAtRef = useRef5(null);
   const messageCountRef = useRef5(0);
@@ -1817,19 +1818,61 @@ function AgentThreadProvider({
       }
       try {
         if (options.streaming !== false && streaming !== false) {
-          client.chat.stream(
-            {
-              threadId,
-              messages: [userMessage],
-              command,
-              custom_run_config: customRunConfig,
-              ...rest
-            },
-            () => {
-            }
-          );
-          if (!stopStreamingRef.current) {
-            loadMessages();
+          if (stopStreamingRef.current) {
+            client.chat.stream(
+              {
+                threadId,
+                messages: [userMessage],
+                command,
+                custom_run_config: customRunConfig,
+                ...rest
+              },
+              () => {
+              },
+              // Empty handler - resumeStream will handle chunks
+              async () => {
+                if (options.enableReturnStateWhenStreamCompleted) {
+                  await fetchAndUpdateAgentState(threadId);
+                }
+              },
+              (error) => {
+                setState((prev) => ({
+                  ...prev,
+                  isLoading: false,
+                  error
+                }));
+              }
+            );
+          } else {
+            const stopStreaming2 = client.chat.stream(
+              {
+                threadId,
+                messages: [userMessage],
+                command,
+                custom_run_config: customRunConfig,
+                ...rest
+              },
+              (chunk) => handleStreamEvent(chunk),
+              async () => {
+                setState((prev) => ({
+                  ...prev,
+                  isLoading: false
+                }));
+                if (options.enableReturnStateWhenStreamCompleted) {
+                  await fetchAndUpdateAgentState(threadId);
+                }
+                sendMessageStreamsRef.current.delete(userMessage.id);
+              },
+              (error) => {
+                setState((prev) => ({
+                  ...prev,
+                  isLoading: false,
+                  error
+                }));
+                sendMessageStreamsRef.current.delete(userMessage.id);
+              }
+            );
+            sendMessageStreamsRef.current.set(userMessage.id, stopStreaming2);
           }
         } else {
           const response = await client.chat.send({
@@ -1865,16 +1908,20 @@ function AgentThreadProvider({
     ]
   );
   const stopStreaming = useCallback7(() => {
+    sendMessageStreamsRef.current.forEach((stopFn) => {
+      stopFn();
+    });
+    sendMessageStreamsRef.current.clear();
     if (stopStreamingRef.current) {
       stopStreamingRef.current();
       stopStreamingRef.current = null;
-      const currentMessages = chunkMessageMerger.current.getMessages();
-      setState((prev) => ({
-        ...prev,
-        messages: currentMessages,
-        isLoading: false
-      }));
     }
+    const currentMessages = chunkMessageMerger.current.getMessages();
+    setState((prev) => ({
+      ...prev,
+      messages: currentMessages,
+      isLoading: false
+    }));
   }, []);
   const resumeStream = useCallback7(
     (messageId, onMessageChunk) => {
@@ -1952,6 +1999,10 @@ function AgentThreadProvider({
     if (!threadId) {
       return;
     }
+    sendMessageStreamsRef.current.forEach((stopFn, messageId) => {
+      stopFn();
+    });
+    sendMessageStreamsRef.current.clear();
     if (stopStreamingRef.current) {
       stopStreamingRef.current();
       stopStreamingRef.current = null;
@@ -1993,7 +2044,7 @@ function AgentThreadProvider({
         ...needUpdateFields,
         isLoading: false
       }));
-      if (options.enableResumeStream && needUpdateFields.messages.length > 0) {
+      if (options.enableResumeStream && needUpdateFields.messages.length > 0 && sendMessageStreamsRef.current.size === 0) {
         const lastMessage = needUpdateFields.messages[needUpdateFields.messages.length - 1];
         let lastMessageIsRemoved = false;
         resumeStream(lastMessage.id, (chunk) => {
@@ -2042,6 +2093,10 @@ function AgentThreadProvider({
       clearMessages();
     }
     return () => {
+      sendMessageStreamsRef.current.forEach((stopFn) => {
+        stopFn();
+      });
+      sendMessageStreamsRef.current.clear();
       if (stopStreamingRef.current) {
         stopStreamingRef.current();
         stopStreamingRef.current = null;
@@ -5784,11 +5839,15 @@ var MarkdownErrorBoundary = class extends React11.Component {
 };
 var SafeXMarkdown = React11.memo(({ content, components, className }) => {
   const deferredContent = useDeferredValue(content);
+  if (typeof deferredContent !== "string") {
+    console.warn("[MDResponse] Content is not a string:", typeof deferredContent, deferredContent);
+  }
+  const safeContent = typeof deferredContent === "string" ? deferredContent : String(deferredContent ?? "");
   return /* @__PURE__ */ jsx22("div", { className, children: /* @__PURE__ */ jsx22(
     XMarkdown,
     {
       components,
-      content: deferredContent,
+      content: safeContent,
       paragraphTag: "div"
     }
   ) });