add test

Signed-off-by: myan <[email protected]>

Author: yanmxa

examples/basic/agent_lifecycle_example.py

@@ -31,7 +31,7 @@ async def on_handoff(self, context: RunContextWrapper, agent: Agent, source: Age
     async def on_tool_start(self, context: RunContextWrapper, agent: Agent, action: Action) -> None:
         self.event_counter += 1
         print(
-            f"### ({self.display_name}) {self.event_counter}: Agent {agent.name} started tool {action.function_tool.name}"
+            f"### ({self.display_name}) {self.event_counter}: Agent {agent.name} started tool {action.function_tool.name} with arguments {action.tool_call.arguments}"
         )
 
     async def on_tool_end(

src/agents/_run_impl.py

@@ -52,7 +52,7 @@
 from .models.interface import ModelTracing
 from .run_context import RunContextWrapper, TContext
 from .stream_events import RunItemStreamEvent, StreamEvent
-from .tool import ComputerTool, FunctionTool, FunctionToolResult, Tool
+from .tool import ComputerTool, FunctionTool, FunctionToolResult, Tool, ToolRunFunction, ToolRunComputerAction
 from .tracing import (
     SpanError,
     Trace,
@@ -99,19 +99,6 @@ class ToolRunHandoff:
     handoff: Handoff
     tool_call: ResponseFunctionToolCall
 
-
-@dataclass
-class ToolRunFunction:
-    tool_call: ResponseFunctionToolCall
-    function_tool: FunctionTool
-
-
-@dataclass
-class ToolRunComputerAction:
-    tool_call: ResponseComputerToolCall
-    computer_tool: ComputerTool
-
-
 @dataclass
 class ProcessedResponse:
     new_items: list[RunItem]
@@ -471,8 +458,7 @@ async def run_single_tool(action: ToolRunFunction) -> Any:
 
         tasks = []
         for tool_run in tool_runs:
-            function_tool = tool_run.function_tool
-            tasks.append(run_single_tool(function_tool, tool_run))
+            tasks.append(run_single_tool(tool_run))
 
         results = await asyncio.gather(*tasks)
 

src/agents/tool.py

@@ -10,6 +10,10 @@
 from openai.types.responses.web_search_tool_param import UserLocation
 from pydantic import ValidationError
 from typing_extensions import Concatenate, ParamSpec
+from openai.types.responses import (
+    ResponseComputerToolCall,
+    ResponseFunctionToolCall,
+)
 
 from . import _debug
 from .computer import AsyncComputer, Computer
@@ -133,6 +137,19 @@ def name(self):
 Tool = Union[FunctionTool, FileSearchTool, WebSearchTool, ComputerTool]
 """A tool that can be used in an agent."""
 
+@dataclass
+class ToolRunFunction:
+    tool_call: ResponseFunctionToolCall
+    function_tool: FunctionTool
+
+
+@dataclass
+class ToolRunComputerAction:
+    tool_call: ResponseComputerToolCall
+    computer_tool: ComputerTool
+
+Action = Union[ToolRunFunction, ToolRunComputerAction]
+"""An action that can be performed by an agent. It contains the tool call and the tool"""
 
 def default_tool_error_function(ctx: RunContextWrapper[Any], error: Exception) -> str:
     """The default tool error function, which just returns a generic error message."""