microrag 0.1.0

A feature-rich, universal RAG library for Python with ONNX-backed embeddings and DuckDB storage

MicroRAG

A feature-rich, universal RAG library for Python with ONNX-backed embeddings and DuckDB storage.

Features

• ONNX-backed embeddings - CPU-optimized inference using sentence-transformers with ONNX runtime
• DuckDB storage - Persistent vector storage with HNSW indexes for fast similarity search
• Three-tier hybrid search - Combines semantic, BM25, and full-text search with RRF fusion
• Query preprocessing - Abbreviation expansion and stopword removal for better search
• Flexible document input - Accept strings, dicts, or Document objects
• Text chunking - Automatic chunking with sentence boundary detection

Why ONNX?

MicroRAG uses ONNX (Open Neural Network Exchange) format for embedding models:

• Faster inference - ONNX Runtime provides optimized CPU execution, often 2-3x faster than PyTorch
• Smaller footprint - No need for full PyTorch/TensorFlow installation in production
• Cross-platform - Same model runs on any platform without framework dependencies
• Quantization support - Easy to use INT8/FP16 quantized models for even faster inference

Installation

# Install with pip
pip install microrag

# Or install with uv
uv add microrag

# For CPU-only PyTorch (recommended for smaller installs)
uv add microrag --extra cpu

Quick Start

from microrag import MicroRAG, RAGConfig

config = RAGConfig(
    model_path="/path/to/all-MiniLM-L6-v2",
    db_path="./rag.duckdb",
    embedding_dim=384,
)

with MicroRAG(config) as rag:
    # Add documents (strings, dicts, or Document objects)
    rag.add_documents([
        "Machine learning is a subset of artificial intelligence.",
        {"content": "Deep learning uses neural networks.", "metadata": {"source": "wiki"}},
    ])

    # Build search indexes
    rag.build_index()

    # Search
    results = rag.search("neural networks", top_k=5)
    for r in results:
        print(f"{r.score:.3f}: {r.content}")

Search Pipeline

MicroRAG uses a three-tier hybrid search architecture that combines multiple retrieval methods for better results:

Query: "ML techniques"
         │
         ▼
┌─────────────────────────────────────┐
│      Query Preprocessing            │
│  • Normalize whitespace             │
│  • Expand abbreviations (ML→machine │
│    learning)                        │
│  • Tokenize for BM25                │
└─────────────────────────────────────┘
         │
         ▼
┌─────────────────────────────────────┐
│      Parallel Search                │
│                                     │
│  ┌──────────┐  ┌──────────┐  ┌────────────┐
│  │ Semantic │  │  BM25    │  │    FTS     │
│  │  Search  │  │  Search  │  │   Search   │
│  │ (Vector) │  │(Keywords)│  │ (Stemmed)  │
│  └────┬─────┘  └────┬─────┘  └─────┬──────┘
│       │             │              │
│       ▼             ▼              ▼
│    Results       Results        Results
│   + scores      + scores       + scores
└─────────────────────────────────────┘
         │
         ▼
┌─────────────────────────────────────┐
│    Reciprocal Rank Fusion (RRF)     │
│                                     │
│  score = Σ 1/(k + rank_i)           │
│                                     │
│  Combines rankings from all methods │
│  with configurable weighting        │
└─────────────────────────────────────┘
         │
         ▼
      Final ranked results

Search Components

• Semantic - HNSW vector similarity; understands meaning and context
• BM25 - Term frequency scoring; exact keyword matching
• FTS - DuckDB full-text search; stemming and linguistic matching

Why Hybrid Search?

Each search method has different strengths:

• Semantic search finds conceptually similar documents even with different wording
• BM25 excels at finding exact keyword matches
• FTS handles word variations through stemming

By combining all three with RRF fusion, MicroRAG achieves better recall and precision than any single method alone.

Configuration

from microrag import RAGConfig

