diff --git a/reme_ai/core/schema/memory_node.py b/reme_ai/core/schema/memory_node.py index fa9bd25b..304dfe5c 100644 --- a/reme_ai/core/schema/memory_node.py +++ b/reme_ai/core/schema/memory_node.py @@ -153,7 +153,7 @@ class MemoryNode(BaseModel): str: Formatted string with when_to_use, content, and ref_memory_id. """ parts: list[str] = [ - f"memory_id={self.memory_id} modified_time={self.time_modified}", + f"memory_id={self.memory_id}", ] if self.when_to_use: diff --git a/reme_ai/mem_tool/wk/vector_retrieve_memory.py b/reme_ai/mem_tool/wk/vector_retrieve_memory.py index 0cfc367e..de7498b4 100644 --- a/reme_ai/mem_tool/wk/vector_retrieve_memory.py +++ b/reme_ai/mem_tool/wk/vector_retrieve_memory.py @@ -12,11 +12,13 @@ class VectorRetrieveMemory(BaseMemoryTool): self, add_memory_type_target: bool = False, top_k: int = 20, + enable_metadata: bool = True, **kwargs, ): super().__init__(**kwargs) self.add_memory_type_target: bool = add_memory_type_target self.top_k: int = top_k + self.enable_metadata: bool = enable_metadata def _build_query_schema(self) -> tuple[dict, list[str]]: properties = {} @@ -39,10 +41,11 @@ class VectorRetrieveMemory(BaseMemoryTool): } required.append("query") - properties["metadata"] = { - "type": "object", - "description": self.get_prompt("metadata"), - } + if self.enable_metadata: + properties["metadata"] = { + "type": "object", + "description": self.get_prompt("metadata"), + } return properties, required @@ -133,7 +136,7 @@ class VectorRetrieveMemory(BaseMemoryTool): for item in query_items: memory_type = item.get("memory_type") or default_memory_type memory_target = item.get("memory_target") or default_memory_target - metadata = item.get("metadata", {}) + metadata = item.get("metadata", {}) if self.enable_metadata else None if not memory_type or not memory_target: logger.warning(f"Skipping query with missing memory_type or memory_target: {item}")