How to Use the DataRobot MCP in Google ADK
Connect DataRobot to Google ADK to keep your Gemini-powered agents updated on ML project status and deployment logs.
Works with every AI agent you already use
…and any MCP-compatible client
Connect DataRobot MCP to Google ADK
Create your Vinkius account to connect DataRobot to Google ADK and route execution through our secure gateway. The platform manages server hosting, runtime updates, and security layers. Configuration requires no manual server provisioning.
Track model deployments via Google ADK
Your agent runs `list_deployments` to see what is currently serving traffic. It maps these deployments to your existing Google Cloud infrastructure. Gemini models ingest this metadata to reason about production health. You get a clear picture of your ML ecosystem without manual reporting.
Search DataRobot datasets with Google ADK
The `list_datasets` tool gives your agent a complete inventory of your training data. It identifies which datasets require attention or updates. You avoid stale data issues by letting your agent verify registry contents. It pulls the latest metadata into your Google ADK workflow.
Access project metadata in Google ADK
Call `get_project` to retrieve specific configuration details for any ML task. Your agent uses this information to align its logic with active project requirements. This bridge between DataRobot and your agent ensures constant parity. You stop jumping between tabs to check project status.
Set up DataRobot MCP in Google ADK
Prerequisites
- Python 3.10+ installed
-
google-adkpackage (pip install google-adk) - Active Vinkius subscription with a valid endpoint token
- 1
Install Google ADK
Run
pip install google-adkto install the Agent Development Kit. MCP support is included via theMcpToolsetclass. - 2
Connect via SSE transport
Use
McpToolset.from_server()withSseServerParamspointing to your Vinkius endpoint. Replace[YOUR_TOKEN_HERE]with your token from cloud.vinkius.com. - 3
Create an LlmAgent
Pass the returned
mcp_toolslist directly toLlmAgent(tools=mcp_tools). The ADK maps each MCP tool to a native Gemini function call — no manual schema definitions required. - 4
Run with any Gemini model
The agent works with any Gemini model (
gemini-2.0-flash,gemini-2.5-pro, etc.). Copy the full example on the right to get started with DataRobot tools in your ADK agent.
from google.adk.agents import LlmAgent
from google.adk.tools.mcp_tool.mcp_toolset import McpToolset
from google.adk.tools.mcp_tool.mcp_session_manager import SseServerParams
# Connect to the MCP via SSE
mcp_tools, exit_stack = await McpToolset.from_server(
connection_params=SseServerParams(
url="https://edge.vinkius.com/[YOUR_TOKEN_HERE]/mcp"
)
)
# Create your agent with auto-discovered tools
agent = LlmAgent(
name="DataRobot_agent",
model="gemini-2.0-flash",
instruction="You have access to DataRobot tools via MCP.",
tools=mcp_tools,
) Independent Platform Disclaimer: Vinkius is an independent platform and is not affiliated with, endorsed by, sponsored by, verified by, or otherwise authorized by DataRobot. 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 DataRobot MCP in Google ADK
Use it with your favorite AI tools
Connect this server to Cursor, Claude, VS Code, and more.
Start using the DataRobot MCP today
We host it, we monitor it, we maintain it. You just paste one token.