How to Use the AI21 Labs MCP in Google ADK

Q: How do I connect AI21 Labs to Google ADK?

Install the framework with pip install google-adk. Create an McpToolset using your Vinkius HTTP endpoint. Hand that toolset directly to your agent.

Q: Can I filter the AI21 Labs MCP tools exposed to Google ADK?

Yes, the toolset accepts an optional toolnames list. Pass an array of specific endpoints if you only want the agent to see the summarization tools.

Connect the AI21 Labs MCP Server to your Google ADK agents. Process text, correct grammar, and summarize long documents alongside Gemini.

See Vinkius in Action

Works with every AI agent you already use

…and any MCP-compatible client

MCP Servers - Free for Subscribers

Connect AI21 Labs MCP to Google ADK

Create your Vinkius account to connect AI21 Labs 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.

GDPR Free for Subscribers

Setup AI21 Labs with Google ADK

Ask AI about this MCP

ChatGPT

Claude

Perplexity

Expand your Google ADK toolbelt

Gemini is great, but sometimes you want a different model architecture for specific tasks. This integration lets you mix Jamba models into your Google Cloud workflows. Just initialize `McpToolset` with your Vinkius HTTP URL. Pass it into your `LlmAgent` constructor. Now your agent can call `ai21_chat_completion` whenever it decides the task requires it.

Structure messy BigQuery data

Enterprise teams dump massive amounts of unstructured text into BigQuery. Reading it manually is impossible. Have your Gemini agent pull the rows, then loop through them using `ai21_segment_text` and `ai21_summarize_text`. The agent breaks down the raw customer feedback or legal text into distinct topics, making it ready for downstream analysis.

Automate corporate editing pipelines

Marketing and legal teams spend hours rewriting drafts. You can build an agent that does the heavy lifting first. Give the agent access to `ai21_text_improvements` and `ai21_grammar_correction`. It will scan your drafts, fix the typos, and suggest stylistic changes before a human ever reviews the document.

Setup guide

Set up AI21 Labs MCP in Google ADK

Prerequisites

Python 3.10+ installed
google-adk package (pip install google-adk)
Active Vinkius subscription with a valid endpoint token

1

Install Google ADK

Run pip install google-adk to install the Agent Development Kit. MCP support is included via the McpToolset class.
2

Connect via SSE transport

Use McpToolset.from_server() with SseServerParams pointing to your Vinkius endpoint. Replace [YOUR_TOKEN_HERE] with your token from cloud.vinkius.com.
3

Create an LlmAgent

Pass the returned mcp_tools list directly to LlmAgent(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 AI21 Labs tools in your ADK agent.

agent.py

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="AI21 Labs_agent",
    model="gemini-2.0-flash",
    instruction="You have access to AI21 Labs tools via MCP.",
    tools=mcp_tools,
)

Get your connection token →

Independent Platform Disclaimer: Vinkius is an independent platform and is not affiliated with, endorsed by, sponsored by, verified by, or otherwise authorized by AI21 Labs. 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.

Connect AI21 Labs now

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 AI21 Labs MCP in Google ADK

Install the framework with `pip install google-adk`. Create an `McpToolset` using your Vinkius HTTP endpoint. Hand that toolset directly to your agent.

Yes, the toolset accepts an optional `tool_names` list. Pass an array of specific endpoints if you only want the agent to see the summarization tools.

They work together perfectly. Gemini can read a million tokens of context, decide what matters, and then pass specific chunks to the specialized rewriting endpoints.

Use the built-in file endpoints. Your agent can upload a file, check its status with `ai21_get_file`, and run queries against it.

Completely. Your unstructured data routes through an ephemeral V8 sandbox managed by Vinkius. The container processes the text and immediately self-destructs.

Use it with your favorite AI tools

Connect this server to Cursor, Claude, VS Code, and more.

OpenAI Agents SDK sdk-python

Google ADK sdk-python

Pydantic AI sdk-python

Vercel AI SDK sdk-typescript