Weights & Biases MCP Server
for AI Agents
Connect your AI agent to StackOne's Weights & Biases MCP server and give it 35 MCP tools out of the box. Auth, tool execution, and security all managed.
StackOne 
Coverage
35 Agent Actions
Create, read, update, and delete across Weights & Biases — and extend your agent's capabilities with custom actions.
Authentication
Agent Tool Authentication
Per-user OAuth in one call. Your Weights & Biases MCP server gets session-scoped tokens with zero credentials stored on your infra.
Agent Auth →
Security
Agent Protection
Every Weights & Biases tool response scanned for prompt injection in milliseconds — 88.7% accuracy, all running on CPU.
Prompt Injection Defense →
Performance
Max Agent Context. Min Cost.
Free up to 96% of your agent's context window to enhance reasoning and reduce cost, on every Weights & Biases call.
Tools Discovery →What is the Weights & Biases MCP Server?
A Weights & Biases MCP server lets AI agents read and write Weights & Biases data through the Model Context Protocol — Anthropic's open standard for connecting LLMs to external tools. StackOne's Weights & Biases MCP server ships with 35 pre-built actions, fully extensible via the Connector Builder — plus managed authentication, prompt injection defense, observability, and agent execution runtime. Connect it from MCP clients like Claude Desktop, Claude Code, Cursor, Goose, and VS Code, or from agent frameworks like OpenAI Agents SDK, LangChain, and Vercel AI SDK.
All Weights & Biases MCP Tools
Every action from Weights & Biases's API, ready for your agent. Create, read, update, and delete — scoped to exactly what you need.
Artifact Collections
- Get Artifact Collection
Get a W&B artifact collection by name.
- List Artifact Collections
List artifact collections under a project's artifact type.
Artifact Types
- Get Artifact Type
Get a W&B artifact type by name within a project.
- List Artifact Types
List all artifact types within a project.
Artifacts
- Get Artifact
Get a W&B artifact by full name within a project.
- Update Artifact
Update metadata on an existing W&B artifact version.
- Delete Artifact
Delete a W&B artifact version by name.
Jobs
- Get Job
Get a W&B Launch job by full artifact name.
- List Jobs
List W&B Launch jobs within a project.
Projects
- Create Project
Create a new W&B project within an entity.
- Get Project
Get a W&B project by name within an entity.
- List Projects
List W&B projects for an entity, with cursor pagination.
Registrys
- Create Registry
Create a new W&B Model Registry within an organization.
- Get Registry
Get a single W&B Model Registry by name within an organization.
Runs
- Create Run
Create a new W&B run (shell) via the upsertBucket mutation.
- Get Run
Get a single W&B run by entity, project, and run id.
- List Runs
List W&B runs in a project with MongoDB-style filters and cursor pagination.
- Update Run
Update metadata on an existing W&B run.
- Delete Run
Delete a W&B run by id, optionally cascading to its artifacts.
Sweeps
- Create Sweep
Create a new W&B hyperparameter sweep via the upsertSweep GraphQL mutation.
- List Sweeps
List W&B hyperparameter sweeps in a project with cursor pagination.
- Get Sweep
Get a W&B hyperparameter sweep by id within a project.
Teams
- Create Team
Create a new W&B team (entity).
- Get Team
Get a W&B team (entity) by name.
Users
- Create User
Provision a new W&B user account by email.
- Get User
Search for a W&B user by username or email (returns the first match).
- List Users
List W&B users, optionally filtered by username/email substring.
Other (8)
- Create Artifact (Empty)
Create a new empty W&B artifact (metadata-only, no files) via the GraphQL API.
- Create Custom Chart
Create a custom Vega chart preset for a W&B entity.
- List Artifact Versions
List all versions of an artifact collection within a project.
- List Registries
List W&B Model Registries within an organization (Enterprise / Teams plans).
- List Reports
List W&B reports within a project.
- Get Viewer
Get the authenticated user (entity owner of the API key).
- Check Artifact Collection Exists
Check whether a W&B artifact collection exists.
- Check Artifact Exists
Check whether a W&B artifact exists.
Set Up Your Weights & Biases MCP Server in Minutes
One endpoint. Any framework. Your agent is talking to Weights & Biases in under 10 lines of code.
Agent Frameworks
JSON{
"mcpServers": {
"stackone": {
"command": "npx",
"args": [
"-y",
"mcp-remote@latest",
"https://api.stackone.com/mcp?x-account-id=<account_id>",
"--header",
"Authorization: Basic <YOUR_BASE64_TOKEN>"
]
}
}
}Platform Resources
MCP Code Mode: Keeping Tool Responses Out of Agent Context
Anthropic's code_execution processes data already in context. Custom MCP code mode keeps raw tool responses in a sandbox. 14K tokens vs 500.
11 min
Comparing BM25, TF-IDF, and Hybrid Search for MCP Tool Discovery
Benchmarking BM25, TF-IDF, and hybrid search for MCP tool discovery across 916 tools. The 80/20 TF-IDF/BM25 hybrid hits 21% Top-1 accuracy in under 1ms.
10 min
Indirect Prompt Injection Defense for MCP Tools: A Technical Guide
MCP tools that read emails, CRM records, and tickets are indirect prompt injection vectors. Here's how we built a two-tier defense that scans tool results in ~11ms.
12 min
MCP vs A2A: Architecture, Security, and When to Use Each
MCP vs A2A: what each protocol standardizes, how they differ, their shared security risks including indirect prompt injection, and when to use one, both, or a hybrid architecture.
12 min
MCP vs API: What 200+ Connector Builds Taught Us
MCP wraps APIs, it doesn't replace them. After building 200+ connectors that serve both, here's when each approach wins.
14 min read
Weights & Biases MCP Server FAQ
Does StackOne have a Weights & Biases MCP server?
Weights & Biases MCP server vs direct API integration — what's the difference?
How does Weights & Biases authentication work for AI agents?
origin_owner_id.Are Weights & Biases MCP tools vulnerable to prompt injection?
What is the context bloat of a Weights & Biases agent and how do I avoid it?
Can I limit which actions my Weights & Biases agent can access?
Can I create custom agent actions for my Weights & Biases MCP server?
When should I NOT use a Weights & Biases MCP server?
What AI frameworks and AI clients does the StackOne Weights & Biases MCP server support?
Put your AI agents to work
All the tools you need to build and scale AI agent integrations, with best-in-class connectivity, execution, and security.