Skip to main content

Deep Agent framework built on Pydantic-ai with planning, filesystem, and subagent capabilities

Project description

pydantic-deep

Pydantic Deep Agents

From framework to terminal -- autonomous AI agents that plan, code, and ship

Docs · PyPI · CLI · DeepResearch · Examples

PyPI version Python 3.10+ License: MIT Coverage Status CI Pydantic AI

Unlimited Context  •  Subagent Delegation  •  Persistent Memory  •  Lifecycle Hooks


Same Architecture as the Best

pydantic-deep implements the deep agent pattern -- the same architecture powering:

Product What They Built
Claude Code Anthropic's AI coding assistant
Manus AI Autonomous task execution
Devin AI software engineer

Now you can build the same thing -- or just use the CLI.

Inspired by: LangChain's Deep Agents research on autonomous agent architectures.


pydantic-deep is four things:

  1. A Python framework for building Claude Code-style agents with planning, filesystem access, subagents, memory, and unlimited context
  2. A CLI that gives you a terminal AI assistant out of the box
  3. An ACP adapter that runs deep agents inside editors like Zed
  4. DeepResearch -- a full-featured research agent with web UI, web search, diagrams, and sandboxed code execution

CLI -- Terminal AI Assistant

pydantic-deep CLI demo

pip install pydantic-deep[cli]
pydantic-deep

That's it. A Textual-based TUI launches with:

  • Streaming chat with tool call visualization
  • File read/write/edit, shell execution, glob, grep
  • Task planning and subagent delegation
  • Persistent memory across sessions
  • Context compression for unlimited conversations
  • Git-aware project context
  • /improve -- learn from past sessions and update context files
  • /skills, /diff, /model, /provider, /compact, and more
  • Customizable themes, skills, and hooks
# Launch TUI (default)
pydantic-deep

# Pick a model
pydantic-deep tui --model anthropic:claude-sonnet-4-6

# Manage config
pydantic-deep config set model anthropic:claude-sonnet-4-6

# List skills
pydantic-deep skills list

See CLI docs for the full reference.


Framework -- Build Your Own Agent

pip install pydantic-deep

Requires pydantic-ai >= 1.77.0.

from pydantic_ai_backends import StateBackend
from pydantic_deep import create_deep_agent, create_default_deps

agent = create_deep_agent()
deps = create_default_deps(StateBackend())

result = await agent.run("Create a todo list for building a REST API", deps=deps)

One function call gives you an agent with planning, filesystem tools, subagents, skills, context management, and cost tracking. Everything is toggleable:

agent = create_deep_agent(
    model="anthropic:claude-sonnet-4-6",
    include_todo=True,          # Task planning
    include_filesystem=True,    # File read/write/edit/execute
    include_subagents=True,     # Delegate to subagents
    include_skills=True,        # Domain-specific skills from SKILL.md files
    include_memory=True,        # Persistent MEMORY.md across sessions
    include_plan=True,          # Structured planning before execution
    include_teams=True,         # Multi-agent teams with shared TODOs
    web_search=True,            # WebSearch capability
    web_fetch=True,             # WebFetch capability
    thinking="high",            # Thinking/reasoning effort
    context_manager=True,       # Auto-summarization for unlimited context
    cost_tracking=True,         # Token/USD budget enforcement
    include_checkpoints=True,   # Save, rewind, and fork conversations
)

Structured Output

from pydantic import BaseModel

class CodeReview(BaseModel):
    summary: str
    issues: list[str]
    score: int

agent = create_deep_agent(output_type=CodeReview)
result = await agent.run("Review the auth module", deps=deps)
print(result.output.score)  # Type-safe!

Context Management

from pydantic_deep import create_summarization_processor

processor = create_summarization_processor(
    trigger=("tokens", 100000),
    keep=("messages", 20),
)
agent = create_deep_agent(history_processors=[processor])

Hooks (Claude Code-Style)

from pydantic_deep import Hook, HookEvent

agent = create_deep_agent(
    hooks=[
        Hook(
            event=HookEvent.PRE_TOOL_USE,
            command="echo 'Tool called: $TOOL_NAME' >> /tmp/audit.log",
        ),
    ],
)

Cost Tracking

agent = create_deep_agent(
    cost_tracking=True,
    cost_budget_usd=5.0,
    on_cost_update=lambda info: print(f"Cost: ${info.total_usd:.4f}"),
)

MCP Servers

Connect to any MCP server via pydantic-ai's MCP capability:

from pydantic_ai.capabilities import MCP

agent = create_deep_agent(
    capabilities=[
        MCP(url="https://mcp.example.com/api"),
    ],
)

Subagents

A built-in research subagent is included by default. Add your own:

agent = create_deep_agent(
    subagents=[
        {
            "name": "code-reviewer",
            "description": "Reviews code for quality issues",
            "instructions": "Check for security, performance, error handling...",
        },
    ],
)
# The main agent delegates: task(description="Review auth.py", subagent_type="code-reviewer")

