Feat/add timerlog (#317)

* feat: add logs for cube * feat: add loggers for mem * add dict timer log * fix: change ci code * update size * fix ci
MemTensor · fridayL · Sep 24, 2025 · Sep 10, 2025 · Sep 10, 2025 · Sep 11, 2025
commit c407987fd127ce1bebf22c704a2c6cb31bf4b41c
diff --git a/src/memos/mem_os/product.py b/src/memos/mem_os/product.py
@@ -2,6 +2,7 @@
 import json
 import os
 import random
+import sys
 import time
 
 from collections.abc import Generator
@@ -1315,6 +1316,9 @@ def search(
         # Load user cubes if not already loaded
         time_start = time.time()
         self._load_user_cubes(user_id, self.default_cube_config)
+        dict_size = sys.getsizeof(self.mem_cubes._dict)
+        size_mb = dict_size / (1024 * 1024)
+        logger.info(f"now search memcubes_size is : {len(self.mem_cubes)} {size_mb}MB")
         load_user_cubes_time_end = time.time()
         logger.info(
             f"time search: load_user_cubes time user_id: {user_id} time is: {load_user_cubes_time_end - time_start}"
@@ -1363,7 +1367,9 @@ def add(
 
         # Load user cubes if not already loaded
         self._load_user_cubes(user_id, self.default_cube_config)
-
+        dict_size = sys.getsizeof(self.mem_cubes._dict)
+        size_mb = dict_size / (1024 * 1024)
+        logger.info(f"now add memcubes_size is : {len(self.mem_cubes)} {size_mb}MB")
         result = super().add(
             messages, memory_content, doc_path, mem_cube_id, user_id, session_id=session_id
         )