Weights & Biases MCP server for AI agents and assistants

Securely connect your AI agents and chatbots (Claude, ChatGPT, Cursor, etc) with Weights & Biases MCP or direct API to inspect projects, summarize runs, compare sweeps, and review artifacts and reports through natural language.

Weights & Biases logoWeights & Biases
Api Key

Weights & Biases is an MLOps platform for tracking experiments, models, sweeps, artifacts, and reports. It helps ML teams monitor runs, compare results, and manage model development in one place.

10 Tools

Try Weights & Biases now

Type what you want done — sign in and watch it run live in the Tool Router playground.

TOOL ROUTER PLAYGROUND
Weights & Biases
Try asking
TOOLS

Supported Tools

Every Weights & Biases action and event your agent gets out of the box.

Get Current Viewer

Return the authenticated W&B user, default entity, and team names for connection context.

Get Organization

Return a W&B organization's identifiers and organization entity by exact name.

Get Team

Return a W&B team's identity and bounded membership metadata by exact team entity name.

List Artifact Types

Return one page of artifact types defined in a W&B project.

List Automations

Return one page of W&B automations accessible under an entity.

List Integrations

List one page of Slack workspace/channel metadata and webhook integration names for a W&B entity.

List Projects

Return one page of W&B projects visible under an entity.

List Reports

Return one page of beta W&B reports for a project.

List Runs

List one page of runs with IDs, display names, states, and creation times.

List Sweeps

Return one page of hyperparameter sweeps from a W&B project when sweeps are enabled for the account.

SETUP GUIDE

Connect Weights & Biases MCP Tool with your Agent

1

Install Composio

typescript
npm install @composio/core ai @ai-sdk/mcp @ai-sdk/openai
Install the Composio SDK and your agent framework
2

Create a session with MCP enabled

typescript
import { Composio } from "@composio/core";

const composio = new Composio();
const { mcp } = await composio.create("your-user-id", {
  toolkits: ["weights_and_biases"],
  mcp: true,
});
Create a session scoped to Weights & Biases and read its MCP URL and headers
3

Connect your agent to the MCP server

typescript
import { createMCPClient } from "@ai-sdk/mcp";
import { openai } from "@ai-sdk/openai";
import { generateText, stepCountIs } from "ai";

const client = await createMCPClient({
  transport: { type: "http", url: mcp.url, headers: mcp.headers },
});

const { text } = await generateText({
  model: openai("gpt-5.6-sol"),
  tools: await client.tools(),
  prompt: "List recent runs for my Weights & Biases project and summarize failed runs",
  stopWhen: stepCountIs(10),
});

console.log(text);
await client.close();
Pass the session's MCP URL and headers to your agent and run a Weights & Biases request
SETUP GUIDE

Connect Weights & Biases API Tool with your Agent

1

Install Composio

typescript
npm install @composio/core @composio/openai openai
Install the Composio SDK, the OpenAI provider, and the OpenAI SDK
2

Create a Composio session

typescript
import OpenAI from "openai";
import { Composio } from "@composio/core";
import { OpenAIResponsesProvider } from "@composio/openai";

const composio = new Composio({ provider: new OpenAIResponsesProvider() });
const client = new OpenAI();

const session = await composio.create("your-user-id", { toolkits: ["weights_and_biases"] });
const tools = await session.tools();
Initialize Composio with the OpenAI Responses provider and create a session scoped to Weights & Biases
3

Run Weights & Biases tools with your agent

typescript
let response = await client.responses.create({
  model: "gpt-5.6-sol",
  tools,
  input: [{ role: "user", content: "List recent runs for my Weights & Biases project and summarize failed runs" }],
});

while (response.output.some((o) => o.type === "function_call")) {
  const outputs = await composio.provider.handleToolCalls(session, response.output);
  response = await client.responses.create({
    model: "gpt-5.6-sol",
    tools,
    previous_response_id: response.id,
    input: outputs,
  });
}

