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):
+                attributes.update(cast('dict[str, Any]', data))
         return attributes
     except Exception:  # pragma: no cover
         log_internal_error()

logfire/experimental/query_client.py

@@ -17,7 +17,7 @@
     raise ImportError('httpx is required to use the Logfire query clients') from e
 
 if TYPE_CHECKING:
-    from pyarrow import Table  # pyright: ignore[reportUnknownVariableType]
+    from pyarrow import Table
 
 DEFAULT_TIMEOUT = Timeout(30.0)  # queries might typically be slower than the 5s default from AsyncClient
 
@@ -226,8 +226,8 @@ def query_arrow(  # pyright: ignore[reportUnknownParameterType]
             max_timestamp=max_timestamp,
             limit=limit,
         )
-        with pyarrow.ipc.open_stream(response.content) as reader:  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
-            arrow_table: Table = reader.read_all()  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
+        with pyarrow.ipc.open_stream(response.content) as reader:
+            arrow_table: Table = reader.read_all()
         return arrow_table  # pyright: ignore[reportUnknownVariableType]
 
     def query_csv(
@@ -365,8 +365,8 @@ async def query_arrow(  # pyright: ignore[reportUnknownParameterType]
             max_timestamp=max_timestamp,
             limit=limit,
         )
-        with pyarrow.ipc.open_stream(response.content) as reader:  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
-            arrow_table: Table = reader.read_all()  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
+        with pyarrow.ipc.open_stream(response.content) as reader:
+            arrow_table: Table = reader.read_all()
         return arrow_table  # pyright: ignore[reportUnknownVariableType]
 
     async def query_csv(

pyproject.toml

@@ -332,6 +332,14 @@ filterwarnings = [
     "ignore:The '__version__' attribute is deprecated and will be removed in Flask 3.2.*:DeprecationWarning",
     # aiohttp web.Application is deprecated and used by OTel
     "ignore:Inheritance class _InstrumentedApplication from web.Application is discouraged:DeprecationWarning",
+    # anthropic SDK warns when using older models that are still recorded in test cassettes
+    "ignore:The model 'claude-.*' is deprecated.*:DeprecationWarning",
+    # openai-agents 0.14+ defines a `DynamicCompactionPolicy` model with a `model_info` field;
+    # pydantic 2.4 emits a UserWarning at class-definition time about its `model_` protected
+    # namespace. Newer pydantic versions don't, but our test matrix still includes 2.4, and
+    # without this filter `import agents` aborts collection entirely because `filterwarnings=error`
+    # above turns it into an exception. Upstream fix would be `model_config['protected_namespaces'] = ()`.
+    'ignore:Field "model_info" has conflict with protected namespace "model_".*:UserWarning',
 ]
 DJANGO_SETTINGS_MODULE = "tests.otel_integrations.django_test_project.django_test_site.settings"