config = RAGConfig(
    # Required
    model_path="/path/to/model",      # Sentence-transformer model path

    # Storage
    db_path=":memory:",               # DuckDB path (":memory:" for in-memory)
    embedding_dim=384,                # Embedding vector dimension

    # Chunking
    chunk_size=1000,                  # Max characters per chunk
    chunk_overlap=200,                # Overlap between chunks

    # Search
    hybrid_enabled=True,              # Enable hybrid search
    hybrid_alpha=0.7,                 # Semantic weight (0-1)
    similarity_threshold=0.4,         # Min score threshold

    # Query processing
    abbreviations={"ML": "machine learning"},  # Query expansion
    remove_stopwords=True,            # Remove stopwords for BM25

    # HNSW tuning
    hnsw_ef_construction=200,         # Build-time parameter
    hnsw_ef_search=100,               # Search-time parameter
)

Configuration Options

Required:

• model_path (str) - Path to sentence-transformer model

Storage:

• model_file (str, default: None) - ONNX model filename (for quantized models)
• db_path (str, default: :memory:) - DuckDB database path
• embedding_dim (int, default: 384) - Embedding vector dimension

Chunking:

• chunk_size (int, default: 1000) - Text chunking size in characters
• chunk_overlap (int, default: 200) - Overlap between chunks

Search:

• hybrid_enabled (bool, default: True) - Enable hybrid search
• hybrid_alpha (float, default: 0.7) - Semantic weight in fusion (0-1)
• similarity_threshold (float, default: 0.4) - Minimum score to return

Query Processing:

• abbreviations (dict, default: None) - Query expansion mapping
• stopwords (set, default: English) - Stopwords for BM25 tokenization
• remove_stopwords (bool, default: True) - Enable stopword removal

HNSW Tuning:

• hnsw_ef_construction (int, default: 200) - HNSW build parameter
• hnsw_ef_search (int, default: 100) - HNSW search parameter

API Reference

MicroRAG

Main class for RAG operations.

from microrag import MicroRAG, RAGConfig

config = RAGConfig(model_path="/path/to/model")

# Use as context manager (recommended)
with MicroRAG(config) as rag:
    rag.add_documents([...])
    rag.build_index()
    results = rag.search("query")

# Or manage lifecycle manually
rag = MicroRAG(config)
try:
    # ... use rag
finally:
    rag.close()

Methods:

• add_documents(docs, chunk=True) - Add documents (str, dict, or Document)
• build_index() - Build HNSW, BM25, and FTS indexes
• search(query, top_k=10, threshold=None, hybrid=None) - Search documents
• get_document(doc_id) - Get document by ID
• get_all_documents() - Get all documents
• count() - Get document count
• clear() - Remove all documents
• close() - Close resources

Document

Document data model.

from microrag import Document

doc = Document(
    id="doc1",                    # Optional, auto-generated if not provided
    content="Document text...",   # Required
    metadata={"source": "wiki"},  # Optional metadata
)

SearchResult

Search result with score and document data.

results = rag.search("query")

for result in results:
    print(result.score)      # Similarity score
    print(result.content)    # Document content
    print(result.metadata)   # Document metadata
    print(result.document)   # Full Document object

Adding Documents

MicroRAG accepts documents in multiple formats:

# Strings
rag.add_documents([
    "First document content",
    "Second document content",
])

# Dicts with metadata
rag.add_documents([
    {"content": "Document text", "metadata": {"source": "file.txt"}},
    {"id": "custom_id", "content": "Another document"},
])

# Document objects
from microrag import Document

rag.add_documents([
    Document(id="doc1", content="Text", metadata={"key": "value"}),
])

# Disable chunking for pre-chunked content
rag.add_documents(["Already chunked text"], chunk=False)

Examples

See the examples/ directory for complete working examples:

• basic_usage.py - Core workflow: adding documents, building indexes, searching
• advanced_config.py - Custom abbreviations, hybrid search tuning, config variants
• faq_search.py - FAQ/knowledge base search with metadata filtering

Run examples with:

make example name=basic_usage
make example name=advanced_config
make example name=faq_search

Development

# Clone and install
git clone https://github.com/yourname/microrag.git
cd microrag
uv sync --group dev

# Run tests
uv run pytest

# Run linting
uv run ruff check src/ tests/
uv run mypy src/

# Format code
uv run ruff format src/ tests/

License

MIT License - see LICENSE file.