mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-11 22:51:10 +00:00
feat(memory): add metadata toggle option to vector retrieval tool
This commit is contained in:
parent
12efcc9224
commit
4d312ea682
2 changed files with 9 additions and 6 deletions
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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}")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue