llama-index tools mcp_discovery integration
Project description
LlamaIndex Tool: MCP Discovery
This tool allows LlamaIndex agents to query a Model Context Protocol (MCP) Discovery server to find and retrieve other tools dynamically.
⚠️ Important Note
This MCP Discovery integration does not work out of the box. It requires a separately deployed MCP Discovery server, which you must self-host locally or deploy to your own cloud.
This tool acts only as a client and assumes an existing, reachable MCP Discovery server.
Required Environment Variables
SUPABASE_URL=your-supabase-url
SUPABASE_SERVICE_ROLE_KEY=your-service-role-key
OPENAI_API_KEY=your-openai-api-key
Deploying the MCP Discovery Server
git clone https://github.com/yksanjo/mcp-discovery.git
cd mcp-discovery
npm install
Features
- 🔍 Autonomous Tool Discovery: Query MCP servers to discover available tools based on natural language descriptions
- ⚡ Async Operations: Built with
aiohttpfor high-performance async operations - 🤖 Seamless Integration: Works directly with LlamaIndex agents via
BaseToolSpec - 🛡️ Error Handling: Graceful error handling with informative messages
Requirements
- Python >= 3.9
- llama-index-core >= 0.13.0
- aiohttp >= 3.8.0
Installation
pip install llama-index-tools-mcp-discovery
Usage
from llama_index.tools.mcp_discovery import MCPDiscoveryTool
from llama_index.core.agent import ReActAgent
# Initialize the tool with the MCP Discovery API
tool_spec = MCPDiscoveryTool(
api_url="https://mcp-discovery-two.vercel.app/api/v1/discover"
)
# Convert the spec to a list of FunctionTools
tools = tool_spec.to_tool_list()
# Create an agent with the discovery tool
agent = ReActAgent.from_tools(tools, verbose=True)
# The agent can now use the 'discover_tools' function to find MCP servers it needs
agent.chat("Find me a server that can send Slack notifications")
API Response Format
This tool uses the standard MCP Discovery response schema as defined in the MCP Discovery. The API should return responses following this format:
{
"recommendations": [
{
"server": "filesystem-server",
"name": "Filesystem Server",
"npm_package": "@modelcontextprotocol/server-filesystem",
"install_command": "npx -y @modelcontextprotocol/server-filesystem",
"confidence": 0.85,
"description": "Secure file operations for MCP...",
"category": "development",
"github_url": "https://github.com/modelcontextprotocol/servers"
}
],
"total_found": 10,
"query_time_ms": 52
}
Note: To optimize context window usage, the tool summarizes the raw JSON into a concise string containing only the name, server, and category. This allows the LLM to efficiently evaluate and select the best tool without being overwhelmed by installation metadata.
Examples
See the examples directory for more usage examples.
Development
Run tests:
make test
Run linters:
make lint
Format code:
make format
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file llama_index_tools_mcp_discovery-0.2.0.tar.gz.
File metadata
- Download URL: llama_index_tools_mcp_discovery-0.2.0.tar.gz
- Upload date:
- Size: 5.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.10.9 {"installer":{"name":"uv","version":"0.10.9","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
43c3e41b62e8c05e12f386d9109ec2e89155b48ccacb16b36f765b07407e1538
|
|
| MD5 |
c086cc4eba675b8b8ffd592149ad5f5f
|
|
| BLAKE2b-256 |
d265b1a2921776d3f0d886226f9020ceceeda57190aa4b995cdf230d9f954ef6
|
File details
Details for the file llama_index_tools_mcp_discovery-0.2.0-py3-none-any.whl.
File metadata
- Download URL: llama_index_tools_mcp_discovery-0.2.0-py3-none-any.whl
- Upload date:
- Size: 5.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.10.9 {"installer":{"name":"uv","version":"0.10.9","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7d4079fa720bce94d8ed0dfbd74fbb815704f510be033057407710bfd6bcb1fe
|
|
| MD5 |
10609a572f1eaa858def4fb33793b008
|
|
| BLAKE2b-256 |
0d35a7bf66aba8ed4464f551f6e11750d65d7fa261c3f62a8a7f2e6ad153f18f
|