From e185d36bdc1c8bb2f46467b62cf1c9d038d2a8fd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=96=B9=E5=BA=94?= Date: Mon, 2 Mar 2026 20:41:50 +0800 Subject: [PATCH] =?UTF-8?q?feat(reme):=20=E6=B7=BB=E5=8A=A0=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E9=80=89=E9=A1=B9=E4=BB=A5=E5=90=AF=E7=94=A8=E6=88=96?= =?UTF-8?q?=E7=A6=81=E7=94=A8=E4=B8=AA=E4=BA=BA=E8=B5=84=E6=96=99=E5=8A=9F?= =?UTF-8?q?=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 ReMe 初始化方法中添加 enable_profile 参数，默认值为 True - 根据 enable_profile 设置决定是否创建 profile 目录和设置 profile_dir - 在 PersonalSummarizer 中根据 enable_profile 条件性地添加个人资料相关工具 - 在 PersonalRetriever 中根据 enable_profile 条件性地添加 ReadAllProfiles 工具 - 修改 profile_path 属性以在禁用个人资料时返回 None - 修改 get_profile_handler 方法以在禁用个人资料时返回 None - 为 enable_profile 参数添加文档说明其用于云向量存储场景 --- reme/reme.py | 102 ++++++++++++++++++++++++++++++++------------------- 1 file changed, 65 insertions(+), 37 deletions(-) diff --git a/reme/reme.py b/reme/reme.py index 2fe99f3d..faec7ed5 100644 --- a/reme/reme.py +++ b/reme/reme.py @@ -53,6 +53,7 @@ class ReMe(Application): target_user_names: list[str] | None = None, target_task_names: list[str] | None = None, target_tool_names: list[str] | None = None, + enable_profile: bool = True, **kwargs, ): """Initialize ReMe with config. @@ -65,6 +66,10 @@ class ReMe(Application): await reme.retrieve_memory(...) await reme.close() ``` + + Args: + enable_profile: Whether to enable profile functionality. Set to False when using + cloud-based vector stores to avoid local file operations. Default is True. """ super().__init__( *args, @@ -83,6 +88,9 @@ class ReMe(Application): default_token_counter_config=default_token_counter_config, **kwargs, ) + + self.enable_profile = enable_profile + memory_target_type_mapping: dict[str, MemoryType] = {} if target_user_names: for name in target_user_names: @@ -101,9 +109,12 @@ class ReMe(Application): self.service_context.memory_target_type_mapping = memory_target_type_mapping - profile_path = Path(self.service_context.service_config.working_dir) / "profile" - profile_path.mkdir(parents=True, exist_ok=True) - self.profile_dir: str = str(profile_path) + if self.enable_profile: + profile_path = Path(self.service_context.service_config.working_dir) / "profile" + profile_path.mkdir(parents=True, exist_ok=True) + self.profile_dir: str = str(profile_path) + else: + self.profile_dir: str = "" def _add_meta_memory(self, memory_type: str | MemoryType, memory_target: str): """Register or validate a memory target with the given memory type.""" @@ -174,34 +185,40 @@ class ReMe(Application): format_messages.append(message) if version == "default": + personal_summarizer_tools = [ + AddDraftAndRetrieveSimilarMemory( + enable_thinking_params=enable_thinking_params, + enable_memory_target=False, + enable_when_to_use=False, + enable_multiple=True, + top_k=retrieve_top_k, + ), + AddMemory( + enable_thinking_params=enable_thinking_params, + enable_memory_target=False, + enable_when_to_use=False, + enable_multiple=True, + ), + ] + if self.enable_profile: + personal_summarizer_tools.extend( + [ + ReadAllProfiles( + enable_thinking_params=False, + enable_memory_target=False, + profile_dir=self.profile_dir, + ), + UpdateProfilesV1( + enable_thinking_params=enable_thinking_params, + enable_memory_target=False, + enable_multiple=True, + profile_dir=self.profile_dir, + ), + ], + ) personal_summarizer: BaseMemoryAgent = PersonalSummarizer( llm=llm_config_name, - tools=[ - AddDraftAndRetrieveSimilarMemory( - enable_thinking_params=enable_thinking_params, - enable_memory_target=False, - enable_when_to_use=False, - enable_multiple=True, - top_k=retrieve_top_k, - ), - AddMemory( - enable_thinking_params=enable_thinking_params, - enable_memory_target=False, - enable_when_to_use=False, - enable_multiple=True, - ), - ReadAllProfiles( - enable_thinking_params=False, - enable_memory_target=False, - profile_dir=self.profile_dir, - ), - UpdateProfilesV1( - enable_thinking_params=enable_thinking_params, - enable_memory_target=False, - enable_multiple=True, - profile_dir=self.profile_dir, - ), - ], + tools=personal_summarizer_tools, ) else: @@ -324,14 +341,17 @@ class ReMe(Application): """Retrieve relevant personal, procedural and tool memories for a query.""" if version == "default": - personal_retriever: BaseMemoryAgent = PersonalRetriever( - llm=llm_config_name, - tools=[ + personal_retriever_tools = [] + if self.enable_profile: + personal_retriever_tools.append( ReadAllProfiles( enable_thinking_params=False, enable_memory_target=False, profile_dir=self.profile_dir, ), + ) + personal_retriever_tools.extend( + [ RetrieveMemory( top_k=retrieve_top_k, enable_thinking_params=enable_thinking_params, @@ -344,6 +364,10 @@ class ReMe(Application): ), ], ) + personal_retriever: BaseMemoryAgent = PersonalRetriever( + llm=llm_config_name, + tools=personal_retriever_tools, + ) else: raise NotImplementedError(f"version={version} is not supported") @@ -601,12 +625,16 @@ class ReMe(Application): return MemoryHandler(memory_target=memory_target, service_context=self.service_context) @property - def profile_path(self) -> Path: - """Get the path to the profile directory.""" + def profile_path(self) -> Path | None: + """Get the path to the profile directory. Returns None if profile is disabled.""" + if not self.enable_profile: + return None return Path(self.profile_dir) / self.default_vector_store.collection_name - def get_profile_handler(self, user_name: str) -> ProfileHandler: - """Get the profile handler for the specified user.""" + def get_profile_handler(self, user_name: str) -> ProfileHandler | None: + """Get the profile handler for the specified user. Returns None if profile is disabled.""" + if not self.enable_profile: + return None return ProfileHandler(memory_target=user_name, profile_path=self.profile_path) @@ -619,4 +647,4 @@ def main(): if __name__ == "__main__": - main() + main() \ No newline at end of file