feat(tools): remove tool_name

Author: Ratish1

src/strands/tools/registry.py

@@ -279,25 +279,23 @@ def register_tool(self, tool: AgentTool) -> None:
                 list(self.dynamic_tools.keys()),
             )
 
-    def replace(self, tool_name: str, new_tool: AgentTool) -> None:
+    def replace(self, new_tool: AgentTool) -> None:
         """Replace an existing tool with a new implementation.
 
         This performs an atomic swap of the tool implementation in the registry.
         The replacement takes effect on the next agent invocation.
 
         Args:
-            tool_name: Name of the tool to replace.
-            new_tool: New tool implementation.
+            new_tool: New tool implementation. Its name must match the tool being replaced.
 
         Raises:
-            ValueError: If the tool doesn't exist or if names don't match.
+            ValueError: If the tool doesn't exist.
         """
+        tool_name = new_tool.tool_name
+
         if tool_name not in self.registry:
             raise ValueError(f"Cannot replace tool '{tool_name}' - tool does not exist")
 
-        if new_tool.tool_name != tool_name:
-            raise ValueError(f"Tool names must match - expected '{tool_name}', got '{new_tool.tool_name}'")
-
         # Atomic replacement in main registry
         self.registry[tool_name] = new_tool
 

tests/strands/tools/test_registry.py

@@ -526,7 +526,7 @@ def test_tool_registry_replace_existing_tool():
 
     registry = ToolRegistry()
     registry.register_tool(old_tool)
-    registry.replace("my_tool", new_tool)
+    registry.replace(new_tool)
 
     assert registry.registry["my_tool"] == new_tool
 
@@ -539,24 +539,7 @@ def test_tool_registry_replace_nonexistent_tool():
     registry = ToolRegistry()
 
     with pytest.raises(ValueError, match="Cannot replace tool 'my_tool' - tool does not exist"):
-        registry.replace("my_tool", new_tool)
-
-
-def test_tool_registry_replace_with_different_name():
-    """Test replacing with different name raises ValueError."""
-    old_tool = MagicMock()
-    old_tool.tool_name = "old_tool"
-    old_tool.is_dynamic = False
-    old_tool.supports_hot_reload = False
-
-    new_tool = MagicMock()
-    new_tool.tool_name = "new_tool"
-
-    registry = ToolRegistry()
-    registry.register_tool(old_tool)
-
-    with pytest.raises(ValueError, match="Tool names must match"):
-        registry.replace("old_tool", new_tool)
+        registry.replace(new_tool)
 
 
 def test_tool_registry_replace_dynamic_tool():
@@ -572,7 +555,7 @@ def test_tool_registry_replace_dynamic_tool():
 
     registry = ToolRegistry()
     registry.register_tool(old_tool)
-    registry.replace("dynamic_tool", new_tool)
+    registry.replace(new_tool)
 
     assert registry.registry["dynamic_tool"] == new_tool
     assert registry.dynamic_tools["dynamic_tool"] == new_tool
@@ -594,7 +577,7 @@ def test_tool_registry_replace_dynamic_with_non_dynamic():
 
     assert "my_tool" in registry.dynamic_tools
 
-    registry.replace("my_tool", new_tool)
+    registry.replace(new_tool)
 
     assert registry.registry["my_tool"] == new_tool
     assert "my_tool" not in registry.dynamic_tools
@@ -616,7 +599,7 @@ def test_tool_registry_replace_non_dynamic_with_dynamic():
 
     assert "my_tool" not in registry.dynamic_tools
 
-    registry.replace("my_tool", new_tool)
+    registry.replace(new_tool)
 
     assert registry.registry["my_tool"] == new_tool
     assert registry.dynamic_tools["my_tool"] == new_tool