How to Use the Dify.AI SDK MCP in Pydantic AI

Q: How do I register the Dify.AI SDK toolset in Pydantic AI?

Use the unified MCPToolset class initialized with your Vinkius server URL. This connects the MCP Server so you can pass this instance into the toolsets argument of your Agent constructor. Do not use the deprecated MCPServerHTTP class.

Q: How do I send structured completions instead of chat messages?

Use the sendcompletion tool to send a text completion request to a Dify completion app. This tool returns the full generated text in a structured format that your Pydantic models can validate instantly.

Q: How is my conversational state and historical message data secured?

All conversational data fetched via getconversationmessages is processed in memory within an isolated V8 sandbox container. No session state is persisted on Vinkius infrastructure. The server uses secure, ephemeral TLS tunnels to communicate with the Dify API.

Run type-safe Dify workflows with Pydantic AI to catch API schema mismatches at runtime before they corrupt your agent pipelines.

See Vinkius in Action

Works with every AI agent you already use

…and any MCP-compatible client

MCP Servers - Free for Subscribers

Connect Dify.AI SDK MCP to Pydantic AI

Create your Vinkius account to connect Dify.AI SDK to Pydantic AI 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 Dify.AI SDK with Pydantic AI

Ask AI about this MCP

ChatGPT

Claude

Perplexity

Catch Dify schema errors at runtime

This MCP Server exposes the `get_workflow_parameters` tool to fetch the exact input schema required by your Dify application. Pydantic AI maps these requirements directly to Python type hints, ensuring your agent never sends malformed payloads. If the API model changes, the runtime validation catches the mismatch immediately. This prevents silent failures that typically plague LLM integrations. Your agent parses the schema, validates the user's input against it, and only then triggers `run_workflow`. You get a guaranteed, type-safe execution path every time.

Validate chat outputs using Pydantic AI MCP Server

The `chat_message` tool sends your queries to the Dify chatbot engine. When the response returns, this MCP Server guarantees that the output conforms exactly to your defined Pydantic schemas. If a model hallucinates a field, the framework raises a validation error instead of passing bad data. You can capture the output and immediately run validation checks on the text. If the validation fails, your agent can catch the error and automatically retry the call. This ensures your downstream databases only ingest clean, structured data.

Manage type-safe conversation histories

The `get_conversations` tool lists active chat histories as structured Python objects. Your agent uses these records to track user sessions without manually parsing messy JSON arrays. You can easily filter, sort, and map these sessions to your local database schemas. When you need to fetch specific messages, `get_conversation_messages` returns a strongly typed list of past interactions. If a session is no longer needed, your agent cleans up using `delete_conversation`. Every step is protected by Pydantic's strict runtime type-checking.

Setup guide

Set up Dify.AI SDK MCP in Pydantic AI

Prerequisites

Python 3.10+ installed
pydantic-ai-slim[fastmcp] package
Active Vinkius subscription with a valid endpoint token

1

Install Pydantic AI with FastMCP
Run pip install "pydantic-ai-slim[fastmcp]". The FastMCP toolset replaces the deprecated MCPServerHTTP class with full protocol support.
2

Configure the FastMCPToolset
Pass a JSON-style config dict to FastMCPToolset with your Vinkius URL. Replace [YOUR_TOKEN_HERE] with your token from cloud.vinkius.com. Supports Streamable HTTP, SSE, and Stdio transports.
3

Create and run your agent
Pass the toolset to Agent(toolsets=[toolset]) and call agent.run(). Swap openai:gpt-4o for any supported model — Anthropic, Google, Mistral, or Groq.

agent.py

from pydantic_ai import Agent
from pydantic_ai.toolsets.fastmcp import FastMCPToolset

toolset = FastMCPToolset({
    "mcpServers": {
        "difyai-sdk-mcp": {
            "url": "https://edge.vinkius.com/[YOUR_TOKEN_HERE]/mcp"
        }
    }
})

agent = Agent(
    "openai:gpt-4o",
    toolsets=[toolset],
    system_prompt="You have access to Dify.AI SDK tools.",
)

result = await agent.run("List recent Dify.AI SDK transactions")
print(result.output)

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 Dify.AI. 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 Dify.AI SDK 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 Dify.AI SDK MCP in Pydantic AI

Use the unified `MCPToolset` class initialized with your Vinkius server URL. This connects the MCP Server so you can pass this instance into the `toolsets` argument of your `Agent` constructor. Do not use the deprecated `MCPServerHTTP` class.

This integration supports both Streamable HTTP and SSE transports. Ensure your external MCP Server is running so your Pydantic AI application can establish a persistent connection.

The MCP Server will immediately raise a validation error. This prevents your agent from processing corrupt payloads or hallucinating fields that do not exist in the Dify application schema.

Use the `send_completion` tool to send a text completion request to a Dify completion app. This tool returns the full generated text in a structured format that your Pydantic models can validate instantly.

All conversational data fetched via `get_conversation_messages` is processed in memory within an isolated V8 sandbox container. No session state is persisted on Vinkius infrastructure. The server uses secure, ephemeral TLS tunnels to communicate with the Dify API.

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