DocumentDB - A2A | AG-UI | MCP

Version: 0.0.12

Overview

DocumentDB + MCP Server + A2A

A FastMCP server and A2A (Agent-to-Agent) agent for DocumentDB. DocumentDB is a MongoDB-compatible open source document database built on PostgreSQL.

This package provides:

1. MCP Server: Exposes DocumentDB functionality (CRUD, Administration) as tools for LLMs.
2. A2A Agent: A specialized agent that uses these tools to help users manage their database.

Features

• CRUD Operations: Insert, Find, Update, Replace, Delete, Count, Distinct, Aggregate.
• Collection Management: Create, Drop, List, Rename collections.
• User Management: Create, Update, Drop users.
• Direct Commands: Run raw database commands.

MCP

MCP Tools

Function Name	Description	Tag(s)
binary_version	Get the binary version of the server (using buildInfo).	system
list_databases	List all databases in the connected DocumentDB/MongoDB instance.	system
run_command	Run a raw command against the database.	system
list_collections	List all collections in a specific database.	collections
create_collection	Create a new collection in the specified database.	collections
drop_collection	Drop a collection from the specified database.	collections
create_database	Explicitly create a database by creating a collection in it (MongoDB creates DBs lazily).	collections
drop_database	Drop a database.	collections
rename_collection	Rename a collection.	collections
create_user	Create a new user on the specified database.	users
drop_user	Drop a user from the specified database.	users
update_user	Update a user's password or roles.	users
users_info	Get information about a user.	users
insert_one	Insert a single document into a collection.	crud
insert_many	Insert multiple documents into a collection.	crud
find_one	Find a single document matching the filter.	crud
find	Find documents matching the filter.	crud
replace_one	Replace a single document matching the filter.	crud
update_one	Update a single document matching the filter. 'update' must contain update operators like $set.	crud
update_many	Update multiple documents matching the filter.	crud
delete_one	Delete a single document matching the filter.	crud
delete_many	Delete multiple documents matching the filter.	crud
count_documents	Count documents matching the filter.	crud
distinct	Find distinct values for a key.	analysis
aggregate	Run an aggregation pipeline.	analysis
find_one_and_update	Finds a single document and updates it. return_document: 'before' or 'after'.	crud
find_one_and_replace	Finds a single document and replaces it. return_document: 'before' or 'after'.	crud
find_one_and_delete	Finds a single document and deletes it.	crud

A2A Agent

Architecture:

---
config:
  layout: dagre
---
flowchart TB
 subgraph subGraph0["Agent Capabilities"]
        C["Agent"]
        B["A2A Server - Uvicorn/FastAPI"]
        D["MCP Tools"]
        F["Agent Skills"]
  end
    C --> D & F
    A["User Query"] --> B
    B --> C
    D --> E["Platform API"]

     C:::agent
     B:::server
     A:::server
    classDef server fill:#f9f,stroke:#333
    classDef agent fill:#bbf,stroke:#333,stroke-width:2px
    style B stroke:#000000,fill:#FFD600
    style D stroke:#000000,fill:#BBDEFB
    style F fill:#BBDEFB
    style A fill:#C8E6C9
    style subGraph0 fill:#FFF9C4

Loading

Component Interaction Diagram

sequenceDiagram
    participant User
    participant Server as A2A Server
    participant Agent as Agent
    participant Skill as Agent Skills
    participant MCP as MCP Tools

    User->>Server: Send Query
    Server->>Agent: Invoke Agent
    Agent->>Skill: Analyze Skills Available
    Skill->>Agent: Provide Guidance on Next Steps
    Agent->>MCP: Invoke Tool
    MCP-->>Agent: Tool Response Returned
    Agent-->>Agent: Return Results Summarized
    Agent-->>Server: Final Response
    Server-->>User: Output

Loading

Usage

MCP CLI

