Pydantic AI + Atulya Memory

Give your Pydantic AI agents persistent long-term memory. Chat with an assistant multiple times and watch it remember what you told it in previous sessions.

What This Demonstrates​

• Memory tools — retain, recall, and reflect via create_atulya_tools()
• Auto-injected context — relevant memories in every run via memory_instructions()
• Persistent memory across sessions — the agent remembers between script runs
• Interactive chat loop with message history reuse

Architecture​

Session 1:
    You: "I'm a Python developer working on a FastAPI project"
    │
    ├─ memory_instructions() ──► recalls prior context (empty on first run)
    ├─ Agent decides to call atulya_retain ──► stores the fact
    └─ Agent responds with acknowledgement

Session 2:
    You: "What do you know about me?"
    │
    ├─ memory_instructions() ──► injects "User is a Python developer..."
    ├─ Agent calls atulya_recall ──► finds stored facts
    └─ Agent responds with everything it remembers

Prerequisites​

1. Atulya running

   export OPENAI_API_KEY=your-key
   
   docker run --rm -it --pull always -p 8888:8888 -p 9999:9999 \
     -e ATULYA_API_LLM_API_KEY=$OPENAI_API_KEY \
     -e ATULYA_API_LLM_MODEL=o3-mini \
     -v $HOME/.atulya-docker:/home/atulya/.pg0 \
     ghcr.io/eight-atulya/atulya:latest
   

2. OpenAI API key (for Pydantic AI's LLM)

   export OPENAI_API_KEY=your-key
   

3. Install dependencies

   cd applications/pydantic-ai-memory
   pip install -r requirements.txt
   

Quick Start​

Interactive Chat​

python personal_assistant.py

Example session:

Personal assistant ready (bank: personal-assistant)
Type 'quit' or 'exit' to stop.

You: I'm a Python developer and I love hiking on weekends
Assistant: I've noted that! You're a Python developer who enjoys weekend hiking.

You: What do you know about me?
Assistant: From my memory, I know that you're a Python developer and you
love hiking on weekends.

You: quit

Run it again — the agent still remembers:

You: What are my hobbies?
Assistant: Based on my memories, you enjoy hiking on weekends!

Single Query​

python personal_assistant.py "What do you remember about my preferences?"

Reset Memory​

python personal_assistant.py --reset

How It Works​

1. Create a Atulya Client​

from atulya_client import Atulya

client = Atulya(base_url="http://localhost:8888")

2. Create Memory Tools​

create_atulya_tools() returns Pydantic AI Tool instances the agent can call:

from atulya_pydantic_ai import create_atulya_tools

tools = create_atulya_tools(client=client, bank_id="personal-assistant")
# Returns: [atulya_retain, atulya_recall, atulya_reflect]

3. Add Memory Instructions​

memory_instructions() returns an async callable that auto-recalls relevant memories and injects them into the system prompt on every run:

from atulya_pydantic_ai import memory_instructions

instructions_fn = memory_instructions(
    client=client,
    bank_id="personal-assistant",
    query="important context about the user",
    max_results=5,
)

4. Wire Up the Agent​

from pydantic_ai import Agent

agent = Agent(
    "openai:gpt-4o-mini",
    system_prompt="You are a helpful assistant with long-term memory...",
    tools=tools,
    instructions=[instructions_fn],
)

result = await agent.run("What do you know about me?")

Core Files​

Customization​

Use Only Tools (No Auto-Injection)​

Let the agent decide when to search memory, rather than always injecting context:

agent = Agent(
    "openai:gpt-4o-mini",
    tools=create_atulya_tools(client=client, bank_id="my-bank"),
)

Use Only Instructions (No Tools)​

Auto-inject memories without giving the agent explicit retain/recall/reflect tools:

agent = Agent(
    "openai:gpt-4o-mini",
    instructions=[memory_instructions(client=client, bank_id="my-bank")],
)

Select Specific Tools​

tools = create_atulya_tools(
    client=client,
    bank_id="my-bank",
    include_retain=True,
    include_recall=True,
    include_reflect=False,  # Omit reflect
)

Use a Different Model​

Any Pydantic AI model works:

agent = Agent(
    "anthropic:claude-sonnet-4-20250514",
    tools=create_atulya_tools(client=client, bank_id="my-bank"),
)

Common Issues​

"Connection refused"

• Make sure Atulya is running on localhost:8888

"OPENAI_API_KEY not set"

export OPENAI_API_KEY=your-key

"No module named 'atulya_pydantic_ai'"

pip install -r requirements.txt

Built with:

• Pydantic AI - Type-safe AI agent framework
• atulya-pydantic-ai - Atulya memory tools for Pydantic AI
• Atulya - Long-term memory for AI agents