swanlab-mcp

<div align="center"> # SwanLab MCP Server [![][github-shield]][github-shield-link] &nbsp; &nbsp; [![][pypi-version-shield]][pypi-version-shield-link] &nbsp; &nbsp; [![][license-shield]][license-shield-link] </div> > A Model Context Protocol (MCP) server implementation for SwanLab, combining SwanLab-OpenAPI & FastMCP. ## ✨ Features ### Core Features - **Workspace Queries** - List accessible workspaces and enumerate workspace projects - **Project Queries** - List projects and inspect a specific project with run summaries - **Run Queries** - Inspect runs with normalized fields (`id`, `state`, `profile`, `user`) - **Metric Queries** - Fetch metric tables with consistent `columns`, `rows`, and `total` - **API Integration** - Provide read-only access through SwanLab OpenAPI (`swanlab.Api`) ### Tech Stack - **Language**: Python 3.12+ - **Core Framework**: FastMCP (v2.14.4+) - **API Client**: SwanLab SDK - **Config Management**: Pydantic Settings ## 🚀 Quick Start ### ❗️Configuration Add the following configuration to your relative mcp config list ```json { "mcpServers": ... { "swanlab-mcp": { "command": "uvx", "args": ["--from", "swanlab-mcp", "swanlab_mcp", "--transport", "stdio"], "env": { "SWANLAB_API_KEY": "your_api_key_here" } } } } ``` For `Claude Code` Users, you can config like this: ```bash claude mcp add --env SWANLAB_API_KEY=<your_api_key> -- swanlab_mcp uvx --from swanlab-mcp swanlab_mcp --transport stdio ``` ### Prerequisites - Python >= 3.12 - SwanLab API Key (get it from [SwanLab](https://swanlab.cn)) ### Installation ```bash # Using uv (recommended) uv sync # Or using pip pip install -e . ``` ### Configuration #### Environment Variables Create a `.env` file and configure your API key: ```bash cp .env.template .env ``` Edit the `.env` file: ```env SWANLAB_API_KEY=your_api_key_here ``` ### Running ```bash # Using stdio transport (default) python -m swanlab_mcp # Or using CLI …