How to integrate Neon MCP with Pydantic AI

This guide walks you through connecting Neon to Pydantic AI using the Composio tool router. By the end, you'll have a working Neon agent that can create a new neon project with autoscaling, add a developer's email permission to a project, spin up a new branch database for testing through natural language commands. This guide will help you understand how to give your Pydantic AI agent real control over a Neon account through Composio's Neon MCP server. Before we dive in, let's take a quick look at the key ideas and tools involved.

Get started for freeGet a demo
Neon logoNeon
Api Key

Neon is a serverless Postgres platform for building scalable apps faster. Its serverless architecture helps you launch, scale, and manage Postgres with zero overhead.

107 Tools

Introduction

This guide walks you through connecting Neon to Pydantic AI using the Composio tool router. By the end, you'll have a working Neon agent that can create a new neon project with autoscaling, add a developer's email permission to a project, spin up a new branch database for testing through natural language commands.

This guide will help you understand how to give your Pydantic AI agent real control over a Neon account through Composio's Neon MCP server.

Before we dive in, let's take a quick look at the key ideas and tools involved.

Also integrate Neon with

OpenAI Agents SDKClaude Agent SDKClaude CodeClaude CoworkCodexOpenClawHermesCLIGoogle ADKLangChainVercel AI SDKMastra AILlamaIndexCrewAI

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 Neon
  • 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 Neon 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 Neon MCP server, and what's possible with it?

The Neon MCP server is an implementation of the Model Context Protocol that connects your AI agent and assistants like Claude, Cursor, etc directly to your Neon account. It provides structured and secure access to your serverless Postgres databases and project infrastructure, so your agent can perform actions like managing projects, configuring authentication, creating branches, and automating database operations on your behalf.

  • Project management and insights: Retrieve detailed information about specific Neon projects, including configurations, workflows, and current status for monitoring or troubleshooting.
  • Automated role and permission setup: Grant email-based permissions, add new roles to branches, and streamline access control across your projects with granular, automated adjustments.
  • Seamless API key and authentication management: Create new API keys for organizations or projects, manage authentication flows, and securely handle access credentials for integrations.
  • Database and branch orchestration: Spin up new Postgres databases within branches, create and configure compute endpoints, and automate the creation of development, testing, or isolated environments as needed.
  • Custom project provisioning: Launch new Neon projects with advanced quota, resource, and networking settings—tailored to your application's requirements and operational best practices.

What is the Composio tool router, and how does it fit here?

What is Composio SDK?

Composio's Composio SDK helps agents find the right tools for a task at runtime. You can plug in multiple toolkits (like Gmail, HubSpot, and GitHub), and the agent will identify the relevant app and action to complete multi-step workflows. This can reduce token usage and improve the reliability of tool calls. Read more here: Getting started with Composio SDK

The tool router generates a secure MCP URL that your agents can access to perform actions.

How the Composio SDK works

The Composio SDK follows a three-phase workflow:

  1. Discovery: Searches for tools matching your task and returns relevant toolkits with their details.
  2. Authentication: Checks for active connections. If missing, creates an auth config and returns a connection URL via Auth Link.
  3. Execution: Executes the action using the authenticated connection.

Step-by-step Guide

Step by step09 STEPS
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
2

Getting API Keys for OpenAI and Composio

OpenAI API Key
  • Go to the OpenAI dashboard 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.
  • Navigate to your API settings and generate a new API key.
  • Store this key securely as you'll need it for authentication.
3

Install dependencies

bash
pip install composio pydantic-ai python-dotenv

Install the required libraries.

What's happening:

  • composio connects your agent to external SaaS tools like Neon
  • pydantic-ai lets you create structured AI agents with tool support
  • python-dotenv loads your environment variables securely from a .env file
4

Set up environment variables

bash
COMPOSIO_API_KEY=your_composio_api_key_here
USER_ID=your_user_id_here
OPENAI_API_KEY=your_openai_api_key

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
5

