Add Memory Tool to AI Agent

Quick Start

1

Install Dependencies

Make sure you have Node.js installed, as the MCP server requires it:

pip install praisonaiagents mcp
2

Set API Key

Set your OpenAI API key as an environment variable in your terminal:

export OPENAI_API_KEY=your_openai_api_key_here
3

Create a file

Create a new file memory_agent.py with the following code:

from praisonaiagents import Agent, MCP
import os

# Get the memory file path from environment
memory_file_path = os.getenv("MEMORY_FILE_PATH", "/path/to/custom/memory.json")

# Use a single string command with Memory configuration
memory_agent = Agent(
    instructions="""You are a helpful assistant that can store and retrieve information.
    Use the available tools when relevant to manage memory operations.""",
    llm="gpt-4o-mini",
    tools=MCP("npx -y @modelcontextprotocol/server-memory",
            env={"MEMORY_FILE_PATH": memory_file_path})
)

memory_agent.start("Store this conversation in memory")
4

Run the Agent

Execute your script:

python memory_agent.py

Requirements

  • Python 3.10 or higher
  • Node.js installed on your system
  • OpenAI API key (for the agent’s LLM)
  • Write access to the specified memory file path

Was this page helpful?