diff --git a/src/memos/api/config.py b/src/memos/api/config.py index 355ee0385..c9ff70d4e 100644 --- a/src/memos/api/config.py +++ b/src/memos/api/config.py @@ -518,6 +518,13 @@ def create_user_config(user_name: str, user_id: str) -> tuple[MOSConfig, General "embedder": APIConfig.get_embedder_config(), "internet_retriever": internet_config, "reranker": APIConfig.get_reranker_config(), + "reorganize": os.getenv("MOS_ENABLE_REORGANIZE", "false").lower() + == "true", + "memory_size": { + "WorkingMemory": os.getenv("NEBULAR_WORKING_MEMORY", 20), + "LongTermMemory": os.getenv("NEBULAR_LONGTERM_MEMORY", 1e6), + "UserMemory": os.getenv("NEBULAR_USER_MEMORY", 1e6), + }, }, }, "act_mem": {} @@ -575,6 +582,11 @@ def get_default_cube_config() -> GeneralMemCubeConfig | None: "reorganize": os.getenv("MOS_ENABLE_REORGANIZE", "false").lower() == "true", "internet_retriever": internet_config, + "memory_size": { + "WorkingMemory": os.getenv("NEBULAR_WORKING_MEMORY", 20), + "LongTermMemory": os.getenv("NEBULAR_LONGTERM_MEMORY", 1e6), + "UserMemory": os.getenv("NEBULAR_USER_MEMORY", 1e6), + }, }, }, "act_mem": {} diff --git a/src/memos/memories/textual/tree_text_memory/organize/manager.py b/src/memos/memories/textual/tree_text_memory/organize/manager.py index c9cd4de8a..5cc714806 100644 --- a/src/memos/memories/textual/tree_text_memory/organize/manager.py +++ b/src/memos/memories/textual/tree_text_memory/organize/manager.py @@ -44,6 +44,7 @@ def __init__( "LongTermMemory": 1500, "UserMemory": 480, } + logger.info(f"MemorySize is {self.memory_size}") self._threshold = threshold self.is_reorganize = is_reorganize self.reorganizer = GraphStructureReorganizer(