feat: Add support for excluding predefined functions in ComputerUseToolset

Co-authored-by: Xuan Yang <xygoogle@google.com>
PiperOrigin-RevId: 904517817

Author: xuanyang15

src/google/adk/tools/computer_use/computer_use_toolset.py

@@ -48,9 +48,11 @@ def __init__(
       self,
       *,
       computer: BaseComputer,
+      excluded_predefined_functions: Optional[list[str]] = None,
   ):
     super().__init__()
     self._computer = computer
+    self._excluded_predefined_functions = excluded_predefined_functions
     self._initialized = False
     self._tools = None
 
@@ -203,6 +205,13 @@ async def get_tools(
       if method_name == "session_state":
         continue
 
+      # Skip methods excluded by configuration
+      if (
+          self._excluded_predefined_functions
+          and method_name in self._excluded_predefined_functions
+      ):
+        continue
+
       # Check if it's a method defined in Computer class
       attr = getattr(BaseComputer, method_name, None)
       if attr is not None and callable(attr):
@@ -259,11 +268,18 @@ async def process_llm_request(
           types.Environment.ENVIRONMENT_BROWSER,
       )
       llm_request.config.tools.append(
-          types.Tool(computer_use=types.ComputerUse(environment=environment))
+          types.Tool(
+              computer_use=types.ComputerUse(
+                  environment=environment,
+                  excluded_predefined_functions=self._excluded_predefined_functions,
+              )
+          )
       )
       logger.debug(
-          "Added computer use tool with environment: %s",
+          "Added computer use tool with environment: %s,"
+          " excluded_functions: %s",
           environment,
+          self._excluded_predefined_functions,
       )
 
     except Exception as e:

tests/unittests/tools/computer_use/test_computer_use_toolset.py

@@ -202,6 +202,23 @@ async def test_get_tools_excludes_utility_methods(self, toolset):
     for method in expected_methods:
       assert method in tool_names
 
+  @pytest.mark.asyncio
+  async def test_get_tools_filters_excluded_functions(self, mock_computer):
+    """Test that get_tools filters out excluded functions."""
+    excluded_funcs = ["drag_and_drop", "key_combination"]
+    toolset = ComputerUseToolset(
+        computer=mock_computer,
+        excluded_predefined_functions=excluded_funcs,
+    )
+
+    tools = await toolset.get_tools()
+    tool_names = [tool.func.__name__ for tool in tools]
+
+    for func in excluded_funcs:
+      assert func not in tool_names
+
+    assert "click_at" in tool_names
+
   @pytest.mark.asyncio
   async def test_get_tools_with_readonly_context(self, toolset):
     """Test get_tools with readonly_context parameter."""
@@ -362,6 +379,41 @@ async def test_process_llm_request(self, toolset, mock_computer):
         == types.Environment.ENVIRONMENT_BROWSER
     )
 
+  @pytest.mark.asyncio
+  async def test_process_llm_request_with_excluded_functions(
+      self, mock_computer
+  ):
+    """Test that process_llm_request passes excluded_predefined_functions."""
+    excluded_funcs = ["drag_and_drop", "key_combination"]
+    toolset = ComputerUseToolset(
+        computer=mock_computer,
+        excluded_predefined_functions=excluded_funcs,
+    )
+
+    llm_request = LlmRequest(
+        model="gemini-1.5-flash",
+        config=types.GenerateContentConfig(),
+    )
+
+    await toolset.process_llm_request(
+        tool_context=MagicMock(), llm_request=llm_request
+    )
+
+    # Should have computer use tool
+    computer_use_tools = [
+        tool
+        for tool in llm_request.config.tools
+        if hasattr(tool, "computer_use") and tool.computer_use
+    ]
+    assert len(computer_use_tools) == 1
+
+    # Should have correct excluded functions
+    computer_use_tool = computer_use_tools[0]
+    assert (
+        computer_use_tool.computer_use.excluded_predefined_functions
+        == excluded_funcs
+    )
+
   @pytest.mark.asyncio
   async def test_process_llm_request_with_existing_computer_use(
       self, toolset, mock_computer