Treasure Data MCP Server
A Model Context Protocol (MCP) server that provides Treasure Data API integration for Claude Code and Claude Desktop, allowing for database management and listing functionality.
Installation
# Install from source
git clone https://github.com/yourusername/td-mcp-server.git
cd td-mcp-server
pip install -e .
Authentication
The client requires a Treasure Data API key for authentication. You can provide this in two ways:
- Set the
TD_API_KEY environment variable:export TD_API_KEY="your-api-key"
- Pass it directly to the command:
python -m td_mcp_server --api-key="your-api-key" list-databases
Usage
Command Line Interface
The package provides a simple command-line interface for common operations:
List all databases
# Show databases in table format (default)
python -m td_mcp_server list-databases
# Get JSON output
python -m td_mcp_server list-databases --format=json
# Specify a different region endpoint
python -m td_mcp_server --endpoint=api.treasuredata.co.jp list-databases
# Get JSON output (default)
python -m td_mcp_server get-database my_database_name
# Get table output
python -m td_mcp_server get-database my_database_name --format=table
Python API
You can also use the client directly in your Python code:
from td_mcp_server.api import TreasureDataClient
# Initialize client with API key from environment variable
client = TreasureDataClient()
# Or provide API key directly
client = TreasureDataClient(api_key="your-api-key")
# Get all databases
databases = client.get_databases()
for db in databases:
print(f"Database: {db.name}, Tables: {db.count}")
# Get information about a specific database
db = client.get_database("my_database_name")
if db:
print(f"Found database: {db.name}")
else:
print("Database not found")
API Endpoints
By default, the client uses the US region endpoint (api.treasuredata.com). If you need to use the Japan region, specify the endpoint:
client = TreasureDataClient(endpoint="api.treasuredata.co.jp")
python -m td_mcp_server --endpoint=api.treasuredata.co.jp list-databases
MCP Server Configuration
This server implements the Model Context Protocol (MCP) to provide Claude with access to Treasure Data API functionality.
Running the MCP Server
You can run the MCP server in several ways:
# Using uv run (recommended)
uv run mcp
# Using Python module
python -m td_mcp_server --mcp-server
# Using the installed script
mcp
The server requires a Treasure Data API key, which should be provided via the TD_API_KEY environment variable.
Setting up with Claude Code
To configure this MCP server for use with Claude Code:
- Install the server
git clone https://github.com/yourusername/td-mcp-server.git
cd td-mcp-server
pip install -e .
- Set your Treasure Data API key as an environment variable
export TD_API_KEY="your-api-key"
- Create a
.claude/plugins.json file in your project directory with the following content:{
"plugins": [
{
"name": "td-mcp",
"description": "Treasure Data API plugin for database management",
"command": {
"args": ["uv", "run", "mcp"],
"env": {
"TD_API_KEY": "${TD_API_KEY}"
}
}
}
]
}
- When using Claude Code in a project with this configuration, you'll have access to the following MCP tools:
mcp__td_list_databases: List all databases in your Treasure Data accountmcp__td_get_database: Get information about a specific database
Setting up with Claude Desktop
To configure this MCP server for use with Claude Desktop:
- Install the server as described above
- In Claude Desktop, create a new MCP tool configuration:
- Go to Settings > MCP Tools > Add New Tool
- Name: Treasure Data API
- Command:
uv run mcp - Environment variables: Add your
TD_API_KEY
- Save the configuration and restart Claude Desktop
- You can now use the Treasure Data API tools in your Claude Desktop conversations
Once configured, you can use commands like:
or
/mcp td_get_database my_database_name
Development
To set up the development environment:
# Clone the repository
git clone https://github.com/yourusername/td-mcp-server.git
cd td-mcp-server
# Install dev dependencies
pip install -e ".[dev]"
# Run tests
pytest