From a6b7da45993df9c57b2b70aca70a652837b5b63e Mon Sep 17 00:00:00 2001 From: davenpi Date: Tue, 13 May 2025 16:54:16 -0400 Subject: [PATCH] chore: unify tool not found error message --- src/fastmcp/tools/tool_manager.py | 2 +- tests/tools/test_tool_manager.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/fastmcp/tools/tool_manager.py b/src/fastmcp/tools/tool_manager.py index 3114fa1ae..509f4e6de 100644 --- a/src/fastmcp/tools/tool_manager.py +++ b/src/fastmcp/tools/tool_manager.py @@ -106,7 +106,7 @@ class ToolManager: if key in self._tools: del self._tools[key] else: - raise NotFoundError(f"Tool {key!r} not found.") + raise NotFoundError(f"Unknown tool: {key}") async def call_tool( self, key: str, arguments: dict[str, Any] diff --git a/tests/tools/test_tool_manager.py b/tests/tools/test_tool_manager.py index 4d61d0520..9d35d177f 100644 --- a/tests/tools/test_tool_manager.py +++ b/tests/tools/test_tool_manager.py @@ -117,7 +117,7 @@ class TestAddTools: def test_remove_tool_missing_key(self): """Test removing a tool that does not exist raises NotFoundError.""" manager = ToolManager() - with pytest.raises(NotFoundError, match="Tool 'missing' not found"): + with pytest.raises(NotFoundError, match=f"Unknown tool: {'missing'}"): manager.remove_tool("missing") def test_warn_on_duplicate_tools(self, caplog):