How to Use the AI21 Studio MCP in Google ADK

Q: How does AI21 Studio integrate with Google ADK infrastructure?

You connect via the McpToolset using your Vinkius endpoint. It functions as a standard toolset that Google ADK agents consume alongside your existing BigQuery or Vertex AI operations.

Q: Can I use AI21 Studio embeddings in Google ADK?

Yes, embedtexts provides high-quality vectors for your documents and queries. You can then pipe these results directly into your BigQuery vector search indexes.

Q: How do I limit which AI21 Studio tools my Google ADK agent uses?

The google-adk framework supports a toolnames filter. You simply list the specific tools you want your agent to see, preventing unauthorized access to other functions.

Connect AI21 Studio to Google ADK to process BigQuery data with Jamba models and specialized language tools.

See Vinkius in Action

Works with every AI agent you already use

…and any MCP-compatible client

MCP Servers - Free for Subscribers

Connect AI21 Studio MCP to Google ADK

Create your Vinkius account to connect AI21 Studio 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 Studio with Google ADK

Ask AI about this MCP

ChatGPT

Claude

Perplexity

Run Jamba models inside Google ADK agents

Invoke the `chat_completion` tool to leverage Jamba-1.5 models within your Gemini-based agent workflows. It allows your Google Cloud agents to process complex reasoning tasks using AI21's architecture. Register the `McpToolset` in your `LlmAgent` definition to make these tools available. You can filter specific tools using `tool_names` to keep your agent focused on its primary objective.

Scale text processing in Google ADK

Use `summarize` to condense massive reports stored in BigQuery before your Gemini model analyzes them. It saves context tokens, which helps you stay under the limits for long-context reasoning tasks. `segmentation` ensures that your agent reads large documents one sentence at a time. This prevents your agent from losing track of details when dealing with lengthy enterprise documentation.

Use AI21 Studio for precise text manipulation

Apply `grammar_corrections` to verify the quality of automated reports generated by your agents. It fixes errors in real-time, ensuring that any data sent to clients is accurate. `paraphrase` lets your agent adjust its tone for different stakeholders. Whether you need a short summary or a long-form explanation, the tool modifies the text to fit your requirements.

Setup guide

Set up AI21 Studio 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 Studio 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 Studio_agent",
    model="gemini-2.0-flash",
    instruction="You have access to AI21 Studio 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 Studio. 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 Studio 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 Studio MCP in Google ADK

You connect via the `McpToolset` using your Vinkius endpoint. It functions as a standard toolset that Google ADK agents consume alongside your existing BigQuery or Vertex AI operations.

The server runs on Vinkius infrastructure, which is built for speed. Your agent calls the tools over HTTP, keeping the overhead minimal even when processing large text inputs.

Yes, `embed_texts` provides high-quality vectors for your documents and queries. You can then pipe these results directly into your BigQuery vector search indexes.

We enforce strict zero-trust protocols. Your text input for tasks like paraphrasing never hits long-term logs and is discarded immediately after the tool completes the operation.

The `google-adk` framework supports a tool_names filter. You simply list the specific tools you want your agent to see, preventing unauthorized access to other functions.

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