From 751f5d5a19ab33f5f92e0b5adb1e77a9e1bdec78 Mon Sep 17 00:00:00 2001 From: Mayank <83648453+spike-spiegel-21@users.noreply.github.com> Date: Wed, 20 Nov 2024 23:18:34 +0530 Subject: [PATCH] [bug_improvement]: Update hash changed and Vector base class improved (#2034) --- mem0/memory/main.py | 2 +- mem0/vector_stores/base.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/mem0/memory/main.py b/mem0/memory/main.py index f0f9cbae..14f980f2 100644 --- a/mem0/memory/main.py +++ b/mem0/memory/main.py @@ -582,7 +582,7 @@ class Memory(MemoryBase): new_metadata = metadata or {} new_metadata["data"] = data - new_metadata["hash"] = existing_memory.payload.get("hash") + new_metadata["hash"] = hashlib.md5(data.encode()).hexdigest() new_metadata["created_at"] = existing_memory.payload.get("created_at") new_metadata["updated_at"] = datetime.now(pytz.timezone("US/Pacific")).isoformat() diff --git a/mem0/vector_stores/base.py b/mem0/vector_stores/base.py index 974f120f..21f47f3a 100644 --- a/mem0/vector_stores/base.py +++ b/mem0/vector_stores/base.py @@ -38,7 +38,7 @@ class VectorStoreBase(ABC): pass @abstractmethod - def delete_col(self, name): + def delete_col(self): """Delete a collection.""" pass @@ -46,3 +46,8 @@ class VectorStoreBase(ABC): def col_info(self, name): """Get information about a collection.""" pass + + @abstractmethod + def list(self, filters=None, limit=None): + """List all memories.""" + pass