Surprise-guided chunking and novelty-aware RAG scoring — split text at knowledge boundaries, filter by information gain
Project description
pymrsf — Model-Relative Semantic Filtering
Split text at knowledge boundaries — then score by information gain.
Semantic chunkers split at sentence or token boundaries. smart_chunk uses the model's own predictive surprise to detect where one topic ends and another begins — producing chunks that align with actual knowledge transitions, not arbitrary token counts.
- Surprise-guided chunking — finds natural topic boundaries using the model's prediction signal
- Novelty scoring — measures whether the model already knows a chunk (surprise-based)
- Relevance scoring — cosine similarity against your query
- Query Ignorance — does the model even know the answer? (probe-based gate)
- Diversity — dedup post-filter to avoid redundant chunks
Quick install
# Start fast with an API provider (no 4 GB model download)
pip install pymrsf[openai]
export OPENAI_API_KEY='sk-...'
# Or for full features (probing, smart_chunk, round-trip):
pip install pymrsf[local]
All providers require Ollama for embeddings:
ollama pull nomic-embed-text
30-second example — surprise-guided chunking
Instead of splitting at fixed sizes or sentence boundaries, smart_chunk uses the model's surprise signal to find natural knowledge transitions:
from pymrsf import smart_chunk
long_article = """
Quantum computing leverages superposition and entanglement to perform
calculations that would be infeasible for classical computers. Unlike
classical bits, qubits can exist in multiple states simultaneously.
...
Machine learning models learn patterns from data through iterative
optimization of a loss function. Neural networks, in particular,
use backpropagation to adjust millions of parameters.
...
"""
# Chunks split at the boundary between "quantum computing" and "ML" —
# where the model's surprise signal drops after absorbing one topic
pieces = smart_chunk(long_article, min_chunk_len=200, max_chunk_len=800)
Works with any provider — uses surprise signals (local) or embedding similarity (API) to detect topic boundaries. Falls back to sentence splitting if neither is available.
60-second example — score and filter chunks
from pymrsf import score_chunk, filter_chunks
chunks = [
"Backpropagation computes gradients using the chain rule.",
"Neural networks are inspired by the human brain.",
"The sky is blue because of Rayleigh scattering.",
]
# Score a single chunk
result = score_chunk(chunks[0], query="How does backpropagation work?")
print(result["rag_score"]) # 0–100
print(result["verdict"]) # "excellent" / "good" / "moderate" / "weak" / "skip"
# Filter to only the useful chunks
useful = filter_chunks(chunks, query="How does backpropagation work?", min_rag_score=50)
# useful ≈ ["Backpropagation computes gradients..."]
With async for production pipelines:
import asyncio
from pymrsf import filter_chunks_async
useful = asyncio.run(filter_chunks_async(chunks, query="...", min_rag_score=50))
Provider matrix
| Feature | local | openai | anthropic |
|---|---|---|---|
| smart_chunk (topic-aware) | Surprise-guided | Embedding similarity | Embedding similarity |
| RAG scoring | Full (novelty + relevance + ignorance) | Relevance-only | Relevance-only |
| Knowledge probing | ✅ Full | ⚠️ Limited | ❌ |
| Delta compression / round-trip | Experimental | ❌ | ❌ |
| Model session (KV-cache) | ✅ Yes | ❌ | ❌ |
| Async scoring | ✅ | ✅ | ✅ |
| Score caching | ✅ | ✅ | ✅ |
Key takeaway: Probing and surprise-guided chunking require the local provider (pip install pymrsf[local] + a GGUF model). Relevance scoring and embedding-based topic chunking work with any provider.
Production configuration
import pymrsf
# Enable pymrsf log output (silent by default)
pymrsf.configure_logging("INFO")
# Tweak runtime settings without touching env vars
pymrsf.configure(
provider="openai",
embed_timeout=60,
default_relevance_cutoff=0.4,
)
Environment variables for container/CI environments:
PYMRSF_PROVIDER=openai
OPENAI_API_KEY=sk-...
PYMRSF_ALLOW_PROVIDER_FALLBACK=true # silently fall back on embed failures
PYMRSF_EMBED_TIMEOUT=30
PYMRSF_ALLOW_PROVIDER_FALLBACK— whentrue, embed failures log a warning and continue instead of raising. Off by default (fail-fast).pymrsf.configure_logging("WARNING")— pymrsf ships with aNullHandlersoimport pymrsfis silent until you opt in.
See ENV_CONFIG.md for all supported variables.
Experimental: MRSF delta-compression storage
A research backend that stores only "surprise" tokens (40–60% compression) and reconstructs text via model inference. Import from pymrsf.experimental:
from pymrsf.experimental import mrsf_write, mrsf_read, save_index
doc = mrsf_write("The Eiffel Tower was built in 1889.")
print(doc["compression"]) # 0.47 — 47% of tokens were predictable
save_index()
results = mrsf_read("famous French landmark", top_k=1)
Score interpretation
| Score | Verdict | Suggested action |
|---|---|---|
| 80–100 | excellent | Prioritise |
| 60–79 | good | Include |
| 40–59 | moderate | Include if space allows |
| 20–39 | weak | Skip if better chunks exist |
| 0–19 | skip | Model already knows this |
Additional documentation
- PROVIDER_SUPPORT.md — full capability matrix with programmatic checks
- ENV_CONFIG.md — all environment variables
- docs/CONCURRENCY.md — threading and process-safety model
- CHANGELOG.md — version history
Paper
The technical approach is described in the MRSF paper (link forthcoming). For now, see CHANGELOG.md for the research lineage and the experimental module for the delta-compression implementation.
License
MIT
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
Built Distribution
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 pymrsf-1.0.1.tar.gz.
File metadata
- Download URL: pymrsf-1.0.1.tar.gz
- Upload date:
- Size: 53.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
d882d1102fb207757ff9f358fb94763385ec6132fb8a4496bd0577dbcbb85cce
|
|
| MD5 |
fda0304765eaad0de63d1bbdb4114c19
|
|
| BLAKE2b-256 |
2d10b33893f103beef5b14fad90900ef34019bea2d608cf4836fa47d9ed6f45a
|
File details
Details for the file pymrsf-1.0.1-py3-none-any.whl.
File metadata
- Download URL: pymrsf-1.0.1-py3-none-any.whl
- Upload date:
- Size: 49.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6bc187b589516426249867b2bab1756572140edbfc608803ec38d958a57ea2da
|
|
| MD5 |
b94f52a91d974cd4fef82defc90ed7ae
|
|
| BLAKE2b-256 |
16f42e1cc7ea24be59500b48a9d15ca6fe473b48cf31ec312161572a48f9de4e
|