wandb-mcp-server
The Weights & Biases MCP Server enables querying of W&B data for enhanced data manipulation and reporting capabilities. It supports a range of functionalities including querying models, generating reports, and interacting with support bots, offering robust analysis and documentation tools for users.
Weights & Biases MCP Server
A Model Context Protocol (MCP) server for querying Weights & Biases data, allowing clients to:
- Query W&B model runs, sweeps, artifacts, and registry.
- Query W&B Weave traces, evaluations, and datasets.
- Write text and charts to W&B reports.
- Consult the W&B support bot for assistance with W&B features.
Installation
A utility is provided for easy installation into applications using a JSON server spec. Requires installing uv and using the add_to_client helper for popular clients. Manual installation instructions are also provided.
Available W&B Tools
- wandb: Execute GraphQL queries against W&B experiment tracking data.
- Weave: Query and count Weave traces with various options.
- W&B Support bot: Ask wandbot for help on W&B products and features.
- Saving Analysis: Create W&B Reports with markdown text and visualizations.
- General W&B helpers: List W&B entities and projects to provide more context for LLM queries.
Usage Tips
- Specify W&B entity and project name to improve query clarity.
- Avoid overly broad questions for best results.
- Ensure all data is retrieved when querying.
Troubleshooting
Error: spawn uv ENOENT
To resolve this error, ensure uv is installed and available system-wide by creating a symlink if necessary. Restart applications after changes.
Testing
Includes unit tests and integration tests; requires setting up an API key in the .env file. Instructions for running tests with pytest are included.