How to Use the AI21 Studio MCP in LangChain

Q: How do I install the AI21 Studio MCP Server for LangChain?

Run pip install langchain-mcp-adapters langgraph in your environment. Then configure a MultiServerMCPClient pointing to the Vinkius endpoint.

Q: Can LangChain ReAct agents use AI21 Studio Jamba models?

Yes. Once you call client.gettools(), pass the resulting list to your agent. The agent will automatically discover the chatcompletion tool and use it for reasoning.

Give LangChain agents direct access to AI21 Studio Jamba models and text processing tools for complex reasoning chains.

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 LangChain

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

Ask AI about this MCP

ChatGPT

Claude

Perplexity

LangChain ReAct Agents with Jamba MCP Server

The `chat_completion` tool connects your LangChain ReAct agents directly to AI21 Jamba models. You pass the conversation context, and the agent decides how to respond. You build multi-step pipelines where the agent routes tasks to specific Jamba variants like jamba-1.5-large or jamba-1.5-mini based on the raw difficulty of the prompt. Building these chains requires exact control over token usage and latency. LangSmith tracing captures every interaction with this MCP Server, giving you actual visibility into tool inputs and outputs. You see exactly how the agent formulated its prompt before hitting the AI21 endpoint.

Pipeline Text Processing

The `summarize` tool feeds long documents directly into your composable chains before passing the condensed output to downstream tasks. Instead of writing custom API wrappers, your agent calls this operation natively to shrink context windows. You feed a massive transcript in, and the agent extracts the core points automatically. Text transformation happens within the same workflow using the `paraphrase` tool. Your agent takes the summarized output and applies specific styles like casual, formal, long, or short based on user preferences. The output of one link in the chain becomes the exact input for the next.

Automated Grammar and Structuring

The `grammar_corrections` tool lets LangChain agents perform grammatical error correction on raw text inputs before storing them. You build a chain that cleans user input first, ensuring the data hitting your database is actually readable. Breaking down large blocks of text requires the `segmentation` tool. Your agent splits massive paragraphs into individual sentences, allowing parallel processing across multiple downstream nodes. This modular approach keeps your text processing pipelines fast and predictable.

Setup guide

Set up AI21 Studio MCP in LangChain

Prerequisites

Python 3.10+ installed
langchain-mcp-adapters + langgraph packages
Active Vinkius subscription with a valid endpoint token

1

Install dependencies
Run pip install langchain-mcp-adapters langgraph langchain-openai. The MCP adapters package converts MCP tools into native LangChain BaseTool objects.
2

Connect via HTTP transport
Use MultiServerMCPClient with "transport": "http" pointing to your Vinkius endpoint. Replace [YOUR_TOKEN_HERE] with your token from cloud.vinkius.com.
3

Create a ReAct agent
Pass the discovered tools to create_react_agent() from LangGraph. The agent automatically routes AI21 Studio tool calls through the MCP protocol.
4

Run with any LLM
Swap ChatOpenAI for ChatAnthropic, ChatGoogleGenerativeAI, or any LangChain-compatible model. The MCP tools work identically across all providers.

agent.py

from langchain_mcp_adapters.client import MultiServerMCPClient
from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI

async with MultiServerMCPClient({
    "ai21-studio-mcp": {
        "transport": "http",
        "url": "https://edge.vinkius.com/[YOUR_TOKEN_HERE]/mcp",
    }
}) as client:
    tools = client.get_tools()

    agent = create_react_agent(
        ChatOpenAI(model="gpt-4o"),
        tools,
    )
    result = await agent.ainvoke({
        "messages": "List recent AI21 Studio transactions"
    })
    print(result["messages"][-1].content)

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 LangChain

Run `pip install langchain-mcp-adapters langgraph` in your environment. Then configure a `MultiServerMCPClient` pointing to the Vinkius endpoint.

Yes. Once you call `client.get_tools()`, pass the resulting list to your agent. The agent will automatically discover the `chat_completion` tool and use it for reasoning.

LangSmith traces the exact inputs and outputs of every tool invocation. You track latency and token usage for all AI21 operations right in your dashboard.

The connection is stateless by default. You maintain conversation history by using `client.session()` to keep persistent context across multiple turns.

Vinkius isolates your connection in an ephemeral V8 sandbox. The raw text documents and chat messages you send for processing are never stored or logged permanently.

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