Short Flag	Long Flag	Description
-h	--help	Display help information
-t	--transport	Transport method: 'stdio', 'http', or 'sse' [legacy] (default: stdio)
-s	--host	Host address for HTTP transport (default: 0.0.0.0)
-p	--port	Port number for HTTP transport (default: 8000)
	--auth-type	Authentication type: 'none', 'static', 'jwt', 'oauth-proxy', 'oidc-proxy', 'remote-oauth' (default: none)
	--token-jwks-uri	JWKS URI for JWT verification
	--token-issuer	Issuer for JWT verification
	--token-audience	Audience for JWT verification
	--token-algorithm	JWT signing algorithm (e.g., HS256, RS256). Required for HMAC or static keys. Auto-detected for JWKS.
	--token-secret	Shared secret for HMAC (HS*) verification. Used with --token-algorithm.
	--token-public-key	Path to PEM public key file or inline PEM string for static asymmetric verification.
	--required-scopes	Comma-separated required scopes (e.g., documentdb.read,documentdb.write). Enforced by JWTVerifier.
	--oauth-upstream-auth-endpoint	Upstream authorization endpoint for OAuth Proxy
	--oauth-upstream-token-endpoint	Upstream token endpoint for OAuth Proxy
	--oauth-upstream-client-id	Upstream client ID for OAuth Proxy
	--oauth-upstream-client-secret	Upstream client secret for OAuth Proxy
	--oauth-base-url	Base URL for OAuth Proxy
	--oidc-config-url	OIDC configuration URL
	--oidc-client-id	OIDC client ID
	--oidc-client-secret	OIDC client secret
	--oidc-base-url	Base URL for OIDC Proxy
	--remote-auth-servers	Comma-separated list of authorization servers for Remote OAuth
	--remote-base-url	Base URL for Remote OAuth
	--allowed-client-redirect-uris	Comma-separated list of allowed client redirect URIs
	--eunomia-type	Eunomia authorization type: 'none', 'embedded', 'remote' (default: none)
	--eunomia-policy-file	Policy file for embedded Eunomia (default: mcp_policies.json)
	--eunomia-remote-url	URL for remote Eunomia server
	--enable-delegation	Enable OIDC token delegation to (default: False)
	--audience	Audience for the delegated token
	--delegated-scopes	Scopes for the delegated token (space-separated)
	--openapi-file	Path to OpenAPI JSON spec to import tools/resources from
	--openapi-base-url	Base URL for the OpenAPI client (defaults to instance URL)

A2A CLI

Endpoints

• Web UI: http://localhost:8000/ (if enabled)
• A2A: http://localhost:8000/a2a (Discovery: /a2a/.well-known/agent.json)
• AG-UI: http://localhost:8000/ag-ui (POST)

Short Flag	Long Flag	Description
-h	--help	Display help information
	--host	Host to bind the server to (default: 0.0.0.0)
	--port	Port to bind the server to (default: 9000)
	--reload	Enable auto-reload
	--provider	LLM Provider: 'openai', 'anthropic', 'google', 'huggingface'
	--model-id	LLM Model ID (default: qwen/qwen3-4b-2507)
	--base-url	LLM Base URL (for OpenAI compatible providers)
	--api-key	LLM API Key
	--mcp-url	MCP Server URL (default: http://localhost:8000/mcp)
	--web	Enable Pydantic AI Web UI

Usage

1. DocumentDB MCP Server

The MCP server connects to your DocumentDB (or MongoDB) instance.

Environment Variables:

• MONGODB_URI: Connection string (e.g., mongodb://localhost:27017/).
• Alternatively: MONGODB_HOST (default: localhost) and MONGODB_PORT (default: 27017).

Running the Server:

# Stdio mode (default)
documentdb-mcp

# HTTP mode
documentdb-mcp --transport http --port 8000

2. DocumentDB A2A Agent

The A2A agent connects to the MCP server to perform tasks.

Environment Variables:

• OPENAI_API_KEY / ANTHROPIC_API_KEY: API key for your chosen LLM provider.
• OPENAI_BASE_URL: (Optional) Base URL for OpenAI-compatible providers (e.g. Ollama).

Running the Agent:

# Start Agent Server (Default: OpenAI/Ollama)
documentdb-agent

# Custom Configuration
documentdb-agent --provider anthropic --model-id claude-3-5-sonnet-20240620 --mcp-url http://localhost:8000/mcp

Installation

pip install documentdb-mcp

Development

# Install dependencies
pip install -e ".[dev]"

# Run tests or verification
python -m build

Repository Owners