@axiom-lattice/react-sdk 2.1.57 → 2.1.58

package/dist/index.js

@@ -1737,6 +1737,7 @@ function AgentThreadProvider({
     }));
   }, [clientAssistantId, tenantId, threadId]);
   const stopStreamingRef = (0, import_react7.useRef)(null);
+  const sendMessageStreamsRef = (0, import_react7.useRef)(/* @__PURE__ */ new Map());
   const chunkMessageMerger = (0, import_react7.useRef)((0, import_client_sdk3.createSimpleMessageMerger)());
   const lastAgentStateCreatedAtRef = (0, import_react7.useRef)(null);
   const messageCountRef = (0, import_react7.useRef)(0);
@@ -1886,19 +1887,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({
@@ -1934,16 +1977,20 @@ function AgentThreadProvider({
     ]
   );
   const stopStreaming = (0, import_react7.useCallback)(() => {
+    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 = (0, import_react7.useCallback)(
     (messageId, onMessageChunk) => {
@@ -2021,6 +2068,10 @@ function AgentThreadProvider({
     if (!threadId) {
       return;
     }
+    sendMessageStreamsRef.current.forEach((stopFn, messageId) => {
+      stopFn();
+    });
+    sendMessageStreamsRef.current.clear();
     if (stopStreamingRef.current) {
       stopStreamingRef.current();
       stopStreamingRef.current = null;
@@ -2062,7 +2113,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) => {
@@ -2111,6 +2162,10 @@ function AgentThreadProvider({
       clearMessages();
     }
     return () => {
+      sendMessageStreamsRef.current.forEach((stopFn) => {
+        stopFn();
+      });
+      sendMessageStreamsRef.current.clear();
       if (stopStreamingRef.current) {
         stopStreamingRef.current();
         stopStreamingRef.current = null;
@@ -5833,11 +5888,15 @@ var MarkdownErrorBoundary = class extends import_react23.default.Component {
 };
 var SafeXMarkdown = import_react23.default.memo(({ content, components, className }) => {
   const deferredContent = (0, import_react23.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__ */ (0, import_jsx_runtime22.jsx)("div", { className, children: /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(
     import_x_markdown.default,
     {
       components,
-      content: deferredContent,
+      content: safeContent,
       paragraphTag: "div"
     }
   ) });