Skip to main content

Blazingly fast text chunkers in Rust with pythonic bindings

Project description

konan logo

konan

Like the paper angel of the Akatsuki, konan folds your documents into perfect pieces — blazingly fast chunkers written in Rust, wrapped for pythonic bliss.

PyPI CI python-ci Python 3.12+ Rust License: MIT


Why konan?

  • 🦀 Rust core — all chunking runs in native code, no Python-loop overhead
  • Multithreaded out of the boxchunk_many() fans out across all cores via rayon and releases the GIL
  • 🌀 First-class async — every chunker ships real async def flavours (chunk_async, chunk_many_async)
  • 🎯 Char-accurate offsetstext[chunk.start:chunk.end] == chunk.text, always (Python slicing semantics, emoji-safe)
  • 🧠 Semantic chunking — splits on topic shifts using any OpenAI-compatible embeddings endpoint, or your own async embedder
  • 🔌 Ports & adapters — the Embedder port is injectable; bring your own backend

Strategies

Chunker Splits by Best for
NaiveChunker fixed word count quick & dirty baselines
FixedSizeChunker chars, sentence-aware, overlap classic RAG pipelines
RecursiveChunker separator hierarchy (\n\n\n → … ) general text, LangChain-compatible
SentenceChunker unicode sentence boundaries prose, multilingual text
MarkdownChunker document structure + heading breadcrumbs docs, wikis, READMEs
TokenChunker exact token counts (cl100k_base, o200k_base) embedding-model token limits
SemanticChunker embedding similarity drops topic-coherent chunks

Installation

uv add konan        # or: pip install konan

Quickstart

from konan import RecursiveChunker

chunker = RecursiveChunker(chunk_size=1000, chunk_overlap=200)

chunks = chunker.chunk(open("moby_dick.txt").read())
print(chunks[0].text, chunks[0].start, chunks[0].end, chunks[0].hash)

Parallel — all cores, zero setup

# rayon work-stealing across every core, GIL released:
all_chunks = chunker.chunk_many(documents)

Async — real async def, no thread-pool wrappers

chunks = await chunker.chunk_async(text)
batches = await chunker.chunk_many_async(documents)

Markdown with breadcrumbs

from konan import MarkdownChunker

chunks = MarkdownChunker(chunk_size=800).chunk(readme_text)
# chunk text is prefixed with its heading trail: "# Guide > ## Install\n\n..."
# code fences are never split

Token-exact chunks

from konan import TokenChunker

chunker = TokenChunker(chunk_size=512, chunk_overlap=64, encoding="o200k_base")

Semantic chunking

Point it at any OpenAI-compatible /embeddings endpoint (OpenAI, vLLM, Ollama, LiteLLM, …):

from konan import OpenAIEmbedder, SemanticChunker

embedder = OpenAIEmbedder(
    base_url="https://api.openai.com/v1",
    model="text-embedding-3-small",
    api_key="sk-...",
    batch_size=128,    # texts per request
    timeout=30.0,      # request timeout, seconds
    max_retries=2,     # exponential backoff on 429/5xx/connect errors
    dimensions=512,    # optional: shorten text-embedding-3-* vectors
)
chunker = SemanticChunker(embedder=embedder, threshold=0.75)
chunks = await chunker.chunk_async(article)

Or inject your own embedder — any async callable works:

async def my_embedder(texts: list[str]) -> list[list[float]]:
    return await my_model.embed(texts)

chunker = SemanticChunker(embedder=my_embedder, percentile=95.0)
chunks = await chunker.chunk_async(article)   # async-only for Python embedders

Python embedders must return list[list[float]] — call .tolist() on numpy arrays. They are async-only: chunk()/chunk_many() raise a RuntimeError pointing you at the _async variants.

The Chunk object

chunk.text    # the chunk's text
chunk.start   # char offset into the source (Python slicing semantics)
chunk.end     # char offset, exclusive
chunk.index   # 0-based position
chunk.hash    # xxh3-64 content hash

Benchmarks

Benchmarked on Apple M3 Pro (arm64), Python 3.12.10. Decimal MB/s, median of 5 runs, measured from Python (the numbers you actually get). Reproduce — tables and plots — with uv run --extra bench benchmarks/bench.py; Rust-level criterion benches: cargo bench -p konan-core.

vs other libraries (same 1 MB document, identical configs)

konan vs other libraries

