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

This commit is contained in:
Mayank
2024-11-09 03:13:24 +05:30
committed by GitHub
parent eaf295756e
commit 6d535951df
2 changed files with 21 additions and 2 deletions

View File

@@ -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)