From 7b411f8aadbc98338261f8e392a6d63c8b769cc1 Mon Sep 17 00:00:00 2001 From: fridayL Date: Wed, 24 Sep 2025 16:59:17 +0800 Subject: [PATCH 1/2] add: change deafult pre_load --- src/memos/api/product_api.py | 2 +- src/memos/mem_os/product.py | 4 ++-- src/memos/mem_user/mysql_persistent_user_manager.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/memos/api/product_api.py b/src/memos/api/product_api.py index 681644a0..709ad74f 100644 --- a/src/memos/api/product_api.py +++ b/src/memos/api/product_api.py @@ -33,6 +33,6 @@ parser = argparse.ArgumentParser() parser.add_argument("--port", type=int, default=8001) - parser.add_argument("--workers", type=int, default=32) + parser.add_argument("--workers", type=int, default=1) args = parser.parse_args() uvicorn.run("memos.api.product_api:app", host="0.0.0.0", port=args.port, workers=args.workers) diff --git a/src/memos/mem_os/product.py b/src/memos/mem_os/product.py index a4ab4ef2..d6464389 100644 --- a/src/memos/mem_os/product.py +++ b/src/memos/mem_os/product.py @@ -179,14 +179,14 @@ def _restore_user_instances( """ try: # Get all user configurations from persistent storage - user_configs = self.user_manager.list_user_configs() + user_configs = self.user_manager.list_user_configs(self.max_user_instances) # Get the raw database records for sorting by updated_at session = self.user_manager._get_session() try: from memos.mem_user.persistent_user_manager import UserConfig - db_configs = session.query(UserConfig).all() + db_configs = session.query(UserConfig).limit(self.max_user_instances).all() # Create a mapping of user_id to updated_at timestamp updated_at_map = {config.user_id: config.updated_at for config in db_configs} diff --git a/src/memos/mem_user/mysql_persistent_user_manager.py b/src/memos/mem_user/mysql_persistent_user_manager.py index f8983c87..367275fc 100644 --- a/src/memos/mem_user/mysql_persistent_user_manager.py +++ b/src/memos/mem_user/mysql_persistent_user_manager.py @@ -188,7 +188,7 @@ def delete_user_config(self, user_id: str) -> bool: finally: session.close() - def list_user_configs(self) -> dict[str, MOSConfig]: + def list_user_configs(self,limit: int = 1) -> dict[str, MOSConfig]: """List all user configurations. Returns: @@ -196,7 +196,7 @@ def list_user_configs(self) -> dict[str, MOSConfig]: """ session = self._get_session() try: - user_configs = session.query(UserConfig).all() + user_configs = session.query(UserConfig).limit(limit).all() result = {} for user_config in user_configs: From eee8d4e0baa6b7d3e7a729dcc6e475ef57ba870e Mon Sep 17 00:00:00 2001 From: fridayL Date: Wed, 24 Sep 2025 17:04:40 +0800 Subject: [PATCH 2/2] fix: code --- src/memos/mem_user/mysql_persistent_user_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/memos/mem_user/mysql_persistent_user_manager.py b/src/memos/mem_user/mysql_persistent_user_manager.py index 367275fc..99e49d20 100644 --- a/src/memos/mem_user/mysql_persistent_user_manager.py +++ b/src/memos/mem_user/mysql_persistent_user_manager.py @@ -188,7 +188,7 @@ def delete_user_config(self, user_id: str) -> bool: finally: session.close() - def list_user_configs(self,limit: int = 1) -> dict[str, MOSConfig]: + def list_user_configs(self, limit: int = 1) -> dict[str, MOSConfig]: """List all user configurations. Returns: