ci: fix issues that surfaced in CI

Author: mjschock

examples/agent_patterns/human_in_the_loop.py

@@ -10,7 +10,7 @@
 import asyncio
 import json
 
-from agents import Agent, Runner, RunState, function_tool
+from agents import Agent, Runner, RunState, ToolApprovalItem, function_tool
 
 
 @function_tool
@@ -101,17 +101,20 @@ async def main():
 
         # Reading state from file (demonstrating deserialization)
         print("Loading state from result.json")
-        with open("result.json", "r") as f:
+        with open("result.json") as f:
             stored_state_json = json.load(f)
 
         state = RunState.from_json(agent, stored_state_json)
 
         # Process each interruption
         for interruption in result.interruptions:
-            print(f"\nTool call details:")
+            if not isinstance(interruption, ToolApprovalItem):
+                continue
+
+            print("\nTool call details:")
             print(f"  Agent: {interruption.agent.name}")
-            print(f"  Tool: {interruption.raw_item.name}")  # type: ignore
-            print(f"  Arguments: {interruption.raw_item.arguments}")  # type: ignore
+            print(f"  Tool: {interruption.raw_item.name}")
+            print(f"  Arguments: {interruption.raw_item.arguments}")
 
             confirmed = await confirm("\nDo you approve this tool call?")
 

examples/agent_patterns/human_in_the_loop_stream.py

@@ -10,7 +10,7 @@
 
 import asyncio
 
-from agents import Agent, Runner, function_tool
+from agents import Agent, Runner, ToolApprovalItem, function_tool
 
 
 async def _needs_temperature_approval(_ctx, params, _call_id) -> bool:
@@ -89,10 +89,13 @@ async def main():
         state = result.to_state()
 
         for interruption in result.interruptions:
-            print(f"\nTool call details:")
+            if not isinstance(interruption, ToolApprovalItem):
+                continue
+
+            print("\nTool call details:")
             print(f"  Agent: {interruption.agent.name}")
-            print(f"  Tool: {interruption.raw_item.name}")  # type: ignore
-            print(f"  Arguments: {interruption.raw_item.arguments}")  # type: ignore
+            print(f"  Tool: {interruption.raw_item.name}")
+            print(f"  Arguments: {interruption.raw_item.arguments}")
 
             confirmed = await confirm("\nDo you approve this tool call?")
 

src/agents/__init__.py

@@ -246,6 +246,7 @@ def enable_verbose_stdout_logging():
     "RunItem",
     "HandoffCallItem",
     "HandoffOutputItem",
+    "ToolApprovalItem",
     "ToolCallItem",
     "ToolCallOutputItem",
     "ReasoningItem",
@@ -262,6 +263,7 @@ def enable_verbose_stdout_logging():
     "RunResult",
     "RunResultStreaming",
     "RunConfig",
+    "RunState",
     "RawResponsesStreamEvent",
     "RunItemStreamEvent",
     "AgentUpdatedStreamEvent",

src/agents/run.py

@@ -6,7 +6,7 @@
 import os
 import warnings
 from dataclasses import dataclass, field