Import dependencies

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()
What's happening:
  • We load environment variables and import required modules
  • Composio manages connections to Neon
  • MCPServerStreamableHTTP connects to the Neon MCP server endpoint
  • Agent from Pydantic AI lets you define and run the AI assistant
6

Create a Tool Router Session

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 Neon
    composio = Composio(api_key=api_key)
    session = composio.create(
        user_id=user_id,
        toolkits=["neon"],
    )
    url = session.mcp.url
    if not url:
        raise ValueError("Composio session did not return an MCP URL")
What's happening:
  • We're creating a Tool Router session that gives your agent access to Neon 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
7

Initialize the Pydantic AI Agent

python
# Attach the MCP server to a Pydantic AI Agent
neon_mcp = MCPServerStreamableHTTP(url, headers={"x-api-key": COMPOSIO_API_KEY})
agent = Agent(
    "openai:gpt-5",
    toolsets=[neon_mcp],
    instructions=(
        "You are a Neon assistant. Use Neon tools to help users "
        "with their requests. Ask clarifying questions when needed."
    ),
)
What's happening:
  • The MCP client connects to the Neon endpoint
  • The agent uses GPT-5 to interpret user commands and perform Neon operations
  • The instructions field defines the agent's role and behavior
8

Build the chat interface

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 Neon.\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()
What's happening:
  • The agent reads input from the terminal and streams its response
  • Neon API calls happen automatically under the hood
  • The model keeps conversation history to maintain context across turns
9

Run the application

python
if __name__ == "__main__":
    asyncio.run(main())
What's happening:
  • The asyncio loop launches the agent and keeps it running until you exit

Complete Code

