update mcp toolset and sample agent based on new tool_filter definition

PiperOrigin-RevId: 757969950
2025-12-18 11:22:22 -06:00 · 2025-05-12 17:04:55 -07:00
parent 722028801a
commit d35b99e6dd
3 changed files with 21 additions and 4 deletions
--- a/contributing/samples/mcp_agent/agent.py
+++ b/contributing/samples/mcp_agent/agent.py
@@ -34,8 +34,12 @@ root_agent = LlmAgent(
                ],
            ),
            # don't want agent to do write operation
-            tool_predicate=lambda tool, ctx=None: tool.name
-            not in ('write_file', 'edit_file', 'create_directory', 'move_file'),
+            tool_filter=[
+                'write_file',
+                'edit_file',
+                'create_directory',
+                'move_file',
+            ],
        )
    ],
 )