@axiom-lattice/react-sdk 2.1.100 → 2.1.102

package/dist/index.js

@@ -2031,6 +2031,7 @@ function AgentThreadProvider({
   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 streamEndedByInterruptRef = (0, import_react7.useRef)(false);
   const messageCountRef = (0, import_react7.useRef)(0);
   const onToolCompletedRef = (0, import_react7.useRef)(onToolCompleted);
   onToolCompletedRef.current = onToolCompleted;
@@ -2116,6 +2117,7 @@ function AgentThreadProvider({
       let interrupt;
       if (chunk.type === "interrupt") {
         const rawChunk = chunk;
+        streamEndedByInterruptRef.current = true;
         interrupt = {
           type: "interrupt",
           id: rawChunk.id,
@@ -2143,11 +2145,12 @@ function AgentThreadProvider({
         }
       }
       const updatedMessages = chunkMessageMerger.current.getMessages();
+      const isStopSignal = interrupt !== void 0 || anyChunk.type === "message_failed";
       setState((prev) => ({
         ...prev,
         todos: todos || prev.todos,
         messages: updatedMessages,
-        isLoading: true,
+        isLoading: isStopSignal ? false : true,
         interrupts: interrupt ? [interrupt] : void 0
       }));
     },
@@ -2158,6 +2161,7 @@ function AgentThreadProvider({
       if (!threadId) {
         throw new Error("Thread ID is required to send messages");
       }
+      streamEndedByInterruptRef.current = false;
       sendMessageStreamsRef.current.forEach((stopFn) => {
         stopFn();
       });
@@ -2218,9 +2222,10 @@ function AgentThreadProvider({
               },
               // Chunks handled by resumeStream
               async () => {
-                if (options.enableReturnStateWhenStreamCompleted) {
+                if (options.enableReturnStateWhenStreamCompleted && !streamEndedByInterruptRef.current) {
                   await fetchAndUpdateAgentState(threadId);
                 }
+                streamEndedByInterruptRef.current = false;
                 onStreamCompletedRef.current?.();
                 sendMessageStreamsRef.current.delete(userMessage.id);
               },
@@ -2230,6 +2235,7 @@ function AgentThreadProvider({
                   isLoading: false,
                   error
                 }));
+                streamEndedByInterruptRef.current = false;
                 sendMessageStreamsRef.current.delete(userMessage.id);
               }
             );
@@ -2250,9 +2256,10 @@ function AgentThreadProvider({
                   ...prev,
                   isLoading: false
                 }));
-                if (options.enableReturnStateWhenStreamCompleted) {
+                if (options.enableReturnStateWhenStreamCompleted && !streamEndedByInterruptRef.current) {
                   await fetchAndUpdateAgentState(threadId);
                 }
+                streamEndedByInterruptRef.current = false;
                 onStreamCompletedRef.current?.();
                 sendMessageStreamsRef.current.delete(userMessage.id);
               },