How to add memory to Claude chatbot

pip install anthropic>=0.20

import os
import anthropic

client = anthropic.Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"])

# Initialize conversation history
conversation = []

# System prompt defining assistant behavior
system_prompt = "You are a helpful assistant."

# Function to send a message and update memory

def chat_with_memory(user_input):
    global conversation
    # Append user message
    conversation.append({"role": "user", "content": user_input})

    # Call Claude with full conversation history
    response = client.messages.create(
        model="claude-3-5-haiku-20241022",
        max_tokens=500,
        system=system_prompt,
        messages=conversation
    )

    assistant_reply = response.content[0].text
    # Append assistant message
    conversation.append({"role": "assistant", "content": assistant_reply})

    return assistant_reply

# Example usage
print(chat_with_memory("Hello, who won the World Series in 2020?"))
print(chat_with_memory("Can you remind me what you just said?"))

import asyncio
import anthropic

client = anthropic.Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"])

conversation = []
system_prompt = "You are a helpful assistant."

async def chat_with_memory_async(user_input):
    global conversation
    conversation.append({"role": "user", "content": user_input})

    response = await client.messages.acreate(
        model="claude-3-5-haiku-20241022",
        max_tokens=500,
        system=system_prompt,
        messages=conversation
    )

    assistant_reply = response.content[0].text
    conversation.append({"role": "assistant", "content": assistant_reply})
    return assistant_reply

# Example async usage
async def main():
    reply1 = await chat_with_memory_async("Hello!")
    print(reply1)
    reply2 = await chat_with_memory_async("What did I just say?")
    print(reply2)

asyncio.run(main())