17 lines
733 B
Python
17 lines
733 B
Python
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) |