Dataset Viewer MCP Server

Dataset Viewer MCP 服务

一个用于与 Hugging Face Dataset Viewer API 交互的 MCP 服务，可浏览和分析托管在 Hugging Face Hub 上的数据集。

功能

资源

• 使用 dataset:// URI 方案访问 Hugging Face 数据集
• 支持数据集配置和拆分
• 提供对数据集内容的分页访问
• 支持私有数据集的身份验证
• 支持搜索和筛选数据集内容
• 提供数据集统计信息和分析

工具

此服务提供以下工具：

1. validate

   • 检查数据集是否存在且可访问
   • 参数：
     • dataset：数据集标识符（例如 'stanfordnlp/imdb'）
     • auth_token（可选）：用于私有数据集

2. get_info

   • 获取数据集的详细信息
   • 参数：
     • dataset：数据集标识符
     • auth_token（可选）：用于私有数据集

3. get_rows

   • 分页获取数据集内容
   • 参数：
     • dataset：数据集标识符
     • config：配置名称
     • split：拆分名称
     • page（可选）：页码（从 0 开始）
     • auth_token（可选）：用于私有数据集

4. get_first_rows

   • 获取数据集拆分中的前几行
   • 参数：
     • dataset：数据集标识符
     • config：配置名称
     • split：拆分名称
     • auth_token（可选）：用于私有数据集

5. get_statistics

   • 获取数据集拆分的统计信息
   • 参数：
     • dataset：数据集标识符
     • config：配置名称
     • split：拆分名称
     • auth_token（可选）：用于私有数据集

6. search_dataset

   • 在数据集中搜索文本
   • 参数：
     • dataset：数据集标识符
     • config：配置名称
     • split：拆分名称
     • query：要搜索的文本
     • auth_token（可选）：用于私有数据集

7. filter

   • 使用类似 SQL 的条件筛选行
   • 参数：
     • dataset：数据集标识符
     • config：配置名称
     • split：拆分名称
     • where：SQL WHERE 子句（例如 "score > 0.5"）
     • orderby（可选）：SQL ORDER BY 子句
     • page（可选）：页码（从 0 开始）
     • auth_token（可选）：用于私有数据集

8. get_parquet

   • 下载 Parquet 格式的完整数据集
   • 参数：
     • dataset：数据集标识符
     • auth_token（可选）：用于私有数据集

安装

前提条件

• Python 3.12 或更高版本
• uv - 快速的 Python 包安装器和解析器

设置

1. 克隆仓库：

git clone https://github.com/privetin/dataset-viewer.git
cd dataset-viewer

2. 创建虚拟环境并安装：

# Create virtual environment
uv venv

# Activate virtual environment
# On Unix:
source .venv/bin/activate
# On Windows:
.venv\Scripts\activate

# Install in development mode
uv add -e .

配置

环境变量

• HUGGINGFACE_TOKEN：用于访问私有数据集的 Hugging Face API 令牌

Claude Desktop 集成

将以下内容添加到 Claude Desktop 配置文件中：

Windows：%APPDATA%\Claude\claude_desktop_config.json

MacOS：~/Library/Application Support/Claude/claude_desktop_config.json

{
  "mcpServers": {
    "dataset-viewer": {
      "command": "uv",
      "args": [
        "--directory",
        "parent_to_repo/dataset-viewer",
        "run",
        "dataset-viewer"
      ]
    }
  }
}

许可证

MIT 许可证 - 详情请参阅 LICENSE