Pydantic AI

Automatically capture agent runs, LLM calls, and tool invocations within the Pydantic AI framework via its native OpenTelemetry instrumentation.

Install

Python

pip install traceroot pydantic-ai

Set up

Python

import traceroot
from traceroot import Integration

traceroot.initialize(integrations=[Integration.PYDANTIC_AI])

Usage

Once initialized, all Pydantic AI agent runs and tool invocations are captured automatically:

import traceroot
from traceroot import Integration
from pydantic_ai import Agent

traceroot.initialize(integrations=[Integration.PYDANTIC_AI])

agent = Agent(
    "openai:gpt-4o-mini",
    system_prompt="You are a helpful assistant.",
)

result = agent.run_sync("What is the capital of France?")
print(result.output)

traceroot.flush()

Verify

Run your code and confirm the trace arrives:

In the TraceRoot UI: Navigate to Tracing > Traces in TraceRoot Cloud. You should see a trace with root span invoke_agent agent, named after the agent — Pydantic AI infers agent from the variable name unless you pass Agent(name=...).
Via the CLI: Run traceroot traces list to inspect your recent traces from the terminal.

What gets captured

Attribute	Description
Agent runs	Each `agent.run()` / `agent.run_sync()` / `agent.run_stream()` invocation
Tool calls	Each tool invocation with input arguments and results
LLM calls	Raw completion requests with input messages, output messages, and model name
Tokens & Cost	Input and output token counts; cost when the model ID matches TraceRoot’s pricing table
Cache tokens	Cache read and write token counts when reported by the provider
Latency	Duration per agent run and per span

Limitations

None known.

Run the example

Clone the repo and run a complete agent end-to-end.

Python

Run the Python example

Next steps

Python SDK Reference

Explore the full Python SDK reference and instrumentation options.

Cost Tracking

Learn how TraceRoot calculates token costs and tracks model spend.

Overview

Native

Frameworks

Model Providers

Internal Model Providers

Install

Set up

Usage

Verify

What gets captured

Limitations

Run the example

Python

Next steps

Python SDK Reference

Cost Tracking

​Install

​Set up

​Usage

​Verify

​What gets captured

​Limitations

​Run the example

Python

​Next steps

Python SDK Reference

Cost Tracking

Install

Set up

Usage

Verify

What gets captured

Limitations

Run the example

Next steps