Update add method and prompts (#1775)

mem0/memory/utils.py

@@ -1,14 +1,16 @@
-from mem0.configs.prompts import UPDATE_MEMORY_PROMPT
+from mem0.configs.prompts import FACT_RETRIEVAL_PROMPT
 
 
-def get_update_memory_prompt(existing_memories, memory, template=UPDATE_MEMORY_PROMPT):
-    return template.format(existing_memories=existing_memories, memory=memory)
+def get_fact_retrieval_messages(message):
+    return FACT_RETRIEVAL_PROMPT, f"Input: {message}"
 
-
-def get_update_memory_messages(existing_memories, memory):
-    return [
-        {
-            "role": "user",
-            "content": get_update_memory_prompt(existing_memories, memory),
-        },
-    ]
+def parse_messages(messages):
+        response = ""
+        for msg in messages:
+            if msg["role"] == "system":
+                response += f"system: {msg['content']}\n"
+            if msg["role"] == "user":
+                response += f"user: {msg['content']}\n"
+            if msg["role"] == "assistant":
+                response += f"assistant: {msg['content']}\n"
+        return response