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.2.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.2-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.2-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.2.tar.gz.

File metadata

  • Download URL: konan-0.2.2.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.2.tar.gz
Algorithm Hash digest
SHA256 d85b676545ea007ef84537e2e82963ea8e221113df1da718f9a11e5b6b4a5b3b
MD5 358fe3288d68d5e640f74f7449617b8e
BLAKE2b-256 62eed068fb3b7c3c829457bc6a5a2784f83fb4e6cb6a4a7516cac68e28eb3bc9

See more details on using hashes here.

Provenance

The following attestation bundles were made for konan-0.2.2.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.2-cp312-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for konan-0.2.2-cp312-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 e77e405ca5c470726ce6d4baabe7393de8334cdbb2307320c1aa4ef08a16363c
MD5 fdb1e947f0b4e116772d9c01bfbd4bf5
BLAKE2b-256 71b6407ced816b13cbf0d1590ab1fb5b942bf508e1a9f56ec7379a4c56773e65

See more details on using hashes here.

Provenance

The following attestation bundles were made for konan-0.2.2-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.2-cp312-abi3-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for konan-0.2.2-cp312-abi3-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 713747000a59a1d2a249995e68fe3e2ec4f7981a55c5c216af4a9865c1b8c3a8
MD5 ae969132695d631b9fa0f64a7108f2c1
BLAKE2b-256 f7ef3bb0c78cb86857ecdb2099e89668e4c05980237069dde79cb3590c9cf21e

See more details on using hashes here.

Provenance

The following attestation bundles were made for konan-0.2.2-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