haiku.rag

Agentic RAG that answers questions about your own documents with citations to page numbers and section headings. It runs on an embedded LanceDB database with open models through Ollama by default, so no server or API key is needed. Any provider Pydantic AI supports works in their place, and the same database can live on S3, GCS, Azure or LanceDB Cloud.

Built on LanceDB, Pydantic AI and Docling. Documentation: ggozad.github.io/haiku.rag.

Features

Ingest PDFs, office documents, HTML, Markdown and images with Docling, in-process or on docling-serve. The stored DoclingDocument keeps headings, tables, pictures and page provenance.
Search with hybrid vector and full-text retrieval, optional reranking (cross-encoders, Jina, Cohere, Zero Entropy, vLLM, OpenRouter), section-aware context expansion, and image search with a multimodal embedder (vLLM, OpenRouter, VoyageAI, Cohere). Across several named databases at once.
Answer with the RAG capability: it searches, runs sandboxed Python over the documents for counting and aggregation, and cites page numbers and headings. Vision models receive the figures. Optional capabilities compact earlier evidence in long conversations and require every answer to declare its grounding.
Check a citation by drawing its chunk on the page image, from the CLI, the chat TUI or Python.
Integrate through the Python API, native Pydantic AI capabilities, an MCP server for Claude Code, Codex and Claude Desktop, and a reference web app.
Operate with the haiku-ingester service (filesystem, HTTP, S3 and WebDAV sources, a SQLite or Postgres job queue with retries, a control plane and dashboard), tags and rollback, vacuum, and haiku-rag doctor health checks.

Installation

Python 3.12 or newer.

pip install haiku.rag        # Docling, the VoyageAI and Cohere embedders, every reranker, the TUI
pip install haiku.rag-slim   # the core, with extras chosen by you

The ingester, S3 access and model providers other than Ollama and OpenAI-compatible endpoints are extras. See Installation.

Quick start

The default configuration uses Ollama for embeddings and answers. The quickstart covers the models to pull and using OpenAI instead.

haiku-rag init                                  # create the database
haiku-rag add-src paper.pdf                     # index a file, URL or directory
haiku-rag search "attention mechanism"
haiku-rag ask "What datasets were used for evaluation?"
haiku-rag ask "How many documents mention transformers?"
haiku-rag ask "Does this figure match the spec?" --image figure.png
haiku-rag chat                                  # multi-turn chat in the terminal

Continuous ingestion from configured sources runs as a separate service, with the ingester extra (pip install 'haiku.rag[ingester]'):

haiku-ingester serve

Python API

from haiku.rag.client import HaikuRAG

async with HaikuRAG("knowledge.lancedb", create=True) as rag:
    await rag.create_document_from_source("paper.pdf")
    await rag.create_document_from_source("https://arxiv.org/pdf/1706.03762")

    results = await rag.search("self-attention")
    for result in results:
        print(f"{result.score:.2f} | p.{result.page_numbers} | {result.content[:100]}")

    answer, citations = await rag.ask("What is the complexity of self-attention?")
    print(answer)
    for cite in citations:
        print(f"  [{cite.chunk_id}] p.{cite.page_numbers}: {cite.content[:80]}")

To compose your own agent, see Capabilities.

MCP server

haiku-rag mcp --stdio

The server gives an assistant search, document reading and a Python sandbox over the documents. In Claude Code, the plugin registers it with a skill:

claude plugin marketplace add ggozad/haiku.rag
claude plugin install haiku-rag

Codex and Claude Desktop setup is in the MCP docs.

Examples

Docker setup: docling-serve, the ingester and the MCP server
Web application: conversational RAG over AG-UI with a CopilotKit frontend

Documentation

Quickstart: install, index, chat
Installation: packages and extras
Architecture: how a document becomes a cited answer
CLI and Chat and inspector
Capabilities: native Pydantic AI capabilities
Configuration: every setting, and tuning
Ingester, MCP and remote processing
Python API and custom pipelines
Benchmarks and the changelog

License

MIT.

mcp-name: io.github.ggozad/haiku-rag

Release files for haiku.rag 0.89.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for haiku.rag 0.89.0
File	Size	Uploaded
haiku_rag-0.89.0.tar.gz	535.6 kB	Sep 25, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for haiku.rag 0.89.0
File	Interpreter	ABI	Platform	Reset
haiku_rag-0.89.0-py3-none-any.whl 7.5 kB Sep 25, 2026	Python 3	none	any	Details

Total release size: 543.1 kB

Release files / haiku_rag-0.89.0.tar.gz

Download URL	haiku_rag-0.89.0.tar.gz
Size	535.6 kB
Tags	Source
SHA-256 checksum How to use checksums	`19a59ad4823d816802f7c4e244ea605e1e8a16da0a4e50833de85358065aaa61`
BLAKE2b-256 checksum How to use checksums	`36db58dc9756e579bbf44afea2d122b9edf93f3d16f5046ff9aece704934a57c`
Upload date	Sep 25, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.13.15`

Release files / haiku_rag-0.89.0-py3-none-any.whl

Download URL	haiku_rag-0.89.0-py3-none-any.whl
Size	7.5 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`751c912f7591af923410a0216f8113441bce4b1a7df1c716e122bc13f62d027a`
BLAKE2b-256 checksum How to use checksums	`f885c5d80cbd4f5e9d98a971f1727d36ae56dc7d1d5688c1219fab52c841360c`
Upload date	Sep 25, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.13.15`