Strategy Library Throughput Chunks
recursive konan 279 MB/s 1298
recursive semantic-text-splitter 190 MB/s 1368
recursive chonkie 90 MB/s 1413
recursive langchain-text-splitters 25 MB/s 1468
token konan 62 MB/s 438
token semchunk 25 MB/s 616
token langchain-text-splitters 21 MB/s 438
token chonkie 18 MB/s 438
token semantic-text-splitter 3 MB/s 505
sentence konan 71 MB/s 1042
sentence chonkie 3 MB/s 2124
recursive (unicode) konan 170 MB/s 1092
recursive (unicode) semantic-text-splitter 165 MB/s 1150
recursive (unicode) chonkie 58 MB/s 1171

Throughput per strategy (1 MB document)

konan throughput per strategy

Chunker Config Throughput Chunks
NaiveChunker 200 words 549 MB/s 804
FixedSizeChunker 1000 chars, 200 overlap 1,065 MB/s 1255
RecursiveChunker 1000 chars, 200 overlap 273 MB/s 1298
SentenceChunker 1000 chars, 1 overlap 71 MB/s 1130
MarkdownChunker 1000 chars, 200 overlap 444 MB/s 1511
TokenChunker 512 tokens, 64 overlap (cl100k) 61 MB/s 438

Parallel scaling — rayon goes brrr

chunk_many parallel scaling

64 docs × 256 KB through RecursiveChunker:

Mode Time Throughput Speedup
sequential chunk() loop 61 ms 267 MB/s 1.0×
chunk_many() (rayon, GIL released) 11 ms 1,454 MB/s 5.5×

(Pure-Rust criterion puts the same workload at ~6 GiB/s; the Python numbers include chunk-object conversion.)

Caveats, honestly: recursive/token use identical configs across libraries (1000 chars / 200 overlap; cl100k, 512 / 64 — note the identical token chunk counts for the libraries with the same windowing semantics). Sentence configs are not directly comparable (konan groups by chars, chonkie by tokens) — read those rows as per-library cost, not head-to-head. The recursive (unicode) rows run mixed German/CJK/Cyrillic/emoji prose, off konan's ASCII fast path. konan tokenizes with bpe-openai (tiktoken-equivalent output, much faster encoder).

Development

uv sync                       # set up the venv (builds the extension)
uv run maturin develop --uv   # rebuild after Rust changes
cargo test --workspace        # rust unit tests
uv run pytest -q              # python integration tests

The workspace is hexagonal: crates/konan-core is pure Rust (no PyO3) with Chunker and Embedder ports; crates/konan-py adapts it to Python.

License

MIT


Named after Konan of the Akatsuki — the only one who could fold paper into anything. 🗞️

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

konan-0.2.1.tar.gz (45.0 kB view details)

Uploaded Source

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

konan-0.2.1-cp312-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (29.2 MB view details)

Uploaded CPython 3.12+manylinux: glibc 2.17+ x86-64

konan-0.2.1-cp312-abi3-macosx_11_0_arm64.whl (29.2 MB view details)

Uploaded CPython 3.12+macOS 11.0+ ARM64

File details

Details for the file konan-0.2.1.tar.gz.

File metadata

  • Download URL: konan-0.2.1.tar.gz
  • Upload date:
  • Size: 45.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for konan-0.2.1.tar.gz
Algorithm Hash digest
SHA256 ceb7ad9ae4b6203d06d7c5d2024174bfdea59ab1af7bebbf8742338b870cbb6b
MD5 4452ac504dfce024305941d0e314731b
BLAKE2b-256 fef85645100a02aa15c4dae1f163dfed10dac6b3c715f2a00e081289f71caa47

See more details on using hashes here.

Provenance

The following attestation bundles were made for konan-0.2.1.tar.gz:

Publisher: release-please.yaml on 4thel00z/konan

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file konan-0.2.1-cp312-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for konan-0.2.1-cp312-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 69bb2a8aa5ff8950b9e0944b72c090b80fb726dac9123c50957990061f0bac21
MD5 bf6e7fd1930f8993b130b858f4ef1567
BLAKE2b-256 cb165db0f60bc7c406452de8671d94976af6ff72b2d1d1a2180a79ada67993da

See more details on using hashes here.

Provenance

The following attestation bundles were made for konan-0.2.1-cp312-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl:

Publisher: release-please.yaml on 4thel00z/konan

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file konan-0.2.1-cp312-abi3-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for konan-0.2.1-cp312-abi3-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 c71f91081e697fae6ee2f3e345f012b03198da371a28cc6449d41591f8d3b682
MD5 4d94ee35471b6ef5f020ba5b11e51743
BLAKE2b-256 da5649693f7b909a5fd4882a0e93ccff8965b4556c4bb9735259f1ed77f1285f

See more details on using hashes here.

Provenance

The following attestation bundles were made for konan-0.2.1-cp312-abi3-macosx_11_0_arm64.whl:

Publisher: release-please.yaml on 4thel00z/konan

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page