add name param to tool

Author: anafalcao

aws_lambda_powertools/event_handler/bedrock_agent_function.py

@@ -80,14 +80,26 @@ def __init__(self) -> None:
         self.current_event: BedrockAgentFunctionEvent | None = None
         self._response_builder_class = BedrockFunctionsResponseBuilder
 
-    def tool(self, description: str | None = None) -> Callable:
-        """Decorator to register a tool function"""
+    def tool(
+        self,
+        description: str | None = None,
+        name: str | None = None,
+    ) -> Callable:
+        """Decorator to register a tool function
+
+        Parameters
+        ----------
+        description : str | None
+            Description of what the tool does
+        name : str | None
+            Custom name for the tool. If not provided, uses the function name
+        """
 
         def decorator(func: Callable) -> Callable:
             if not description:
                 raise ValueError("Tool description is required")
 
-            function_name = func.__name__
+            function_name = name or func.__name__
             if function_name in self._tools:
                 raise ValueError(f"Tool '{function_name}' already registered")
 

tests/functional/event_handler/required_dependencies/test_bedrock_agent_functions.py

@@ -128,3 +128,23 @@ def test_bedrock_agent_function_invalid_event():
     # THEN raise ValueError
     with pytest.raises(ValueError, match="Missing required field"):
         app.resolve({}, {})
+
+
+def test_bedrock_agent_function_with_custom_name():
+    # GIVEN a Bedrock Agent Function resolver
+    app = BedrockAgentFunctionResolver()
+
+    @app.tool(name="customName", description="Function with custom name")
+    def test_function():
+        return "Hello from custom named function"
+
+    # WHEN calling the event handler
+    raw_event = load_event("bedrockAgentFunctionEvent.json")
+    raw_event["function"] = "customName"  # Use custom name instead of function name
+    result = app.resolve(raw_event, {})
+
+    # THEN process event correctly
+    assert result["messageVersion"] == "1.0"
+    assert result["response"]["actionGroup"] == raw_event["actionGroup"]
+    assert result["response"]["function"] == "customName"
+    assert result["response"]["functionResponse"]["responseBody"]["TEXT"]["body"] == "Hello from custom named function"