fix: try to fix 'never clsoe' issue, try-1

Author: J1awei-Yang

examples/mem_os/locomo_shared_database_memos.py

@@ -185,3 +185,4 @@ def get_user_configs(db_name, user_name):
     i += 1
 
 print("\n✅ All messages added for both roles.\n")
+mos_product.mem_reorganizer_off()

src/memos/mem_os/core.py

@@ -147,6 +147,17 @@ def mem_scheduler_off(self) -> bool:
             logger.error(f"Failed to stop scheduler: {e!s}")
             return False
 
+    def mem_reorganizer_on(self) -> bool:
+        pass
+
+    def mem_reorganizer_off(self) -> bool:
+        """temporally implement"""
+        for mem_cube in self.mem_cubes.values():
+            logger.info(f"try to close reorganizer for {mem_cube.text_mem.config.cube_id}")
+            if mem_cube.text_mem and mem_cube.text_mem.is_reorganize:
+                logger.info(f"close reorganizer for {mem_cube.text_mem.config.cube_id}")
+                mem_cube.text_mem.memory_manager.close()
+
     def _register_chat_history(self, user_id: str | None = None) -> None:
         """Initialize chat history with user ID."""
         if user_id is None: