From 46ce7d8cb713af31be241ee30e6c29d0a9da099c Mon Sep 17 00:00:00 2001 From: "xianzhe.xxz" Date: Tue, 23 Jul 2024 15:08:05 +0800 Subject: [PATCH] add conditional delete --- memory_scope/storage/llama_index_es_memory_store.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/memory_scope/storage/llama_index_es_memory_store.py b/memory_scope/storage/llama_index_es_memory_store.py index cb16b682..a75d2f3c 100644 --- a/memory_scope/storage/llama_index_es_memory_store.py +++ b/memory_scope/storage/llama_index_es_memory_store.py @@ -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)