[improvement]: Graph nodes extraction improved (#2035)

2024-11-21 12:27:39 +05:30
parent 62ca0ddbe2
commit 4b8e32830a
4 changed files with 171 additions and 142 deletions
--- a/mem0/memory/utils.py
+++ b/mem0/memory/utils.py
@@ -24,11 +24,7 @@ def format_entities(entities):
    
    formatted_lines = []
    for entity in entities:
-        simplified = {
-            "source": entity["source"],
-            "relation": entity["relation"],
-            "destination": entity["destination"]
-        }
-        formatted_lines.append(json.dumps(simplified))
+        simplified = f"{entity['source']} -- {entity['relation'].upper()} -- {entity['destination']}"
+        formatted_lines.append(simplified)

    return "\n".join(formatted_lines)