# How to integrate Cody MCP with Pydantic AI ```json { "title": "How to integrate Cody MCP with Pydantic AI", "toolkit": "Cody", "toolkit_slug": "cody", "framework": "Pydantic AI", "framework_slug": "pydantic-ai", "url": "https://composio.dev/toolkits/cody/framework/pydantic-ai", "markdown_url": "https://composio.dev/toolkits/cody/framework/pydantic-ai.md", "updated_at": "2026-03-29T06:28:13.118Z" } ``` ## Introduction This guide walks you through connecting Cody to Pydantic AI using the Composio tool router. By the end, you'll have a working Cody agent that can summarize key findings from q2 reports, find policy details on employee benefits, draft onboarding checklist for new hires through natural language commands. This guide will help you understand how to give your Pydantic AI agent real control over a Cody account through Composio's Cody MCP server. Before we dive in, let's take a quick look at the key ideas and tools involved. ## Also integrate Cody with - [OpenAI Agents SDK](https://composio.dev/toolkits/cody/framework/open-ai-agents-sdk) - [Claude Agent SDK](https://composio.dev/toolkits/cody/framework/claude-agents-sdk) - [Claude Code](https://composio.dev/toolkits/cody/framework/claude-code) - [Claude Cowork](https://composio.dev/toolkits/cody/framework/claude-cowork) - [Codex](https://composio.dev/toolkits/cody/framework/codex) - [OpenClaw](https://composio.dev/toolkits/cody/framework/openclaw) - [Hermes](https://composio.dev/toolkits/cody/framework/hermes-agent) - [CLI](https://composio.dev/toolkits/cody/framework/cli) - [Google ADK](https://composio.dev/toolkits/cody/framework/google-adk) - [LangChain](https://composio.dev/toolkits/cody/framework/langchain) - [Vercel AI SDK](https://composio.dev/toolkits/cody/framework/ai-sdk) - [Mastra AI](https://composio.dev/toolkits/cody/framework/mastra-ai) - [LlamaIndex](https://composio.dev/toolkits/cody/framework/llama-index) - [CrewAI](https://composio.dev/toolkits/cody/framework/crew-ai) ## TL;DR Here's what you'll learn: - How to set up your Composio API key and User ID - How to create a Composio Tool Router session for Cody - How to attach an MCP Server to a Pydantic AI agent - How to stream responses and maintain chat history - How to build a simple REPL-style chat interface to test your Cody workflows ## What is Pydantic AI? Pydantic AI is a Python framework for building AI agents with strong typing and validation. It leverages Pydantic's data validation capabilities to create robust, type-safe AI applications. Key features include: - Type Safety: Built on Pydantic for automatic data validation - MCP Support: Native support for Model Context Protocol servers - Streaming: Built-in support for streaming responses - Async First: Designed for async/await patterns ## What is the Cody MCP server, and what's possible with it? The Cody MCP server is an implementation of the Model Context Protocol that connects your AI agent and assistants like Claude, Cursor, etc directly to your Cody account. It provides structured and secure access so your agent can perform Cody operations on your behalf. ## Supported Tools | Tool slug | Name | Description | |---|---|---| | `CODY_CREATE_CONVERSATION` | Create Conversation | Tool to create a new conversation with a specified bot. Use when starting a new conversation thread with optional focus mode to limit bot's knowledge base to specific documents. | | `CODY_CREATE_DOCUMENT` | Create Document | Tool to create a new document with text or HTML content in Cody AI. Use when you need to add documents to Cody's knowledge base with up to 768 KB of content. | | `CODY_CREATE_DOCUMENT_FROM_FILE` | Create Document From File | Tool to create a document by uploading a file (up to 100 MB). Supports txt, md, rtf, pdf, ppt, pptx, pptm, doc, docx, docm formats. Use when you need to add file-based documents to Cody's knowledge base. The file is processed asynchronously. | | `CODY_CREATE_DOCUMENT_FROM_WEBPAGE` | Create Document from Webpage | Tool to create a document from a publicly accessible webpage URL. Use when you need to import content from a webpage into Cody AI. The webpage must be accessible without login. If request fails, ensure the URL is publicly accessible and not blocked by a firewall. | | `CODY_CREATE_FOLDER` | Create Folder | Tool to create a new folder in Cody AI for organizing content. Use when you need to create a folder to organize documents or conversations. | | `CODY_DELETE_CONVERSATION` | Delete Conversation | Tool to delete a conversation by its ID. Use when you need to permanently remove a conversation from the system. | | `CODY_DELETE_DOCUMENT` | Delete Document | Tool to delete a document by id. Use when removing a document that is no longer needed. | | `CODY_GET_CONVERSATION` | Get Conversation | Tool to fetch a conversation by its ID from Cody AI. Use when you need to retrieve details about a specific conversation. Supports optional includes parameter to filter response to list document IDs. | | `CODY_GET_DOCUMENT` | Get Document | Tool to retrieve a specific document by its identifier from Cody AI. Use when you need to get details about a particular document including its status, content URL, and metadata. | | `CODY_GET_FOLDER` | Get Folder | Tool to retrieve a specific folder by its identifier. Use when you need to get details about a folder. | | `CODY_GET_MESSAGE` | Get Message | Tool to fetch a specific message by its ID from Cody AI. Use when you need to retrieve details about a particular message, with optional includes for sources or usage metrics. | | `CODY_GET_UPLOAD_SIGNED_URL` | Get Upload Signed URL | Tool to get an AWS S3 signed upload URL for file uploads. Use when you need to obtain a signed URL to upload a file to Cody's storage. | | `CODY_LIST_BOTS` | List Bots | Tool to get all bots with optional keyword filtering. Use when you need to retrieve the list of available bots in a Cody account. | | `CODY_LIST_CONVERSATIONS` | List Conversations | Tool to get all conversations with optional filtering by bot, keyword, or includes. Use when you need to retrieve conversation history, filter by bot, search by name, or get document associations. | | `CODY_LIST_DOCUMENTS` | List Documents | Tool to retrieve all documents from Cody AI account with optional filtering. Use when you need to list documents by folder, conversation, or search by keyword. Returns document details including learning status and content URL. | | `CODY_LIST_FOLDERS` | List Folders | Tool to retrieve all folders with optional keyword filtering. Use when you need to list or search for folders in the account. | | `CODY_LIST_MESSAGES` | List Messages | Tool to retrieve a paginated list of messages from Cody, optionally filtered by conversation. Use when you need to list messages, with optional filtering by conversation_id and extra attributes (sources or usage). | | `CODY_SEND_MESSAGE` | Send Message | Tool to send a message to Cody AI and receive an AI-generated response. Use when you need to send a user message to a conversation and get the AI's reply. | | `CODY_SEND_MESSAGE_FOR_STREAM` | Send Message for Stream | Tool to send a message to Cody AI and receive a Server-Sent Events (SSE) stream URL for the AI response. Use when you need streaming responses instead of waiting for the complete message. The response contains a stream_url that can be used to connect to the SSE stream and receive the AI's response in real-time chunks. | | `CODY_UPDATE_CONVERSATION` | Update Conversation | Tool to update a conversation by its ID including name, bot_id, and document_ids. Use when you need to modify an existing conversation's properties. | | `CODY_UPDATE_FOLDER` | Update Folder | Tool to update a folder by its ID. Use when you need to modify an existing folder's name. | ## Supported Triggers None listed. ## Creating MCP Server - Stand-alone vs Composio SDK The Cody MCP server is an implementation of the Model Context Protocol that connects your AI agent to Cody. It provides structured and secure access so your agent can perform Cody operations on your behalf through a secure, permission-based interface. With Composio's managed implementation, you don't have to create your own developer app. For production, if you're building an end product, we recommend using your own credentials. The managed server helps you prototype fast and go from 0-1 faster. ## Step-by-step Guide ### 1. Prerequisites Before starting, make sure you have: - Python 3.9 or higher - A Composio account with an active API key - Basic familiarity with Python and async programming ### 1. Getting API Keys for OpenAI and Composio OpenAI API Key - Go to the [OpenAI dashboard](https://platform.openai.com/settings/organization/api-keys) and create an API key. You'll need credits to use the models, or you can connect to another model provider. - Keep the API key safe. Composio API Key - Log in to the [Composio dashboard](https://dashboard.composio.dev?utm_source=toolkits&utm_medium=framework_docs). - Navigate to your API settings and generate a new API key. - Store this key securely as you'll need it for authentication. ### 2. Install dependencies Install the required libraries. What's happening: - composio connects your agent to external SaaS tools like Cody - pydantic-ai lets you create structured AI agents with tool support - python-dotenv loads your environment variables securely from a .env file ```bash pip install composio pydantic-ai python-dotenv ``` ### 3. Set up environment variables Create a .env file in your project root. What's happening: - COMPOSIO_API_KEY authenticates your agent to Composio's API - USER_ID associates your session with your account for secure tool access - OPENAI_API_KEY to access OpenAI LLMs ```bash COMPOSIO_API_KEY=your_composio_api_key_here USER_ID=your_user_id_here OPENAI_API_KEY=your_openai_api_key ``` ### 4. Import dependencies What's happening: - We load environment variables and import required modules - Composio manages connections to Cody - MCPServerStreamableHTTP connects to the Cody MCP server endpoint - Agent from Pydantic AI lets you define and run the AI assistant ```python import asyncio import os from dotenv import load_dotenv from composio import Composio from pydantic_ai import Agent from pydantic_ai.mcp import MCPServerStreamableHTTP load_dotenv() ``` ### 5. Create a Tool Router Session What's happening: - We're creating a Tool Router session that gives your agent access to Cody tools - The create method takes the user ID and specifies which toolkits should be available - The returned session.mcp.url is the MCP server URL that your agent will use ```python async def main(): api_key = os.getenv("COMPOSIO_API_KEY") user_id = os.getenv("USER_ID") if not api_key or not user_id: raise RuntimeError("Set COMPOSIO_API_KEY and USER_ID in your environment") # Create a Composio Tool Router session for Cody composio = Composio(api_key=api_key) session = composio.create( user_id=user_id, toolkits=["cody"], ) url = session.mcp.url if not url: raise ValueError("Composio session did not return an MCP URL") ``` ### 6. Initialize the Pydantic AI Agent What's happening: - The MCP client connects to the Cody endpoint - The agent uses GPT-5 to interpret user commands and perform Cody operations - The instructions field defines the agent's role and behavior ```python # Attach the MCP server to a Pydantic AI Agent cody_mcp = MCPServerStreamableHTTP(url, headers={"x-api-key": COMPOSIO_API_KEY}) agent = Agent( "openai:gpt-5", toolsets=[cody_mcp], instructions=( "You are a Cody assistant. Use Cody tools to help users " "with their requests. Ask clarifying questions when needed." ), ) ``` ### 7. Build the chat interface What's happening: - The agent reads input from the terminal and streams its response - Cody API calls happen automatically under the hood - The model keeps conversation history to maintain context across turns ```python # Simple REPL with message history history = [] print("Chat started! Type 'exit' or 'quit' to end.\n") print("Try asking the agent to help you with Cody.\n") while True: user_input = input("You: ").strip() if user_input.lower() in {"exit", "quit", "bye"}: print("\nGoodbye!") break if not user_input: continue print("\nAgent is thinking...\n", flush=True) async with agent.run_stream(user_input, message_history=history) as stream_result: collected_text = "" async for chunk in stream_result.stream_output(): text_piece = None if isinstance(chunk, str): text_piece = chunk elif hasattr(chunk, "delta") and isinstance(chunk.delta, str): text_piece = chunk.delta elif hasattr(chunk, "text"): text_piece = chunk.text if text_piece: collected_text += text_piece result = stream_result print(f"Agent: {collected_text}\n") history = result.all_messages() ``` ### 8. Run the application What's happening: - The asyncio loop launches the agent and keeps it running until you exit ```python if __name__ == "__main__": asyncio.run(main()) ``` ## Complete Code ```python import asyncio import os from dotenv import load_dotenv from composio import Composio from pydantic_ai import Agent from pydantic_ai.mcp import MCPServerStreamableHTTP load_dotenv() async def main(): api_key = os.getenv("COMPOSIO_API_KEY") user_id = os.getenv("USER_ID") if not api_key or not user_id: raise RuntimeError("Set COMPOSIO_API_KEY and USER_ID in your environment") # Create a Composio Tool Router session for Cody composio = Composio(api_key=api_key) session = composio.create( user_id=user_id, toolkits=["cody"], ) url = session.mcp.url if not url: raise ValueError("Composio session did not return an MCP URL") # Attach the MCP server to a Pydantic AI Agent cody_mcp = MCPServerStreamableHTTP(url, headers={"x-api-key": COMPOSIO_API_KEY}) agent = Agent( "openai:gpt-5", toolsets=[cody_mcp], instructions=( "You are a Cody assistant. Use Cody tools to help users " "with their requests. Ask clarifying questions when needed." ), ) # Simple REPL with message history history = [] print("Chat started! Type 'exit' or 'quit' to end.\n") print("Try asking the agent to help you with Cody.\n") while True: user_input = input("You: ").strip() if user_input.lower() in {"exit", "quit", "bye"}: print("\nGoodbye!") break if not user_input: continue print("\nAgent is thinking...\n", flush=True) async with agent.run_stream(user_input, message_history=history) as stream_result: collected_text = "" async for chunk in stream_result.stream_output(): text_piece = None if isinstance(chunk, str): text_piece = chunk elif hasattr(chunk, "delta") and isinstance(chunk.delta, str): text_piece = chunk.delta elif hasattr(chunk, "text"): text_piece = chunk.text if text_piece: collected_text += text_piece result = stream_result print(f"Agent: {collected_text}\n") history = result.all_messages() if __name__ == "__main__": asyncio.run(main()) ``` ## Conclusion You've built a Pydantic AI agent that can interact with Cody through Composio's Tool Router. With this setup, your agent can perform real Cody actions through natural language. You can extend this further by: - Adding other toolkits like Gmail, HubSpot, or Salesforce - Building a web-based chat interface around this agent - Using multiple MCP endpoints to enable cross-app workflows (for example, Gmail + Cody for workflow automation) This architecture makes your AI agent "agent-native", able to securely use APIs in a unified, composable way without custom integrations. ## How to build Cody MCP Agent with another framework - [OpenAI Agents SDK](https://composio.dev/toolkits/cody/framework/open-ai-agents-sdk) - [Claude Agent SDK](https://composio.dev/toolkits/cody/framework/claude-agents-sdk) - [Claude Code](https://composio.dev/toolkits/cody/framework/claude-code) - [Claude Cowork](https://composio.dev/toolkits/cody/framework/claude-cowork) - [Codex](https://composio.dev/toolkits/cody/framework/codex) - [OpenClaw](https://composio.dev/toolkits/cody/framework/openclaw) - [Hermes](https://composio.dev/toolkits/cody/framework/hermes-agent) - [CLI](https://composio.dev/toolkits/cody/framework/cli) - [Google ADK](https://composio.dev/toolkits/cody/framework/google-adk) - [LangChain](https://composio.dev/toolkits/cody/framework/langchain) - [Vercel AI SDK](https://composio.dev/toolkits/cody/framework/ai-sdk) - [Mastra AI](https://composio.dev/toolkits/cody/framework/mastra-ai) - [LlamaIndex](https://composio.dev/toolkits/cody/framework/llama-index) - [CrewAI](https://composio.dev/toolkits/cody/framework/crew-ai) ## Related Toolkits - [Google Sheets](https://composio.dev/toolkits/googlesheets) - Google Sheets is a cloud-based spreadsheet tool for real-time collaboration and data analysis. It lets teams work together from anywhere, updating information instantly. - [Composio](https://composio.dev/toolkits/composio) - Composio is an integration platform that connects AI agents with hundreds of business tools. It streamlines authentication and lets you trigger actions across services—no custom code needed. - [Notion](https://composio.dev/toolkits/notion) - Notion is a collaborative workspace for notes, docs, wikis, and tasks. It streamlines team knowledge, project tracking, and workflow customization in one place. - [Airtable](https://composio.dev/toolkits/airtable) - Airtable combines the flexibility of spreadsheets with the power of a database for easy project and data management. Teams use Airtable to organize, track, and collaborate with custom views and automations. - [Asana](https://composio.dev/toolkits/asana) - Asana is a collaborative work management platform for teams to organize and track projects. It streamlines teamwork, boosts productivity, and keeps everyone aligned on goals. - [Google Tasks](https://composio.dev/toolkits/googletasks) - Google Tasks is a to-do list and task management tool integrated into Gmail and Google Calendar. It helps you organize, track, and complete tasks across your Google ecosystem. - [Linear](https://composio.dev/toolkits/linear) - Linear is a modern issue tracking and project planning tool for fast-moving teams. It helps streamline workflows, organize projects, and boost productivity. - [Composio search](https://composio.dev/toolkits/composio_search) - Composio search is a unified web search toolkit spanning travel, e-commerce, news, financial markets, images, and more. It lets you and your apps tap into up-to-date web data from a single, easy-to-integrate service. - [Jira](https://composio.dev/toolkits/jira) - Jira is Atlassian’s platform for bug tracking, issue tracking, and agile project management. It helps teams organize work, prioritize tasks, and deliver projects efficiently. - [Clickup](https://composio.dev/toolkits/clickup) - ClickUp is an all-in-one productivity platform for managing tasks, docs, goals, and team collaboration. It streamlines project workflows so teams can work smarter and stay organized in one place. - [Monday](https://composio.dev/toolkits/monday) - Monday.com is a customizable work management platform for project planning and collaboration. It helps teams organize tasks, automate workflows, and track progress in real time. - [Perplexityai](https://composio.dev/toolkits/perplexityai) - Perplexityai delivers natural, conversational AI models for generating human-like text. Instantly get context-aware, high-quality responses for chat, search, or complex workflows. - [Browser tool](https://composio.dev/toolkits/browser_tool) - Browser tool is a virtual browser integration that lets AI agents interact with the web programmatically. It enables automated browsing, scraping, and action-taking from any AI workflow. - [Addressfinder](https://composio.dev/toolkits/addressfinder) - Addressfinder is a data quality platform for verifying addresses, emails, and phone numbers. It helps you ensure accurate customer and contact data every time. - [Agiled](https://composio.dev/toolkits/agiled) - Agiled is an all-in-one business management platform for CRM, projects, and finance. It helps you streamline workflows, consolidate client data, and manage business processes in one place. - [Ai ml api](https://composio.dev/toolkits/ai_ml_api) - Ai ml api is a suite of AI/ML models for natural language and image tasks. It provides fast, scalable access to advanced AI capabilities for your apps and workflows. - [Aivoov](https://composio.dev/toolkits/aivoov) - Aivoov is an AI-powered text-to-speech platform offering 1,000+ voices in over 150 languages. Instantly turn written content into natural, human-like audio for any application. - [All images ai](https://composio.dev/toolkits/all_images_ai) - All-Images.ai is an AI-powered image generation and management platform. It helps you create, search, and organize images effortlessly with advanced AI capabilities. - [Anthropic administrator](https://composio.dev/toolkits/anthropic_administrator) - Anthropic administrator is an API for managing Anthropic organizational resources like members, workspaces, and API keys. It helps you automate admin tasks and streamline resource management across your Anthropic organization. - [Api labz](https://composio.dev/toolkits/api_labz) - Api labz is a platform offering a suite of AI-driven APIs and workflow tools. It helps developers automate tasks and build smarter, more efficient applications. ## Frequently Asked Questions ### What are the differences in Tool Router MCP and Cody MCP? With a standalone Cody MCP server, the agents and LLMs can only access a fixed set of Cody tools tied to that server. However, with the Composio Tool Router, agents can dynamically load tools from Cody and many other apps based on the task at hand, all through a single MCP endpoint. ### Can I use Tool Router MCP with Pydantic AI? Yes, you can. Pydantic AI fully supports MCP integration. You get structured tool calling, message history handling, and model orchestration while Tool Router takes care of discovering and serving the right Cody tools. ### Can I manage the permissions and scopes for Cody while using Tool Router? Yes, absolutely. You can configure which Cody scopes and actions are allowed when connecting your account to Composio. You can also bring your own OAuth credentials or API configuration so you keep full control over what the agent can do. ### How safe is my data with Composio Tool Router? All sensitive data such as tokens, keys, and configuration is fully encrypted at rest and in transit. Composio is SOC 2 Type 2 compliant and follows strict security practices so your Cody data and credentials are handled as safely as possible. --- [See all toolkits](https://composio.dev/toolkits) · [Composio docs](https://docs.composio.dev/llms.txt)