Implements a server with MCP endpoints using FastAPI, providing a way to expose the vector database retrieval functionality via API.
Allows searching and extracting Move files from GitHub repositories based on search queries, with support for both GitHub API and web scraping fallback methods.
Enables integration with OpenAI models for the RAG (Retrieval-Augmented Generation) pipeline, allowing enhanced responses based on retrieved information from the vector database.
Provides specialized functionality for working with Sui Move files, including searching, indexing, and retrieving relevant documents related to Sui Move programming.
Click on "Install Server".
Wait a few minutes for the server to deploy. Once ready, it will show a "Started" state.
In the chat, type
@followed by the MCP server name and your instructions, e.g., "@MCP RAG Serverexplain how to create a coin module in Sui Move"
That's it! The server will respond to your query, and you can continue using it as needed.
Here is a step-by-step guide with screenshots.
MCP Server with FAISS for RAG
This project provides a proof-of-concept implementation of a Machine Conversation Protocol (MCP) server that allows an AI agent to query a vector database and retrieve relevant documents for Retrieval-Augmented Generation (RAG).
Features
FastAPI server with MCP endpoints
FAISS vector database integration
Document chunking and embedding
GitHub Move file extraction and processing
LLM integration for complete RAG workflow
Simple client example
Sample documents
Related MCP server: AivisSpeech MCP Server
Installation
Using pipx (Recommended)
pipx is a tool to help you install and run Python applications in isolated environments.
First, install pipx if you don't have it:
# On macOS
brew install pipx
pipx ensurepath
# On Ubuntu/Debian
sudo apt update
sudo apt install python3-pip python3-venv
python3 -m pip install --user pipx
python3 -m pipx ensurepath
# On Windows with pip
pip install pipx
pipx ensurepathInstall the MCP Server package directly from the project directory:
# Navigate to the directory containing the mcp_server folder
cd /path/to/mcp-server-project
# Install in editable mode
pipx install -e .(Optional) Configure environment variables:
Copy
.env.exampleto.envAdd your GitHub token for higher rate limits:
GITHUB_TOKEN=your_token_hereAdd your OpenAI or other LLM API key for RAG integration:
OPENAI_API_KEY=your_key_here
Manual Installation
If you prefer not to use pipx:
Clone the repository
Install dependencies:
cd mcp_server
pip install -r requirements.txtUsage with pipx
After installing with pipx, you'll have access to the following commands:
Downloading Move Files from GitHub
# Download Move files with default settings
mcp-download --query "use sui" --output-dir docs/move_files
# Download with more options
mcp-download --query "module sui::coin" --max-results 50 --new-index --verboseImproved GitHub Search and Indexing (Recommended)
# Search GitHub and index files with default settings
mcp-search-index --keywords "sui move"
# Search multiple keywords and customize options
mcp-search-index --keywords "sui move,move framework" --max-repos 30 --output-results --verbose
# Save search results and use a custom index location
mcp-search-index --keywords "sui coin,sui::transfer" --index-file custom/path/index.bin --output-resultsThe mcp-search-index command provides enhanced GitHub repository search capabilities:
Searches repositories first, then recursively extracts Move files
Supports multiple search keywords (comma-separated)
Intelligently filters for Move files containing "use sui" references
Always rebuilds the vector database after downloading
Indexing Move Files
# Index files in the default location
mcp-index
# Index with custom options
mcp-index --docs-dir path/to/files --index-file path/to/index.bin --verboseQuerying the Vector Database
# Basic query
mcp-query "What is a module in Sui Move?"
# Advanced query with options
mcp-query "How do I define a struct in Sui Move?" -k 3 -fUsing RAG with LLM Integration
# Basic RAG query (will use simulated LLM if no API key is provided)
mcp-rag "What is a module in Sui Move?"
# Using with a specific LLM API
mcp-rag "How do I define a struct in Sui Move?" --api-key your_api_key --top-k 3
# Output as JSON for further processing
mcp-rag "What are the benefits of sui::coin?" --output-json > rag_response.jsonRunning the Server
# Start the server with default settings
mcp-server
# Start with custom settings
mcp-server --host 127.0.0.1 --port 8080 --index-file custom/path/index.binManual Usage (without pipx)
Starting the server
cd mcp_server
python main.pyThe server will start on http://localhost:8000
Downloading Move Files from GitHub
To download Move files from GitHub and populate your vector database:
# Download Move files with default query "use sui"
./run.sh --download-move
# Customize the search query
./run.sh --download-move --github-query "module sui::coin" --max-results 50
# Download, index, and start the server
./run.sh --download-move --indexYou can also use the Python script directly:
python download_move_files.py --query "use sui" --output-dir docs/move_filesIndexing documents
Before querying, you need to index your documents. You can place your text files (.txt), Markdown files (.md), or Move files (.move) in the docs directory.
To index the documents, you can either:
Use the run script with the
--indexflag:
./run.sh --indexUse the index script directly:
python index_move_files.py --docs-dir docs/move_files --index-file data/faiss_index.binQuerying documents
You can use the local query script:
python local_query.py "What is RAG?"
# With more options
python local_query.py -k 3 -f "How to define a struct in Sui Move?"Using RAG with LLM Integration
# Direct RAG query with an LLM
python rag_integration.py "What is a module in Sui Move?" --index-file data/faiss_index.bin
# With API key (if you have one)
OPENAI_API_KEY=your_key_here python rag_integration.py "How do coins work in Sui?"MCP API Endpoint
The MCP API endpoint is available at /mcp/action. You can use it to perform different actions:
retrieve_documents: Retrieve relevant documents for a queryindex_documents: Index documents from a directory
Example:
curl -X POST "http://localhost:8000/mcp/action" -H "Content-Type: application/json" -d '{"action_type": "retrieve_documents", "payload": {"query": "What is RAG?", "top_k": 3}}'Complete RAG Pipeline
The full RAG (Retrieval-Augmented Generation) pipeline works as follows:
Search Query: The user submits a question
Retrieval: The system searches the vector database for relevant documents
Context Formation: Retrieved documents are formatted into a prompt
LLM Generation: The prompt is sent to an LLM with the retrieved context
Enhanced Response: The LLM provides an answer based on the retrieved information
This workflow is fully implemented in the rag_integration.py module, which can be used either through the command line or as a library in your own applications.
GitHub Move File Extraction
The system can extract Move files from GitHub based on search queries. It implements two methods:
GitHub API (preferred): Requires a GitHub token for higher rate limits
Web Scraping fallback: Used when API method fails or when no token is provided
To configure your GitHub token, set it in the .env file or as an environment variable:
GITHUB_TOKEN=your_github_token_hereProject Structure
mcp_server/
├── __init__.py # Package initialization
├── main.py # Main server file
├── mcp_api.py # MCP API implementation
├── index_move_files.py # File indexing utility
├── local_query.py # Local query utility
├── download_move_files.py # GitHub Move file extractor
├── rag_integration.py # LLM integration for RAG
├── pyproject.toml # Package configuration
├── requirements.txt # Dependencies
├── .env.example # Example environment variables
├── README.md # This file
├── data/ # Storage for the FAISS index
├── docs/ # Sample documents
│ └── move_files/ # Downloaded Move files
├── models/ # Model implementations
│ └── vector_store.py # FAISS vector store implementation
└── utils/
├── document_processor.py # Document processing utilities
└── github_extractor.py # GitHub file extraction utilitiesExtending the Project
To extend this proof-of-concept:
Add authentication and security features
Implement more sophisticated document processing
Add support for more document types
Integrate with other LLM providers
Add monitoring and logging
Improve the Move language parsing for more structured data extraction
License
MIT
This server cannot be installed
Resources
Looking for Admin?
Admins can modify the Dockerfile, update the server description, and track usage metrics. If you are the server author, to access the admin panel.