fix (ai/react): useObject stop & isLoading reset at end of stream (#2126)

Author: lgrammel

.changeset/gentle-dolphins-boil.md

@@ -0,0 +1,5 @@
+---
+'@ai-sdk/react': patch
+---
+
+fix (ai/react): useObject stop & isLoading reset at end of stream

packages/react/src/use-object.ts

@@ -148,15 +148,17 @@ function useObject<RESULT, INPUT = any>({
               mutate(currentObject);
             }
           },
+
+          close() {
+            setIsLoading(false);
+            abortControllerRef.current = null;
+          },
         }),
       );
 
       setError(undefined);
     } catch (error) {
       setError(error);
-    } finally {
-      setIsLoading(false);
-      abortControllerRef.current = null;
     }
   };
 

packages/react/src/use-object.ui.test.tsx

@@ -111,8 +111,10 @@ describe('text stream', () => {
       streamController.enqueue('"Hello, world!"}');
       streamController.close();
 
-      await screen.findByTestId('loading');
-      expect(screen.getByTestId('loading')).toHaveTextContent('false');
+      // wait for element "loading" to have text content "false":
+      await waitFor(() => {
+        expect(screen.getByTestId('loading')).toHaveTextContent('true');
+      });
     });
   });