pai-agent-sdk 0.4.1

Production-ready SDK for building AI agents with Pydantic AI

pai-agent-sdk

Production-ready SDK for building AI agents with Pydantic AI.

Early Stage Warning: This project is in active development. APIs may change frequently between versions. We welcome feedback and contributions to help stabilize the SDK.

Key Features

• Environment-based Architecture: Inject file operations, shell access, and resumable resources via Environment for clean separation of concerns(check agent-environment)
• Resumable Sessions: Export and restore AgentContext state for multi-turn conversations across restarts
• Hierarchical Agents: Delegate specialized tasks to subagents with automatic tool inheritance
• Human-in-the-Loop: Built-in approval workflows for sensitive tool operations
• Streaming Support: Real-time streaming of agent responses and tool executions

Installation

# Recommended: install with all optional dependencies
pip install pai-agent-sdk[all]
uv add pai-agent-sdk[all]

# Or install individual extras as needed
pip install pai-agent-sdk[docker]    # Docker sandbox support
pip install pai-agent-sdk[web]       # Web tools (tavily, firecrawl, markitdown)
pip install pai-agent-sdk[document]  # Document processing (pymupdf, markitdown)

Quick Start

from pai_agent_sdk.agents import create_agent, stream_agent

# create_agent returns AgentRuntime (not a context manager)
runtime = create_agent("openai:gpt-4o")

# stream_agent manages runtime lifecycle automatically
async with stream_agent(runtime, "Hello") as streamer:
    async for event in streamer:
        print(event)

Examples

Check out the examples/ directory for production-ready patterns:

Example	Description
general.py	Complete production pattern with streaming, HITL approval, and session persistence
deepresearch.py	Autonomous research agent with web search and content extraction
browser_use.py	Browser automation with Docker-based headless Chrome sandbox

For Agent Users

If you're using an AI agent (e.g., Claude, Cursor) that supports skills:

• Clone this repo: The SKILL.md file in the repository root provides comprehensive guidance for agents
• Download release package: Get the latest SKILL.zip from the Releases page (automatically built during each release)

Configuration

Copy examples/.env.example to examples/.env and configure your API keys.

Documentation

• AgentContext & Sessions - Session state, resumable sessions, extending context
• Toolset Architecture - Create tools, use hooks, handle errors, extend Toolset
• Subagent System - Hierarchical agents, builtin presets, markdown configuration
• Custom Environments - Environment lifecycle, resource management
• Resumable Resources - Export and restore resource states across restarts
• Model Configuration - Provider setup, gateway mode
• Logging Configuration - Configure SDK logging levels

Development

See CONTRIBUTING.md for development setup and guidelines.