보물 데이터 MCP 서버
Claude Code와 Claude Desktop에 대한 Treasure Data API 통합을 제공하는 MCP(Model Context Protocol) 서버입니다.
면책 조항 : 본 소프트웨어는 개인 개발 프로젝트이며, Treasure Data Inc.와 어떠한 제휴, 보증 또는 관련이 없습니다. 본 소프트웨어는 어떠한 보증도 없이 "있는 그대로" 제공되며, 사용 시 발생하는 모든 책임은 사용자에게 있습니다. 본 소프트웨어 사용으로 인해 발생하는 모든 결과에 대해 작성자는 책임을 지지 않습니다.
사용 가능한 MCP 도구
이 MCP 서버는 Treasure Data와 상호 작용하기 위한 다음과 같은 도구를 제공합니다.
데이터베이스 관리
- td_list_databases지엑스피1
- 페이지 매김 지원을 통해 Treasure Data 계정에서 데이터베이스를 가져오세요
- 매개변수 :
verbose : True이면 전체 세부 정보를 반환하고 False이면 이름만 반환합니다(기본값)limit : 검색할 데이터베이스의 최대 개수(기본값은 30)offset : 검색을 시작할 인덱스(기본값은 0)all_results : True인 경우 제한 및 오프셋을 무시하고 모든 데이터베이스를 검색합니다.
- 예시 :
# Get only database names (default, first 30 databases)
td_list_databases
# Get full database details
td_list_databases verbose=True
# Pagination options
td_list_databases limit=10 offset=20
# Get all databases regardless of the number
td_list_databases all_results=True
- td_get_database
td_get_database(database_name)
- 특정 데이터베이스에 대한 자세한 정보를 얻으세요
- 매개변수 :
database_name : 정보를 검색할 데이터베이스의 이름
- 예 :
# Get information about a specific database
td_get_database database_name=my_database_name
- td_list_tables
td_list_tables(database_name, verbose=False, limit=30, offset=0, all_results=False)
- 페이지 매김 지원을 통해 특정 Treasure Data 데이터베이스에서 테이블 가져오기
- 매개변수 :
database_name : 테이블을 검색할 데이터베이스의 이름verbose : True이면 전체 세부 정보를 반환하고 False이면 이름만 반환합니다(기본값)limit : 검색할 테이블의 최대 개수(기본값은 30)offset : 검색을 시작할 인덱스(기본값은 0)all_results : True인 경우 제한 및 오프셋을 무시하고 모든 테이블을 검색합니다.
- 예시 :
# Get only table names in a database (default, first 30 tables)
td_list_tables database_name=my_database_name
# Get detailed information about tables in a database
td_list_tables database_name=my_database_name verbose=True
# Pagination options
td_list_tables database_name=my_database_name limit=10 offset=20
# Get all tables in a database
td_list_tables database_name=my_database_name all_results=True
워크플로 프로젝트 관리
- td_list_projects
td_list_projects(verbose=False, limit=30, offset=0, all_results=False, include_system=False)
- 페이지 매김 지원을 통해 Treasure Data 계정에서 워크플로 프로젝트를 가져옵니다.
- 매개변수 :
verbose : True이면 전체 세부 정보를 반환하고 False이면 이름과 ID만 반환합니다(기본값)limit : 검색할 최대 프로젝트 수(기본값은 30)offset : 검색을 시작할 인덱스(기본값은 0)all_results : True인 경우 제한 및 오프셋을 무시하고 모든 프로젝트를 검색합니다.include_system : True인 경우 시스템 생성 프로젝트("sys" 메타데이터 포함)가 포함됩니다. 기본값은 False입니다.
- 예시 :
# Get basic project info (default, first 30 projects)
td_list_projects
# Get detailed project information
td_list_projects verbose=True
# Pagination options
td_list_projects limit=10 offset=20
# Get all projects regardless of the number
td_list_projects all_results=True
# Include system-generated projects
td_list_projects include_system=True
- td_get_project
td_get_project(project_id)
- 특정 워크플로 프로젝트에 대한 자세한 정보를 얻으세요
- 참고: 이 기능은 기본적인 프로젝트 메타데이터만 제공합니다. 자세한 내용과 파일을 보려면 td_download_project_archive 다음에 td_list_project_files와 td_read_project_file을 사용하세요.
- 매개변수 :
project_id : 정보를 검색할 워크플로 프로젝트의 ID
- 예 :
# Get information about a specific project
td_get_project project_id=123456
- td_다운로드_프로젝트_아카이브
td_download_project_archive(project_id)
- 프로젝트 아카이브(tar.gz)를 다운로드하고 다운로드에 대한 정보를 반환합니다.
- SQL 쿼리 및 워크플로 정의를 포함한 자세한 프로젝트 내용을 검토하는 데 권장됩니다.
- 매개변수 :
project_id : 다운로드할 워크플로 프로젝트의 ID
- 예 :
# Download a project's archive
td_download_project_archive project_id=123456
- td_list_project_files
td_list_project_files(archive_path)
- 프로젝트 아카이브에 포함된 모든 파일을 나열합니다.
- 매개변수 :
archive_path : 다운로드한 프로젝트 아카이브(.tar.gz 파일)의 경로
- 예 :
# List files in a downloaded project archive
td_list_project_files archive_path=/tmp/td_project_123/project_123456.tar.gz
- td_read_project_file
td_read_project_file(archive_path, file_path)
- 프로젝트 아카이브에서 특정 파일의 내용을 읽습니다.
- 매개변수 :
archive_path : 다운로드한 프로젝트 아카이브(.tar.gz 파일)의 경로file_path : 읽을 아카이브 내의 파일 경로
- 예 :
# Read a specific file from a project archive
td_read_project_file archive_path=/tmp/td_project_123/project_123456.tar.gz file_path=workflow.dig
설치 지침
입증
이 MCP 서버는 인증을 위해 Treasure Data API 키가 필요하며, 이 키는 TD_API_KEY 환경 변수를 통해 제공해야 합니다. TD_ENDPOINT 환경 변수(기본값: api.treasuredata.com )를 사용하여 Treasure Data 엔드포인트를 지정할 수도 있습니다.
Claude Code 설정
- 저장소를 복제합니다
git clone https://github.com/knishioka/td-mcp-server.git
- Claude Code CLI를 사용하여 MCP 서버 추가
# Navigate to your project directory
cd your-project-directory
# Add the MCP server (use absolute path to server.py)
claude mcp add td -e TD_API_KEY=${TD_API_KEY} -e TD_ENDPOINT=api.treasuredata.com -- mcp run /absolute/path/to/td-mcp-server/td_mcp_server/server.py
Claude Desktop 설정
Claude Desktop과 함께 사용하도록 이 MCP 서버를 구성하려면 구성 파일( claude_desktop_config.json )을 편집하세요.
{
"mcpServers": {
"td": {
"command": "uv",
"args": [
"--directory",
"/absolute/path/to/td-mcp-server",
"run",
"td_mcp_server/server.py"
],
"env": {
"TD_API_KEY": "YOUR_API_KEY",
"TD_ENDPOINT": "api.treasuredata.com"
}
}
}
}
설치 및 요구 사항
이 프로젝트에는 Python 3.11 이상과 다음 패키지가 필요합니다.
pip를 사용하여 종속성을 설치합니다.
pip install -r requirements.txt
또는 uv를 사용하여:
서버를 직접 실행
MCP 서버를 직접 실행할 수 있습니다.
# Set your API key
export TD_API_KEY="your-api-key"
# For US region (default)
export TD_ENDPOINT="api.treasuredata.com"
# For Japan region
# export TD_ENDPOINT="api.treasuredata.co.jp"
# Run with MCP CLI
mcp run td_mcp_server/server.py
개발
테스트 실행
# Run all tests
pytest
# Run tests with coverage report
pytest --cov=td_mcp_server
# Run tests for a specific module
pytest tests/unit/test_api.py
코드 포맷팅 및 린팅
# Run linting with Ruff
uv run ruff check td_mcp_server tests
# Format code with Ruff
uv run ruff format td_mcp_server tests
# Run pre-commit hooks on all files
uv run pre-commit run --all-files