[GMLP-5459] Add tool restrictions (#6)

Author: dvlpjrs

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gumloop-mcp"
-version = "0.1.6"
+version = "0.1.8"
 description = "guMCP (Gumloop Model Context Protocol) SDK"
 readme = "README.md"
 requires-python = ">=3.10"

src/mcp/server/lowlevel/server.py

@@ -519,6 +519,14 @@ async def handler(req: types.ListToolsRequest):
                 # Filter deprecated properties from all tools
                 tools = [filter_deprecated_properties_from_tool(tool) for tool in tools]
 
+                # Filter restricted tools
+                if hasattr(self, "config") and self.config is not None:
+                    restricted_tools = self.config.get("restricted_tools", [])
+                    if restricted_tools:
+                        tools = [
+                            tool for tool in tools if tool.name not in restricted_tools
+                        ]
+
                 # Filter for external clients
                 if hasattr(self, "config") and self.config is not None:
                     if self.config.get("external_client", False):
@@ -606,6 +614,14 @@ async def handler(req: types.CallToolRequest):
                                 f"Input validation error: {e.message}"
                             )
 
+                    # Check if tool is restricted
+                    if hasattr(self, "config") and self.config is not None:
+                        restricted_tools = self.config.get("restricted_tools", [])
+                        if tool_name in restricted_tools:
+                            return self._make_error_result(
+                                f"Tool '{tool_name}' is restricted and cannot be used"
+                            )
+
                     # tool call
                     results = await func(tool_name, arguments)
                     content = list(results)