from langchain.memory import ConversationBufferMemory

# Memory for storing conversation history
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)