dataio.mcp.types

Type definitions for the DataIO MCP server.

Module Contents

Classes

ToolName

Available MCP tools.

ToolResult

Result from a tool execution.

ToolError

Error from a tool execution.

UserContext

Context about the user making the request.

ToolDefinition

Definition of an MCP tool.

BedrockToolSpec

Tool specification in AWS Bedrock format.

API

class dataio.mcp.types.ToolName[source]

Bases: str, enum.Enum

Available MCP tools.

Initialization

Initialize self. See help(type(self)) for accurate signature.

SEARCH_DATASETS

‘search_datasets’

GET_DATASET_DETAILS

‘get_dataset_details’

LIST_CATEGORIES

‘list_categories’

LIST_DATA_OWNERS

‘list_data_owners’

GET_DOWNLOAD_INFO

‘get_download_info’

GET_DATASET_SCHEMA

‘get_dataset_schema’

class dataio.mcp.types.ToolResult[source]

Result from a tool execution.

success: bool

None

data: Any

None

error: str | None

None

class dataio.mcp.types.ToolError[source]

Error from a tool execution.

code: str

None

message: str

None

details: dict | None

None

class dataio.mcp.types.UserContext[source]

Context about the user making the request.

email: str

None

is_admin: bool

False

groups: list[str]

‘field(…)’

class dataio.mcp.types.ToolDefinition[source]

Definition of an MCP tool.

name: str

None

description: str

None

input_schema: dict

None

class dataio.mcp.types.BedrockToolSpec[source]

Tool specification in AWS Bedrock format.

name: str

None

description: str

None

input_schema: dict

None

to_bedrock_format() dict[source]

Convert to Bedrock API format.