braintrust-adk 0.2.0rc2

Braintrust Google ADK integration

braintrust-adk

SDK for integrating Braintrust with Google ADK (Agent Development Kit). This package provides automatic tracing and logging of ADK agent executions to Braintrust.

Installation

pip install braintrust-adk

Requirements

• Python >= 3.9
• Google ADK >= 1.9.0
• Braintrust Dataplane >= 1.1.19 (applicable only for hybrid deployment customers)
• Braintrust (with Otel support) >= 0.2.1

Quick Start

The braintrust-adk integration automatically traces your ADK agents' execution, including:

• Agent invocations and responses
• Tool calls and their results
• Parallel execution flows
• Multi-step agent reasoning

Basic Usage

from google.adk.agents import LlmAgent
from braintrust_adk import setup_braintrust

# Initialize Braintrust tracing
setup_braintrust(
    api_key="your-api-key",  # Or set BRAINTRUST_API_KEY env var
    project_name="my-adk-project"  # Optional: defaults to "default-google-adk-py"
)

# Create your ADK agent as normal
agent = LlmAgent(
    tools=[get_weather, get_current_time],
    model="gemini-2.0-flash-exp",
    system_instruction="You are a helpful assistant that can check weather and time."
)

# Use the agent - all interactions are automatically traced to Braintrust
response = agent.send_message("What's the weather like in New York?")
print(response.text)

Environment Variables

You can configure the integration using environment variables:

# Required for Braintrust tracing
export BRAINTRUST_API_KEY="your-api-key"

# Optional: specify project name or ID
export BRAINTRUST_PARENT="project_name:my-project"
# or
export BRAINTRUST_PARENT="project_id:your-project-id"

# Optional: enable debug logging
export OTEL_DEBUG="true"

Advanced Configuration

Using Project ID

If you know your Braintrust project ID, you can use it directly:

setup_braintrust(
    api_key="your-api-key",
    project_id="your-project-id"  # Use project ID instead of name
)

Custom Tools with Tracing

The integration automatically traces all tool calls made by your agents:

def get_weather(city: str) -> dict:
    """Get weather for a city."""
    # Your implementation here
    return {"status": "success", "temperature": 72, "city": city}

def search_flights(origin: str, destination: str, date: str) -> dict:
    """Search for flights."""
    # Your implementation here
    return {"flights": [...]}

# Create agent with multiple tools
agent = LlmAgent(
    tools=[get_weather, search_flights],
    model="gemini-2.0-flash-exp",
    system_instruction="You are a travel assistant."
)

# All tool calls are automatically traced
response = agent.send_message(
    "I need to fly from NYC to LA tomorrow. What's the weather like in LA?"
)

Examples

The examples/ directory contains complete working examples:

Viewing Traces in Braintrust

Once you've set up the integration, you can view your traces in the Braintrust dashboard:

1. Navigate to your project in Braintrust
2. Click on "Logs" to see all agent executions
3. Click on any log entry to see the full trace including:
   • Agent reasoning steps
   • Tool calls and responses
   • Token usage and latency metrics
   • Any errors or warnings

Development

To contribute to this integration:

# Clone the repository
git clone https://github.com/braintrustdata/sdk.git
cd sdk/integrations/adk-py

# Install in development mode
pip install -e ".[dev]"

# Run examples
cd examples
make dev

Related Resources

• Braintrust Documentation
• Google ADK Documentation