Can I check the latest metrics for a specific ML run?

Yes. Using the `get_run_details` tool, your AI agent can pull the latest logged metrics (like accuracy or loss) and hyperparameters for any specific run ID within your projects.

Is it possible to list versioned datasets and models?

Absolutely. The `list_project_artifacts` tool allows you to see all artifacts, including datasets and models, helping you track data lineage and versioning directly through conversation.

Can I monitor hyperparameter search sweeps via chat?

Yes. Use the `list_project_sweeps` tool to monitor automated optimization tasks. Your agent will return a list of sweeps in the project so you can track progress without leaving your workspace.

Weights & Biases MCP Server for Google ADK 6 tools — connect in under 2 minutes

Name: Weights & Biases
Availability: InStock
Author: Vinkius

google.github.io/adk-docs/tools/mcp-tools

Built by Vinkius GDPR 6 Tools SDK

Google Agent Development Kit (ADK) is Google's framework for building production AI agents. Add Weights & Biases as an MCP tool provider through the Vinkius and your ADK agents can call every tool with full schema introspection.

Get MCP Server for AI Agents

ASK AI ABOUT THIS MCP SERVER

Open in ChatGPT Open in Claude Open in Perplexity

Vinkius supports streamable HTTP and SSE.

python

from google.adk.agents import Agent
from google.adk.tools.mcp_tool import McpToolset
from google.adk.tools.mcp_tool.mcp_session_manager import (
    StreamableHTTPConnectionParams,
)

# Your Vinkius token — get it at cloud.vinkius.com
mcp_tools = McpToolset(
    connection_params=StreamableHTTPConnectionParams(
        url="https://edge.vinkius.com/[YOUR_TOKEN_HERE]/mcp",
    )
)

agent = Agent(
    model="gemini-2.5-pro",
    name="weights_biases_agent",
    instruction=(
        "You help users interact with Weights & Biases "
        "using 6 available tools."
    ),
    tools=[mcp_tools],
)

Fully ManagedVinkius Servers

60%Token savings

High SecurityEnterprise-grade

IAMAccess control

EU AI ActCompliant

DLPData protection

V8 IsolateSandboxed

Ed25519Audit chain

<40msKill switch

Stream every event to Splunk, Datadog, or your own webhook in real-time

* Every MCP server runs on Vinkius-managed infrastructure inside AWS - a purpose-built runtime with per-request V8 isolates, Ed25519 signed audit chains, and sub-40ms cold starts optimized for native MCP execution. See our infrastructure

About Weights & Biases MCP Server

Connect your Weights & Biases (WandB) account to any AI agent and manage your machine learning experiments through natural conversation.

Google ADK natively supports Weights & Biases as an MCP tool provider — declare the Vinkius Edge URL and the framework handles discovery, validation, and execution automatically. Combine 6 tools with Gemini's long-context reasoning for complex multi-tool workflows, with production-ready session management and evaluation built in.

What you can do

Project Management — List all projects within your WandB entity (user or team) to browse your experiment folders
Run Monitoring — List and track individual experiment runs within a project to monitor real-time activity
Deep Run Analysis — Retrieve full details for any run, including latest accuracies, losses, and hyperparameters
Artifact Management — List versioned datasets, models, and other artifacts to track data lineage and dependencies
Sweep Tracking — Monitor automated hyperparameter search sweeps to see optimization progress
Reports & Collaboration — List saved analysis reports and dashboards to access collaborative documentation

The Weights & Biases MCP Server exposes 6 tools through the Vinkius. Connect it to Google ADK in under two minutes — no API keys to rotate, no infrastructure to provision, no vendor lock-in. Your configuration, your data, your control.

How to Connect Weights & Biases to Google ADK via MCP

Follow these steps to integrate the Weights & Biases MCP Server with Google ADK.

Install Google ADK

Run pip install google-adk

Replace the token

Replace [YOUR_TOKEN_HERE] with your Vinkius token

Create the agent

Save the code above and integrate into your ADK workflow

Explore tools

The agent will discover 6 tools from Weights & Biases via MCP

Why Use Google ADK with the Weights & Biases MCP Server

Google ADK provides unique advantages when paired with Weights & Biases through the Model Context Protocol.

Google ADK natively supports MCP tool servers — declare a tool provider and the framework handles discovery, validation, and execution

Built on Gemini models, ADK provides long-context reasoning ideal for complex multi-tool workflows with Weights & Biases

Production-ready features like session management, evaluation, and deployment come built-in — not bolted on

Seamless integration with Google Cloud services means you can combine Weights & Biases tools with BigQuery, Vertex AI, and Cloud Functions

Weights & Biases + Google ADK Use Cases

Practical scenarios where Google ADK combined with the Weights & Biases MCP Server delivers measurable value.

Enterprise data agents: ADK agents query Weights & Biases and cross-reference results with internal databases for comprehensive analysis

Multi-modal workflows: combine Weights & Biases tool responses with Gemini's vision and language capabilities in a single agent

Automated compliance checks: schedule ADK agents to query Weights & Biases regularly and flag policy violations or configuration drift

Internal tool platforms: build self-service agent platforms where teams connect their own MCP servers including Weights & Biases

Weights & Biases MCP Tools for Google ADK (6)

These 6 tools become available when you connect Weights & Biases to Google ADK via MCP:

get_run_details

Retrieves full details for a specific W&B run, including summary metrics and config

list_project_artifacts

Lists all artifacts (datasets, models, etc.) in a project

list_project_reports

Lists all saved analysis reports in a project

list_project_runs

Lists all experiment runs within a specific W&B project

list_project_sweeps

Lists hyperparameter search sweeps within a project

list_wandb_projects

Lists all projects within a Weights & Biases entity (user or team)

Example Prompts for Weights & Biases in Google ADK

Ready-to-use prompts you can give your Google ADK agent to start working with Weights & Biases immediately.

"List all runs in my 'transformer-nmt' project for entity 'ai-team'."

"Get the final accuracy and config for run ID 'vibrant-sweep-1'."

"What artifacts are available in the 'resnet-training' project?"

Troubleshooting Weights & Biases MCP Server with Google ADK

Common issues when connecting Weights & Biases to Google ADK through the Vinkius, and how to resolve them.

McpToolset not found

Update: pip install --upgrade google-adk

Weights & Biases + Google ADK FAQ

Common questions about integrating Weights & Biases MCP Server with Google ADK.

How does Google ADK connect to MCP servers?

Import the MCP toolset class and pass the server URL. ADK discovers and registers all tools automatically, making them available to your agent's tool-use loop.

Can ADK agents use multiple MCP servers?

Yes. Declare multiple MCP tool providers in your agent configuration. ADK merges all tool schemas and the agent can call tools from any server in a single turn.

Which Gemini models work best with MCP tools?

Gemini 2.0 Flash and Pro models both support function calling required for MCP tools. Flash is recommended for latency-sensitive use cases, Pro for complex reasoning.