klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes for new_memories_with_actions (#2326)

Browse Source
This commit is contained in:
Dev Khant
2025-03-07 13:13:00 +05:30
committed by GitHub
parent 07d2f11081
commit 6cf7ac3e30
1 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
mem0/memory/main.py
View File

@@ -212,45 +212,45 @@ class Memory(MemoryBase):
returned_memories = [] returned_memories = []
try: try:
for resp in new_memories_with_actions["memory"]: for resp in new_memories_with_actions.get("memory", []):
logging.info(resp) logging.info(resp)
try: try:
if resp["event"] == "ADD": if resp.get("event", "") == "ADD":
memory_id = self._create_memory( 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( returned_memories.append(
{ {
"id": memory_id, "id": memory_id,
"memory": resp["text"], "memory": resp.get("text", ""),
"event": resp["event"], "event": resp.get("event", ""),
} }
) )
elif resp["event"] == "UPDATE": elif resp.get("event", "") == "UPDATE":
self._update_memory( self._update_memory(
memory_id=temp_uuid_mapping[resp["id"]], memory_id=temp_uuid_mapping[resp["id"]],
data=resp["text"], data=resp.get("text", ""),
existing_embeddings=new_message_embeddings, existing_embeddings=new_message_embeddings,
metadata=metadata, metadata=metadata,
) )
returned_memories.append( returned_memories.append(
{ {
"id": temp_uuid_mapping[resp["id"]], "id": temp_uuid_mapping[resp.get("id", "")],
"memory": resp["text"], "memory": resp.get("text", ""),
"event": resp["event"], "event": resp.get("event", ""),
"previous_memory": resp["old_memory"], "previous_memory": resp.get("old_memory", ""),
} }
) )
elif resp["event"] == "DELETE": elif resp.get("event", "") == "DELETE":
self._delete_memory(memory_id=temp_uuid_mapping[resp["id"]]) self._delete_memory(memory_id=temp_uuid_mapping[resp.get("id", "")])
returned_memories.append( returned_memories.append(
{ {
"id": temp_uuid_mapping[resp["id"]], "id": temp_uuid_mapping[resp.get("id", "")],
"memory": resp["text"], "memory": resp.get("text", ""),
"event": resp["event"], "event": resp.get("event", ""),
} }
) )
elif resp["event"] == "NONE": elif resp.get("event", "") == "NONE":
logging.info("NOOP for Memory.") logging.info("NOOP for Memory.")
except Exception as e: except Exception as e:
logging.error(f"Error in new_memories_with_actions: {e}") logging.error(f"Error in new_memories_with_actions: {e}")