dbt-mcp

from enum import Enum from typing import Literal from dbt_mcp.tools.tool_names import ToolName class Toolset(Enum): SQL = "sql" SEMANTIC_LAYER = "semantic_layer" DISCOVERY = "discovery" DBT_CLI = "dbt_cli" ADMIN_API = "admin_api" DBT_CODEGEN = "dbt_codegen" DBT_LSP = "dbt_lsp" proxied_tools: set[Literal[ToolName.TEXT_TO_SQL, ToolName.EXECUTE_SQL]] = set( [ ToolName.TEXT_TO_SQL, ToolName.EXECUTE_SQL, ] ) toolsets = { Toolset.SQL: { ToolName.TEXT_TO_SQL, ToolName.EXECUTE_SQL, }, Toolset.SEMANTIC_LAYER: { ToolName.LIST_METRICS, ToolName.LIST_SAVED_QUERIES, 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, ToolName.GET_MODEL_HEALTH, ToolName.GET_ALL_SOURCES, ToolName.GET_EXPOSURES, ToolName.GET_EXPOSURE_DETAILS, }, Toolset.DBT_CLI: { ToolName.BUILD, ToolName.COMPILE, ToolName.DOCS, ToolName.LIST, ToolName.PARSE, ToolName.RUN, ToolName.TEST, ToolName.SHOW, }, Toolset.ADMIN_API: { ToolName.LIST_JOBS, ToolName.GET_JOB_DETAILS, ToolName.TRIGGER_JOB_RUN, ToolName.LIST_JOBS_RUNS, ToolName.GET_JOB_RUN_DETAILS, ToolName.CANCEL_JOB_RUN, ToolName.RETRY_JOB_RUN, ToolName.LIST_JOB_RUN_ARTIFACTS, ToolName.GET_JOB_RUN_ARTIFACT, ToolName.GET_JOB_RUN_ERROR, }, Toolset.DBT_CODEGEN: { ToolName.GENERATE_SOURCE, ToolName.GENERATE_MODEL_YAML, ToolName.GENERATE_STAGING_MODEL, }, Toolset.DBT_LSP: { ToolName.GET_COLUMN_LINEAGE, }, }