Pydantic AI Integration

import agentbasis
from agentbasis.frameworks.pydanticai import instrument

# Initialize AgentBasis first
agentbasis.init(api_key="your-api-key", agent_id="your-agent-id")

# Enable global instrumentation for all Pydantic AI agents
instrument()

from pydantic_ai import Agent

agent = Agent("openai:gpt-4")
result = agent.run_sync("Hello!")

print(result.data)

from agentbasis.frameworks.pydanticai import instrument

# Don't log prompts and completions (for sensitive data)
instrument(include_content=False)

# Include binary content like images (increases trace size)
instrument(include_content=True, include_binary_content=True)

from agentbasis.frameworks.pydanticai import get_instrumentation_settings
from pydantic_ai import Agent

# Agent with full tracing
agent_full = Agent(
    "openai:gpt-4",
    instrument=get_instrumentation_settings()
)

# Agent without content logging (privacy mode)
agent_private = Agent(
    "openai:gpt-4",
    instrument=get_instrumentation_settings(include_content=False)
)

from agentbasis.frameworks.pydanticai import create_traced_agent
import agentbasis

# Create an agent pre-configured with tracing and context
agent = create_traced_agent(
    "openai:gpt-4",
    system_prompt="You are a helpful assistant."
)

# Set user context - automatically included in traces
agentbasis.set_user("user-123")
agentbasis.set_session("session-456")

result = agent.run_sync("Hello!")

from agentbasis.frameworks.pydanticai import get_metadata_callback
from pydantic_ai import Agent
import agentbasis

agent = Agent(
    "openai:gpt-4",
    metadata=get_metadata_callback()
)

# Set context before running
agentbasis.set_user("user-123")

# This run will include user_id in the trace metadata
result = agent.run_sync("Hello!")

from pydantic_ai import Agent
import asyncio

agent = Agent("openai:gpt-4")

async def main():
    result = await agent.run("Hello!")
    print(result.data)

asyncio.run(main())