diff --git a/src/agents/__init__.py b/src/agents/__init__.py index 820616437..b6a8a0791 100644 --- a/src/agents/__init__.py +++ b/src/agents/__init__.py @@ -35,9 +35,19 @@ ModelResponse, ReasoningItem, RunItem, + TImageGenerationCall, + TLocalShellCall, + TMcpCall, ToolCallItem, ToolCallOutputItem, + TResponseCodeInterpreterToolCall, + TResponseComputerToolCall, + TResponseFileSearchToolCall, + TResponseFunctionToolCall, + TResponseFunctionWebSearch, TResponseInputItem, + TResponseOutputItem, + TResponseStreamEvent, ) from .lifecycle import AgentHooks, RunHooks from .model_settings import ModelSettings @@ -201,6 +211,16 @@ def enable_verbose_stdout_logging(): "ReasoningItem", "ModelResponse", "ItemHelpers", + "TResponseOutputItem", + "TResponseStreamEvent", + "TLocalShellCall", + "TMcpCall", + "TResponseCodeInterpreterToolCall", + "TResponseComputerToolCall", + "TResponseFileSearchToolCall", + "TResponseFunctionToolCall", + "TResponseFunctionWebSearch", + "TImageGenerationCall", "RunHooks", "AgentHooks", "RunContextWrapper", diff --git a/src/agents/items.py b/src/agents/items.py index 64797ad22..d5226f97d 100644 --- a/src/agents/items.py +++ b/src/agents/items.py @@ -56,6 +56,31 @@ TResponseStreamEvent = ResponseStreamEvent """A type alias for the ResponseStreamEvent type from the OpenAI SDK.""" +# Tool call type aliases +TLocalShellCall = LocalShellCall +"""A type alias for the LocalShellCall type from the OpenAI SDK.""" + +TMcpCall = McpCall +"""A type alias for the McpCall type from the OpenAI SDK.""" + +TResponseCodeInterpreterToolCall = ResponseCodeInterpreterToolCall +"""A type alias for the ResponseCodeInterpreterToolCall type from the OpenAI SDK.""" + +TResponseComputerToolCall = ResponseComputerToolCall +"""A type alias for the ResponseComputerToolCall type from the OpenAI SDK.""" + +TResponseFileSearchToolCall = ResponseFileSearchToolCall +"""A type alias for the ResponseFileSearchToolCall type from the OpenAI SDK.""" + +TResponseFunctionToolCall = ResponseFunctionToolCall +"""A type alias for the ResponseFunctionToolCall type from the OpenAI SDK.""" + +TResponseFunctionWebSearch = ResponseFunctionWebSearch +"""A type alias for the ResponseFunctionWebSearch type from the OpenAI SDK.""" + +TImageGenerationCall = ImageGenerationCall +"""A type alias for the ImageGenerationCall type from the OpenAI SDK.""" + T = TypeVar("T", bound=Union[TResponseOutputItem, TResponseInputItem])