auto_instrument openai agents sdk

Author: Mahhheshh

py/src/braintrust/auto.py

@@ -35,6 +35,7 @@ def auto_instrument(
     google_genai: bool = True,
     agno: bool = True,
     claude_agent_sdk: bool = True,
+    openai_agent_sdk: bool = True,
     dspy: bool = True,
     adk: bool = True,
 ) -> dict[str, bool]:
@@ -55,6 +56,7 @@ def auto_instrument(
         google_genai: Enable Google GenAI instrumentation (default: True)
         agno: Enable Agno instrumentation (default: True)
         claude_agent_sdk: Enable Claude Agent SDK instrumentation (default: True)
+        openai_agent_sdk: Enable OpenAI Agent SDK instrumentation (default: True)
         dspy: Enable DSPy instrumentation (default: True)
         adk: Enable Google ADK instrumentation (default: True)
 
@@ -118,6 +120,8 @@ def auto_instrument(
         results["agno"] = _instrument_integration(AgnoIntegration)
     if claude_agent_sdk:
         results["claude_agent_sdk"] = _instrument_integration(ClaudeAgentSDKIntegration)
+    if openai_agent_sdk:
+        results["openai_agent_sdk"] = _instrument_openai_agent_sdk()
     if dspy:
         results["dspy"] = _instrument_dspy()
     if adk:
@@ -134,6 +138,16 @@ def _instrument_openai() -> bool:
     return False
 
 
+def _instrument_openai_agent_sdk() -> bool:
+    with _try_patch():
+        from agents import set_trace_processors
+        from braintrust.wrappers.openai import BraintrustTracingProcessor
+
+        set_trace_processors([BraintrustTracingProcessor()])
+        return True
+    return False
+
+
 def _instrument_integration(integration) -> bool:
     with _try_patch():
         return integration.setup()

py/src/braintrust/integrations/auto_test_scripts/test_auto_openai.py

@@ -1,21 +1,29 @@
 """Test auto_instrument for OpenAI."""
 
 import openai
+from agents.tracing import get_trace_provider
 from braintrust.auto import auto_instrument
+from braintrust.wrappers.openai import BraintrustTracingProcessor
 from braintrust.wrappers.test_utils import autoinstrument_test_context
 
 
 # 1. Verify not patched initially
 assert not getattr(openai, "__braintrust_wrapped__", False)
 
+
 # 2. Instrument
 results = auto_instrument()
 assert results.get("openai") == True
 assert getattr(openai, "__braintrust_wrapped__", False)
 
+assert results.get("openai_agent_sdk") == True
+bt_processor = get_trace_provider()._multi_processor._processors[0]
+assert isinstance(bt_processor, BraintrustTracingProcessor)
+
 # 3. Idempotent
 results2 = auto_instrument()
 assert results2.get("openai") == True
+assert results2.get("openai_agent_sdk") == True
 
 # 4. Make API call and verify span
 with autoinstrument_test_context("test_auto_openai") as memory_logger: