diff --git a/src/agents/__init__.py b/src/agents/__init__.py
index e3bc96abff..74106c3932 100644
--- a/src/agents/__init__.py
+++ b/src/agents/__init__.py
@@ -129,11 +129,16 @@
     StreamEvent,
 )
 from .tool import (
+    ApplyPatchOnApprovalFunction,
+    ApplyPatchOnApprovalFunctionResult,
     ApplyPatchTool,
     CodeInterpreterTool,
     ComputerProvider,
     ComputerTool,
+    ComputerToolSafetyCheckData,
     CustomTool,
+    CustomToolOnApprovalFunction,
+    CustomToolOnApprovalFunctionResult,
     FileSearchTool,
     FunctionTool,
     FunctionToolResult,
@@ -481,7 +486,12 @@ def enable_verbose_stdout_logging():
     "ApplyPatchEditor",
     "ApplyPatchOperation",
     "ApplyPatchResult",
+    "ApplyPatchOnApprovalFunction",
+    "ApplyPatchOnApprovalFunctionResult",
     "ApplyPatchTool",
+    "ComputerToolSafetyCheckData",
+    "CustomToolOnApprovalFunction",
+    "CustomToolOnApprovalFunctionResult",
     "Tool",
     "WebSearchTool",
     "HostedMCPTool",