Treasure Data MCP-Server
Ein Model Context Protocol (MCP)-Server, der die Treasure Data API-Integration für Claude Code und Claude Desktop bereitstellt.
HAFTUNGSAUSSCHLUSS : Dies ist ein persönliches Entwicklungsprojekt und steht in keiner Verbindung zu Treasure Data Inc., wird von Treasure Data Inc. unterstützt oder steht in irgendeiner Verbindung zu Treasure Data Inc. Diese Software wird ohne jegliche Gewährleistung bereitgestellt und erfolgt auf eigene Gefahr. Der Autor haftet nicht für die Folgen, die sich aus der Nutzung dieser Software ergeben.
Dieser MCP-Server bietet die folgenden Tools zur Interaktion mit Treasure Data:
Datenbankverwaltung
- td_list_databases
td_list_databases(verbose=False, limit=30, offset=0, all_results=False)
- Holen Sie sich Datenbanken in Ihr Treasure Data-Konto mit Paginierungsunterstützung
- Parameter :
verbose : Wenn „True“, werden alle Details zurückgegeben; wenn „False“, werden nur Namen zurückgegeben (Standard)limit : Maximale Anzahl abzurufender Datenbanken (Standardwert: 30)offset : Index, ab dem mit dem Abrufen begonnen werden soll (Standardwert: 0)all_results : Wenn True, ruft alle Datenbanken ab und ignoriert dabei Limit und Offset
- Beispiele :
# 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)
- Erhalten Sie detaillierte Informationen zu einer bestimmten Datenbank
- Parameter :
database_name : Der Name der Datenbank, für die Informationen abgerufen werden sollen
- Beispiel :
# 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)
- Holen Sie sich Tabellen in einer bestimmten Treasure Data-Datenbank mit Paginierungsunterstützung
- Parameter :
database_name : Der Name der Datenbank, aus der Tabellen abgerufen werden sollenverbose : Wenn „True“, werden alle Details zurückgegeben; wenn „False“, werden nur Namen zurückgegeben (Standard)limit : Maximale Anzahl abzurufender Tabellen (Standardwert: 30)offset : Index, ab dem mit dem Abrufen begonnen werden soll (Standardwert: 0)all_results : Wenn True, werden alle Tabellen abgerufen, wobei Limit und Offset ignoriert werden
- Beispiele :
# 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
Workflow-Projektmanagement
- td_list_projects
td_list_projects(verbose=False, limit=30, offset=0, all_results=False, include_system=False)
- Holen Sie sich Workflow-Projekte in Ihr Treasure Data-Konto mit Paginierungsunterstützung
- Parameter :
verbose : Wenn „True“, werden alle Details zurückgegeben; wenn „False“, werden nur Namen und IDs zurückgegeben (Standard).limit : Maximale Anzahl abzurufender Projekte (Standard: 30)offset : Index, ab dem mit dem Abrufen begonnen werden soll (Standardwert: 0)all_results : Wenn True, ruft alle Projekte ab, wobei Limit und Offset ignoriert werdeninclude_system : Wenn „True“, schließe systemgenerierte Projekte ein (mit „sys“-Metadaten); standardmäßig „False“
- Beispiele :
# 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)
- Erhalten Sie detaillierte Informationen zu einem bestimmten Workflow-Projekt
- Hinweis: Hier werden nur grundlegende Projektmetadaten bereitgestellt. Für detaillierte Inhalte und Dateien verwenden Sie td_download_project_archive, gefolgt von td_list_project_files und td_read_project_file.
- Parameter :
project_id : Die ID des Workflow-Projekts, für das Informationen abgerufen werden sollen
- Beispiel :
# Get information about a specific project
td_get_project project_id=123456
- td_download_project_archive
td_download_project_archive(project_id)
- Laden Sie das Archiv eines Projekts (tar.gz) herunter und geben Sie Informationen zum Download zurück
- Empfohlen für die Untersuchung detaillierter Projektinhalte einschließlich SQL-Abfragen und Workflow-Definitionen
- Parameter :
project_id : Die ID des herunterzuladenden Workflow-Projekts
- Beispiel :
# Download a project's archive
td_download_project_archive project_id=123456
- td_list_project_files
td_list_project_files(archive_path)
- Alle in einem Projektarchiv enthaltenen Dateien auflisten
- Parameter :
archive_path : Der Pfad zum heruntergeladenen Projektarchiv (.tar.gz-Datei)
- Beispiel :
# 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)
- Lesen Sie den Inhalt einer bestimmten Datei aus einem Projektarchiv
- Parameter :
archive_path : Der Pfad zum heruntergeladenen Projektarchiv (.tar.gz-Datei)file_path : Der Pfad der zu lesenden Datei im Archiv
- Beispiel :
# 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
Installationsanweisungen
Authentifizierung
Dieser MCP-Server benötigt zur Authentifizierung einen Treasure Data API-Schlüssel, der über die Umgebungsvariable TD_API_KEY bereitgestellt werden sollte. Sie können den Treasure Data-Endpunkt auch mit der Umgebungsvariable TD_ENDPOINT angeben (standardmäßig api.treasuredata.com ).
Einrichten mit Claude Code
- Klonen Sie das Repository
git clone https://github.com/knishioka/td-mcp-server.git
- Fügen Sie den MCP-Server mithilfe der Claude Code CLI hinzu
# 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
Einrichten mit Claude Desktop
Konfigurieren Sie diesen MCP-Server für die Verwendung mit Claude Desktop, indem Sie Ihre Konfigurationsdatei ( claude_desktop_config.json ) bearbeiten:
{
"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"
}
}
}
}
Installation und Anforderungen
Dieses Projekt erfordert Python 3.11+ und die folgenden Pakete:
Installieren Sie die Abhängigkeiten mit pip:
pip install -r requirements.txt
Oder mit UV:
Direktes Ausführen des Servers
Sie können den MCP-Server direkt ausführen:
# 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
Entwicklung
Ausführen von Tests
# 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