Define toolsets (#285)

Just like `tools/tool_names.py` and `tools/policy.py`, it is helpful to
statically define toolsets for them to be referenced.

Author: DevonFulcher

.changes/unreleased/Under the Hood-20250813-160036.yaml

@@ -0,0 +1,3 @@
+kind: Under the Hood
+body: Define toolsets
+time: 2025-08-13T16:00:36.907595-05:00

src/dbt_mcp/tools/toolsets.py

@@ -0,0 +1,42 @@
+from enum import Enum
+
+from dbt_mcp.tools.tool_names import ToolName
+
+
+class Toolset(Enum):
+    SQL = "sql"
+    SEMANTIC_LAYER = "semantic_layer"
+    DISCOVERY = "discovery"
+    DBT_CLI = "dbt_cli"
+
+
+toolsets = {
+    Toolset.SQL: {
+        ToolName.TEXT_TO_SQL,
+        ToolName.EXECUTE_SQL,
+    },
+    Toolset.SEMANTIC_LAYER: {
+        ToolName.LIST_METRICS,
+        ToolName.GET_DIMENSIONS,
+        ToolName.GET_ENTITIES,
+        ToolName.QUERY_METRICS,
+        ToolName.GET_METRICS_COMPILED_SQL,
+    },
+    Toolset.DISCOVERY: {
+        ToolName.GET_MART_MODELS,
+        ToolName.GET_ALL_MODELS,
+        ToolName.GET_MODEL_DETAILS,
+        ToolName.GET_MODEL_PARENTS,
+        ToolName.GET_MODEL_CHILDREN,
+    },
+    Toolset.DBT_CLI: {
+        ToolName.BUILD,
+        ToolName.COMPILE,
+        ToolName.DOCS,
+        ToolName.LIST,
+        ToolName.PARSE,
+        ToolName.RUN,
+        ToolName.TEST,
+        ToolName.SHOW,
+    },
+}

tests/unit/tools/test_toolsets.py

@@ -0,0 +1,29 @@
+from dbt_mcp.config.config import load_config
+from dbt_mcp.mcp.server import create_dbt_mcp
+from dbt_mcp.tools.toolsets import toolsets
+from tests.env_vars import default_env_vars_context
+
+
+async def test_toolsets_match_server_tools():
+    """Test that the defined toolsets match the tools registered in the server."""
+    sql_tool_names = {"text_to_sql", "execute_sql"}
+
+    with default_env_vars_context():
+        config = load_config()
+        dbt_mcp = await create_dbt_mcp(config)
+
+        # Get all tools from the server
+        server_tools = await dbt_mcp.list_tools()
+        # Manually adding SQL tools here because the server doesn't get them
+        # in this unit test.
+        server_tool_names = {tool.name for tool in server_tools} | sql_tool_names
+        defined_tools = set()
+        for toolset_tools in toolsets.values():
+            defined_tools.update({t.value for t in toolset_tools})
+
+        if server_tool_names != defined_tools:
+            raise ValueError(
+                f"Tool name mismatch:\n"
+                f"In server but not in enum: {server_tool_names - defined_tools}\n"
+                f"In enum but not in server: {defined_tools - server_tool_names}"
+            )