From 6cf7ac3e30bb83d59dc29813bef0e1e567894afe Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Fri, 7 Mar 2025 13:13:00 +0530 Subject: [PATCH] Fixes for new_memories_with_actions (#2326) --- mem0/memory/main.py | 34 +++++++++++++++++----------------- 1 file changed, 17 insertions(+), 17 deletions(-) diff --git a/mem0/memory/main.py b/mem0/memory/main.py index 6bb1ac54..839112f6 100644 --- a/mem0/memory/main.py +++ b/mem0/memory/main.py @@ -212,45 +212,45 @@ class Memory(MemoryBase): returned_memories = [] try: - for resp in new_memories_with_actions["memory"]: + for resp in new_memories_with_actions.get("memory", []): logging.info(resp) try: - if resp["event"] == "ADD": + if resp.get("event", "") == "ADD": memory_id = self._create_memory( - data=resp["text"], existing_embeddings=new_message_embeddings, metadata=metadata + data=resp.get("text", ""), existing_embeddings=new_message_embeddings, metadata=metadata ) returned_memories.append( { "id": memory_id, - "memory": resp["text"], - "event": resp["event"], + "memory": resp.get("text", ""), + "event": resp.get("event", ""), } ) - elif resp["event"] == "UPDATE": + elif resp.get("event", "") == "UPDATE": self._update_memory( memory_id=temp_uuid_mapping[resp["id"]], - data=resp["text"], + data=resp.get("text", ""), existing_embeddings=new_message_embeddings, metadata=metadata, ) returned_memories.append( { - "id": temp_uuid_mapping[resp["id"]], - "memory": resp["text"], - "event": resp["event"], - "previous_memory": resp["old_memory"], + "id": temp_uuid_mapping[resp.get("id", "")], + "memory": resp.get("text", ""), + "event": resp.get("event", ""), + "previous_memory": resp.get("old_memory", ""), } ) - elif resp["event"] == "DELETE": - self._delete_memory(memory_id=temp_uuid_mapping[resp["id"]]) + elif resp.get("event", "") == "DELETE": + self._delete_memory(memory_id=temp_uuid_mapping[resp.get("id", "")]) returned_memories.append( { - "id": temp_uuid_mapping[resp["id"]], - "memory": resp["text"], - "event": resp["event"], + "id": temp_uuid_mapping[resp.get("id", "")], + "memory": resp.get("text", ""), + "event": resp.get("event", ""), } ) - elif resp["event"] == "NONE": + elif resp.get("event", "") == "NONE": logging.info("NOOP for Memory.") except Exception as e: logging.error(f"Error in new_memories_with_actions: {e}")