大模型的“记忆”不应仅仅只依靠向量数据库，mem0是一个很好的“融合架构”实践方向

pip install mem0ai

import os
from mem0 import Memory

# 假设你已经有了OpenAI API密钥
os.environ['OPENAI_API_KEY'] = 'sk-proj-V7DGXzoKsCZVKMGSq3otY4ir2ip8vUwpI8ec_nT3BlbkFJAOz9PVs3oe-6Qq8gW0DRBGBOqTmcsfWP4FDkdXymrdTN9kSUXFqmEdrycA'

m = Memory()

def store_practice_memory(user_id, practice_details):
    """
    存储用户的网球训练记忆。
    """
    m.add(practice_details, user_id=user_id, metadata={"category": "tennis_practice"})
    all_memories = m.get_all(user_id=user_id)
    print("memorie0------: ", all_memories[0])
    memory_id = all_memories[0]["id"] # get a memory_id
    return memory_id


def retrieve_practice_memory(user_id, query):
    """
    检索用户的网球训练记忆。
    """
    related_memories = m.search(query=query, user_id=user_id)
    return related_memories

def provide_training_advice(user_id, query):
    """
    提供个性化的网球训练建议。
    """
    training_history = retrieve_practice_memory(user_id, query)
    
    if training_history:
        response = "基于你之前的训练，建议你今天专注于提升反手击球技巧。"
    else:
        response = "欢迎开始你的网球训练之旅，让我们从基础的正手和反手击球开始吧。"
    
    return response

def update_practice_memory(memory_id, new_details):
    """
    更新用户的网球训练记忆。
    """

    m.update(memory_id=memory_id, data=new_details)
    return memory_id

def get_memory_history(user_id, memory_id):
    """
    获取特定记忆的历史记录。
    """
    all_memories = m.get_all(user_id=user_id)
    print(all_memories)
    memory_id = all_memories[0]["id"] # get a memory_id
    history = m.history(memory_id=memory_id)
    return history

def delete_memory(memory_id):
    """
    删除特定记忆。
    """
    result = m.delete(memory_id=memory_id)
    return result

def delete_all_memory(user_id):
    """
    删除所有记忆。
    """
    result = m.delete_all(user_id=user_id)
    return result


user_id = "alice"
practice_details = "今天练习了正手击球，感觉力量控制有所提升。"
memory_id = store_practice_memory(user_id, practice_details)
print(f"memory_id: {memory_id}\n")
#memory_id: 66b96ee0-dab6-4347-bcf6-280845f87983


# 用户请求训练建议
query = "正手击球训练"
advice = provide_training_advice(user_id, query)
print(f"advice: {advice}\n")
#advice: 基于你之前的训练，建议你今天专注于提升反手击球技巧。

# 用户在某项技能上取得了显著进步，更新记忆
new_details = "正手击球技巧已显著提升，可以开始练习截击了。"
update_practice_memory(memory_id, new_details)

# 获取记忆的历史变化
memory_history = get_memory_history(user_id, memory_id)
print(f"memory_history: {memory_history}\n")

#memory_history: [{'id': '1d1e7ac4-6cd5-431a-b931-b71d4940c30b', 'memory_id': '66b96ee0-dab6-4347-bcf6-280845f87983', 'old_memory': None, 'new_memory': 'Practiced forehand strokes today. Feels that power control has improved.', 'event': 'ADD', 'created_at': '2024-09-01T22:38:57.313817-07:00', 'updated_at': None}, {'id': 'fd36b9a3-4d50-44e8-9997-1d5a1a9f09f4', 'memory_id': '66b96ee0-dab6-4347-bcf6-280845f87983', 'old_memory': 'Practiced forehand strokes today. Feels that power control has improved.', 'new_memory': '正手击球技巧已显著提升，可以开始练习截击了。', 'event': 'UPDATE', 'created_at': '2024-09-01T22:38:57.313817-07:00', 'updated_at': '2024-09-01T22:38:57.752296-07:00'}]

# 删除记忆
delete_memory(memory_id)
all_memories = m.get_all(user_id)
print(f"all_memories: {all_memories}\n")

#all_memories: []

from mem0 import Memory

config = {
    "llm": {
        "provider": "openai",
        "config": {
            "model": "gpt-4o-mini",
            "temperature": 0
        }
    },
    "graph_store": {
        "provider": "neo4j",
        "config": {
            "url": URL,
            "username": USERNAME,
            "password": PASSWORD
        }
    },
    "version": "v1.1"
}

m = Memory.from_config(config_dict=config)

user_id = "alice123"
m.add("I like painting", user_id=user_id)
m.add("I love to play badminton", user_id=user_id)
m.add("I hate playing badminton", user_id=user_id)
m.add("My friend name is john and john has a dog named tommy", user_id=user_id)
m.add("My name is alice", user_id=user_id)
m.add("John loves to hike and Harry loves to hike as well", user_id=user_id)
m.add("My friend peter is the spiderman", user_id=user_id)

m.get_all(user_id=user_id)
m.search("Who is spiderman?", user_id=user_id)
{'memories': [{'id': '44c054fc-a671-4b51-baee-76f4f73b8135',
   'memory': "Friend named Peter is referred to as 'the spiderman.'",
   'hash': '3990cbff8c4252e1ea3435a9f0eebf3d',
   'metadata': None,
   'score': 0.657225732037368,
   'created_at': '2024-08-27T15:53:17.605817-07:00',
   'updated_at': None,
   'user_id': 'alice123'}],
 'entities': [{'source': 'peter',
   'relation': 'identity',
   'destination': 'spiderman'}]}