diff --git a/src/agents/extensions/handoff_filters.py b/src/agents/extensions/handoff_filters.py index de44f1566a..14fd303f47 100644 --- a/src/agents/extensions/handoff_filters.py +++ b/src/agents/extensions/handoff_filters.py @@ -96,6 +96,8 @@ def _remove_tool_types_from_input( "shell_call_output", "apply_patch_call", "apply_patch_call_output", + "custom_tool_call", + "custom_tool_call_output", ] filtered_items: list[TResponseInputItem] = [] diff --git a/tests/test_extension_filters.py b/tests/test_extension_filters.py index 97924d2852..257f3fed5e 100644 --- a/tests/test_extension_filters.py +++ b/tests/test_extension_filters.py @@ -1041,6 +1041,8 @@ def test_removes_hosted_tool_types_from_input_history() -> None: "shell_call_output", "apply_patch_call", "apply_patch_call_output", + "custom_tool_call", + "custom_tool_call_output", ] input_items: list[TResponseInputItem] = [_get_message_input_item("Hello")] for t in hosted_types: