braintrust 0.5.4__py3-none-any.whl → 0.5.5__py3-none-any.whl

braintrust/span_types.py

@@ -16,6 +16,7 @@ class SpanTypeAttribute(str, Enum):
     AUTOMATION = "automation"
     FACET = "facet"
     PREPROCESSOR = "preprocessor"
+    REVIEW = "review"
 
 
 class SpanPurpose(str, Enum):

braintrust/version.py

@@ -1,4 +1,4 @@
-VERSION = "0.5.4"
+VERSION = "0.5.5"
 
 # this will be templated during the build
-GIT_COMMIT = "449f6eeddd54f73d88e185fcf499087d3e61fbe6"
+GIT_COMMIT = "a88b44d6852fae824705415d2795f5c5201f2dde"

braintrust/wrappers/openai.py

@@ -11,11 +11,11 @@ from braintrust.logger import NOOP_SPAN
 
 
 def _span_type(span: tracing.Span[Any]) -> braintrust.SpanTypeAttribute:
-    if span.span_data.type in ["agent", "handoff", "custom"]:
+    if span.span_data.type in ["agent", "handoff", "custom", "speech_group"]:
         return braintrust.SpanTypeAttribute.TASK
-    elif span.span_data.type in ["function", "guardrail"]:
+    elif span.span_data.type in ["function", "guardrail", "mcp_tools"]:
         return braintrust.SpanTypeAttribute.TOOL
-    elif span.span_data.type in ["generation", "response"]:
+    elif span.span_data.type in ["generation", "response", "transcription", "speech"]:
         return braintrust.SpanTypeAttribute.LLM
     else:
         return braintrust.SpanTypeAttribute.TASK
@@ -36,6 +36,16 @@ def _span_name(span: tracing.Span[Any]) -> str:
         return "Response"
     elif isinstance(span.span_data, tracing.HandoffSpanData):
         return "Handoff"
+    elif isinstance(span.span_data, tracing.MCPListToolsSpanData):
+        if span.span_data.server:
+            return f"List Tools ({span.span_data.server})"
+        return "MCP List Tools"
+    elif isinstance(span.span_data, tracing.TranscriptionSpanData):
+        return "Transcription"
+    elif isinstance(span.span_data, tracing.SpeechSpanData):
+        return "Speech"
+    elif isinstance(span.span_data, tracing.SpeechGroupSpanData):
+        return "Speech Group"
     else:
         return "Unknown"
 
@@ -202,6 +212,39 @@ class BraintrustTracingProcessor(tracing.TracingProcessor):
     def _custom_log_data(self, span: tracing.Span[tracing.CustomSpanData]) -> dict[str, Any]:
         return span.span_data.data
 
+    def _mcp_list_tools_log_data(self, span: tracing.Span[tracing.MCPListToolsSpanData]) -> dict[str, Any]:
+        return {
+            "output": span.span_data.result,
+            "metadata": {
+                "server": span.span_data.server,
+            }
+        }
+
+    def _transcription_log_data(self, span: tracing.Span[tracing.TranscriptionSpanData]) -> dict[str, Any]:
+        return {
+            "input": span.span_data.input,
+            "output": span.span_data.output,
+            "metadata": {
+                "model": span.span_data.model,
+                "model_config": span.span_data.model_config,
+            }
+        }
+
+    def _speech_log_data(self, span: tracing.Span[tracing.SpeechSpanData]) -> dict[str, Any]:
+        return {
+            "input": span.span_data.input,
+            "output": span.span_data.output,
+            "metadata": {
+                "model": span.span_data.model,
+                "model_config": span.span_data.model_config,
+            }
+        }
+
+    def _speech_group_log_data(self, span: tracing.Span[tracing.SpeechGroupSpanData]) -> dict[str, Any]:
+        return {
+            "input": span.span_data.input,
+        }
+
     def _log_data(self, span: tracing.Span[Any]) -> dict[str, Any]:
         if isinstance(span.span_data, tracing.AgentSpanData):
             return self._agent_log_data(span)
@@ -217,6 +260,14 @@ class BraintrustTracingProcessor(tracing.TracingProcessor):
             return self._generation_log_data(span)
         elif isinstance(span.span_data, tracing.CustomSpanData):
             return self._custom_log_data(span)
