Team with Memory Manager

This example demonstrates how to use persistent memory with a team. After each run, user memories are created and updated, allowing the team to remember information about users across sessions and provide personalized experiences.

Code

cookbook/examples/teams/memory/01_team_with_memory_manager.py

"""
This example shows you how to use persistent memory with an Agent.

After each run, user memories are created/updated.

To enable this, set `enable_user_memories=True` in the Agent config.
"""

from uuid import uuid4

from agno.agent import Agent
from agno.db.postgres import PostgresDb
from agno.memory import MemoryManager  # noqa: F401
from agno.models.openai import OpenAIChat
from agno.team import Team

db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
db = PostgresDb(db_url=db_url)

session_id = str(uuid4())
john_doe_id = "[e-mail]"

# 1. Create memories by setting `enable_user_memories=True` in the Agent
agent = Agent(
    model=OpenAIChat(id="gpt-5-mini"),
)
team = Team(
    model=OpenAIChat(id="gpt-5-mini"),
    members=[agent],
    db=db,
    enable_user_memories=True,
)

team.print_response(
    "My name is John Doe and I like to hike in the mountains on weekends.",
    stream=True,
    user_id=john_doe_id,
    session_id=session_id,
)

team.print_response(
    "What are my hobbies?", stream=True, user_id=john_doe_id, session_id=session_id
)

# 2. Set a custom MemoryManager on the agent
# memory_manager = MemoryManager(model=OpenAIChat(id="gpt-5-mini"))

# memory_manager.clear()

# agent = Agent(
#     model=OpenAIChat(id="gpt-5-mini"),
#     memory_manager=memory_manager,
# )

# team = Team(
#     model=OpenAIChat(id="gpt-5-mini"),
#     members=[agent],
#     db=db,
#     enable_user_memories=True,
# )

# team.print_response(
#     "My name is John Doe and I like to hike in the mountains on weekends.",
#     stream=True,
#     user_id=john_doe_id,
#     session_id=session_id,
# )

# # You can also get the user memories from the agent
# memories = agent.get_user_memories(user_id=john_doe_id)
# print("John Doe's memories:")
# pprint(memories)

Usage

1. Create a virtual environment

   Open the Terminal and create a python virtual environment.

   Mac

   python3 -m venv .venv
   source .venv/bin/activate
   
2. Set up PostgreSQL database

   ./cookbook/run_pgvector.sh
   
3. Install required libraries

   pip install agno openai psycopg sqlalchemy
   
4. Set environment variables

   export OPENAI_API_KEY=****
   
5. Run the agent

   python cookbook/examples/teams/memory/01_team_with_memory_manager.py