diff --git a/components/use-x-chat/index.ts b/components/use-x-chat/index.ts
index b72a02d34..3d6f1a8a7 100644
--- a/components/use-x-chat/index.ts
+++ b/components/use-x-chat/index.ts
@@ -289,6 +289,11 @@ export default function useXChat<
       },
       transformStream,
     );
+
+    return {
+      loadingMsgId,
+      updatingMsgId
+    }
   });
 
   return {