From 3684eb25eb2091e62a7770df173bf141561fc616 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Wed, 28 Jan 2026 20:47:04 +0800 Subject: [PATCH] feat(memory): update profile management with time-based filtering --- reme/reme.py | 4 ++-- .../tool/memory/add_draft_and_read_all_profiles.py | 14 +++++++++++--- 2 files changed, 13 insertions(+), 5 deletions(-) diff --git a/reme/reme.py b/reme/reme.py index 08ca92ca..30d97bf0 100644 --- a/reme/reme.py +++ b/reme/reme.py @@ -214,7 +214,7 @@ class ReMe(Application): enable_thinking_params: bool = False, version: str = "default", retrieve_top_k: int = 20, - enable_memory_target: bool = True, + enable_time_filter: bool = True, return_dict: bool = False, **kwargs, ) -> str | dict: @@ -231,7 +231,7 @@ class ReMe(Application): RetrieveMemory( enable_thinking_params=enable_thinking_params, top_k=retrieve_top_k, - enable_memory_target=enable_memory_target, + enable_time_filter=enable_time_filter, ), ReadHistory(enable_thinking_params=enable_thinking_params), ], diff --git a/reme/tool/memory/add_draft_and_read_all_profiles.py b/reme/tool/memory/add_draft_and_read_all_profiles.py index da4e559c..0f9ba241 100644 --- a/reme/tool/memory/add_draft_and_read_all_profiles.py +++ b/reme/tool/memory/add_draft_and_read_all_profiles.py @@ -17,12 +17,20 @@ class AddDraftAndReadAllProfiles(BaseMemoryTool): def _build_query_parameters(self) -> dict: """Build the query parameters schema""" properties = { - "profile_draft": { + "message_time": { "type": "string", - "description": "profile_draft", + "description": "Message time, e.g. '2020-01-01 00:00:00'", + }, + "profile_key": { + "type": "string", + "description": "Profile key or category, e.g. 'name'", + }, + "profile_value": { + "type": "string", + "description": "Profile value or content, e.g. 'John Smith'", }, } - required = ["profile_draft"] + required = ["message_time", "profile_key", "profile_value"] if self.enable_memory_target: properties["memory_target"] = {