diff --git a/src/memos/mem_os/product.py b/src/memos/mem_os/product.py index c2eccfe9..4082c38a 100644 --- 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 )