How to Use the Weights & Biases MCP in Mastra AI
Automate ML operations and manage W&B resources using the Mastra AI agent framework.
Works with every AI agent you already use
…and any MCP-compatible client
Connect Weights & Biases MCP to Mastra AI
Create your Vinkius account to connect Weights & Biases to Mastra AI and route execution through our secure gateway. The platform manages server hosting, runtime updates, and security layers. Configuration requires no manual server provisioning.
Discovering W&B Resources for Workflows
To start a workflow, you first need to locate the right project. Use `list_wandb_projects` to get all accessible projects in your W&B account. This output allows your agent framework to branch logic based on which resource is available. The Mastra AI client then knows exactly where to look for artifacts or reports, making complex multi-step workflows reliable.
Analyzing Run Details for Decision Making
When a workflow hits a decision point, it needs metrics. The `get_run_details` tool fetches full run data, including summary performance and config details. Your Mastra AI agent can read these results to decide if the next step should be retrying or failing. This is critical for building robust workflows that handle failure gracefully based on W&B's actual output.
Managing Sweeps and Runs in Automation
You can systematically check both runs (`list_project_runs`) and hyperparameter sweeps (`list_project_sweeps`). These tools give the agent a complete picture of experimental history within a W&B project. The Mastra AI framework uses this data to ensure that before deploying, all necessary optimization steps have been completed and documented.
Set up Weights & Biases MCP in Mastra AI
Prerequisites
- Node.js 18+ and a TypeScript project
-
@mastra/mcp+@mastra/corepackages - Active Vinkius subscription with a valid endpoint token
- 1
Install dependencies
Run
npm install @mastra/mcp @mastra/coreplus your preferred model provider (e.g.@ai-sdk/openai). - 2
Configure the MCPClient
Create an
MCPClientwith your Vinkius endpoint as aURLobject. Replace[YOUR_TOKEN_HERE]with your token from cloud.vinkius.com. - 3
Discover and inject tools
Call
mcpClient.listTools()and spread the result into your agent'stoolsobject. All Weights & Biases tools become native Mastra tools. - 4
Run with any model
Swap
openai("gpt-4o")for any AI SDK-compatible provider. Callagent.generate()and the agent routes tool calls through MCP automatically.
import { MCPClient } from "@mastra/mcp";
import { Agent } from "@mastra/core/agent";
import { openai } from "@ai-sdk/openai";
const mcpClient = new MCPClient({
id: "weights-biases-mcp-client",
servers: {
"weights-biases-mcp": {
url: new URL(
"https://edge.vinkius.com/[YOUR_TOKEN_HERE]/mcp"
),
},
},
});
const agent = new Agent({
name: "Weights & Biases Agent",
model: openai("gpt-4o"),
instructions: "You have access to Weights & Biases tools.",
tools: {
...(await mcpClient.listTools()),
},
});
const result = await agent.generate(
"List recent Weights & Biases transactions"
);
console.log(result.text); Independent Platform Disclaimer: Vinkius is an independent platform and is not affiliated with, endorsed by, sponsored by, verified by, or otherwise authorized by Weights & Biases. All third-party trademarks, logos, and brand names are the property of their respective owners. Their use on this website is strictly for informational purposes to identify service compatibility and interoperability.
Why Choose Vinkius
Vinkius connects your tools to AI with real-time monitoring and automatic cost savings — all from one dashboard.
Real-time monitoring
Live
visibility into every interaction
Connect your favorite tools to your AI and see exactly what's happening — every request, every response, in real time.
Built-in savings
60%
lower AI costs
Vinkius compresses data between your apps and your AI automatically. Lower bills every month — no configuration required.
Single dashboard
One
place for every integration
Every tool your AI connects to, managed from a single screen. One account, complete control.
Common questions about Weights & Biases MCP in Mastra AI
Use it with your favorite AI tools
Connect this server to Cursor, Claude, VS Code, and more.
Start using the Weights & Biases MCP today
We host it, we monitor it, we maintain it. You just paste one token.