Support new TaskSpanData and TurnSpanData from openai-agents 0.14

logfire/_internal/integrations/openai_agents.py

@@ -7,7 +7,7 @@
 from contextlib import nullcontext
 from dataclasses import dataclass
 from types import TracebackType
-from typing import TYPE_CHECKING, Any, Generic, TypeVar
+from typing import TYPE_CHECKING, Any, Generic, TypeVar, cast
 
 import agents
 from agents import (
@@ -27,7 +27,7 @@
     TranscriptionSpanData,
 )
 from agents.models.openai_responses import OpenAIResponsesModel
-from agents.tracing import ResponseSpanData, response_span
+from agents.tracing import ResponseSpanData, TaskSpanData, TurnSpanData, response_span
 from agents.tracing.scope import Scope
 from agents.tracing.spans import NoOpSpan, SpanError, TSpanData
 from agents.tracing.traces import NoOpTrace
@@ -110,6 +110,10 @@ def create_span(
                 msg_template = 'Speech → Text with {gen_ai.request.model!r}'
             elif isinstance(span_data, MCPListToolsSpanData):
                 msg_template = 'MCP: list tools from server {server}'
+            elif isinstance(span_data, TaskSpanData):
+                msg_template = 'Task: {name}'
+            elif isinstance(span_data, TurnSpanData):
+                msg_template = 'Turn {turn} for agent {agent_name}'
             else:
                 msg_template = 'OpenAI agents: {type} span'
 
@@ -382,6 +386,10 @@ def attributes_from_span_data(span_data: SpanData, msg_template: str) -> dict[st
                 attributes['output'] = {k: v for k, v in attributes['output'].items() if k != 'data'}
         elif isinstance(span_data, FunctionSpanData):
             attributes['output'] = span_data.output
+        elif isinstance(span_data, (TaskSpanData, TurnSpanData)):
+            data = attributes.pop('data', None)
+            if isinstance(data, dict):  # pragma: no branch
+                attributes.update(cast('dict[str, Any]', data))
         return attributes
     except Exception:  # pragma: no cover
         log_internal_error()

pyproject.toml

@@ -180,7 +180,7 @@ dev = [
     "botocore >= 1.31.57",
     "greenlet >= 3.1.1",
     "pytest-xdist>=3.6.1",
-    "openai-agents[voice]>=0.9.0; python_version >= '3.10'",
+    "openai-agents[voice]>=0.14.0; python_version >= '3.10'",
     "pydantic-ai-slim>=0.0.39",
     "pydantic-evals>=1.0.0; python_version >= '3.10'",
     "langchain>=0.0.27",