@mastra/react 1.2.3-alpha.2 → 1.2.3-alpha.4

package/dist/index.js

@@ -2067,6 +2067,13 @@ var useChat = ({
       const subscriptionAbort = new AbortController();
       _threadSubscriptionAbortRef.current = subscriptionAbort;
       _threadSubscriptionKeyRef.current = subscriptionKey;
+      const releaseSubscriptionRefs = () => {
+        if (_threadSubscriptionAbortRef.current !== subscriptionAbort) return;
+        _threadSubscriptionRef.current = null;
+        _threadSubscriptionAbortRef.current = null;
+        _threadSubscriptionKeyRef.current = void 0;
+        _threadSubscriptionPromiseRef.current = null;
+      };
       const clientWithAbort = new MastraClient({
         ...baseClient.options,
         abortSignal: subscriptionAbort.signal
@@ -2090,21 +2097,12 @@ var useChat = ({
           if (_threadSubscriptionRef.current === subscription) {
             _threadSubscriptionRef.current = null;
           }
-          if (_threadSubscriptionAbortRef.current === subscriptionAbort) {
-            _threadSubscriptionAbortRef.current = null;
-            _threadSubscriptionKeyRef.current = void 0;
-            _threadSubscriptionPromiseRef.current = null;
-          }
+          releaseSubscriptionRefs();
         });
       }).catch((error) => {
+        releaseSubscriptionRefs();
         if (isThreadSignalUnsupportedError(error)) {
           markThreadSignalsUnsupported();
-          if (_threadSubscriptionAbortRef.current === subscriptionAbort) {
-            _threadSubscriptionRef.current = null;
-            _threadSubscriptionAbortRef.current = null;
-            _threadSubscriptionKeyRef.current = void 0;
-            _threadSubscriptionPromiseRef.current = null;
-          }
           return;
         }
         if (!isAbortError(error)) {
@@ -2641,12 +2639,17 @@ var useChat = ({
     } else {
       setMessages((s) => [...s, dbUserMessage]);
     }
-    if (mode === "generate") {
-      await generate({ ...args, coreUserMessages });
-    } else if (mode === "stream") {
-      await stream({ ...args, coreUserMessages, signalId, clientMessageId });
-    } else if (mode === "network") {
-      await network({ ...args, coreUserMessages });
+    try {
+      if (mode === "generate") {
+        await generate({ ...args, coreUserMessages });
+      } else if (mode === "stream") {
+        await stream({ ...args, coreUserMessages, signalId, clientMessageId });
+      } else if (mode === "network") {
+        await network({ ...args, coreUserMessages });
+      }
+    } catch (error) {
+      setIsRunning(false);
+      throw error;
     }
   };
   return {