feat: add cleanup code

ant-design · Dec 18, 2024 · 6831da6 · 6831da6
1 parent 3599871
commit 6831da6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/components/useXChat/demo/stream-cancel.tsx b/components/useXChat/demo/stream-cancel.tsx
@@ -53,6 +53,12 @@ const App = () => {
 
   const abortRef = useRef(() => {});
 
+  useEffect(() => {
+    return () => {
+      abortRef.current();
+    };
+  }, []);
+
   // Agent for request
   const [agent] = useXAgent({
     request: async ({}, { onSuccess, onUpdate }) => {