molaco/rust-code-mcp

rust-code-mcp

An MCP server for semantic code search in Rust codebases. Combines BM25 full-text search with vector embeddings for hybrid search, plus rust-analyzer based code analysis.

Architecture

Features

• Hybrid search - BM25 keyword search + semantic vector similarity
• Symbol navigation - Find definitions and references across the codebase
• Call graph analysis - Trace function call relationships
• Complexity metrics - LOC, cyclomatic complexity, function counts
• Incremental indexing - Merkle tree change detection for fast re-indexing
• Background sync - Automatic index updates every 5 minutes

Tools

Building

cargo build --release

Nix

A Nix flake is provided for easy setup:

# Enter dev shell with all dependencies
nix develop github:molaco/rust-code-mcp

# Build the binary
nix build github:molaco/rust-code-mcp

The dev shell includes nightly Rust and CUDA support.

Configuration

The server uses stdio transport. Add to your MCP client config:

{
  "mcpServers": {
    "rust-code-mcp": {
      "command": "/path/to/file-search-mcp"
    }
  }
}

GPU Acceleration

Embedding generation uses ONNX Runtime with CUDA support for 10-15x faster indexing on NVIDIA GPUs.

Requirements

• NVIDIA GPU (Maxwell or newer)
• CUDA 12.x + cuDNN 9.x
• The ort crate downloads ONNX Runtime binaries to ~/.cache/ort.pyke.io/

MCP Server CUDA Configuration

For CUDA to work when the MCP server is spawned by Claude Code (or other MCP clients), the LD_LIBRARY_PATH must include:

1. ORT cache - Contains libonnxruntime_providers_shared.so and libonnxruntime_providers_cuda.so
2. CUDA libraries - libcudart.so, libcublas.so, libcublasLt.so
3. cuDNN libraries - libcudnn.so

Option 1: Using flake.nix (recommended)

The included flake.nix automatically generates .mcp.json with the correct LD_LIBRARY_PATH:

nix develop
# Generates .mcp.json with dynamically discovered ORT cache path

Option 2: Manual configuration

First, find your ORT cache path:

find ~/.cache/ort.pyke.io/dfbin -name "libonnxruntime_providers_shared.so" -printf '%h\n' | head -1
# Example output: /home/user/.cache/ort.pyke.io/dfbin/x86_64-unknown-linux-gnu/8BBB.../onnxruntime/lib

Then configure your MCP client (e.g., ~/.claude.json for Claude Code):

{
  "mcpServers": {
    "rust-code-mcp": {
      "command": "/path/to/file-search-mcp",
      "args": [],
      "env": {
        "RUST_LOG": "info",
        "CUDA_HOME": "/usr/local/cuda",
        "CUDA_PATH": "/usr/local/cuda",
        "LD_LIBRARY_PATH": "/home/user/.cache/ort.pyke.io/dfbin/x86_64-unknown-linux-gnu/<HASH>/onnxruntime/lib:/usr/local/cuda/lib64:/usr/lib/x86_64-linux-gnu"
      }
    }
  }
}

Replace:

• /path/to/file-search-mcp with your binary path
• <HASH> with the hash from the find command above
• CUDA paths with your system's CUDA installation

Note: The ORT cache hash changes when ONNX Runtime is updated. If CUDA stops working, re-run the find command to get the new path.

Performance

Stack

• tantivy - Full-text search
• fastembed - Local embeddings (ONNX)
• lancedb - Embedded vector storage
• ra_ap_syntax - AST parsing
• ra_ap_ide - Semantic analysis (goto definition, find references)
• rmcp - MCP protocol

Screenshots

License

MIT