AI 어시스턴트와 에이전트를 위한 지능형 메모리 레이어로, 사용자 선호도를 기억하고 시간에 따라 학습합니다.

pip install mem0ai

pip install mem0ai[nlp]
python -m spacy download en_core_web_sm

from openai import OpenAI
from mem0 import Memory

openai_client = OpenAI()
memory = Memory()

# 메모리 추가
memory.add("사용자 이름은 Alice이고 다크 모드를 선호합니다", user_id="alice")

# 메모리 검색
results = memory.search("Alice의 선호도는?", user_id="alice")
print(results)

from openai import OpenAI
from mem0 import Memory

openai_client = OpenAI()
memory = Memory()

def chat_with_memories(message: str, user_id: str = "default_user") -> str:
    relevant_memories = memory.search(query=message, filters={"user_id": user_id}, top_k=3)
    memories_str = "\n".join(f"- {entry['memory']}" for entry in relevant_memories["results"])
    
    system_prompt = f"당신은 도움이 되는 AI입니다. 쿼리와 메모리를 바탕으로 답변하세요.\n사용자 메모리:\n{memories_str}"
    messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": message}]
    response = openai_client.chat.completions.create(model="gpt-4o-mini", messages=messages)
    assistant_response = response.choices[0].message.content
    
    messages.append({"role": "assistant", "content": assistant_response})
    memory.add(messages, user_id=user_id)
    
    return assistant_response

if __name__ == "__main__":
    while True:
        user_input = input("당신: ").strip()
        if user_input.lower() == 'exit':
            break
        print(f"AI: {chat_with_memories(user_input)}")

npm install -g @mem0/cli

mem0 init
mem0 add "어두운 모드와 vim 단축키를 선호합니다" --user-id alice
mem0 search "Alice가 뭘 선호하나?" --user-id alice