+        elif isinstance(span.span_data, tracing.MCPListToolsSpanData):
+            return self._mcp_list_tools_log_data(span)
+        elif isinstance(span.span_data, tracing.TranscriptionSpanData):
+            return self._transcription_log_data(span)
+        elif isinstance(span.span_data, tracing.SpeechSpanData):
+            return self._speech_log_data(span)
+        elif isinstance(span.span_data, tracing.SpeechGroupSpanData):
+            return self._speech_group_log_data(span)
         else:
             return {}
 

braintrust/wrappers/test_openai.py

@@ -1537,6 +1537,7 @@ async def test_braintrust_tracing_processor_concurrency_bug(memory_logger):
 
 @pytest.mark.asyncio
 @pytest.mark.vcr
+@pytest.mark.skip(reason="OAI Implementation changed, skipping until update")
 async def test_agents_tool_openai_nested_spans(memory_logger):
     """Test that OpenAI calls inside agent tools are properly nested under the tool span."""
     pytest.importorskip("agents", reason="agents package not available")

{braintrust-0.5.4.dist-info → braintrust-0.5.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braintrust
-Version: 0.5.4
+Version: 0.5.5
 Summary: SDK for integrating Braintrust
 Home-page: https://www.braintrust.dev
 Author: Braintrust

{braintrust-0.5.4.dist-info → braintrust-0.5.5.dist-info}/RECORD

@@ -30,7 +30,7 @@ braintrust/span_identifier_v1.py,sha256=eR-dHda0MurdOlghv7-CLSh7eVpNQigbDSQxPpLC
 braintrust/span_identifier_v2.py,sha256=2dLc-Vz8iWLISmL_-ebCyWnY-ysA7sMnBsQtKqzMHYY,8981
 braintrust/span_identifier_v3.py,sha256=RAvOK0lK0huH952kI5X1Q9TaAloD5to8jgTCuYMMw6o,10356
 braintrust/span_identifier_v4.py,sha256=uFT-OdzySo4uCeAaJC3VqH55Oy433xZGBdK2hiEsm2w,10044
-braintrust/span_types.py,sha256=cpTzCwUj4yBPbPLnzR23-VXIU2E9nl_dsVCSVMvtSkc,376
+braintrust/span_types.py,sha256=TtUnrXTVmB1-9_ybaloCoVH9wIbhH0EDEqKMEA5kTDI,398
 braintrust/test_bt_json.py,sha256=pokqmFSQ3m8aB0XN1EEsS-zbv_N4KCYzad6VyOFtwPw,25188
 braintrust/test_context.py,sha256=4wZOhwzGgAxt-CcN2izAxhgcY975oh_HUMkZXRwwTys,42754
 braintrust/test_framework.py,sha256=rPjJYfVzRSbjf8e0irVfCnE3q5Rl1O_2Dy2dPyC9-lo,18674
@@ -51,7 +51,7 @@ braintrust/test_util.py,sha256=SuSKTmvNyaR9Rbgf2TYCUWxJpZHoA2BMx6n4nQfV_pM,8221
 braintrust/test_version.py,sha256=hk5JKjEFbNJ_ONc1VEkqHquflzre34RpFhCEYLTK8iA,1051
 braintrust/trace.py,sha256=PHxfaHApGP_MPMIndZw7atIa2iwKPETUoG-TbF2dv6A,13767
 braintrust/util.py,sha256=S-qMBNsT36r_3pJ4LNKZ-vvHRlJwy8Wy7M7NAdfNOug,8919
-braintrust/version.py,sha256=9Ag0YVXSTg_ksT5pb04Gdog87NeKSiNXtmslh5ZQyhQ,117
+braintrust/version.py,sha256=YMyThQLGiDPCF_KsLHczZxIgOIFFyy6xDhZycKleyAo,117
 braintrust/xact_ids.py,sha256=bdyp88HjlyIkglgLSqYlCYscdSH6EWVyE14sR90Xl1s,658
 braintrust/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 braintrust/cli/__main__.py,sha256=wCBKHGVmn3IT_yMXk5qfDwyI2SV2gf1tLr0NTxm9T8k,1519
@@ -100,7 +100,7 @@ braintrust/wrappers/dspy.py,sha256=DGvDkSteOqhNhD5R3XgYcIKAI1-Ui2WYUbjqBl88C2Y,1
 braintrust/wrappers/langchain.py,sha256=0aY5LuVA7BPkgWA0N6CwPG9EaPqRmVVfEPaM1kN4XZY,5028
 braintrust/wrappers/langsmith_wrapper.py,sha256=mWhBnY6KypOlSQw4NrCwzif0Oe6SHV6LYGGiwiz-EJQ,17956
 braintrust/wrappers/litellm.py,sha256=SxyDrji84OReBGiIf7aP51iIIF8ocD44QmmnYJwVCbo,25168
-braintrust/wrappers/openai.py,sha256=SZuT4ouJd8FRNXxy9zM_OGb2HNL9XGsFnwkHCk3LDAM,10563
+braintrust/wrappers/openai.py,sha256=nrWSK7iyG7Atkl5OeVUljGfHCj625HMgtAfFSD94KKk,12754
 braintrust/wrappers/pydantic_ai.py,sha256=dukJCdcfkRc0ue8nPHpI5DogwNvL9kooGJEbri2W97w,48435
 braintrust/wrappers/test_agno.py,sha256=ad5w6CEWzNwmzbF4NwSvHWhw1SpSjrt-X29pCmBRkhY,3772
 braintrust/wrappers/test_anthropic.py,sha256=VkmlP3_FrZ12x5YWxcbx11A-TEZUghGQeQ4OuT3R4Oc,21409
@@ -109,7 +109,7 @@ braintrust/wrappers/test_google_genai.py,sha256=BKkPfLFk2JG1UNJ3mFKzv5OjioWqo7EQ
 braintrust/wrappers/test_langsmith_wrapper.py,sha256=wEbPNy4o7VVvcuHcsCJ-sy2EATvBxhUXTYFBQNkKCjs,10449
 braintrust/wrappers/test_litellm.py,sha256=MKukVH-C-Mwc0-cVJZhkLdXwiXupXHc8EiWYKq0X8V0,23620
 braintrust/wrappers/test_oai_attachments.py,sha256=_EtNXjQxPgqXmj6UYMZn9GF4GDZf8m_1_TrwiEk7HWQ,11100
-braintrust/wrappers/test_openai.py,sha256=dBphKJT2-xSnK4mJ_xbKLFPBfpHzK07TmHxPTQ8ZQDg,73355
+braintrust/wrappers/test_openai.py,sha256=wJjsFRZFTFgePrxgPwzUxNNg5ZW6djX25qxGQDX09Us,73433
 braintrust/wrappers/test_openrouter.py,sha256=8HUfILPugOMqcvttpq76KQrynFb0xZpazvta7TTSF6A,3849
 braintrust/wrappers/test_pydantic_ai_integration.py,sha256=xUSkiY5HUI-Z9G_etjyX7rGlOjBEdEY9CwVHyWM3xEE,104460
 braintrust/wrappers/test_pydantic_ai_wrap_openai.py,sha256=OO5NrbothkMr4v2sZ-EZLH7-yLj3k6TfdLG4uzXAsQk,5090
@@ -125,8 +125,8 @@ braintrust/wrappers/claude_agent_sdk/__init__.py,sha256=4FUE59ii39jVfhMAfkOcU-TJ
 braintrust/wrappers/claude_agent_sdk/_wrapper.py,sha256=UNZBcgTu7X71zvJKy6e-QQFuz9j8kRS6Kd1VOh3OOXA,17755
 braintrust/wrappers/claude_agent_sdk/test_wrapper.py,sha256=RmSzTfDC3tkepEbc9S_KYFITsjGVyllYE1fF9roSqBk,10779
 braintrust/wrappers/google_genai/__init__.py,sha256=C7MzKUr17CEcIbP0kg9L9IjN-6suE3NQ-oSYAjXdR_g,15832
-braintrust-0.5.4.dist-info/METADATA,sha256=Z3DmXXIkkIpZ5XQ7I6ECeVg5G3SW29onbNWyoLlvOZM,3753
-braintrust-0.5.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-braintrust-0.5.4.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
-braintrust-0.5.4.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
-braintrust-0.5.4.dist-info/RECORD,,
+braintrust-0.5.5.dist-info/METADATA,sha256=-AmNJoVHW9evtVRxgy1OTIJvZHN7zVbfMEugdDEEPVU,3753
+braintrust-0.5.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+braintrust-0.5.5.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
+braintrust-0.5.5.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
+braintrust-0.5.5.dist-info/RECORD,,