-from typing import Any, Callable, Generic, cast, get_args
+from typing import Any, Callable, Generic, Union, cast, get_args
 
 from openai.types.responses import (
     ResponseCompletedEvent,
@@ -537,7 +537,7 @@ async def run(
                 context = run_state._context.context
         else:
             # Keep original user input separate from session-prepared input
-            raw_input = cast(str | list[TResponseInputItem], input)
+            raw_input = cast(Union[str, list[TResponseInputItem]], input)
             original_user_input = raw_input
             prepared_input = await self._prepare_input_with_session(
                 raw_input, session, run_config.session_input_callback
@@ -901,7 +901,7 @@ def run_streamed(
             # Use context wrapper from RunState
             context_wrapper = cast(RunContextWrapper[TContext], run_state._context)
         else:
-            input_for_result = cast(str | list[TResponseInputItem], input)
+            input_for_result = cast(Union[str, list[TResponseInputItem]], input)
             context_wrapper = RunContextWrapper(context=context)  # type: ignore
 
         streamed_result = RunResultStreaming(

src/agents/run_state.py

@@ -8,16 +8,17 @@
 
 from typing_extensions import TypeVar
 
+from ._run_impl import NextStepInterruption
 from .exceptions import UserError
+from .items import ToolApprovalItem
 from .logger import logger
 from .run_context import RunContextWrapper
 from .usage import Usage
 
 if TYPE_CHECKING:
-    from ._run_impl import NextStepInterruption
     from .agent import Agent
     from .guardrail import InputGuardrailResult, OutputGuardrailResult
-    from .items import ModelResponse, RunItem, ToolApprovalItem
+    from .items import ModelResponse, RunItem
 
 TContext = TypeVar("TContext", default=Any)
 TAgent = TypeVar("TAgent", bound="Agent[Any]", default="Agent[Any]")
@@ -105,8 +106,6 @@ def get_interruptions(self) -> list[RunItem]:
         Returns:
             List of tool approval items awaiting approval, or empty list if no interruptions.
         """
-        from ._run_impl import NextStepInterruption
-
         if self._current_step is None or not isinstance(self._current_step, NextStepInterruption):
             return []
         return self._current_step.interruptions
@@ -235,8 +234,6 @@ def to_json(self) -> dict[str, Any]:
 
     def _serialize_current_step(self) -> dict[str, Any] | None:
         """Serialize the current step if it's an interruption."""
-        from ._run_impl import NextStepInterruption
-
         if self._current_step is None or not isinstance(self._current_step, NextStepInterruption):
             return None
 
@@ -245,10 +242,15 @@ def _serialize_current_step(self) -> dict[str, Any] | None:
             "interruptions": [
                 {
                     "type": "tool_approval_item",
-                    "rawItem": item.raw_item.model_dump(exclude_unset=True),
+                    "rawItem": (
+                        item.raw_item.model_dump(exclude_unset=True)
+                        if hasattr(item.raw_item, "model_dump")
+                        else item.raw_item
+                    ),
                     "agent": {"name": item.agent.name},
                 }
                 for item in self._current_step.interruptions
+                if isinstance(item, ToolApprovalItem)
             ],
         }
 
@@ -366,10 +368,7 @@ def from_string(initial_agent: Agent[Any], state_string: str) -> RunState[Any, A
         if current_step_data and current_step_data.get("type") == "next_step_interruption":
             from openai.types.responses import ResponseFunctionToolCall
 
-            from ._run_impl import NextStepInterruption
-            from .items import ToolApprovalItem
-
-            interruptions = []
+            interruptions: list[RunItem] = []
             for item_data in current_step_data.get("interruptions", []):
                 agent_name = item_data["agent"]["name"]
                 agent = agent_map.get(agent_name)
@@ -458,10 +457,7 @@ def from_json(
         if current_step_data and current_step_data.get("type") == "next_step_interruption":
             from openai.types.responses import ResponseFunctionToolCall
 
-            from ._run_impl import NextStepInterruption
-            from .items import ToolApprovalItem
-
-            interruptions = []
+            interruptions: list[RunItem] = []
             for item_data in current_step_data.get("interruptions", []):
                 agent_name = item_data["agent"]["name"]
                 agent = agent_map.get(agent_name)

tests/extensions/memory/test_advanced_sqlite_session.py

@@ -74,6 +74,7 @@ def create_mock_run_result(
         tool_output_guardrail_results=[],
         context_wrapper=context_wrapper,
         _last_agent=agent,
+        interruptions=[],
     )
 
 

tests/test_result_cast.py

@@ -18,6 +18,7 @@ def create_run_result(final_output: Any) -> RunResult:
         tool_output_guardrail_results=[],
         _last_agent=Agent(name="test"),
         context_wrapper=RunContextWrapper(context=None),
+        interruptions=[],
     )