scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-11 22:51:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

add conditional delete

Browse source
This commit is contained in:
xianzhe.xxz 2024-07-23 15:08:05 +08:00
parent 2c14b958a9
commit 46ce7d8cb7
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
memory_scope/storage/llama_index_es_memory_store.py
View file

@ -95,6 +95,10 @@ class LlamaIndexEsMemoryStore(BaseMemoryStore):
def delete(self, node: MemoryNode):
return self.es_store.delete(node.memory_id)
def delete_conditional(self, filter_dict: Dict = {}):
nodes = self.retrieve_memories(filter_dict=filter_dict, top_k=10000)
self.batch_delete(nodes)
def update(self, node: MemoryNode, update_embedding: bool = True):
# TODO update without embedding?
self.delete(node)