Update client in README (#2215)

This commit is contained in:
Dev Khant
2025-02-17 00:14:23 +05:30
committed by GitHub
parent 064d28626b
commit 095189d39a

View File

@@ -82,11 +82,11 @@ from openai import OpenAI
from mem0 import Memory
openai_client = OpenAI()
mem0 = Memory()
memory = Memory()
def chat_with_memories(message: str, user_id: str = "default_user") -> str:
# Retrieve relevant memories
relevant_memories = mem0.search(query=message, user_id=user_id, limit=3)
relevant_memories = memory.search(query=message, user_id=user_id, limit=3)
memories_str = "\n".join(f"- {entry['memory']}" for entry in relevant_memories)
# Generate Assistant response
@@ -97,7 +97,7 @@ def chat_with_memories(message: str, user_id: str = "default_user") -> str:
# Create new memories from the conversation
messages.append({"role": "assistant", "content": assistant_response})
mem0.add(messages, user_id=user_id)
memory.add(messages, user_id=user_id)
return assistant_response