Rust-backed Python chunking library
Project description
py-chunks
Fast, framework-agnostic document chunking library backed by Rust. Extract meaningful content segments from DOCX, DOC, PDF, PPTX, TXT, Markdown, HTML, CSV, XLSX, and XLS files — optimised for production use.
Features
- 10 Document Formats: PDF, DOCX, DOC (Word 97–2003), PPTX, Markdown, HTML, TXT, CSV, XLSX, XLS
- 7 Chunking Modes for document formats:
default,structural,section,semantic,sliding_window,sentence,page_aware - 6 Chunking Modes for spreadsheet formats (XLSX / XLS):
row,table,sheet,sliding_window,page_aware,semantic - 4 Chunking Modes for CSV:
row,default,sliding_window,page_aware - Streaming for every format via a single
stream_chunks()entry point- PDF: background Rust thread +
mpscchannel (all 7 modes, true one-chunk-at-a-time) - Markdown / HTML / TXT: block-by-block state machine for
structural+semantic; batch-drain for the rest - DOCX: all 7 modes —
DocxStructuralIteratorfordefault/structural; dedicated per-mode iterators for the remaining 5 modes (lazy chunk emission after a single upfront parse) - DOC: all 7 modes —
DocStructuralIteratorfordefault/structural; dedicated per-mode iterators for the remaining 5 modes - PPTX: batch-drain (ZIP must be read upfront, then chunks are yielded one at a time)
- XLSX / XLS:
rowandsliding_windowuse true state machines (one chunk per__next__, O(parsed_rows) memory);table,sheet,page_aware, andsemanticuse batch-drain (global sheet analysis required before first chunk) - CSV: true line-by-line worker for
row/default,sliding_window, andpage_aware; delimiter auto-detection and encoding-aware decoding are supported
- PDF: background Rust thread +
- Markdown conversion via
get_markdown()— converts any supported document to a Markdown string (11 extensions:.doc,.docx,.pptx,.pdf,.html,.htm,.xlsx,.xls,.csv,.txt,.md) - Multiple Input Sources: local file paths, raw
bytes/bytearray/memoryview, file-like objects (BytesIO, open files), FastAPI / StarletteUploadFile, HTTP(S) / S3 pre-signed URLs - Consistent Output Schema: every chunk is a
dictwithcontent,content_type, andmetadatakeys - Minimal Python dependencies: all parsing happens in the Rust extension; PDF support uses
pypdfium2(installed automatically), which bundles the PDFium native binary for every platform
Installation
pip install py-chunks
Requirements: Python 3.9+
PDF native library
PDF chunking uses the PDFium native library. It is provided by pypdfium2, which is installed automatically as a dependency and bundles the correct PDFium binary for your platform (macOS, Linux, Windows) — no separate installation needed.
To use a custom PDFium binary instead, set the environment variable before importing:
export PDFIUM_LIBRARY_PATH=/path/to/libpdfium.dylib # macOS
export PDFIUM_LIBRARY_PATH=/path/to/libpdfium.so # Linux
set PDFIUM_LIBRARY_PATH=C:\path\to\pdfium.dll # Windows
Quick Start
from py_chunks import get_chunks, stream_chunks, get_markdown
# Batch — works for every supported format
chunks = get_chunks("document.pdf")
chunks = get_chunks("notes.md", mode="semantic")
chunks = get_chunks("page.html", mode="section")
chunks = get_chunks("deck.pptx", mode="sliding_window", window_size=3, overlap=1)
chunks = get_chunks("report.docx", mode="sentence", sentences_per_chunk=3)
chunks = get_chunks("legacy.doc", mode="default") # DOC (Word 97-2003)
chunks = get_chunks("data.xlsx", mode="row", rows_per_chunk=5)
chunks = get_chunks("legacy.xls", mode="row", rows_per_chunk=5)
chunks = get_chunks("data.csv", mode="row", rows_per_chunk=10)
chunks = get_chunks("data.csv", mode="sliding_window", window_size=5, overlap=1)
for chunk in chunks:
print(chunk["content"])
print(chunk["content_type"]) # e.g. "heading", "plain_paragraph", "semantic"
print(chunk["metadata"]) # format- and mode-specific
# Streaming — works for every supported format
for chunk in stream_chunks("large.pdf", mode="section"):
handle(chunk)
# Markdown conversion — get a Markdown string from any supported document
md = get_markdown("report.docx")
md = get_markdown("legacy.doc")
md = get_markdown("data.csv")
md = get_markdown("notes.txt")
md = get_markdown(file_bytes, filename="report.pdf") # bytes also supported
Chunking Modes
The same seven modes are accepted by every format. The implementation is format-specific (e.g. PDF uses font-size analysis, PPTX uses slide structure, MD/HTML use block parsing), but the surface API is uniform:
| Mode | What it does |
|---|---|
default / structural |
One chunk per structural unit (heading, paragraph, list, table, code block, slide…). For PDF specifically, they are not aliases: default uses chunk_pdf_fast (lightweight extraction) while structural uses chunk_pdf (full font-size-weighted analysis), and output may differ. |
section |
Groups everything under a heading into a single chunk (≤ 2 000 chars). Adds section_heading, section_level, heading_path. |
semantic |
Heuristically merges adjacent blocks by topic continuity using 10 signals (reference pronouns, transition words, elaboration cues, examples, cause/effect, contrast continuation, question/answer, definition expansion, short-paragraph absorption, keyword overlap). Adds merge_reasons, primary_merge_reason, paragraph_count, keyword_density. |
sliding_window |
Overlapping windows of N blocks. Params: window_size (default 3), overlap (default 1, must be < window_size). |
sentence |
N sentences per chunk, detected without NLP (handles abbreviations like Mr., Dr., e.g., numeric markers, initials). Param: sentences_per_chunk (default 3, must be > 0). |
page_aware |
Groups by page boundary where available (PDF page breaks, DOCX w:pageBreak / w:sectPr, PPTX slides), with a paragraph-count fallback. Param: paragraphs_per_page (default 15 for most formats, 5 for PPTX where it means slides-per-chunk). |
DOCX modes
Pass mode to any get_chunks / stream_chunks call:
from py_chunks import get_chunks
chunks = get_chunks("file.docx", mode="default") # structural (default)
chunks = get_chunks("file.docx", mode="structural") # same as default
chunks = get_chunks("file.docx", mode="section")
chunks = get_chunks("file.docx", mode="semantic")
chunks = get_chunks("file.docx", mode="sliding_window", window_size=3, overlap=1)
chunks = get_chunks("file.docx", mode="sentence", sentences_per_chunk=3)
chunks = get_chunks("file.docx", mode="page_aware", paragraphs_per_page=15)
| Mode | Description |
|---|---|
default / structural |
One chunk per document element: heading, paragraph, table, list. Each element typed via content_type. |
section |
All content under a heading grouped into a single chunk (≤ 2 000 chars). metadata includes section_heading, section_level, heading_path. |
semantic |
Paragraphs merged by topic continuity using pure heuristics — reference pronouns, transition words, keyword overlap, short-paragraph absorption (≤ 1 500 chars). metadata includes paragraph_count, merge_reason. |
sliding_window |
Overlapping paragraph windows. Params: window_size (default 3), overlap (default 1). metadata includes window_size, overlap, window_index, paragraph_indices. |
sentence |
N sentences per chunk, detected without NLP. Handles common abbreviations. Param: sentences_per_chunk (default 3). metadata includes sentences_per_chunk, actual_sentence_count, chunk_index, source_paragraph_index. |
page_aware |
Chunks by explicit page breaks (w:pageBreak), section breaks (w:sectPr), then paragraph count fallback. Param: paragraphs_per_page (default 15). metadata includes page_number, page_break_type, paragraph_count. |
Streaming: all 7 DOCX modes are supported by
stream_chunks.default/structuraluseDocxStructuralIterator; the other five modes (section,semantic,sliding_window,sentence,page_aware) each have a dedicated Rust iterator that parses the document once upfront and emits chunks one at a time. Output is byte-for-byte identical toget_chunksfor every mode.
DOC modes (Word 97–2003)
.doc files use the same 7-mode API as DOCX. The parser is a pure Rust implementation using the Compound Binary File (cfb) crate — no LibreOffice, no external processes.
from py_chunks import get_chunks, stream_chunks
chunks = get_chunks("file.doc", mode="default") # structural (default)
chunks = get_chunks("file.doc", mode="structural") # same as default
chunks = get_chunks("file.doc", mode="section")
chunks = get_chunks("file.doc", mode="semantic")
chunks = get_chunks("file.doc", mode="sliding_window", window_size=3, overlap=1)
chunks = get_chunks("file.doc", mode="sentence", sentences_per_chunk=3)
chunks = get_chunks("file.doc", mode="page_aware", paragraphs_per_page=15)
# Streaming — all 7 modes supported
for chunk in stream_chunks("file.doc", mode="semantic"):
process(chunk)
| Mode | Description |
|---|---|
default / structural |
One chunk per paragraph element: heading, normal paragraph, table row, list item. Short paragraphs (< 80 chars) are merged into a single short_disconnected_paragraph chunk. |
section |
All paragraphs under a heading grouped into one chunk (≤ 2 000 chars, split if longer). |
semantic |
Paragraphs merged by keyword overlap and reference continuity (≤ 1 200 chars). |
sliding_window |
Overlapping windows of N paragraphs. Params: window_size (default 3), overlap (default 1). |
sentence |
N sentences per chunk, split without NLP. Param: sentences_per_chunk (default 3). |
page_aware |
Chunks by explicit page-break markers in the binary stream, with paragraph-count fallback. Param: paragraphs_per_page (default 15). |
Format notes:
- Only Word 97–2003 (
.doc) binary format is supported. Pre-Word 97 files raiseRuntimeError: Pre-Word 97 .doc files are not supported. Convert to .docx first. - Text is reconstructed from the piece table (CLX), supporting both compressed CP1252 and Unicode UTF-16LE pieces.
- Heading levels are inferred from the stylesheet (
Stshf) style index. All-caps or title-case short lines are promoted toHeading(2)as a fallback. - Table cells (separated by
\x07in the binary stream) are joined with|in chunk content. - Page breaks (
\x0C) flush the current page group; they are not emitted as content.
Markdown conversion:
get_markdown("file.doc")converts the.docto Markdown — headings become#/##/###, list items become- item, table rows become| cell | cell |, and page breaks become---.
Streaming: all 7 modes are supported.
default/structuraluseDocStructuralIterator; the other five modes each have a dedicated Rust iterator.
PDF modes
All 7 modes are supported for both batch (get_chunks) and streaming (stream_chunks):
For PDF, default and structural are intentionally different modes (not aliases): default uses a fast lightweight path, while structural uses the full font-size-weighted pipeline, so outputs can differ on the same file.
from py_chunks import get_chunks, stream_chunks
# Batch
chunks = get_chunks("file.pdf", mode="default")
chunks = get_chunks("file.pdf", mode="structural")
chunks = get_chunks("file.pdf", mode="section")
chunks = get_chunks("file.pdf", mode="semantic")
chunks = get_chunks("file.pdf", mode="sliding_window", window_size=3, overlap=1)
chunks = get_chunks("file.pdf", mode="sentence", sentences_per_chunk=3)
chunks = get_chunks("file.pdf", mode="page_aware", paragraphs_per_page=15)
# Streaming — same modes, same parameters
for chunk in stream_chunks("file.pdf", mode="section"):
print(chunk["content"])
| Mode | Rust function | Description |
|---|---|---|
default |
chunk_pdf_fast |
Fast page-by-page text extraction with block splitting. Minimal font analysis. |
structural |
chunk_pdf |
Font-size-weighted span pipeline. Heading detection via font size relative to document average. |
section |
chunk_pdf_section |
Groups content under each heading into one chunk (≤ 2 000 chars). metadata includes section_heading, section_level, heading_path, heading_font_size. |
semantic |
chunk_pdf_semantic |
Heuristic merging by reference pronouns, transition words, and keyword overlap (≤ 1 500 chars). metadata includes paragraph_count, merge_reason. |
sentence |
chunk_pdf_sentence |
N sentences per chunk. metadata includes sentences_per_chunk, actual_sentence_count, chunk_index, source_paragraph_index. |
sliding_window |
chunk_pdf_sliding_window |
Overlapping paragraph windows. metadata includes window_size, overlap, window_index, paragraph_range. |
page_aware |
chunk_pdf_page_aware |
Chunks by real page boundaries; falls back to paragraph count for dense pages. metadata includes page_number, page_break_type, paragraph_count. |
Note: PDFs without a text layer (scanned / image-only) will raise
RuntimeError: PDF appears to contain no extractable text. PDFium can only extract text that is embedded as actual text, not rendered as images.
PPTX modes
PPTX supports all 7 modes via the unified mode parameter:
from py_chunks import get_chunks, stream_chunks
chunks = get_chunks("deck.pptx", mode="default") # one chunk per slide (with short-slide merging)
chunks = get_chunks("deck.pptx", mode="structural") # alias for default
chunks = get_chunks("deck.pptx", mode="section") # group by PPTX sections / title-divider heuristic
chunks = get_chunks("deck.pptx", mode="semantic") # merge consecutive slides by topic continuity
chunks = get_chunks("deck.pptx", mode="sliding_window", window_size=3, overlap=1)
chunks = get_chunks("deck.pptx", mode="sentence", sentences_per_chunk=3)
chunks = get_chunks("deck.pptx", mode="page_aware", paragraphs_per_page=5) # slides per chunk
for chunk in stream_chunks("deck.pptx", mode="section"):
...
Note: For PPTX,
paragraphs_per_pageis interpreted as slides per chunk (default 5, not 15).Legacy API:
chunk_pptx_with_strategy(path, strategy=...)still works and is a thin wrapper aroundchunk_pptx(path, mode=...). It is not re-exported at the top level — import it from the submodule:from py_chunks.chunkers.pptx import chunk_pptx_with_strategyNew code should use
chunk_pptx(..., mode=...)instead.
Markdown, HTML, TXT modes
All three formats accept the full set of 7 modes:
from py_chunks import get_chunks
chunks = get_chunks("notes.md", mode="default") # one chunk per block element
chunks = get_chunks("notes.md", mode="semantic") # topic-continuity merging (10 signals)
chunks = get_chunks("notes.md", mode="section") # grouped under each heading
chunks = get_chunks("notes.md", mode="sliding_window", window_size=4, overlap=1)
chunks = get_chunks("notes.md", mode="sentence", sentences_per_chunk=3)
chunks = get_chunks("notes.md", mode="page_aware", paragraphs_per_page=15)
chunks = get_chunks("page.html", mode="semantic") # same modes for HTML
chunks = get_chunks("readme.txt", mode="section") # same modes for plain text
These three formats also support streaming in every mode — see the Streaming section below.
XLSX / XLS modes
Both .xlsx and .xls files are handled by the same chunker. All 6 modes are available for batch and streaming:
from py_chunks import get_chunks, stream_chunks
from py_chunks.chunkers.xlsx import chunk_xlsx, stream_chunk_xlsx
# Batch — via unified API
chunks = get_chunks("data.xlsx", mode="row", rows_per_chunk=5)
chunks = get_chunks("legacy.xls", mode="row", rows_per_chunk=5)
# Batch — via format-specific chunker (returns chunks + timing)
chunks, timing = chunk_xlsx("data.xlsx", mode="row", rows_per_chunk=5)
chunks, timing = chunk_xlsx("data.xlsx", mode="table", max_chunk_chars=3000)
chunks, timing = chunk_xlsx("data.xlsx", mode="sheet", max_chunk_chars=5000)
chunks, timing = chunk_xlsx("data.xlsx", mode="sliding_window", window_size=4, overlap=1)
chunks, timing = chunk_xlsx("data.xlsx", mode="page_aware", max_chunk_chars=3000)
chunks, timing = chunk_xlsx("data.xlsx", mode="semantic", rows_per_chunk=10)
# Filter to specific sheets
chunks, _ = chunk_xlsx("data.xlsx", mode="row", sheet_names=["Sales", "Q4"])
# Streaming — identical output to batch
for chunk in stream_chunks("data.xlsx", mode="row", rows_per_chunk=5):
print(chunk["content"])
for chunk in stream_chunk_xlsx("data.xlsx", mode="sliding_window", window_size=4, overlap=1):
embed_and_store(chunk)
| Mode | content_type |
Description |
|---|---|---|
row |
row_document |
Groups N consecutive data rows into one chunk. Header row is auto-detected and excluded from content. Param: rows_per_chunk (default 1). |
table |
table_region |
Named Excel tables (XLSX only) or heuristic contiguous data regions per sheet. For XLS and sheets without named tables, falls back to bounding-box detection. Param: max_chunk_chars. |
sheet |
sheet |
One chunk per sheet (split by max_chunk_chars if needed). Includes named-table metadata. Param: max_chunk_chars. |
sliding_window |
row_window |
Overlapping windows of N rows. Params: window_size (default 3), overlap (default 1, must be < window_size). |
page_aware |
sheet_region |
Chunks by Excel print areas (XLSX only); falls back to the full sheet when no print area is defined. For XLS, always uses the full-sheet fallback. Param: max_chunk_chars. |
semantic |
semantic_group |
Detects the column with the lowest cardinality of string values, sorts by it, and groups rows sharing the same category value. Falls back to fixed-size chunking when no suitable column is found. Param: rows_per_chunk (used for the fallback). |
Parameters accepted by chunk_xlsx and stream_chunk_xlsx:
| Parameter | Type | Default | Description |
|---|---|---|---|
file_path |
str | — | Path to .xlsx or .xls file |
mode |
str | "row" |
One of the 6 modes above |
rows_per_chunk |
int | 1 | Rows per chunk (row mode and semantic fallback). Must be > 0. |
window_size |
int | 3 | Window size in rows (sliding_window mode). Must be >= 1. |
overlap |
int | 1 | Overlapping rows between windows. Must be < window_size. |
include_headers |
bool | True | Prefix each row value with its column header (key: value format). |
sheet_names |
list[str] | None | None | Process only the named sheets; processes all sheets when None or []. |
skip_empty_rows |
bool | True | Skip rows where every cell is empty. |
max_chunk_chars |
int | 2000 | Character limit per chunk (table, sheet, page_aware modes). |
XLS vs XLSX differences:
| Feature | XLSX | XLS |
|---|---|---|
Named table detection (table mode) |
ZIP XML (table1.xml) — full named-table metadata |
Not available — heuristic bounding-box only; is_named_table is always false |
Print area detection (page_aware mode) |
Parsed from xl/workbook.xml |
Not available — always uses full-sheet fallback; has_print_area is always false |
Named table metadata in sheet mode |
has_named_tables: true/false, named_tables: [...] |
Always has_named_tables: false, named_tables: [] |
| All other modes | Identical | Identical |
XLSX / XLS metadata fields by mode:
| Mode | Notable metadata keys |
|---|---|
row |
sheet_name, sheet_index, row_index, header_row, col_count, rows_per_chunk, actual_row_count, chunk_index |
table |
sheet_name, sheet_index, table_name, is_named_table, header_row, start_row, end_row, start_col, end_col, row_count, col_count, chunk_index, is_split, split_part |
sheet |
sheet_name, sheet_index, row_count, col_count, header_row, has_named_tables, named_tables, chunk_index, is_split, split_part |
sliding_window |
sheet_name, sheet_index, window_size, overlap, actual_row_count, window_index, start_row, end_row, header_row, col_count, chunk_index |
page_aware |
sheet_name, sheet_index, has_print_area, print_area_ref, start_row, end_row, start_col, end_col, row_count, col_count, header_row, region_index, chunk_index, is_split, split_part |
semantic |
sheet_name, sheet_index, category_column, category_value, used_fallback, low_grouping_quality, avg_group_size, start_row, end_row, actual_row_count, header_row, col_count, group_index, chunk_index |
Streaming memory profile:
rowandsliding_windowpre-parse all sheet data once (calamine reads the entire file on open — there is no incremental I/O at the format level), then build and yield one chunk per__next__. The other four modes require global sheet analysis before the first chunk can be emitted, so they materialise all chunks at construction time and drain them lazily. In both cases the streaming iterator yields one chunk at a time.
Header detection: the first all-string row in each sheet is automatically detected as the header row and excluded from chunk content. Columns without a header label are named
Column 1,Column 2, etc.
CSV modes
CSV files support a smaller mode set than the spreadsheet formats, but the API shape is the same for batch and streaming:
from py_chunks import get_chunks, stream_chunks
from py_chunks.chunkers.csv import chunk_csv, stream_chunk_csv
chunks = get_chunks("data.csv", mode="default")
chunks = get_chunks("data.csv", mode="sliding_window", window_size=4, overlap=1)
chunks = get_chunks("data.csv", mode="page_aware", paragraphs_per_page=3)
chunks, timing = chunk_csv("data.csv", mode="row", rows_per_chunk=10, delimiter=",")
for chunk in stream_chunk_csv("data.csv", mode="row", rows_per_chunk=10):
print(chunk["content"])
| Mode | content_type |
Description |
|---|---|---|
row / default |
row_group |
Groups N consecutive data rows into one chunk. Header row is preserved in metadata and included in content when include_headers=True. |
sliding_window |
row_window |
Overlapping windows of N rows. Params: window_size and overlap. |
page_aware |
row_group |
CSV-friendly alias for row chunking. The unified API maps paragraphs_per_page to CSV row count for this mode. |
CSV-specific options:
delimiter: one ofNone,,,\t,;, or|. When omitted, the first non-empty non-comment line is scanned to detect the delimiter.encoding: one ofutf-8,utf-8-bom,latin-1, orwindows-1252.skip_empty_rows: skips rows whose cells are all empty or whitespace-only.
Markdown Conversion
get_markdown() converts a document to a Markdown string in a single call. Use it when you want the full document as Markdown rather than split into chunks — for example, to feed into an LLM context window, display in a UI, or pipe into another tool.
from py_chunks import get_markdown
# From a file path
md = get_markdown("report.docx")
md = get_markdown("legacy.doc")
md = get_markdown("deck.pptx")
md = get_markdown("paper.pdf")
md = get_markdown("page.html")
md = get_markdown("notes.md") # returned as-is
md = get_markdown("readme.txt") # returned as-is
md = get_markdown("data.xlsx")
md = get_markdown("data.csv")
# From bytes (e.g. API upload)
md = get_markdown(file_bytes, filename="report.docx")
# From a file-like object
from io import BytesIO
md = get_markdown(BytesIO(data), filename="document.pdf")
Supported extensions for get_markdown
| Extension(s) | What is produced |
|---|---|
.docx |
Full fidelity: headings (#–###### from Word heading styles / outline levels), unordered lists (- item), ordered lists (1. item) with per-level indentation, pipe tables, fenced code blocks, images as [Image: alt] / [Image], hyperlinks as [text](url), page/section breaks as ---, footnotes and endnotes as [^id]: text appended at the end |
.doc |
H1 → #, H2–H3 → ##, H4+ → ###; lists → - item; each table paragraph → pipe row with | --- | separator; page breaks → ---; plain paragraphs as-is |
.pptx |
Presentation title → # Title; PPTX sections → # Section Name; each slide → ## Slide N: Title (or ## Slide N); paragraphs as plain text; unordered bullets (- item) and ordered bullets (1. item) with per-level indentation; pipe tables; image placeholders ([Image: alt]); speaker notes as > **Notes:** …; slides/sections separated by --- |
.pdf |
Headings inferred from font size vs document average → # / ## / ###; bullet lists preserved or normalized to - item; tables detected by tab/multi-space alignment → pipe tables; page boundaries → --- |
.html, .htm |
H1–H6 → #–######; paragraphs as plain text; ordered lists → 1. item; unordered lists → - item; code blocks → fenced ```; pipe tables with auto-detected header row; | in cells escaped |
.xlsx, .xls |
Each non-empty sheet → ## SheetName heading + pipe table (auto-detected header row); sheets separated by ---; | in cells escaped |
.csv |
Single pipe table; first row = header with | --- | separator; | in cells escaped; empty rows skipped; delimiter auto-detected or manually set; accepts delimiter and encoding params — see csv_to_markdown below |
.md |
Returned as-is (already Markdown — no transformation) |
.txt |
Returned as-is (plain text — no transformation) |
get_markdown signature
get_markdown(
source,
*,
filename: str | None = None,
) -> str
| Parameter | Type | Description |
|---|---|---|
source |
str, Path, bytes, bytearray, memoryview, file-like | Document source. Local file path, raw bytes, or file-like object. |
filename |
str | None | Required when source is bytes, bytearray, memoryview, or a file-like object without a .name attribute. |
Returns — str. The full document rendered as a Markdown string.
Raises
| Exception | Condition |
|---|---|
FileNotFoundError |
Path does not exist |
ValueError |
Unsupported extension or missing filename for bytes / fileobj inputs |
TypeError |
Unsupported source type |
RuntimeError |
Rust-level failure (e.g. scanned PDF with no text layer) |
Note:
get_markdowndoes not support URLs. For URL sources, download the bytes first and pass them with afilename.
Streaming
When to use streaming
Use stream_chunks (or the stream_chunks_from_* variants) when:
- Processing large documents and you want to forward / persist / embed each chunk before the whole document is parsed
- Building pipelines where chunks flow into a queue, vector store, database, or HTTP response
- You want bounded memory regardless of document size (PDF and the MD/HTML/TXT state machines)
Streaming support matrix
| Format | Modes streamable | Mechanism | Notes |
|---|---|---|---|
| All 7 | Background Rust thread + mpsc channel |
Owns the PdfDocument on the worker thread, sends one RawChunk at a time. Output is byte-for-byte identical to get_chunks. |
|
| Markdown | All 7 | Block-by-block state machine (structural, semantic) + batch-drain (others) |
structural / semantic use O(blocks) memory; the other four modes compute the chunk list once and drain it one chunk per __next__. |
| HTML | All 7 | Same as Markdown | Identical hybrid model: state machine for structural / semantic, batch-drain for section / sliding_window / sentence / page_aware. |
| TXT | All 7 | Same as Markdown | Pure Rust, no threads. |
| DOCX | All 7 | DocxStructuralIterator for default/structural; dedicated per-mode Rust iterators for the other 5 |
Full document parsed once upfront; chunks emitted lazily. Peak memory ≈ file size + chunk vec. Output equals get_chunks for every mode. |
| DOC | All 7 | DocStructuralIterator for default/structural; dedicated per-mode Rust iterators for the other 5 |
Binary stream parsed once upfront via piece table reconstruction; chunks emitted lazily. Output equals get_chunks for every mode. |
| PPTX | All 7 | Batch-drain | PPTX requires the full ZIP up front, so chunks are computed once at construction and yielded one per __next__. |
| XLSX / XLS | All 6 | State machine for row / sliding_window; batch-drain for table / sheet / page_aware / semantic |
calamine reads the full file on open (no incremental I/O at format level). row and sliding_window build one chunk per __next__ from pre-parsed row data. The other four modes require global analysis first and materialise all chunks at iterator construction. Output is identical to chunk_xlsx for every mode. |
| CSV | All 3 | Background thread + mpsc channel for row / default / page_aware; VecDeque rolling buffer for sliding_window |
True line-by-line worker — never loads the full file. sliding_window streaming uses an O(window_size) rolling buffer. Output is identical to chunk_csv for every mode. |
Parity guarantee: streaming output equals
list(get_chunks(...))for every format and every supported mode (this is exercised bytest_pdf_streaming.pyfor PDF and by the tests inpy_chunks/tests/test_source_apis.py).
Streaming examples
from py_chunks import stream_chunks
from py_chunks.chunkers.csv import stream_chunk_csv
# PDF — all 7 modes
for chunk in stream_chunks("large.pdf", mode="section"):
store_in_db(chunk)
for chunk in stream_chunks("report.pdf", mode="sliding_window", window_size=4, overlap=1):
embed_and_index(chunk)
# Markdown / HTML / TXT — all 7 modes
for chunk in stream_chunks("book.md", mode="semantic"): ...
for chunk in stream_chunks("page.html", mode="section"): ...
for chunk in stream_chunks("log.txt", mode="sentence", sentences_per_chunk=2): ...
# DOCX — all 7 modes
for chunk in stream_chunks("document.docx", mode="structural"): send_to_queue(chunk)
for chunk in stream_chunks("document.docx", mode="semantic"): process(chunk)
for chunk in stream_chunks("document.docx", mode="section"): index(chunk)
for chunk in stream_chunks("document.docx", mode="sentence", sentences_per_chunk=3): embed(chunk)
for chunk in stream_chunks("document.docx", mode="sliding_window", window_size=3, overlap=1): embed(chunk)
for chunk in stream_chunks("document.docx", mode="page_aware", paragraphs_per_page=15): store(chunk)
# DOC (Word 97-2003) — all 7 modes
for chunk in stream_chunks("legacy.doc", mode="structural"): send_to_queue(chunk)
for chunk in stream_chunks("legacy.doc", mode="semantic"): process(chunk)
for chunk in stream_chunks("legacy.doc", mode="section"): index(chunk)
for chunk in stream_chunks("legacy.doc", mode="sentence", sentences_per_chunk=3): embed(chunk)
for chunk in stream_chunks("legacy.doc", mode="page_aware", paragraphs_per_page=15): store(chunk)
# PPTX — any mode
for chunk in stream_chunks("deck.pptx", mode="semantic"):
...
# XLSX / XLS — all 6 modes
for chunk in stream_chunks("data.xlsx", mode="row", rows_per_chunk=10):
embed_and_index(chunk)
for chunk in stream_chunks("report.xls", mode="sliding_window", window_size=5, overlap=2):
process(chunk)
for chunk in stream_chunks("data.xlsx", mode="table", max_chunk_chars=3000):
store_in_db(chunk)
for chunk in stream_chunks("data.xlsx", mode="semantic", rows_per_chunk=20):
handle(chunk)
# CSV — all 3 modes
for chunk in stream_chunks("data.csv", mode="row", rows_per_chunk=50):
embed_and_index(chunk)
for chunk in stream_chunks("data.csv", mode="sliding_window", window_size=5, overlap=1):
process(chunk)
for chunk in stream_chunk_csv("data.csv", mode="page_aware", rows_per_chunk=100, delimiter="\t"):
store_in_db(chunk)
# From bytes (e.g. FastAPI body)
for chunk in stream_chunks(request_body, filename="report.pdf", mode="semantic"):
process(chunk)
# As a context manager (temp file cleanup for bytes sources)
with stream_chunks(data, filename="big.pdf", mode="section") as it:
for chunk in it:
...
Supported Input Sources
The unified get_chunks / stream_chunks entry points accept any of these automatically:
| Source | Example |
|---|---|
| Local file path (str or Path) | get_chunks("report.pdf") |
| HTTP / S3 presigned URL | get_chunks("https://bucket.s3.amazonaws.com/file.pdf?sig=...") |
| Raw bytes | get_chunks(data, filename="report.pdf") |
bytearray / memoryview |
get_chunks(bytearray_data, filename="doc.docx") |
File-like object (BytesIO, open file) |
get_chunks(BytesIO(data), filename="doc.md") |
FastAPI / Starlette UploadFile |
get_chunks(upload_file) |
Or use the explicit source-specific helpers:
| Function | Source |
|---|---|
get_chunks_from_path(file_path) |
Local path |
get_chunks_from_bytes(data, filename) |
Raw bytes |
get_chunks_from_fileobj(file_obj, filename=None) |
File-like object |
get_chunks_from_upload(upload_file) |
FastAPI UploadFile |
get_chunks_from_s3_presigned_url(url, filename=None, timeout=60) |
Presigned URL |
stream_chunks_from_path(file_path, ...) |
Local path (streaming) |
stream_chunks_from_bytes(data, filename, ...) |
Raw bytes (streaming) |
stream_chunks_from_fileobj(file_obj, ...) |
File-like object (streaming) |
stream_chunks_from_upload(upload_file, ...) |
FastAPI UploadFile (streaming) |
stream_chunks_from_s3_presigned_url(url, ...) |
Presigned URL (streaming) |
Note:
get_markdownaccepts file paths, bytes, and file-like objects, but does not support URLs. Download the content first and pass as bytes with afilename.
Supported Formats
| Format | Extensions | Batch modes | Streaming modes |
|---|---|---|---|
.pdf |
All 7 | All 7 (background thread) | |
| DOCX | .docx |
All 7 | All 7 (dedicated iterator per mode) |
| DOC | .doc |
All 7 | All 7 (dedicated iterator per mode) |
| PPTX | .pptx |
All 7 | All 7 (batch-drain) |
| Markdown | .md |
All 7 | All 7 (state machine for structural / semantic) |
| HTML | .html, .htm |
All 7 | All 7 (state machine for structural / semantic) |
| Plain Text | .txt |
All 7 | All 7 (state machine for structural / semantic) |
| Excel | .xlsx, .xls |
All 6 | All 6 (row / sliding_window state machine; others batch-drain) |
| CSV | .csv |
All 3 | All 3 (background thread; VecDeque rolling buffer for sliding_window) |
The 7 document modes are: default, structural, section, semantic, sliding_window, sentence, page_aware.
The 6 spreadsheet modes are: row, table, sheet, sliding_window, page_aware, semantic.
The 3 CSV modes are: row / default, sliding_window, page_aware.
get_markdown supported extensions: .doc, .docx, .pptx, .pdf, .html, .htm, .xlsx, .xls, .csv, .txt, .md
API Reference
Unified entry points
get_chunks(
source,
*,
filename: str | None = None,
mode: str = "default",
window_size: int = 3,
overlap: int = 1,
sentences_per_chunk: int = 3,
paragraphs_per_page: int = 15,
) -> list[dict]
stream_chunks(
source,
*,
filename: str | None = None,
mode: str = "default",
window_size: int = 3,
overlap: int = 1,
sentences_per_chunk: int = 3,
paragraphs_per_page: int = 15,
) -> Iterator[dict]
get_markdown(
source,
*,
filename: str | None = None,
) -> str
Parameters
| Parameter | Type | Default | Description |
|---|---|---|---|
source |
str, Path, bytes, file-like, upload, URL | — | Document source. Auto-detected. (get_markdown does not support URLs.) |
filename |
str | None | None | Required when source is bytes or a file object without a .name attribute. |
mode |
str | "default" |
Chunking mode. Applies to every supported format (PDF, DOCX, DOC, PPTX, MD, HTML, TXT). One of default, structural, section, semantic, sliding_window, sentence, page_aware. |
window_size |
int | 3 | Number of blocks per window (sliding_window mode). Must be > 0. |
overlap |
int | 1 | Overlapping blocks between windows (sliding_window mode). Must be < window_size. |
sentences_per_chunk |
int | 3 | Sentences per chunk (sentence mode). Must be > 0. |
paragraphs_per_page |
int | 15 | Block / paragraph quota before a page flush (page_aware mode). Must be > 0. For PPTX this means slides per chunk and the format-level default is 5. |
Returns — list[dict] (batch) or Iterator[dict] (streaming) or str (get_markdown). Each chunk dict:
{
"content": str, # extracted text
"content_type": str, # see content types below
"metadata": dict # format- and mode-specific fields
}
Raises
| Exception | Condition |
|---|---|
FileNotFoundError |
Path does not exist |
ValueError |
Unsupported extension, invalid mode, or bad parameter |
TypeError |
Unsupported source type or async .read() on upload |
RuntimeError |
Rust-level failure (e.g. no extractable text in PDF, pre-Word 97 .doc file) |
NotImplementedError |
Streaming requested for an unsupported format/mode |
Format-specific chunkers (advanced)
Each format also has a direct module that returns (chunks, timing), where timing is {"rust_ms": float, "python_ms": float}. Use these when you want per-call timing data or when you only need one format and want to skip source-type detection.
from py_chunks.chunkers.pdf import chunk_pdf, stream_chunk_pdf, pdf_to_markdown
from py_chunks.chunkers.docx import chunk_docx, stream_chunk_docx, docx_to_markdown
from py_chunks.chunkers.doc import chunk_doc, stream_chunk_doc, doc_to_markdown
from py_chunks.chunkers.pptx import chunk_pptx, stream_chunk_pptx, chunk_pptx_with_strategy, pptx_to_markdown
from py_chunks.chunkers.html import chunk_html, stream_chunk_html, html_to_markdown
from py_chunks.chunkers.md import chunk_md, stream_chunk_md, md_to_markdown
from py_chunks.chunkers.txt import chunk_txt, stream_chunk_txt, txt_to_markdown
from py_chunks.chunkers.xlsx import chunk_xlsx, stream_chunk_xlsx, xlsx_to_markdown # handles both .xlsx and .xls
from py_chunks.chunkers.csv import chunk_csv, stream_chunk_csv, csv_to_markdown
# Batch with timing
chunks, timing = chunk_pdf("file.pdf", mode="section")
print(f"Rust: {timing['rust_ms']} ms Python: {timing['python_ms']} ms")
chunks, timing = chunk_md("notes.md", mode="semantic")
chunks, timing = chunk_html("page.html", mode="sliding_window", window_size=4, overlap=1)
chunks, timing = chunk_txt("log.txt", mode="sentence", sentences_per_chunk=2)
chunks, timing = chunk_pptx("deck.pptx", mode="page_aware", paragraphs_per_page=5)
chunks, timing = chunk_doc("legacy.doc", mode="section")
# Legacy PPTX strategy wrapper (kept for backward compatibility)
chunks, timing = chunk_pptx_with_strategy("deck.pptx", strategy="structural")
# Streaming — all formats
for chunk in stream_chunk_pdf("report.pdf", mode="semantic"): ...
for chunk in stream_chunk_docx("doc.docx", mode="structural"): ...
for chunk in stream_chunk_docx("doc.docx", mode="semantic"): ...
for chunk in stream_chunk_docx("doc.docx", mode="section"): ...
for chunk in stream_chunk_docx("doc.docx", mode="sentence", sentences_per_chunk=3): ...
for chunk in stream_chunk_doc("legacy.doc", mode="structural"): ...
for chunk in stream_chunk_doc("legacy.doc", mode="semantic"): ...
for chunk in stream_chunk_doc("legacy.doc", mode="section"): ...
for chunk in stream_chunk_doc("legacy.doc", mode="sentence", sentences_per_chunk=3): ...
for chunk in stream_chunk_md("book.md", mode="sentence", sentences_per_chunk=2): ...
for chunk in stream_chunk_html("page.html", mode="section"): ...
for chunk in stream_chunk_txt("log.txt", mode="page_aware", paragraphs_per_page=20): ...
for chunk in stream_chunk_pptx("deck.pptx", mode="semantic"): ...
# XLSX / XLS — all 6 modes, batch and streaming
chunks, timing = chunk_xlsx("data.xlsx", mode="row", rows_per_chunk=5)
chunks, timing = chunk_xlsx("data.xlsx", mode="table", max_chunk_chars=3000)
chunks, timing = chunk_xlsx("data.xlsx", mode="sheet", max_chunk_chars=5000)
chunks, timing = chunk_xlsx("data.xlsx", mode="sliding_window", window_size=4, overlap=1)
chunks, timing = chunk_xlsx("data.xlsx", mode="page_aware", max_chunk_chars=3000)
chunks, timing = chunk_xlsx("data.xlsx", mode="semantic", rows_per_chunk=10)
chunks, timing = chunk_xlsx("legacy.xls", mode="row", rows_per_chunk=5) # XLS works identically
for chunk in stream_chunk_xlsx("data.xlsx", mode="row", rows_per_chunk=10): ...
for chunk in stream_chunk_xlsx("data.xlsx", mode="sliding_window", window_size=4, overlap=1): ...
for chunk in stream_chunk_xlsx("legacy.xls", mode="semantic", rows_per_chunk=20): ...
# CSV — batch with timing
chunks, timing = chunk_csv("data.csv", mode="row", rows_per_chunk=10)
chunks, timing = chunk_csv("data.csv", mode="sliding_window", window_size=5, overlap=1)
chunks, timing = chunk_csv("data.csv", mode="page_aware", rows_per_chunk=100)
chunks, timing = chunk_csv("data.csv", mode="row", delimiter="\t", encoding="utf-8")
# CSV — streaming
for chunk in stream_chunk_csv("data.csv", mode="row", rows_per_chunk=50): ...
for chunk in stream_chunk_csv("data.csv", mode="sliding_window", window_size=5, overlap=1): ...
for chunk in stream_chunk_csv("data.csv", mode="page_aware", rows_per_chunk=100): ...
# Markdown conversion — direct format wrappers
md = docx_to_markdown("report.docx") # full DOCX → Markdown (headings, lists, tables, footnotes, hyperlinks…)
md = doc_to_markdown("legacy.doc") # DOC binary → Markdown (headings, lists, tables, page breaks)
md = pdf_to_markdown("paper.pdf") # PDF → Markdown (headings by font size, lists, tables, page separators)
md = pptx_to_markdown("deck.pptx") # PPTX → Markdown (title, sections, slides as ##, notes as blockquote)
md = html_to_markdown("page.html") # HTML → Markdown (H1-H6, lists, tables, code blocks)
md = xlsx_to_markdown("data.xlsx") # each sheet → ## heading + pipe table, separated by ---
md = csv_to_markdown("data.csv") # pipe table, first row = header
md = csv_to_markdown("data.csv", delimiter=",", encoding="utf-8") # explicit delimiter + encoding
md = txt_to_markdown("notes.txt") # returned as-is
md = md_to_markdown("readme.md") # returned as-is
Output Schema
Chunk structure
{
"content": "The extracted text segment.",
"content_type": "plain_paragraph",
"metadata": { ... } # keys depend on format and mode — see below
}
content_type values
| Value | Description |
|---|---|
heading |
Section heading (H1–H6, bold text, ALLCAPS line, etc.) |
plain_paragraph |
Regular prose paragraph |
bullet_list |
Unordered or numbered list |
table |
Tabular data |
code_block |
Code or preformatted text |
long_single_paragraph |
Paragraph > 500 characters |
short_disconnected_paragraph |
Paragraph < 80 characters (also used for merged short paragraphs in DOC/DOCX structural mode) |
mixed_content |
DOCX structural block that merges a heading with its immediately following body element (e.g. a heading run that shares a <w:p> with body text) |
section |
Heading-scoped grouped content (section mode) |
semantic |
Heuristic topic-continuity group (semantic mode) |
sliding_window |
Fixed-size overlapping window (sliding_window mode) |
sentence |
Sentence-count group (sentence mode) |
page_aware |
Page boundary group (page_aware mode for document formats) |
row_document |
XLSX/XLS: N consecutive data rows (row mode) |
table_region |
XLSX/XLS: named table or heuristic data region (table mode) |
sheet |
XLSX/XLS: full sheet or split part (sheet mode) |
row_window |
XLSX/XLS and CSV: overlapping row window (sliding_window mode) |
sheet_region |
XLSX/XLS: print area or full sheet (page_aware mode) |
semantic_group |
XLSX/XLS: category-grouped rows or fallback fixed-size group (semantic mode) |
row_group |
CSV: N consecutive data rows (row, default, page_aware modes) |
Metadata fields by mode
Metadata is a dict whose keys depend on both the format and the mode. The most useful keys are listed below; treat any field as optional and use chunk["metadata"].get(key).
| Mode | Format(s) | Notable metadata keys |
|---|---|---|
default / structural |
page_number, is_heading, avg_font_size |
|
default / structural |
DOCX | section_heading, section_heading_level, footnotes (list of {id, text}), endnotes (list of {id, text}), page_number, document_metadata (header_text, footer_text, image_count). Inline images are emitted as [Image: <alt>] (or [Image] when no alt text). Footnote / endnote ids reference word/footnotes.xml / word/endnotes.xml and are anchored to the chunk that contains the referring paragraph. |
default / structural |
DOC | source, chunk_index, total_chunks, paragraph_type (heading/normal/table/list_item), heading_level (1–6 or null), page_number (always null — not available in the binary format) |
default / structural |
MD / HTML / TXT | section_heading, document_metadata.source_type |
default / structural |
PPTX | slide_number, section_heading (when detectable) |
section |
page_number, section_heading, section_level, heading_path, paragraph_count, heading_font_size |
|
section |
DOCX | section_heading, section_heading_level, section_level, heading_path, document_metadata |
section |
DOC | source, chunk_index, total_chunks, paragraph_type, heading_level, page_number |
section |
MD / HTML / TXT / PPTX | section_heading, section_level, heading_path, paragraph_count |
semantic |
page_number, paragraph_count, merge_reason |
|
semantic |
DOCX | section_heading, section_heading_level, paragraph_count, merge_reason, document_metadata |
semantic |
DOC | source, chunk_index, total_chunks, paragraph_type, heading_level, page_number |
semantic |
MD / HTML / TXT / PPTX | paragraph_count, merge_reasons (list), primary_merge_reason, keyword_density, avg_block_length (MD/TXT), section_heading, heading_path, chunk_index, document_metadata |
sentence |
sentences_per_chunk, actual_sentence_count, chunk_index, source_paragraph_index |
|
sentence |
DOCX | sentences_per_chunk, actual_sentence_count, chunk_index, source_paragraph_index, source_paragraph_is_heading, source_paragraph_heading_level, source_paragraph_is_list, source_paragraph_is_table, document_metadata |
sentence |
DOC | source, chunk_index, total_chunks, paragraph_type, heading_level, page_number |
sentence |
MD / HTML / TXT / PPTX | sentences_per_chunk, actual_sentence_count, chunk_index, source_paragraph_index |
sliding_window |
window_size, overlap, window_index, paragraph_count, paragraph_range, page_number |
|
sliding_window |
DOCX | window_size, overlap, window_index, paragraph_indices, list_item_count, heading_count, paragraph_meta, document_metadata |
sliding_window |
DOC | source, chunk_index, total_chunks, paragraph_type, heading_level, page_number |
sliding_window |
MD / HTML / TXT / PPTX | window_size, overlap, window_index, paragraph_count, paragraph_range |
page_aware |
page_number, page_break_type, paragraph_count, document_metadata |
|
page_aware |
DOCX | page_number, page_break_type, paragraph_count, section_heading_level, headings, list_item_count, table_count, document_metadata |
page_aware |
DOC | source, chunk_index, total_chunks, paragraph_type, heading_level, page_number |
page_aware |
MD / HTML / TXT | page_number, page_break_type (heading-boundary or paragraph-count), paragraph_count |
page_aware |
PPTX | slide_numbers, paragraph_count |
CSV metadata fields by mode:
| Mode | Notable metadata keys |
|---|---|
row / default / page_aware |
row_start, row_end, row_count, col_count, header_row, delimiter_detected, encoding, chunk_index |
sliding_window |
window_index, window_size, overlap, row_start, row_end, actual_row_count, col_count, header_row, delimiter_detected, encoding, chunk_index |
The DOCX semantic merge_reason is one of: heading_merge, keyword_overlap, reference_continuity, short_paragraph, transition_break.
The MD / HTML / TXT / PPTX semantic primary_merge_reason is one of: reference_continuity, elaboration, example, cause_effect, contrast_continuation, question_answer, definition_expansion, short_paragraph, keyword_overlap, or initial (singleton chunks).
Usage Examples
Local file
from py_chunks import get_chunks
chunks = get_chunks("report.pdf")
for chunk in chunks:
print(chunk["content"][:120])
Legacy .doc file (Word 97-2003)
from py_chunks import get_chunks, get_markdown
# Chunk the document
chunks = get_chunks("contract.doc", mode="section")
for chunk in chunks:
print(chunk["metadata"]["heading_level"], chunk["content"][:80])
# Or convert the whole thing to Markdown
md = get_markdown("contract.doc")
print(md)
Convert any document to Markdown
from py_chunks import get_markdown
# Works for all 11 supported extensions
for path in ["report.docx", "legacy.doc", "deck.pptx", "paper.pdf",
"data.xlsx", "data.csv", "notes.txt", "readme.md"]:
md = get_markdown(path)
print(f"--- {path} ---")
print(md[:500])
Streaming a large PDF section-by-section
from py_chunks import stream_chunks
for chunk in stream_chunks("large.pdf", mode="section"):
heading = chunk["metadata"].get("section_heading", "")
print(f"[{heading}] {chunk['content'][:80]}")
From bytes (API upload)
from py_chunks import get_chunks_from_bytes, get_markdown
file_bytes = request.files['document'].read()
chunks = get_chunks_from_bytes(file_bytes, filename="report.pdf")
# Or get Markdown from bytes
md = get_markdown(file_bytes, filename="report.docx")
From file-like object
from py_chunks import get_chunks_from_fileobj
from io import BytesIO
bio = BytesIO(file_data)
chunks = get_chunks_from_fileobj(bio, filename="document.md")
From S3 presigned URL
from py_chunks import get_chunks_from_s3_presigned_url
url = "https://bucket.s3.amazonaws.com/file.docx?AWSAccessKeyId=..."
chunks = get_chunks_from_s3_presigned_url(url)
Framework Integration
FastAPI
from fastapi import FastAPI, File, UploadFile
from py_chunks import get_chunks_from_upload
app = FastAPI()
@app.post("/chunk/")
async def chunk_document(file: UploadFile = File(...)):
chunks = get_chunks_from_upload(file)
return {"chunks": chunks}
FastAPI — streaming response
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import StreamingResponse
from py_chunks import stream_chunks_from_upload
import json
app = FastAPI()
@app.post("/chunk/stream/")
async def chunk_stream(file: UploadFile = File(...)):
def generate():
for chunk in stream_chunks_from_upload(file):
yield json.dumps(chunk) + "\n"
return StreamingResponse(generate(), media_type="application/x-ndjson")
FastAPI — Markdown conversion endpoint
from fastapi import FastAPI, File, UploadFile
from py_chunks import get_markdown
app = FastAPI()
@app.post("/markdown/")
async def to_markdown(file: UploadFile = File(...)):
data = await file.read()
md = get_markdown(data, filename=file.filename)
return {"markdown": md}
Flask
from flask import Flask, request
from py_chunks import get_chunks_from_bytes
app = Flask(__name__)
@app.post("/chunk")
def chunk_document():
file = request.files['document']
chunks = get_chunks_from_bytes(file.read(), file.filename)
return {"chunks": chunks}
Django
from django.http import JsonResponse
from py_chunks import get_chunks_from_upload
def chunk_view(request):
if request.FILES:
chunks = get_chunks_from_upload(request.FILES['document'])
return JsonResponse({"chunks": chunks})
return JsonResponse({"error": "No file"}, status=400)
Celery background job
import celery
from py_chunks import get_chunks
@celery.task
def process_document(file_path: str):
chunks = get_chunks(file_path)
# persist to database
return len(chunks)
Architecture
┌──────────────────────────────────────────────┐
│ Python Public API │
│ (py_chunks/__init__.py) │
│ get_chunks() / stream_chunks() │
│ get_markdown() │
│ *_from_path / _from_bytes / _from_fileobj │
│ *_from_upload / _from_s3_presigned_url │
└──────────────┬───────────────────────────────┘
│ source detection + temp-file management + cleanup
↓
┌──────────────────────────────────────────────┐
│ Format Dispatcher │
│ (py_chunks/chunkers/*.py) │
│ chunk_pdf / chunk_docx / chunk_doc / │
│ chunk_pptx / chunk_md / chunk_html / │
│ chunk_txt / chunk_xlsx (xlsx + xls) / │
│ chunk_csv + │
│ matching stream_chunk_* variants + │
│ *_to_markdown conversion wrappers │
└──────────────┬───────────────────────────────┘
│ validates args, dispatches to the right Rust function,
│ measures Python-side timing
↓
┌──────────────────────────────────────────────────────────────────┐
│ Rust Extension (_rust.so) │
│ (src/extensions/<format>/*.rs) │
│ │
│ Each format submodule contains: │
│ structural.rs — default / structural chunker │
│ section.rs — section-grouped chunker │
│ semantic.rs — 10-signal topic-continuity chunker │
│ sliding_window.rs │
│ sentence.rs │
│ page_aware.rs │
│ stream_iter.rs — streaming iterator(s) │
│ to_markdown.rs — Markdown conversion function │
│ │
│ PDF stream — background thread owns PdfDocument; sends │
│ RawChunk through mpsc channel; __next__ recvs │
│ MD/HTML/TXT — block-by-block state machine for structural / │
│ semantic; batch-drain for the other 4 modes │
│ DOCX stream — DocxStructuralIterator (default/structural) + │
│ per-mode iterators for all other 5 modes │
│ DOC — cfb crate opens Compound Binary File; │
│ FIB → piece table (CLX) → paragraph props │
│ (PlcfBtePapx) → stylesheet (Stshf); │
│ DocStructuralIterator + per-mode iterators │
│ PPTX stream — batch-drain (ZIP must be read upfront) │
│ XLSX/XLS — open_workbook_auto() handles both formats; │
│ row / sliding_window: state machine, one chunk per __next__ │
│ table / sheet / page_aware / semantic: batch-drain │
│ table mode: ZIP XML for named tables (XLSX) or heuristic │
│ page_aware: print-area XML (XLSX) or full-sheet fallback │
│ CSV — csv crate (not calamine); encoding_rs decoding │
│ row / default / page_aware: background thread + mpsc channel │
│ sliding_window: VecDeque rolling buffer, O(window_size) memory│
└──────────────────────────────────────────────────────────────────┘
Design principles
- Single responsibility — each format has its own Rust submodule; modes never leak between formats
- Framework-agnostic Python layer — source detection (path / URL / bytes / file-like / upload) lives in
py_chunks/__init__.py; the Rust layer only sees a file path - Temp-file strategy for bytes — bytes / file-like / URL inputs are written to a
NamedTemporaryFile(with the original extension), passed to Rust, then deleted; streaming variants wrap the iterator in_StreamingFileCleanupso the temp file is removed even on early exit - PDF streaming safety — the background worker owns the
PdfDocumentfor its full lifetime; chunks cross the thread boundary as plainRawChunkstructs throughmpsc, so nounsafeis needed - Streaming parity — every streaming iterator yields the same chunks (and metadata) as the corresponding batch function
- Pure Rust DOC parsing — the
.docbinary format is parsed entirely in Rust using thecfbcrate with no external processes; text reconstruction, heading detection, and paragraph classification all happen at compile-time-checked byte offsets
Error Handling
from py_chunks import get_chunks, get_markdown
# File not found
try:
chunks = get_chunks("missing.pdf")
except FileNotFoundError as e:
print(e) # File not found: missing.pdf
# Unsupported format
try:
chunks = get_chunks("image.png")
except ValueError as e:
print(e) # Unsupported file type '.png'. Supported: .csv, .doc, .docx, .htm, .html, .md, .pdf, .pptx, .txt, .xls, .xlsx
# Pre-Word 97 .doc file (not supported)
try:
chunks = get_chunks("ancient.doc")
except RuntimeError as e:
print(e) # Pre-Word 97 .doc files are not supported. Convert to .docx first.
# Scanned / image-only PDF (no text layer)
try:
chunks = get_chunks("scanned.pdf")
except RuntimeError as e:
print(e) # PDF appears to contain no extractable text
# Bytes source requires a filename so the extension can be detected
try:
chunks = get_chunks(b"hello")
except ValueError as e:
print(e) # filename is required when source is bytes
# Invalid sliding_window parameters
try:
chunks = get_chunks("notes.md", mode="sliding_window", window_size=2, overlap=2)
except ValueError as e:
print(e) # overlap must be less than window_size
# get_markdown with unsupported extension
try:
md = get_markdown("image.png")
except ValueError as e:
print(e) # get_markdown does not support '.png'. Supported: ['.csv', '.doc', ...]
Exceptions raised
| Exception | When |
|---|---|
FileNotFoundError |
A path was given but does not exist on disk. |
ValueError |
Unsupported extension, unknown mode, empty bytes, invalid window_size / overlap / sentences_per_chunk / paragraphs_per_page, missing filename for bytes / fileobj / URL inputs. |
TypeError |
Unsupported source type, or upload_file.read() returned a coroutine (async). Pass upload_file.file instead, or await it yourself. |
RuntimeError |
Rust-level failure (e.g. PDF with no extractable text, malformed DOCX/PPTX ZIP, pre-Word 97 .doc file, unreadable file). |
NotImplementedError |
A streaming mode/format combination that is not supported. |
Development & Testing
Build from source
cd py_chunks
pip install maturin
maturin develop
Running tests
cd py_chunks
python -m pytest -v
Full PDF strategy test (batch + streaming parity across all modes)
python test_pdf_streaming.py
Tests all 7 strategies × batch + streaming on every PDF in test_files/. Validates chunk count parity between batch and streaming paths.
Code quality
python -m pylint py_chunks tests/test_source_apis.py
Expected: 10.00/10
License
MIT
Built with Rust (performance) + Python (simplicity)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distributions
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file py_chunks-0.4.0.tar.gz.
File metadata
- Download URL: py_chunks-0.4.0.tar.gz
- Upload date:
- Size: 224.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
650ffebc1d4c6990eb976d74821e4528c4a23474af91350bc1149787224ebcf9
|
|
| MD5 |
3c1e013f0006da5ea15f8ad66a57d68e
|
|
| BLAKE2b-256 |
ff69672616abd8d9fca4d8a780d948e615b5ea4881eac8f0a81f93b0315698fc
|
Provenance
The following attestation bundles were made for py_chunks-0.4.0.tar.gz:
Publisher:
release.yml on RanjanKudesia/py-chunks
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
py_chunks-0.4.0.tar.gz -
Subject digest:
650ffebc1d4c6990eb976d74821e4528c4a23474af91350bc1149787224ebcf9 - Sigstore transparency entry: 1642679066
- Sigstore integration time:
-
Permalink:
RanjanKudesia/py-chunks@349803f73474f7bf8f77d7030de9c25d19130a06 -
Branch / Tag:
refs/tags/v0.4.0 - Owner: https://github.com/RanjanKudesia
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yml@349803f73474f7bf8f77d7030de9c25d19130a06 -
Trigger Event:
push
-
Statement type:
File details
Details for the file py_chunks-0.4.0-cp313-cp313-win_amd64.whl.
File metadata
- Download URL: py_chunks-0.4.0-cp313-cp313-win_amd64.whl
- Upload date:
- Size: 2.2 MB
- Tags: CPython 3.13, Windows x86-64
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
2727e789b9faf73550e999450eaff40980b741b345ada8b18066e8a8fd4a829c
|
|
| MD5 |
0723c2ef96c256ffdc5778604d4d5b48
|
|
| BLAKE2b-256 |
39b1bc70015150a74bab21ee5c12c6e8e2b71f9d04dd12adf6abd659992c765a
|
Provenance
The following attestation bundles were made for py_chunks-0.4.0-cp313-cp313-win_amd64.whl:
Publisher:
release.yml on RanjanKudesia/py-chunks
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
py_chunks-0.4.0-cp313-cp313-win_amd64.whl -
Subject digest:
2727e789b9faf73550e999450eaff40980b741b345ada8b18066e8a8fd4a829c - Sigstore transparency entry: 1642679223
- Sigstore integration time:
-
Permalink:
RanjanKudesia/py-chunks@349803f73474f7bf8f77d7030de9c25d19130a06 -
Branch / Tag:
refs/tags/v0.4.0 - Owner: https://github.com/RanjanKudesia
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yml@349803f73474f7bf8f77d7030de9c25d19130a06 -
Trigger Event:
push
-
Statement type:
File details
Details for the file py_chunks-0.4.0-cp313-cp313-macosx_11_0_arm64.whl.
File metadata
- Download URL: py_chunks-0.4.0-cp313-cp313-macosx_11_0_arm64.whl
- Upload date:
- Size: 2.3 MB
- Tags: CPython 3.13, macOS 11.0+ ARM64
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
2de68534dafac47f0ca3b66810ad964649e5576999d750b9fc6c211a820e095b
|
|
| MD5 |
26961e3debd16320894dd2c539b56aae
|
|
| BLAKE2b-256 |
4fd31918776caf2f59264341217a3c337da3bea0d77b2ea6c9146f3e0f41ade7
|
Provenance
The following attestation bundles were made for py_chunks-0.4.0-cp313-cp313-macosx_11_0_arm64.whl:
Publisher:
release.yml on RanjanKudesia/py-chunks
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
py_chunks-0.4.0-cp313-cp313-macosx_11_0_arm64.whl -
Subject digest:
2de68534dafac47f0ca3b66810ad964649e5576999d750b9fc6c211a820e095b - Sigstore transparency entry: 1642679571
- Sigstore integration time:
-
Permalink:
RanjanKudesia/py-chunks@349803f73474f7bf8f77d7030de9c25d19130a06 -
Branch / Tag:
refs/tags/v0.4.0 - Owner: https://github.com/RanjanKudesia
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yml@349803f73474f7bf8f77d7030de9c25d19130a06 -
Trigger Event:
push
-
Statement type:
File details
Details for the file py_chunks-0.4.0-cp313-cp313-macosx_10_12_x86_64.whl.
File metadata
- Download URL: py_chunks-0.4.0-cp313-cp313-macosx_10_12_x86_64.whl
- Upload date:
- Size: 2.4 MB
- Tags: CPython 3.13, macOS 10.12+ x86-64
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0bb5281dffeccbfc2bcf7fe369994bb18d91ddf3a7300554f8d7bb9b4b6cb8f5
|
|
| MD5 |
bfb70475a4b67b16757bf974fce00eb3
|
|
| BLAKE2b-256 |
52a251e841a42980c01fdeb6fe0e944a8a6466b66417630783fd8d983a6abb53
|
Provenance
The following attestation bundles were made for py_chunks-0.4.0-cp313-cp313-macosx_10_12_x86_64.whl:
Publisher:
release.yml on RanjanKudesia/py-chunks
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
py_chunks-0.4.0-cp313-cp313-macosx_10_12_x86_64.whl -
Subject digest:
0bb5281dffeccbfc2bcf7fe369994bb18d91ddf3a7300554f8d7bb9b4b6cb8f5 - Sigstore transparency entry: 1642679721
- Sigstore integration time:
-
Permalink:
RanjanKudesia/py-chunks@349803f73474f7bf8f77d7030de9c25d19130a06 -
Branch / Tag:
refs/tags/v0.4.0 - Owner: https://github.com/RanjanKudesia
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yml@349803f73474f7bf8f77d7030de9c25d19130a06 -
Trigger Event:
push
-
Statement type:
File details
Details for the file py_chunks-0.4.0-cp311-cp311-manylinux_2_28_x86_64.whl.
File metadata
- Download URL: py_chunks-0.4.0-cp311-cp311-manylinux_2_28_x86_64.whl
- Upload date:
- Size: 5.5 MB
- Tags: CPython 3.11, manylinux: glibc 2.28+ x86-64
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
e490e5387df494f5ca1bbd96cee52ef4a7ad4bc899c4a05fc0471034539009b9
|
|
| MD5 |
a0f3af259e8baf4f139427c2c2f90676
|
|
| BLAKE2b-256 |
5d5cb1510596e025f5f5c86449586031aea26796e325b97b5edbc8396f3d971b
|
Provenance
The following attestation bundles were made for py_chunks-0.4.0-cp311-cp311-manylinux_2_28_x86_64.whl:
Publisher:
release.yml on RanjanKudesia/py-chunks
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
py_chunks-0.4.0-cp311-cp311-manylinux_2_28_x86_64.whl -
Subject digest:
e490e5387df494f5ca1bbd96cee52ef4a7ad4bc899c4a05fc0471034539009b9 - Sigstore transparency entry: 1642679401
- Sigstore integration time:
-
Permalink:
RanjanKudesia/py-chunks@349803f73474f7bf8f77d7030de9c25d19130a06 -
Branch / Tag:
refs/tags/v0.4.0 - Owner: https://github.com/RanjanKudesia
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
release.yml@349803f73474f7bf8f77d7030de9c25d19130a06 -
Trigger Event:
push
-
Statement type: