[graph_improvement]: Unique Id removed from update prompt (#2020)

mem0/memory/graph_memory.py

@@ -1,5 +1,7 @@
 import logging
 
+from mem0.memory.utils import format_entities
+
 try:
     from langchain_community.graphs import Neo4jGraph
 except ImportError:
@@ -94,8 +96,8 @@ class MemoryGraph:
             extracted_entities = []
 
         logger.debug(f"Extracted entities: {extracted_entities}")
-
-        update_memory_prompt = get_update_memory_messages(search_output, extracted_entities)
+        search_output_string = format_entities(search_output)
+        update_memory_prompt = get_update_memory_messages(search_output_string, extracted_entities)
 
         _tools = [UPDATE_MEMORY_TOOL_GRAPH, ADD_MEMORY_TOOL_GRAPH, NOOP_TOOL]
         if self.llm_provider in ["azure_openai_structured", "openai_structured"]:

mem0/memory/utils.py

@@ -1,3 +1,5 @@
+import json
+
 from mem0.configs.prompts import FACT_RETRIEVAL_PROMPT
 
 
@@ -15,3 +17,18 @@ def parse_messages(messages):
         if msg["role"] == "assistant":
             response += f"assistant: {msg['content']}\n"
     return response
+
+def format_entities(entities):
+    if not entities:
+        return ""
+    
+    formatted_lines = []
+    for entity in entities:
+        simplified = {
+            "source": entity["source"],
+            "relation": entity["relation"],
+            "destination": entity["destination"]
+        }
+        formatted_lines.append(json.dumps(simplified))
+
+    return "\n".join(formatted_lines)