fix: Add local method override

Author: Unshure

src/strands/agent/agent.py

@@ -100,11 +100,17 @@ def __getattr__(self, name: str) -> Callable[..., Any]:
                 AttributeError: If no tool with the given name exists or if multiple tools match the given name.
             """
 
-            def caller(user_message_override: Optional[str] = None, **kwargs: Any) -> Any:
+            def caller(
+                user_message_override: Optional[str] = None,
+                record_direct_tool_call: Optional[bool] = None,
+                **kwargs: Any,
+            ) -> Any:
                 """Call a tool directly by name.
 
                 Args:
                     user_message_override: Optional custom message to record instead of default
+                    record_direct_tool_call: Whether to record direct tool calls in message history. Overrides class
+                        attribute if provided.
                     **kwargs: Keyword arguments to pass to the tool.
 
                 Returns:
@@ -134,7 +140,12 @@ def caller(user_message_override: Optional[str] = None, **kwargs: Any) -> Any:
                     kwargs=kwargs,
                 )
 
-                if self._agent.record_direct_tool_call:
+                if record_direct_tool_call is not None:
+                    should_record_direct_tool_call = record_direct_tool_call
+                else:
+                    should_record_direct_tool_call = self._agent.record_direct_tool_call
+
+                if should_record_direct_tool_call:
                     # Create a record of this tool execution in the message history
                     self._agent._record_tool_execution(
                         tool_use, tool_result, user_message_override, self._agent.messages

tests/strands/agent/test_agent.py

@@ -739,6 +739,18 @@ def test_agent_tool_do_not_record_tool(agent):
     assert tru_messages == exp_messages
 
 
+def test_agent_tool_do_not_record_tool_with_method_override(agent):
+    agent.record_direct_tool_call = True
+    agent.tool.tool_decorated(
+        random_string="abcdEfghI123", user_message_override="test override", record_direct_tool_call=False
+    )
+
+    tru_messages = agent.messages
+    exp_messages = []
+
+    assert tru_messages == exp_messages
+
+
 def test_agent_tool_tool_does_not_exist(agent):
     with pytest.raises(AttributeError):
         agent.tool.does_not_exist()