Support Custom Prompt for Memory Action Decision (#2371)

tests/configs/test_prompts.py

@@ -0,0 +1,17 @@
+from mem0.configs import prompts
+
+
+def test_get_update_memory_messages():
+    retrieved_old_memory_dict = [{"id": "1", "text": "old memory 1"}]
+    response_content = ["new fact"]
+    custom_update_memory_prompt = "custom prompt determining memory update"
+    
+    ## When custom update memory prompt is provided
+    ##
+    result = prompts.get_update_memory_messages(retrieved_old_memory_dict, response_content, custom_update_memory_prompt)
+    assert result.startswith(custom_update_memory_prompt)
+    
+    ## When custom update memory prompt is not provided
+    ##
+    result = prompts.get_update_memory_messages(retrieved_old_memory_dict, response_content, None)
+    assert result.startswith(prompts.DEFAULT_UPDATE_MEMORY_PROMPT)