How to trace Pydantic AI with HoneyHive

pip install honeyhive pydantic-ai

import os
from honeyhive import HoneyHiveTracer
from pydantic_ai import Agent

tracer = HoneyHiveTracer.init(api_key=os.getenv("HH_API_KEY"))
Agent.instrument_all()

# Your existing PydanticAI code works unchanged

import asyncio
import os
from dataclasses import dataclass

from pydantic_ai import Agent, RunContext

from honeyhive import HoneyHiveTracer

tracer = HoneyHiveTracer.init(api_key=os.getenv("HH_API_KEY"))
Agent.instrument_all()


def lookup_order_status(order_id: str) -> dict:
    return {
        "ORD-1001": {"state": "shipped", "eta_days": 2},
        "ORD-1002": {"state": "processing", "eta_days": 5},
    }.get(order_id.upper(), {"state": "not_found"})


@dataclass
class CustomerContext:
    customer_name: str
    order_id: str


agent = Agent(
    "anthropic:claude-haiku-4-5",
    name="support_agent",
    deps_type=CustomerContext,
    instructions=(
        "You are a customer support agent. Use check_order to look up "
        "order status. Address the customer by name. Keep responses concise."
    ),
)


@agent.tool
def check_order(ctx: RunContext[CustomerContext], order_id: str) -> str:
    """Look up order status."""
    result = lookup_order_status(order_id)
    return f"[{ctx.deps.customer_name}] {result}"


async def main():
    customer = CustomerContext(customer_name="Alex Kim", order_id="ORD-1002")
    await agent.run(
        "My order ORD-1002 has been processing for a while. What's the status?",
        deps=customer,
    )


asyncio.run(main())

import asyncio
import os
from dataclasses import dataclass

from pydantic_ai import Agent, RunContext

from honeyhive import HoneyHiveTracer

tracer = HoneyHiveTracer.init(api_key=os.getenv("HH_API_KEY"))
Agent.instrument_all()


@dataclass
class CustomerContext:
    customer_name: str
    order_id: str


policy_agent = Agent(
    "anthropic:claude-haiku-4-5",
    name="policy_specialist",
    instructions=(
        "You are a policy specialist. Give concise answers about "
        "refund, cancellation, and shipping policies."
    ),
)

coordinator = Agent(
    "anthropic:claude-haiku-4-5",
    name="support_coordinator",
    deps_type=CustomerContext,
    instructions=(
        "You are a support coordinator. For policy questions, "
        "delegate to ask_policy_specialist. Combine the answer "
        "with any order context to form your response. Be concise."
    ),
)


@coordinator.tool
async def ask_policy_specialist(
    ctx: RunContext[CustomerContext], question: str
) -> str:
    """Delegate policy questions to the specialist."""
    result = await policy_agent.run(question, usage=ctx.usage)
    return str(result.output)


async def main():
    customer = CustomerContext(customer_name="Jordan Lee", order_id="ORD-1003")
    await coordinator.run(
        "Order ORD-1003 is delayed. Can I cancel and get a refund?",
        deps=customer,
    )


asyncio.run(main())

from honeyhive import HoneyHiveTracer
from pydantic_ai import Agent

tracer = HoneyHiveTracer.init(api_key=os.getenv("HH_API_KEY"))

# ✅ Correct — instrument before running
Agent.instrument_all()

result = await agent.run("Hello")