console.log(response.output_text);
Send a request, execute the Weights & Biases tool calls through the session, and print the final answer

Why Use Composio?

AI Native Weights & Biases Integration

  • Supports both Weights & Biases MCP and direct API based integrations
  • Structured, LLM-friendly schemas for reliable W&B Cloud GraphQL queries
  • Rich coverage for reading projects, runs, sweeps, artifact types, reports, teams, and organizations

Managed Auth

  • Secure API key handling so your agent never needs hard-coded Weights & Biases credentials
  • Central place to manage, scope, and revoke W&B access for each user
  • Per user and per environment credentials for safer experiment and model tracking workflows

Agent Optimized Design

  • Tools are tuned for language models, so agents can query W&B runs and projects with fewer mistakes
  • Clear tool schemas help agents fetch the right experiment data instead of guessing GraphQL shapes
  • Comprehensive execution logs so you always know what ran, when, and on whose behalf

Enterprise Grade Security

  • Fine-grained RBAC so you control which agents and users can access Weights & Biases
  • Scoped, least privilege access to W&B resources like teams, orgs, projects, and reports
  • Full audit trail of agent actions to support model governance, review, and compliance
FAQ

Frequently asked questions

Yes, Weights & Biases requires you to configure your own API key credentials. Once set up, Composio handles secure credential storage and API request handling for you.

Yes! Composio's Tool Router enables agents to use multiple toolkits. Learn more.

Yes. Composio is SOC 2 Type II compliant and is built to keep your Weights & Biases connection and credentials secure. OAuth tokens and API keys are encrypted, and sensitive customer data is protected at rest and in transit.

Composio also undergoes independent security testing and continuously monitors its systems for security threats. You can review the latest reports and policies in the Composio Trust Center.

Composio maintains and updates all toolkit integrations automatically, so your agents always work with the latest API versions.

Create the key in your Weights & Biases account settings, then paste it once on Composio's connection page. Composio stores it encrypted and uses it only for the Weights & Biases actions your agent runs. Nobody else in your workspace can read it, and you can revoke it inside Weights & Biases at any time.

When you connect a Weights & Biases account through Composio, every action runs under that account, so anything the agent creates, sends, or changes shows up in Weights & Biases as done by you. Keep an approval step in your prompt for actions with side effects, such as sending or deleting, and have the agent draft first.

Whatever the credentials you connect allow inside Weights & Biases. If Weights & Biases lets you scope a key to specific permissions, create a scoped one so the agent can only do what you intend. You can revoke the key inside Weights & Biases at any time.

Yes. Composio supports multiple connected accounts for the same app, and that works in Claude, ChatGPT, or any other assistant you connect through Composio. Give each Weights & Biases connection a name such as work or personal, and the assistant uses the one you mention in the request. Each account keeps its own credentials and nothing is merged.

The connection stops working the moment Weights & Biases rejects the old key. Create a new key in Weights & Biases and reconnect the account from the Composio dashboard or by asking your agent to reconnect Weights & Biases. Nothing else changes.

Composio's free Hobby plan includes 100,000 tool calls per month with no credit card, which covers most personal Weights & Biases use. Paid plans add higher limits and team features. Your Weights & Biases plan and its API limits still apply as usual.

Built-in connectors usually give one AI access to a limited set of apps. Many people use Composio because it lets their AI connect to more apps than it normally supports, or connect to multiple accounts for the same app (e.g. connect Claude to multiple Weights & Biases accounts).

With Composio, you connect Weights & Biases once and then use it across different AI assistants without setting it up separately in each one. Connect your apps to Composio once, then connect Composio to whichever AI you use, whether that's Claude, ChatGPT, Hermes, or your own custom assistant.

Start with Weights & Biases.It takes 30 seconds.

Managed auth, hosted MCP servers, and every Weights & Biases tool your agent needs.Free to start.

Start building