fix: filter custom_tool_call types in remove_all_tools handoff filter (#3095)

Quratulain-bilal · web-flow · commit 4bb4400731e9 · 2026-05-04T09:55:20.000+09:00
diff --git 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
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -96,6 +96,8 @@ def _remove_tool_types_from_input(`
`96`	`96`	`"shell_call_output",`
`97`	`97`	`"apply_patch_call",`
`98`	`98`	`"apply_patch_call_output",`
	`99`	`+ "custom_tool_call",`
	`100`	`+ "custom_tool_call_output",`
`99`	`101`	`]`
`100`	`102`
`101`	`103`	`filtered_items: list[TResponseInputItem] = []`
Original file line number	Diff line number	Diff line change
`@@ -1041,6 +1041,8 @@ def test_removes_hosted_tool_types_from_input_history() -> None:`
`1041`	`1041`	`"shell_call_output",`
`1042`	`1042`	`"apply_patch_call",`
`1043`	`1043`	`"apply_patch_call_output",`
	`1044`	`+ "custom_tool_call",`
	`1045`	`+ "custom_tool_call_output",`
`1044`	`1046`	`]`
`1045`	`1047`	`input_items: list[TResponseInputItem] = [_get_message_input_item("Hello")]`
`1046`	`1048`	`for t in hosted_types:`