Weights & Biases MCP Server logo

Weights & Biases MCP Server

by wandb

The Weights & Biases MCP Server is a Model Context Protocol server designed for querying Weights & Biases Weave traces. It enables a MCP Client to query traces and write to W&B Reports.

View on GitHub

Last updated: N/A

What is Weights & Biases MCP Server?

The Weights & Biases MCP Server is a server that allows querying Weights & Biases Weave traces and writing text and charts to W&B Reports using a Model Context Protocol (MCP) client.

How to use Weights & Biases MCP Server?

  1. Install the server using the provided installation instructions (git clone, uv venv, uv pip install). 2. Configure the server by creating a .env file with your WANDB_API_KEY. 3. Run the server using uv run src/mcp_server/server.py. 4. Configure your MCP client (e.g., Claude Desktop) to connect to the server.

Key features of Weights & Biases MCP Server

  • Query W&B Weave traces

  • Write text and charts to W&B Reports

  • Execute GraphQL queries against wandb experiment tracking data

  • Efficiently count Weave traces matching given filters

Use cases of Weights & Biases MCP Server

  • Analyzing and visualizing experiment results

  • Creating shareable reports with findings and generated charts

  • Integrating W&B data into LLM-powered applications

  • Querying and filtering Weave traces for specific insights

FAQ from Weights & Biases MCP Server

What is the purpose of the MCP server?

The MCP server allows querying W&B Weave traces and writing to W&B Reports from an MCP client.

How do I install the MCP server?

Clone the repository, create a virtual environment using uv venv, activate the environment, and install the server using uv pip install -e ..

How do I configure the MCP server?

Create a .env file in the root directory with your Weights & Biases API key: WANDB_API_KEY=your_api_key_here.

How do I run the MCP server?

Run the server using the command: uv run src/mcp_server/server.py.

What do I do if I encounter a 'spawn uv ENOENT' error?

Install uv using the official installation script and create a symlink to make uv available system-wide.