Add ShellTool support to TemporalStreamingModel

openai-agents introduced a next-generation ShellTool (replacing
LocalShellTool) that carries an environment config like
{"type": "local", "skills": [...]}. The Temporal streaming model was
dropping it with "Unknown tool type: ShellTool, skipping", so agents
running through AgentEx/Temporal lost the tool entirely even though
plain Runner.run(...) worked.

Serialize ShellTool to the Responses API "shell" payload, defaulting
environment to {"type": "local"} when unset. Import is guarded so
users on older openai-agents versions (ShellTool not yet exported)
continue to work.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: danielmillerp

src/agentex/lib/core/temporal/plugins/openai_agents/models/temporal_streaming_model.py

@@ -27,6 +27,11 @@
     CodeInterpreterTool,
     ImageGenerationTool,
 )
+
+try:
+    from agents.tool import ShellTool  # type: ignore[attr-defined]
+except ImportError:
+    ShellTool = None  # type: ignore[assignment,misc]
 from agents.usage import Usage, InputTokensDetails, OutputTokensDetails  # type: ignore[attr-defined]
 from agents.model_settings import MCPToolChoice
 from openai.types.responses import (
@@ -326,6 +331,13 @@ def _convert_tools(self, tools: list[Tool], handoffs: list[Handoff]) -> tuple[Li
                     "type": "local_shell",
                 })
 
+            elif ShellTool is not None and isinstance(tool, ShellTool):
+                environment = dict(tool.environment) if tool.environment else {"type": "local"}
+                response_tools.append({
+                    "type": "shell",
+                    "environment": environment,
+                })
+
             else:
                 logger.warning(f"Unknown tool type: {type(tool).__name__}, skipping")
 

src/agentex/lib/core/temporal/plugins/openai_agents/tests/test_convert_tools.py

@@ -0,0 +1,61 @@
+"""Unit tests for TemporalStreamingModel._convert_tools tool serialization."""
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from agentex.lib.core.temporal.plugins.openai_agents.models import (
+    temporal_streaming_model as tsm_module,
+)
+from agentex.lib.core.temporal.plugins.openai_agents.models.temporal_streaming_model import (
+    TemporalStreamingModel,
+)
+
+
+@pytest.fixture
+def model():
+    with patch(
+        "agentex.lib.core.temporal.plugins.openai_agents.models.temporal_streaming_model.create_async_agentex_client"
+    ):
+        return TemporalStreamingModel(model_name="gpt-4o", openai_client=MagicMock())
+
+
+class _FakeShellTool:
+    """Stand-in for agents.tool.ShellTool for environments where it isn't installed."""
+
+    def __init__(self, environment):
+        self.environment = environment
+
+
+def test_shell_tool_local_environment(model, monkeypatch):
+    """ShellTool with a local environment should serialize to a 'shell' payload."""
+    monkeypatch.setattr(tsm_module, "ShellTool", _FakeShellTool)
+
+    tool = _FakeShellTool(environment={"type": "local", "skills": ["git"]})
+    response_tools, _ = model._convert_tools([tool], handoffs=[])
+
+    assert response_tools == [{"type": "shell", "environment": {"type": "local", "skills": ["git"]}}]
+
+
+def test_shell_tool_defaults_environment_when_missing(model, monkeypatch):
+    """ShellTool with environment=None should fall back to {'type': 'local'}."""
+    monkeypatch.setattr(tsm_module, "ShellTool", _FakeShellTool)
+
+    tool = _FakeShellTool(environment=None)
+    response_tools, _ = model._convert_tools([tool], handoffs=[])
+
+    assert response_tools == [{"type": "shell", "environment": {"type": "local"}}]
+
+
+def test_shell_tool_unavailable_falls_through(model, monkeypatch, caplog):
+    """If ShellTool isn't installed, an unknown tool should log a warning and be skipped."""
+    monkeypatch.setattr(tsm_module, "ShellTool", None)
+
+    class _NotAShellTool:
+        pass
+
+    with caplog.at_level("WARNING"):
+        response_tools, _ = model._convert_tools([_NotAShellTool()], handoffs=[])
+
+    assert response_tools == []
+    assert any("Unknown tool type" in rec.message for rec in caplog.records)