Skip to content

datarobot_genai.drmcp.core.tool_config

tool_config

Tool configuration and enablement logic.

ToolType

Bases: StrEnum

Enumeration of available tool types.

Source code in datarobot_genai/drmcp/core/tool_config.py 
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class ToolType(StrEnum):
    """Enumeration of available tool types."""

    PREDICTIVE = "predictive"
    JIRA = "jira"
    CONFLUENCE = "confluence"
    GDRIVE = "gdrive"
    MICROSOFT_GRAPH = "microsoft_graph"
    PERPLEXITY = "perplexity"
    TAVILY = "tavily"
    DR_DOCS = "dr_docs"
    USE_CASE = "use_case"
    CODE_EXECUTION = "code_execution"
    OPTIMIZATION = "optimization"
    VDB = "vdb"
    PANELS = "panels"
    WORKLOAD = "workload"
    FILES_API = "files_api"

ToolConfig

Bases: TypedDict

Configuration for a tool type.

Source code in datarobot_genai/drmcp/core/tool_config.py 
45
46
47
48
49
50
51
class ToolConfig(TypedDict):
    """Configuration for a tool type."""

    name: str
    directory: str
    package_prefix: str
    config_field_name: str

get_tool_enable_config_name 

get_tool_enable_config_name(tool_type: ToolType) -> str

Get the configuration field name for enabling a tool.

Source code in datarobot_genai/drmcp/core/tool_config.py 
149
150
151
def get_tool_enable_config_name(tool_type: ToolType) -> str:
    """Get the configuration field name for enabling a tool."""
    return TOOL_CONFIGS[tool_type]["config_field_name"]

is_tool_enabled 

is_tool_enabled(tool_type: ToolType, config: MCPServerConfig) -> bool

Return whether tool_type is enabled in config.

Source code in datarobot_genai/drmcp/core/tool_config.py 
154
155
156
157
def is_tool_enabled(tool_type: ToolType, config: "MCPServerConfig") -> bool:
    """Return whether *tool_type* is enabled in *config*."""
    enable_config_name = TOOL_CONFIGS[tool_type]["config_field_name"]
    return bool(getattr(config.tool_config, enable_config_name))