fix tests

Author: jhills20

tests/test_agent_runner.py

@@ -226,9 +226,9 @@ async def test_structured_output():
 
     assert result.final_output == Foo(bar="baz")
     assert len(result.raw_responses) == 4, "should have four model responses"
-    assert len(result.to_input_list()) == 11, (
-        "should have input: 2 orig inputs, function call, function call result, message, handoff, "
-        "handoff output, preamble message, tool call, tool call result, final output"
+    assert len(result.to_input_list()) == 10, (
+        "should have input: conversation summary, function call, function call result, message, "
+        "handoff, handoff output, preamble message, tool call, tool call result, final output"
     )
 
     assert result.last_agent == agent_1, "should have handed off to agent_1"

tests/test_agent_runner_streamed.py

@@ -231,9 +231,9 @@ async def test_structured_output():
 
     assert result.final_output == Foo(bar="baz")
     assert len(result.raw_responses) == 4, "should have four model responses"
-    assert len(result.to_input_list()) == 11, (
-        "should have input: 2 orig inputs, function call, function call result, message, handoff, "
-        "handoff output, preamble message, tool call, tool call result, final output"
+    assert len(result.to_input_list()) == 10, (
+        "should have input: conversation summary, function call, function call result, message, "
+        "handoff, handoff output, preamble message, tool call, tool call result, final output"
     )
 
     assert result.last_agent == agent_1, "should have handed off to agent_1"
@@ -717,9 +717,9 @@ async def test_streaming_events():
 
     assert result.final_output == Foo(bar="baz")
     assert len(result.raw_responses) == 4, "should have four model responses"
-    assert len(result.to_input_list()) == 10, (
-        "should have input: 2 orig inputs, function call, function call result, message, handoff, "
-        "handoff output, tool call, tool call result, final output"
+    assert len(result.to_input_list()) == 9, (
+        "should have input: conversation summary, function call, function call result, message, "
+        "handoff, handoff output, tool call, tool call result, final output"
     )
 
     assert result.last_agent == agent_1, "should have handed off to agent_1"

tests/test_extension_filters.py

@@ -350,7 +350,11 @@ def test_nest_handoff_history_honors_custom_wrappers() -> None:
         summary = _as_message(nested.input_history[0])
         summary_content = summary["content"]
         assert isinstance(summary_content, str)
-        assert summary_content.startswith("<<START>>")
+        lines = summary_content.splitlines()
+        assert lines[0] == (
+            "For context, here is the conversation so far between the user and the previous agent:"
+        )
+        assert lines[1].startswith("<<START>>")
         assert summary_content.endswith("<<END>>")
 
         # Ensure the custom markers are parsed correctly when nesting again.

tests/test_run_step_processing.py

@@ -1,11 +1,12 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, cast
 
 import pytest
 from openai.types.responses import (
     ResponseComputerToolCall,
     ResponseFileSearchToolCall,
+    ResponseFunctionToolCall,
     ResponseFunctionWebSearch,
 )
 from openai.types.responses.response_computer_tool_call import ActionClick
@@ -30,7 +31,7 @@
     handoff,
 )
 from agents._run_impl import RunImpl, ToolRunHandoff
-from agents.lifecycle import RunHooksBase
+from agents import RunHooks
 from agents.run import AgentRunner
 
 from .test_responses import (
@@ -215,11 +216,11 @@ async def test_handoff_can_disable_run_level_history_nesting(monkeypatch: pytest
     source_agent = Agent(name="source")
     target_agent = Agent(name="target")
     override_handoff = handoff(target_agent, nest_handoff_history=False)
-    tool_call = get_handoff_tool_call(target_agent)
+    tool_call = cast(ResponseFunctionToolCall, get_handoff_tool_call(target_agent))
     run_handoffs = [ToolRunHandoff(handoff=override_handoff, tool_call=tool_call)]
     run_config = RunConfig(nest_handoff_history=True)
     context_wrapper = RunContextWrapper(context=None)
-    hooks = RunHooksBase()
+    hooks = RunHooks()
     original_input = [get_text_input_item("hello")]
     pre_step_items: list[RunItem] = []
     new_step_items: list[RunItem] = []
@@ -258,11 +259,11 @@ async def test_handoff_can_enable_history_nesting(monkeypatch: pytest.MonkeyPatc
     source_agent = Agent(name="source")
     target_agent = Agent(name="target")
     override_handoff = handoff(target_agent, nest_handoff_history=True)
-    tool_call = get_handoff_tool_call(target_agent)
+    tool_call = cast(ResponseFunctionToolCall, get_handoff_tool_call(target_agent))
     run_handoffs = [ToolRunHandoff(handoff=override_handoff, tool_call=tool_call)]
     run_config = RunConfig(nest_handoff_history=False)
     context_wrapper = RunContextWrapper(context=None)
-    hooks = RunHooksBase()
+    hooks = RunHooks()
     original_input = [get_text_input_item("hello")]
     pre_step_items: list[RunItem] = []
     new_step_items: list[RunItem] = []