Treasure Data MCP 服务

基本能力

产品定位

Treasure Data MCP 服务是一个为 Claude Code 和 Claude Desktop 提供 Treasure Data API 集成的 MCP 服务器。

核心功能

1. 数据库管理

   • td_list_databases: 获取 Treasure Data 账户中的数据库列表，支持分页和详细模式。
   • td_get_database: 获取特定数据库的详细信息。
   • td_list_tables: 获取特定数据库中的表列表，支持分页和详细模式。

2. 工作流项目管理

   • td_list_projects: 获取工作流项目列表，支持分页和详细模式。
   • td_get_project: 获取特定项目的详细信息。
   • td_download_project_archive: 下载项目的存档文件。
   • td_list_project_files: 列出项目存档中的所有文件。
   • td_read_project_file: 读取项目存档中的特定文件内容。

适用场景

• 需要批量查询和管理 Treasure Data 数据库和表的场景。
• 需要下载和分析 Treasure Data 工作流项目内容的场景。
• 需要自动化操作 Treasure Data API 的场景。

使用教程

使用依赖

1. 确保已安装 Python 3.11+。
2. 安装依赖库：
   复制

   pip install requests>=2.28.0 pydantic>=2.0.0 mcp[cli]>=1.8.1
   

安装教程

1. 克隆仓库：

   复制

   git clone https://github.com/knishioka/td-mcp-server.git
   cd td-mcp-server
   

2. 安装依赖：

   复制

   pip install -r requirements.txt
   

   或使用 uv（推荐）：

   复制

   uv pip install -e .
   

3. 设置环境变量并运行：

   复制

   export TD_API_KEY="your-api-key"
   export TD_ENDPOINT="api.treasuredata.com"
   mcp run td_mcp_server/server.py
   

调试方式

1. 运行测试：
   复制

   pytest
   

2. 检查代码格式：
   复制

   uv run ruff check td_mcp_server tests
   

3. 格式化代码：
   复制

   uv run ruff format td_mcp_server tests