ref(adk): migrate ADK wrapper to integrations API

Convert the Google ADK (Agent Development Kit) instrumentation from the
legacy wrappers pattern to the new integrations API introduced in #118.

- Extract tracing helpers into integrations/adk/tracing.py
- Create ADK patchers (agent, runner, flow, mcp_tool) in patchers.py
- Add AdkIntegration class with proper registration
- Move tests and cassettes from wrappers/adk/ to integrations/adk/
- Move auto_instrument test script to integrations/auto_test_scripts/
- Slim down wrappers/adk/__init__.py to delegate to integrations
- Update noxfile.py with ADK integration test sessions
- Update integrations base and registry for ADK support

Author: AbhiPrasad

py/noxfile.py

@@ -41,8 +41,6 @@ def _pinned_python_version():
 SRC_DIR = "braintrust"
 WRAPPER_DIR = "braintrust/wrappers"
 INTEGRATION_DIR = "braintrust/integrations"
-INTEGRATION_AUTO_TEST_DIR = "braintrust/integrations/auto_test_scripts"
-ANTHROPIC_INTEGRATION_DIR = "braintrust/integrations/anthropic"
 CONTRIB_DIR = "braintrust/contrib"
 DEVSERVER_DIR = "braintrust/devserver"
 
@@ -198,8 +196,8 @@ def test_google_adk(session, version):
     """Test Google ADK integration."""
     _install_test_deps(session)
     _install(session, "google-adk", version)
-    _run_tests(session, f"{WRAPPER_DIR}/adk/test_adk.py")
-    _run_tests(session, f"{WRAPPER_DIR}/adk/test_adk_mcp_tool.py")
+    _run_tests(session, f"{INTEGRATION_DIR}/adk/test_adk.py")
+    _run_tests(session, f"{INTEGRATION_DIR}/adk/test_adk_mcp_tool.py")
     _run_core_tests(session)
 
 
@@ -407,7 +405,12 @@ def _run_core_tests(session):
     _run_tests(
         session,
         SRC_DIR,
-        ignore_paths=[WRAPPER_DIR, INTEGRATION_AUTO_TEST_DIR, ANTHROPIC_INTEGRATION_DIR, CONTRIB_DIR, DEVSERVER_DIR],
+        ignore_paths=[
+            WRAPPER_DIR,
+            INTEGRATION_DIR,
+            CONTRIB_DIR,
+            DEVSERVER_DIR,
+        ],
     )
 
 

py/src/braintrust/auto.py

@@ -9,7 +9,7 @@
 import logging
 from contextlib import contextmanager
 
-from braintrust.integrations import AnthropicIntegration, IntegrationPatchConfig
+from braintrust.integrations import ADKIntegration, AnthropicIntegration, IntegrationPatchConfig
 
 
 __all__ = ["auto_instrument"]