All subagents are full deep agents with filesystem, web, and memory tools. You only provide the specialized instructions — the framework adds BASE_PROMPT automatically.

Project Files

pydantic-deep recognizes three special markdown files:

File Purpose Who Sees It
AGENTS.md Project instructions, conventions, architecture Main agent + subagents
SOUL.md Agent personality, style, user preferences Main agent only
MEMORY.md Persistent memory across sessions (read/write/update tools) Per-agent (isolated)
agent = create_deep_agent(
    context_discovery=True,  # Auto-discover AGENTS.md and SOUL.md at backend root
    include_memory=True,     # MEMORY.md with read/write/update tools (on by default)
)

AGENTS.md follows the agents.md spec — compatible with other agent frameworks.

See the full API reference for all options.


ACP -- Editor Integration (Zed)

zed.png

Run pydantic-deep agents inside Zed via the Agent Client Protocol:

pip install pydantic-deep[acp]
python -m apps.acp

Add to Zed settings (Cmd+,):

{
  "agent_servers": {
    "pydantic-deep": {
      "type": "custom",
      "command": "/path/to/venv/bin/python",
      "args": ["-m", "apps.acp"],
      "cwd": "/path/to/pydantic-deep"
    }
  }
}

API keys are loaded from ~/.pydantic-deep/.env (global) or .pydantic-deep/.env (per-project):

mkdir -p ~/.pydantic-deep
echo 'OPENROUTER_API_KEY=sk-or-your-key' > ~/.pydantic-deep/.env

See ACP README for full configuration.


DeepResearch -- Reference App

A full-featured research agent with web UI, built entirely on pydantic-deep.

Planner subagent asks clarifying questions

Plan Mode -- planner asks clarifying questions

Parallel subagent research

Parallel Subagents -- 5 agents researching simultaneously

Excalidraw canvas

Excalidraw Canvas -- live diagrams synced with agent

File browser

File Browser -- workspace files with inline preview

Web search (Tavily, Brave, Jina), sandboxed code execution, Excalidraw diagrams, subagents, plan mode, report export, and more.

cd apps/deepresearch
uv sync
cp .env.example .env  # Add your API keys
uv run deepresearch    # Open http://localhost:8080

See apps/deepresearch/README.md for full setup.


Architecture

pydantic-deep v0.3.0 uses pydantic-ai's native Capabilities API (Agent(capabilities=[...])) for all cross-cutting concerns. This replaces the previous middleware wrapping approach and provides a cleaner, more composable architecture.

Capabilities

All lifecycle features are implemented as capabilities that extend AbstractCapability from pydantic-ai:

Capability Package What It Does
CostTracking pydantic-ai-shields Token/USD budget enforcement and real-time cost callbacks
ContextManagerCapability summarization-pydantic-ai Auto-compression when approaching token budget
HooksCapability pydantic-deep Claude Code-style lifecycle hooks on tool events
CheckpointMiddleware pydantic-deep Save, rewind, and fork conversation state
WebSearch / WebFetch pydantic-ai (built-in) Web search and URL fetching

pydantic-deep also provides 5 internal capabilities that are automatically wired up when their corresponding include_* flags are set:

Internal Capability Activated By What It Does
SkillsCapability include_skills=True Domain-specific skills from SKILL.md files
ContextFilesCapability context_files / context_discovery Auto-discover and inject DEEP.md, AGENTS.md, CLAUDE.md
MemoryCapability include_memory=True Persistent MEMORY.md across sessions
TeamCapability include_teams=True Multi-agent teams with shared TODOs and message bus
PlanCapability include_plan=True Structured planning before execution

Component Packages

Every component is modular and works standalone:

Component Package What It Does
Backends pydantic-ai-backend File storage, Docker/Daytona sandbox
Planning pydantic-ai-todo Task tracking with dependencies
Subagents subagents-pydantic-ai Sync/async delegation, cancellation
Summarization summarization-pydantic-ai LLM summaries or sliding window
Shields pydantic-ai-shields Cost tracking, input/output/tool blocking
                              pydantic-deep
+---------------------------------------------------------------------+
|                                                                     |
|   +----------+ +----------+ +----------+ +----------+ +---------+   |
|   | Planning | |Filesystem| | Subagents| |  Skills  | |  Teams  |   |
|   +----+-----+ +----+-----+ +----+-----+ +----+-----+ +----+----+   |
|        |            |            |            |            |        |
|        +------------+-----+------+------------+------------+        |
|                           |                                         |
|                           v                                         |
|  Summarization --> +------------------+ <-- Capabilities            |
|  Checkpointing --> |    Deep Agent    | <-- Hooks                   |
|  Cost Tracking --> |   (pydantic-ai)  | <-- Memory                  |
|                    +--------+---------+                             |
|                             |                                       |
|           +-----------------+-----------------+                     |
|           v                 v                 v                     |
|    +------------+    +------------+    +------------+               |
|    |   State    |    |   Local    |    |   Docker   |               |
|    |  Backend   |    |  Backend   |    |  Sandbox   |               |
|    +------------+    +------------+    +------------+               |
|                                                                     |
+---------------------------------------------------------------------+

