Skip to content

datarobot_genai.drmcpbase.feature_flags

feature_flags 

check_mcp_tools_gallery_support(datarobot_api_client: DataRobotClientWithAsyncAPI | None) -> bool

Return True when the MCP tools-gallery feature is enabled for the request's user.

Shared entry point for everything gated on ENABLE_MCP_TOOLS_GALLERY_SUPPORT: the tool providers (which pass their lifespan client) and the global-mcp /toolGallery/tools/ route gate (which passes a short-lived client). None-safe — returns False when the client is not yet available (e.g. a request that races a provider's lifespan), so callers no longer need a separate is_datarobot_api_client_initialized guard. Otherwise it delegates to :meth:FeatureFlagEvaluation.is_mcp_tools_gallery_support_enabled_for_user_in_mcp_request, which reads the user's bearer token from the request context and fails closed to False.

Source code in datarobot_genai/drmcpbase/feature_flags.py 
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
async def check_mcp_tools_gallery_support(
    datarobot_api_client: DataRobotClientWithAsyncAPI | None,
) -> bool:
    """Return True when the MCP tools-gallery feature is enabled for the request's user.

    Shared entry point for everything gated on ``ENABLE_MCP_TOOLS_GALLERY_SUPPORT``: the
    tool providers (which pass their lifespan client) and the global-mcp ``/toolGallery/tools/``
    route gate (which passes a short-lived client). None-safe — returns False when the
    client is not yet available (e.g. a request that races a provider's lifespan), so
    callers no longer need a separate ``is_datarobot_api_client_initialized`` guard.
    Otherwise it delegates to
    :meth:`FeatureFlagEvaluation.is_mcp_tools_gallery_support_enabled_for_user_in_mcp_request`,
    which reads the user's bearer token from the request context and fails closed to False.
    """
    if datarobot_api_client is None:
        return False
    return await FeatureFlagEvaluation.is_mcp_tools_gallery_support_enabled_for_user_in_mcp_request(
        datarobot_api_client
    )