diff --git a/reme/reme.py b/reme/reme.py index 5e5b9caf..cab3a2c4 100644 --- a/reme/reme.py +++ b/reme/reme.py @@ -287,20 +287,8 @@ class ReMe(Application): raise NotImplementedError procedural_summarizer: BaseMemoryAgent - if version in ["default", "v1"]: - procedural_summarizer = ProceduralSummarizer( - tools=[ - VectorRetrieveMemory( - add_memory_type_target=False, - enable_thinking_params=True, - metadata_desc=None, - top_k=5, - ), - AddMemory(enable_thinking_params=True,add_when_to_use=True, metadata_desc=metadata_summary), - DeleteMemory(enable_thinking_params=True), - UpdateMemory(enable_thinking_params=True,add_when_to_use=True, metadata_desc=metadata_summary), - ], - ) + if version in ["default", "v1", "v2", "halumem"]: + procedural_summarizer = ProceduralSummarizer(tools=[]) else: raise NotImplementedError @@ -478,18 +466,8 @@ class ReMe(Application): raise NotImplementedError procedural_retriever: BaseMemoryAgent - if version in ["default", "v1"]: - procedural_retriever = ProceduralRetriever( - tools=[ - VectorRetrieveMemory( - add_memory_type_target=True, - enable_thinking_params=True, - metadata_desc=metadata_retrieve, - top_k=top_k, - ), - ReadHistory(enable_thinking_params=True), - ], - ) + if version in ["default", "v1", "v2", "halumem"]: + procedural_retriever = ProceduralRetriever(tools=[]) else: raise NotImplementedError