Here's the complete code to get you started with Neon and Pydantic AI:

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 Neon
    composio = Composio(api_key=api_key)
    session = composio.create(
        user_id=user_id,
        toolkits=["neon"],
    )
    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
    neon_mcp = MCPServerStreamableHTTP(url, headers={"x-api-key": COMPOSIO_API_KEY})
    agent = Agent(
        "openai:gpt-5",
        toolsets=[neon_mcp],
        instructions=(
            "You are a Neon assistant. Use Neon 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 Neon.\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 Neon through Composio's Tool Router. With this setup, your agent can perform real Neon 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 + Neon for workflow automation)
This architecture makes your AI agent "agent-native", able to securely use APIs in a unified, composable way without custom integrations.
TOOLS

Supported Tools

Every Neon action and event your agent gets out of the box.

Accept project transfer request

Tool to accept a transfer request for a Neon project.

Access project details by id

Retrieves detailed information about a specific Neon serverless Postgres project.

Add new jwks to project endpoint

Adds a new JSON Web Key Set (JWKS) URL to a Neon project for JWT-based authentication.

Add project email permission

Adds permissions for a specified email address to a particular project within the Neon B2B SaaS integration platform.

Add role to branch

Creates a new PostgreSQL role within a specific branch of a Neon project.

Count project branches

Tool to get the total number of branches in a Neon project.

Create API Key for Organization

Creates a new API key for the specified organization in Neon.

Create Auth Provider SDK Keys

Tool to generate SDK or API keys for authentication providers.

Create auth user

Tool to create a new user in Neon Auth for a specific project branch.

Create branch database

Creates a new database within a specified project and branch in the Neon platform.

Enable Neon Auth for branch

Tool to enable Neon Auth integration for a branch.

Create Neon Data API

Creates a new instance of Neon Data API in the specified branch.

Create branch snapshot

Creates a snapshot from the specified branch at a specific point in time.

Create compute endpoint

Creates a new compute endpoint for a specified branch within a Neon project.

Create New API Key

Creates a new personal API key for the authenticated Neon user account.

Create new project branch

Creates a new branch in a Neon project.

Create anonymized branch

Creates a new branch with anonymized data using PostgreSQL Anonymizer for static masking.

Create project transfer request

Tool to create a transfer request for a Neon project.

Create project with quota and settings

Creates a new Neon project with specified configuration settings.

Set VPC endpoint restriction

Sets or updates a VPC endpoint restriction for a Neon project.

Assign or Update VPC Endpoint

Assigns an AWS VPC endpoint to a Neon organization for Private Networking, or updates the label of an existing VPC endpoint assignment.

Delete api key by id

Deletes a specific API key from the Neon platform.

Delete auth domains

Deletes domains from the redirect_uri whitelist for a specific branch's authentication configuration.

Delete OAuth provider

Deletes an OAuth provider configuration from a specific branch in a Neon project.

Delete auth user

Deletes an authentication user from a specified branch within a Neon project.

Delete branch data API

Deletes the Neon Data API for a specified branch and database.

Delete database from branch

Deletes a specific database from a designated branch within a project in the Neon platform.

Delete organization api key

Deletes a specific API key associated with an organization in the Neon platform.

Delete organization member

Removes a specific member from a Neon organization.

Delete project branch by id

Deletes a specific branch within a project in the Neon B2B SaaS integration platform.

Delete project branch role

Deletes a PostgreSQL role from a specific branch within a Neon project.

Delete project by id

Permanently deletes a Neon PostgreSQL project and all its associated resources.

Delete project endpoint

Deletes a specific endpoint within a Neon project.

Delete project jwks by id

Deletes a specific JSON Web Key Set (JWKS) associated with a given project in the Neon B2B SaaS integration platform.

Delete project permission

Deletes a specific permission associated with a project in the Neon B2B SaaS integration platform.

Delete project snapshot

Deletes a specific snapshot for a project in Neon.

Delete vpc endpoint by ids

Permanently deletes a VPC endpoint from a Neon organization in a specified AWS region.

Delete vpc endpoint by project id

Deletes a specific VPC endpoint within a designated project in the Neon platform.

Disable branches auth

Tool to disable Neon Auth for a specific branch in a Neon project.

Fetch database for branch

Retrieves a list of databases associated with a specific project and branch in the Neon platform.

Fetch organization members by id

Retrieves a list of all members associated with a specific organization in the Neon B2B SaaS integration platform.

Fetch vpcendpoint details by id

Retrieves the current state and configuration details of a specific VPC endpoint assigned to a Neon organization.

Finalize branch restore

Finalizes the restore operation for a branch created from a snapshot.

Get auth

Retrieves authentication information about the current API request credentials.

Get auth allow localhost

Retrieves the allow localhost configuration for a specific branch in a Neon project.

Get auth email provider

Retrieves the email provider configuration for Neon Auth on a specific branch.

Get available preload libraries

Retrieves the list of available shared preload libraries for a Neon project.

Get branches anonymized status

Retrieves the current status of an anonymized branch, including its state and progress information.

Get branches backup schedule

Retrieves the backup schedule configuration for a specified branch within a Neon project.

Get Neon Data API for branch

Retrieves the Neon Data API information for a specific branch and database.

Get branches for project

Retrieves a list of branches associated with a specific project in the Neon B2B SaaS integration platform.

Get branches masking rules

Retrieves the masking rules for the specified anonymized branch.

Get branch roles for project

Retrieves the roles associated with a specific branch within a project in the Neon B2B SaaS integration platform.

Get consumption history projects

Retrieves consumption history for projects within a Neon organization.

Get current user information

Retrieves the profile information for the currently authenticated user.

Get organization api keys

Retrieves a list of all API keys associated with a specific organization in the Neon B2B SaaS integration platform.

Get project branches

Retrieves detailed information about a specific branch within a Neon project.

Get project branch role

Retrieves detailed information about a specific role within a particular branch of a Neon project.

Get project branch schema comparison

Compares the database schema of a branch with another branch's schema in a Neon project.

Get project connection uri

Retrieves the connection URI for a specified project within the Neon B2B SaaS integration platform.

Get project endpoint information

Retrieves a list of all compute endpoints associated with a specific Neon project.

Get project operation by id

Retrieves detailed information about a specific operation within a Neon serverless PostgreSQL project.

Get schema for project branch

Retrieves the PostgreSQL database schema (DDL) for a specific database on a branch within a Neon project.

Get user organizations

Retrieves a list of organizations associated with the currently authenticated user in the Neon platform.

Get vpc region endpoints

Retrieves a list of VPC endpoints for a specified organization within a particular AWS region.

List api keys

Retrieves a list of API keys associated with the authenticated user's Neon account.

List auth domains

Lists the domains in the redirect_uri whitelist for the specified project and branch.

List Auth OAuth Providers

Retrieves the OAuth providers configured for Neon Auth on a specific branch.

List project snapshots

Tool to list all snapshots for a specified Neon project.

List shared projects

Retrieves a list of Neon Postgres projects shared with your account.

List VPC endpoints for organization

Tool to retrieve the list of VPC endpoints for a specified Neon organization across all regions.

Modify branch details in project

Updates the details of a specific branch within a project in the Neon platform.

Patch branch database information

Updates the properties of a specific database within a project branch in the Neon platform.

Reset role password for branch

Resets the password for a PostgreSQL role on a specific Neon branch and generates a new password.

Restart project endpoint

Restarts a Neon compute endpoint (Postgres instance) within a project.

Restore project branch

This endpoint restores a branch to a specific state or point in time.

Restore snapshot to new branch

Tool to restore a Neon snapshot to a new branch.

Retrieve account consumption history

Retrieves the consumption history for a specified account within the Neon platform.

Retrieve all regions

Retrieves a list of available geographic regions supported by the Neon B2B SaaS integration platform.

Retrieve branch database details

Retrieves detailed information about a specific database within a Neon project and branch.

Retrieve branch endpoints

Retrieves all compute endpoints associated with a specific branch in a Neon project.

Retrieve jwks for project

Retrieves the JSON Web Key Set (JWKS) for a specified project.

Retrieve organization by id

Retrieves detailed information about a specific organization within the Neon platform.

Retrieve organization invitations

Retrieves a list of all pending invitations for a specified organization.

Retrieve organization member info

Retrieves detailed information about a specific member within an organization in the Neon B2B SaaS integration platform.

Retrieve project endpoint details

Retrieves detailed information about a specific compute endpoint within a Neon serverless PostgreSQL project.

Retrieve project operations

Retrieves a list of operations associated with a specific project in the Neon B2B SaaS integration platform.

Retrieve project permissions

Retrieves the current permissions settings for a specific project within the Neon B2B SaaS integration platform.

Retrieve projects list

Retrieves a list of all Neon projects associated with the authenticated user's account.

Retrieve vpc endpoints for project

Retrieves a list of VPC (Virtual Private Cloud) endpoints associated with a specific project in the Neon B2B SaaS integration platform.

Reveal role password in branch

Reveals the password for a specific role within a branch of a Neon project.

Send auth test email

Tool to send a test email using specified SMTP configuration for Neon Auth.

Send organization invitations

Creates and sends invitations to join an organization in the Neon B2B SaaS integration platform.

Set branch as default

Sets a specified branch as the default branch for a given project in Neon.

Start branch anonymization

Starts the anonymization process for an anonymized branch that is in the initialized, error, or anonymized state.

Start endpoint for project

Starts a suspended or idle compute endpoint for a Neon PostgreSQL project.

Suspend project endpoint by id

Suspends a specific endpoint within a project in the Neon B2B SaaS integration platform.

Transfer projects between organizations

Transfers selected projects from one organization to another organization.

Transfer user projects to organization

Transfers multiple projects from the authenticated user's personal account to a specified organization within the Neon B2B SaaS integration platform.

Update auth allow localhost

Updates the allow localhost configuration for a specific branch in a Neon project.

Update branch auth email provider

Updates the email provider configuration for a specific branch's authentication system.

Update branches data API

Updates the Neon Data API configuration for the specified branch.

Update branches masking rules

Updates the masking rules for the specified anonymized branch.

Update organization member role

Updates the role of a specific member within an organization in the Neon B2B SaaS integration platform.

Update project compute endpoint settings

Updates the configuration of a specific compute endpoint within a Neon project.

Update project settings by id

Updates the configuration and settings of a specific Neon project.

Update project snapshot

Tool to update the name of a specific snapshot in a Neon project.

FRAMEWORKS

How to build Neon MCP Agent with another framework

OpenAI Agents SDK logo

OpenAI Agents SDK

Use Neon MCP with OpenAI Agents SDK

Claude Agent SDK logo

Claude Agent SDK

Use Neon MCP with Claude Agent SDK

Claude Code logo

Claude Code

Use Neon MCP with Claude Code

Claude Cowork logo

Claude Cowork

Use Neon MCP with Claude Cowork

Codex logo

Codex

Use Neon MCP with Codex

OpenClaw logo

OpenClaw

Use Neon MCP with OpenClaw

Hermes logo

Hermes

Use Neon MCP with Hermes

CLI logo

CLI

Use Neon MCP with CLI

Google ADK logo

Google ADK

Use Neon MCP with Google ADK

LangChain logo

LangChain

Use Neon MCP with LangChain

Vercel AI SDK logo

Vercel AI SDK

Use Neon MCP with Vercel AI SDK

Mastra AI logo

Mastra AI

Use Neon MCP with Mastra AI

LlamaIndex logo

LlamaIndex

Use Neon MCP with LlamaIndex

CrewAI logo

CrewAI

Use Neon MCP with CrewAI

FAQ

Frequently asked questions

With a standalone Neon MCP server, the agents and LLMs can only access a fixed set of Neon tools tied to that server. However, with the Composio Tool Router, agents can dynamically load tools from Neon and many other apps based on the task at hand, all through a single MCP endpoint.

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 Neon tools.

Yes, absolutely. You can configure which Neon 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.

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 Neon data and credentials are handled as safely as possible.

Start with Neon.It takes 30 seconds.

Managed auth, hosted MCP servers, and every Neon tool your agent needs.Free to start.

Start building
Customer stories

Teams ship real agents on Composio

GET STARTED FOR FREEBOOK A DEMO
agent.aiZoomLettaGleanHubSpotAWSWabi

Karan skipped his own birthday party to fix our critical issue. It was 10pm and he diverted his Waymo to help us instead. This really sets the bar — it shows the commitment you need when users rely on your software.

Harsha Gaddipati
Harsha Gaddipati
Co-founder, Slashy

A lot of students tell us that the moment their connected tools start talking to each other inside Opennote feels almost magical. The agent just knows them, and it's immensely helped keep new users on the platform.

Abhi Arya
Abhi Arya
Co-founder, Opennote

We chose Composio over Pipedream because it delivered depth where it mattered — niche tools and tricky edge cases other platforms simply ignored. That gave us confidence to scale without compromising.

Nirman Dave
Nirman Dave
CEO, Zams

As a solo builder, shipping fast is life or death. The only way I can outcompete incumbents is by outmanoeuvring them. Getting bogged down managing agent auth would have been a death sentence.

Ryan Yu
Ryan Yu
Founder, Extra Thursday

Before Composio, adding tool integrations was slow and resource-intensive. Each one could take weeks of engineering time, and maintaining them meant constantly keeping up with API changes.

Tomisin Jenrola
Tomisin Jenrola
Founder & CEO, SwarmZero

With hands-on help from their founder, we integrated Gmail and Google Drive in just 30 minutes. This level of personal support and commitment is exactly what startups should strive for.

Jerome Leclanche
Jerome Leclanche
Co-Founder, Ingram Technologies