Does this support AI21's Jamba models?

Yes. You can invoke the chat completion tool and instruct your agent to use specific model parameters like 'jamba-1.5-large' or 'jamba-1.5-mini' directly to leverage their native SSM-Transformer hybrid architecture.

Why use the specific summarization endpoint instead of a generic prompt?

AI21 has trained dedicated models purely for tasks like summarization and grammatical correction. Using these specialized task endpoints often yields much higher reliability, speed, and fidelity than trying to force a conversational chatbot to do the exact same task.

What languages are supported for grammar corrections?

Currently, AI21's dedicated Grammar and Paraphrase tools perform spectacularly on English, with varying outcomes for other localized languages. It is highly recommended to check their official documentation for the exact status of multi-language support boundaries.

AI21 Studio MCP Server for Google ADK 7 tools — connect in under 2 minutes

Name: AI21 Studio
Availability: InStock
Author: Vinkius

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

Built by Vinkius GDPR 7 Tools SDK

Google Agent Development Kit (ADK) is Google's framework for building production AI agents. Add AI21 Studio as an MCP tool provider through 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="ai21_studio_agent",
    instruction=(
        "You help users interact with AI21 Studio "
        "using 7 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 AI21 Studio MCP Server

Connect AI21 Studio to your AI agent and unlock access to the powerful Jamba models alongside a suite of highly-specialized linguistic tools. Give your agent the ability to process texts with enterprise-grade precision via natural conversation.

Google ADK natively supports AI21 Studio as an MCP tool provider. declare Vinkius Edge URL and the framework handles discovery, validation, and execution automatically. Combine 7 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

Chat & Instruct — Generate chat completions natively utilizing AI21's advanced Jamba models
Summarization — Produce accurate and concise summaries of long text payloads instantly
Paraphrasing — Rewrite text passages into different specific styles (general, casual, formal, long, short)
Grammar & Spelling — Catch and correct grammatical and spelling errors using AI21's dedicated correction engine
Text Segmentation — Split continuous text into distinct sentences based on complex grammatical boundaries
Embeddings — Convert pieces of text into high-dimensional numerical embeddings for RAG workflows

The AI21 Studio MCP Server exposes 7 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 AI21 Studio to Google ADK via MCP

Follow these steps to integrate the AI21 Studio 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 7 tools from AI21 Studio via MCP

Why Use Google ADK with the AI21 Studio MCP Server

Google ADK provides unique advantages when paired with AI21 Studio 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 AI21 Studio

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 AI21 Studio tools with BigQuery, Vertex AI, and Cloud Functions

AI21 Studio + Google ADK Use Cases

Practical scenarios where Google ADK combined with the AI21 Studio MCP Server delivers measurable value.

Enterprise data agents: ADK agents query AI21 Studio and cross-reference results with internal databases for comprehensive analysis

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

Automated compliance checks: schedule ADK agents to query AI21 Studio regularly and flag policy violations or configuration drift

Internal tool platforms: build self-service agent platforms where teams connect their own MCP servers including AI21 Studio

AI21 Studio MCP Tools for Google ADK (7)

These 7 tools become available when you connect AI21 Studio to Google ADK via MCP:

chat_completion

g., jamba-1.5-large, jamba-1.5-mini) and the conversation messages as a JSON array. Generate chat completions using AI21 Jamba models

embed_texts

Pass a JSON array of texts and specify whether they are for "query" or "document". Generate text embeddings using AI21 Studio embeddings

grammar_corrections

Perform Grammatical Error Correction (GEC) on text

paraphrase

Available styles: general, casual, formal, long, short. Paraphrase text using AI21 Studio

segmentation

Segment texts into sentences using AI21 Studio process

summarize

Text length must depend on API limits. Summarize long texts using AI21 Studio process

text_completion

Generate text completions using AI21 Studio models

Example Prompts for AI21 Studio in Google ADK

Ready-to-use prompts you can give your Google ADK agent to start working with AI21 Studio immediately.

"Paraphrase this paragraph into a highly formal tone."

"Run a grammatical error correction over my draft and output the fixed version."

"Segment this long block of text into individual sentences."

Troubleshooting AI21 Studio MCP Server with Google ADK

Common issues when connecting AI21 Studio to Google ADK through the Vinkius, and how to resolve them.

McpToolset not found

Update: pip install --upgrade google-adk

AI21 Studio + Google ADK FAQ

Common questions about integrating AI21 Studio 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.