@@ -39,7 +39,7 @@ def auto_instrument(
     agno: bool = True,
     claude_agent_sdk: bool = True,
     dspy: bool = True,
-    adk: bool = True,
+    adk: InstrumentOption = True,
 ) -> dict[str, bool]:
     """
     Auto-instrument supported AI/ML libraries for Braintrust tracing.
@@ -109,14 +109,14 @@ def auto_instrument(
     results = {}
 
     openai_enabled = _normalize_bool_option("openai", openai)
-    anthropic_enabled, anthropic_config = _normalize_anthropic_option(anthropic)
+    anthropic_enabled, anthropic_config = _normalize_instrument_option("anthropic", anthropic)
     litellm_enabled = _normalize_bool_option("litellm", litellm)
     pydantic_ai_enabled = _normalize_bool_option("pydantic_ai", pydantic_ai)
     google_genai_enabled = _normalize_bool_option("google_genai", google_genai)
     agno_enabled = _normalize_bool_option("agno", agno)
     claude_agent_sdk_enabled = _normalize_bool_option("claude_agent_sdk", claude_agent_sdk)
     dspy_enabled = _normalize_bool_option("dspy", dspy)
-    adk_enabled = _normalize_bool_option("adk", adk)
+    adk_enabled, adk_config = _normalize_instrument_option("adk", adk)
 
     if openai_enabled:
         results["openai"] = _instrument_openai()
@@ -135,7 +135,7 @@ def auto_instrument(
     if dspy_enabled:
         results["dspy"] = _instrument_dspy()
     if adk_enabled:
-        results["adk"] = _instrument_adk()
+        results["adk"] = _instrument_integration(ADKIntegration, patch_config=adk_config)
 
     return results
 
@@ -164,15 +164,15 @@ def _normalize_bool_option(name: str, option: bool) -> bool:
     raise TypeError(f"auto_instrument option {name!r} must be a bool, got {type(option).__name__}")
 
 
-def _normalize_anthropic_option(option: InstrumentOption) -> tuple[bool, IntegrationPatchConfig | None]:
+def _normalize_instrument_option(name: str, option: InstrumentOption) -> tuple[bool, IntegrationPatchConfig | None]:
     if isinstance(option, bool):
         return option, None
 
     if isinstance(option, IntegrationPatchConfig):
         return True, option
 
     raise TypeError(
-        f"auto_instrument option 'anthropic' must be a bool or IntegrationPatchConfig, got {type(option).__name__}"
+        f"auto_instrument option {name} must be a bool or IntegrationPatchConfig, got {type(option).__name__}"
     )
 
 
@@ -222,11 +222,3 @@ def _instrument_dspy() -> bool:
 
         return patch_dspy()
     return False
-
-
-def _instrument_adk() -> bool:
-    with _try_patch():
-        from braintrust.wrappers.adk import setup_adk
-
-        return setup_adk()
-    return False

py/src/braintrust/integrations/__init__.py

@@ -1,5 +1,6 @@
+from .adk import ADKIntegration
 from .anthropic import AnthropicIntegration
 from .base import IntegrationPatchConfig
 
 
-__all__ = ["AnthropicIntegration", "IntegrationPatchConfig"]
+__all__ = ["ADKIntegration", "AnthropicIntegration", "IntegrationPatchConfig"]

py/src/braintrust/integrations/adk/__init__.py

@@ -0,0 +1,64 @@
+"""Braintrust integration for Google ADK."""
+
+import logging
+import warnings
+
+from braintrust.logger import NOOP_SPAN, current_span, init_logger
+
+from .integration import ADKIntegration
+from .patchers import (
+    wrap_agent,
+    wrap_flow,
+    wrap_mcp_tool,
+    wrap_runner,
+)
+from .tracing import _create_thread_wrapper
+
+
+logger = logging.getLogger(__name__)
+
+__all__ = [
+    "ADKIntegration",
+    "_create_thread_wrapper",
+    "setup_adk",
+    "setup_braintrust",
+    "wrap_agent",
+    "wrap_runner",
+    "wrap_flow",
+    "wrap_mcp_tool",
+]
+
+
+def setup_braintrust(*args, **kwargs):
+    warnings.warn("setup_braintrust is deprecated, use setup_adk instead", DeprecationWarning, stacklevel=2)
+    return setup_adk(*args, **kwargs)
+
+
+def setup_adk(
+    api_key: str | None = None,
+    project_id: str | None = None,
+    project_name: str | None = None,
+    SpanProcessor: type | None = None,
+) -> bool:
+    """
+    Setup Braintrust integration with Google ADK. Will automatically patch Google ADK agents, runners, flows, and MCP tools for automatic tracing.
+
+    If you prefer manual patching take a look at `wrap_agent`, `wrap_runner`, `wrap_flow`, and `wrap_mcp_tool`.
+
+    Args:
+        api_key (Optional[str]): Braintrust API key.
+        project_id (Optional[str]): Braintrust project ID.
+        project_name (Optional[str]): Braintrust project name.
+        SpanProcessor (Optional[type]): Deprecated parameter.
+
+    Returns:
+        bool: True if setup was successful, False otherwise.
+    """
+    if SpanProcessor is not None:
+        warnings.warn("SpanProcessor parameter is deprecated and will be ignored", DeprecationWarning, stacklevel=2)
+
+    span = current_span()
+    if span == NOOP_SPAN:
+        init_logger(project=project_name, api_key=api_key, project_id=project_id)
+
+    return ADKIntegration.setup()