All Features

Click to expand full feature list

Core Toolsets

  • Planning -- Task tracking with subtasks, dependencies, cycle detection. PostgreSQL storage. Event system.
  • Filesystem -- ls, read_file, write_file, edit_file, glob, grep, execute. Docker sandbox. Permission system.
  • Subagents -- Sync/async delegation. Background task management. Soft/hard cancellation.
  • Summarization -- LLM-based summaries or zero-cost sliding window. Trigger on tokens, messages, or fraction.

Capabilities

  • CostTracking -- Token/USD budgets with automatic enforcement and real-time callbacks (from pydantic-ai-shields).
  • ContextManagerCapability -- Auto-compression when approaching token budget (from summarization-pydantic-ai).
  • HooksCapability -- Claude Code-style lifecycle hooks. Shell commands on tool events. Audit logging, safety gates.
  • CheckpointMiddleware -- Save state at intervals. Rewind or fork sessions. In-memory and file-based stores. Extends AbstractCapability.
  • WebSearch / WebFetch -- Built-in pydantic-ai capabilities for web search and URL fetching.
  • SkillsCapability -- Domain-specific skills loaded from SKILL.md files.
  • ContextFilesCapability -- Auto-discover and inject DEEP.md, AGENTS.md, CLAUDE.md, SOUL.md into the system prompt.
  • MemoryCapability -- Persistent MEMORY.md across sessions, auto-injected into system prompt.
  • TeamCapability -- Multi-agent teams with shared TODO lists, claiming, dependency tracking, and peer-to-peer message bus.
  • PlanCapability -- Dedicated planner subagent for structured planning before execution.

Advanced

  • Agent Teams -- Shared TODO lists with claiming and dependency tracking. Peer-to-peer message bus.
  • Persistent Memory -- MEMORY.md that persists across sessions. Auto-injected into system prompt.
  • Context Files -- Auto-discover and inject AGENT.md into the system prompt.
  • Output Styles -- Built-in (concise, explanatory, formal, conversational) or custom from files.
  • Plan Mode -- Dedicated planner subagent for structured planning before execution.
  • Eviction Processor -- Evict large tool outputs to files. Keep context lean while preserving data.
  • Patch Tool Calls -- On resume, patch stale tool call results for clean history.
  • Custom Tool Descriptions -- Override any tool's description via descriptions parameter.
  • Custom Commands -- /commit, /pr, /review, /test, /fix, /explain. Three-scope discovery: built-in, user, project.
  • Structured Output -- Type-safe responses with Pydantic models via output_type.
  • Human-in-the-Loop -- Confirmation workflows for sensitive operations.
  • Streaming -- Full streaming support for real-time responses.
  • Image Support -- Multi-modal analysis with image inputs.

Contributing

git clone https://github.com/vstorm-co/pydantic-deepagents.git
cd pydantic-deepagents
make install
make test  # 100% coverage required
make all   # lint + typecheck + test

Star History

Star History


License

MIT -- see LICENSE


Need help implementing this in your company?

We're Vstorm -- an Applied Agentic AI Engineering Consultancy
with 30+ production AI agent implementations.

Talk to us



Made with care by Vstorm

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pydantic_deep-0.3.4.tar.gz (14.4 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pydantic_deep-0.3.4-py3-none-any.whl (2.1 MB view details)

Uploaded Python 3

File details

Details for the file pydantic_deep-0.3.4.tar.gz.

File metadata

  • Download URL: pydantic_deep-0.3.4.tar.gz
  • Upload date:
  • Size: 14.4 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for pydantic_deep-0.3.4.tar.gz
Algorithm Hash digest
SHA256 e756ca7549f1256a6539e3fe40b2f7f4e6f4c4f2e2821eb4b52716ca256fe979
MD5 5872a1e9cd550f990003d8139b40b031
BLAKE2b-256 cd166f4b24d4e5c8746eb645007208c274c52a76013c6284946b4160e825c647

See more details on using hashes here.

Provenance

The following attestation bundles were made for pydantic_deep-0.3.4.tar.gz:

Publisher: publish.yml on vstorm-co/pydantic-deepagents

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file pydantic_deep-0.3.4-py3-none-any.whl.

File metadata

  • Download URL: pydantic_deep-0.3.4-py3-none-any.whl
  • Upload date:
  • Size: 2.1 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for pydantic_deep-0.3.4-py3-none-any.whl
Algorithm Hash digest
SHA256 6c3138350d44faaa8d7ad1e539de85c70638a2b009726349a680d3f5eadc8d57
MD5 00723cd979569a4e52be5e78b56db0f4
BLAKE2b-256 4f8bb5d59bed75f70be81a8d0d32fb0b0a28246c7b063253e6874da43c90916e

See more details on using hashes here.

Provenance

The following attestation bundles were made for pydantic_deep-0.3.4-py3-none-any.whl:

Publisher: publish.yml on vstorm-co/pydantic-deepagents

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page