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

Replace UUID with indexes to reduce hallucinations (#2004)

Browse Source
This commit is contained in:
Dev Khant
2024-11-02 12:40:55 +05:30
committed by GitHub
parent e7cc8b9552
commit a6ac4a6698
2 changed files with 29 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
mem0/memory/main.py
View File

@@ -172,7 +172,14 @@ class Memory(MemoryBase):
logging.info(f"Total existing memories: {len(retrieved_old_memory)}")
# mapping UUIDs with integers for handling UUID hallucinations
temp_uuid_mapping = {}
for idx, item in enumerate(retrieved_old_memory):
temp_uuid_mapping[str(idx)] = item["id"]
retrieved_old_memory[idx]["id"] = str(idx)
function_calling_prompt = get_update_memory_messages(retrieved_old_memory, new_retrieved_facts)
new_memories_with_actions = self.llm.generate_response(
messages=[{"role": "user", "content": function_calling_prompt}],
response_format={"type": "json_object"},
@@ -197,24 +204,24 @@ class Memory(MemoryBase):
)
elif resp["event"] == "UPDATE":
self._update_memory(
memory_id=resp["id"],
memory_id=temp_uuid_mapping[resp["id"]],
data=resp["text"],
existing_embeddings=new_message_embeddings,
metadata=metadata,
)
returned_memories.append(
{
"id": resp["id"],
"id": temp_uuid_mapping[resp["id"]],
"memory": resp["text"],
"event": resp["event"],
"previous_memory": resp["old_memory"],
}
)
elif resp["event"] == "DELETE":
self._delete_memory(memory_id=resp["id"])
self._delete_memory(memory_id=temp_uuid_mapping[resp["id"]])
returned_memories.append(
{
"id": resp["id"],
"id": temp_uuid_mapping[resp["id"]],
"memory": resp["text"],
"event": resp["event"],
}