DSPM Memory
Compress multi-turn LLM conversations by 80%+ while every constraint and decision survives.
100% critical retention across our benchmark suite — under impossibly small budgets, criticals are trimmed to their word floor and dropped only as a documented last resort.
The Problem
Long conversations eat your context window. Naive truncation drops the constraint from turn 3 that the entire system depends on. DSPM fixes this.
How It Works
DSPM converts each conversation turn into typed semantic patches — constraint, decision, code, entity, structure — and compresses them under a fixed token budget. Critical patches (constraints and decisions) are structurally protected: they survive compression even when everything else is trimmed.
Raw conversation (452 tokens, 18 turns)
↓
Semantic extraction → 28 patches, 18 critical
↓
7-stage compression pipeline
↓
Compressed context (249 tokens) — 100% of criticals intact
The 7-stage pipeline: dedup → slot fusion → delta encoding → causal pruning → utility scoring → shadow selection → adaptive budgeting.
Install
pip install dspm-memory
Requires Python 3.9+. Works with any OpenAI-compatible LLM provider.
Optional: semantic ranking
pip install "dspm-memory[semantic]"
Adds sentence-transformer-based query alignment for better non-critical patch ranking (downloads PyTorch and a small embedding model on first use). Without it, DSPM falls back to a neutral alignment score — criticals are fully protected either way.
Quickstart
from openai import OpenAI
from dspm import DSPMMemory
# Works with OpenAI, Groq, Together, Ollama, or any OpenAI-compatible endpoint
llm = OpenAI(
api_key="sk-...",
# base_url="https://api.groq.com/openai/v1" # uncomment for Groq
)
# budget=60 so compression is visible even in this small example;
# use 200-300 for real conversations
memory = DSPMMemory(budget=60, llm_client=llm, model="gpt-4o-mini")
memory.add_turn("user", "Building a payment API. Hard rules: PCI-DSS compliant, max fee 0.5%, deadline Friday.")
memory.add_turn("assistant", "PCI-DSS needs tokenized card storage and quarterly ASV scans. Tracking the 0.5% fee cap.")
memory.add_turn("user", "Webhook timeout must be 30 seconds.")
memory.add_turn("assistant", "Webhook timeout set to 30 seconds with automatic retries.")
memory.add_turn("user", "Change the webhook timeout to 10 seconds instead — 30 is too slow.")
memory.add_turn("assistant", "Updated: webhook timeout is now 10 seconds; the 30s setting is superseded.")
context = memory.get_context(query="What are the hard requirements?")
print(context)
print(memory.stats)
Output:
[DEC] Track the 0.5% fee cap.
[CON] PCI-DSS requires tokenized card storage and ASV
[DEC] webhook timeout updated to 10 seconds
[CON] PCI-DSS compliant, max fee 0.5%, deadline Friday
{'turns': 6, 'total_patches': 4, 'critical_total': 4, 'critical_selected': 4, 'crr': 100, 'budget': 60, 'raw_tokens': 61, 'context_tokens': 56, 'trr': 9}
Note the revision: only the 10-second entry remains — the 30s setting is fully superseded. The fee cap and deadline survived compression, trr: 9 shows real token reduction even at this small scale, and crr: 100 throughout.
Persistence: Memory Across Chats and Sessions
Save the notebook when a chat ends, load it when the next one starts — Chat 2 remembers Chat 1, and revisions supersede old values across sessions:
# Chat 1 — Monday
memory = DSPMMemory(budget=250, llm_client=llm, model="gpt-4o-mini")
memory.add_turn("user", "Building a budgeting app. Hard rules: must work offline, deadline Oct 20.")
# ... chat ...
memory.save("user_dhruv.json")
# Chat 2 — Thursday, new process, fresh start
memory = DSPMMemory(budget=250, llm_client=llm, model="gpt-4o-mini")
memory.load("user_dhruv.json")
memory.get_context(query="What were my hard rules?")
# → [CON] must work offline
# → [CON] deadline Oct 20 ← recalled from Chat 1, zero API cost
# Revisions work across chats too:
memory.add_turn("user", "The deadline moved to November 5.")
# → Oct 20 is superseded. November 5 replaces it.
Save files are portable JSON, written atomically (a crash mid-save can't corrupt the notebook), and merge-on-load means saved revisions supersede stale values. One file per user = each person's long-term memory.
The Guarantee
[CON] and [DEC] patches are structurally protected:
- Never dropped by deduplication, fusion, or pruning
- Under budget pressure, payloads are trimmed numbers-first — and units bind to their numbers (
30 secondsstays30 seconds, never just30) - When a constraint is revised mid-conversation (e.g. TTL 60s → 300s), the new value supersedes the old — including cross-type revisions (a constraint revised by a decision) and terse patches
- A critical is only dropped as a last resort: every critical already at its 2-word floor and budget still cannot hold them
- Transient API failures (rate limits, 5xx) are retried automatically with backoff — three attempts before the error surfaces
Ablation result: Removing the shadow-selection mechanism collapses CRR from 100% to 37.9%, isolating the guarantee to a single identifiable component.
Results
Tested across 7 domains × 40 turns each:
| Budget | Tokens Used | TRR | CRR |
|---|---|---|---|
| 150 | 150 | 66.8% | 100% |
| 250 | 249 | 82.8% | 100% |
| 400 | 395 | 72.4% | 100% |
CRR = Critical Retention Rate. TRR = Token Reduction Ratio.
Provenance: measured on 7 hand-authored 40-turn technical dialogues (API design, ML ops, IoT, security IR, supply chain, clinical workflow, project management); extraction and judging by GLM via OpenRouter (the since-renamed "ox-alpha" model). Numbers vary by extraction model and domain — check your own with memory.stats.
Supported Providers
# OpenAI
llm = OpenAI(api_key="sk-...")
# Groq (free tier available)
llm = OpenAI(base_url="https://api.groq.com/openai/v1", api_key="gsk-...")
# Together AI
llm = OpenAI(base_url="https://api.together.xyz/v1", api_key="...")
# Ollama (local, no key needed)
llm = OpenAI(base_url="http://localhost:11434/v1", api_key="ollama")
API Reference
DSPMMemory(budget, llm_client, model)
| Parameter | Type | Default | Description |
|---|---|---|---|
budget |
int | 250 | Maximum tokens in the compressed output |
llm_client |
OpenAI | None | Any OpenAI-compatible client |
model |
str | "gpt-4o-mini" |
Model used for patch extraction |
The budget can be changed at any time:
memory.budget = 100 # takes effect on the next get_context() call
memory.set_budget(100) # explicit alternative — same effect
Methods
| Method | Description |
|---|---|
add_turn(role, text) |
Add a conversation turn. Returns extracted patches. |
get_context(query="") |
Returns compressed context string, ready for your prompt. |
set_budget(budget) |
Update the token budget. Takes effect on the next get_context() call. |
save(path) |
Persist the memory notebook to JSON (atomic write). |
load(path) |
Load a notebook into memory (merge semantics, supersession on load). |
reset() |
Clear all memory and start fresh. |
Properties
| Property | Description |
|---|---|
memory.stats |
Dict with token counts, patch counts, active budget, CRR |
memory.critical_patches |
List of all critical patches currently in memory |
memory.all_patches |
List of every patch in memory |
Paper
DSPM: A Critical-Retention Approach to Long-Context Memory Compression for LLM Conversations
Dhruv Dubey, 2026
Zenodo: 10.5281/zenodo.19438636
Version History
| Version | Changes |
|---|---|
| 0.1.7 | README-first release (0.1.6's PyPI page was frozen pre-update — built after README finalization this time). Compound units ("per minute") and key constraint nouns ("timeout", "limit") survive trimming as atomic number-spans. "now"/"moved" recognized as revision verbs, and the extractor preserves revision verbs in payloads (real-model quickstart caught a stale-value contradiction). Claims scoped to benchmarks; retry behavior accurately described. Test scripts moved to scripts/ with exposed API key revoked and stripped. semantic extra documented. Python 3.13 classifier; status → Beta. 3 new tests (27 total). |
| 0.1.6 | Cross-type revision supersession fixed: revisions sharing only 3 content words previously survived as contradictions (found in external review). Units now bind to their numbers during trimming (30 seconds, never bare 30). Automatic retry with backoff on rate limits and transient 5xx errors in add_turn(). Homepage added to PyPI metadata. 5 regression tests (24 total). |
| 0.1.5 | Budget fix: changing memory.budget was silently ignored — the engine kept an independent budget copy. Now synced on every get_context(). New set_budget() method. stats now measures the actual joined context and reports the active budget. 2 regression tests (19 total). |
| 0.1.4 | Persistence: memory.save() / memory.load() — cross-session long-term memory as portable JSON. Atomic writes, merge-on-load, revisions supersede stale values. 8 new tests (17 total). |
| 0.1.3 | Revision supersession fix: stale same-type criticals now removed when superseded. Robust normalized content-word matching. |
| 0.1.2 | Fixed critical-patch ID collisions (CRR 36% → 100% in 18-turn live test). Budget enforced on joined context string. |
| 0.1.1 | Fixed T4 dropping criticals with dependencies. Fixed T3 payload mangling. Fixed extractor schema mismatch. |
| 0.1.0 | Initial release. |
License
MIT © 2026 Dhruv Dubey
Metadata
Release files for dspm-memory 0.1.8
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| dspm_memory-0.1.8.tar.gz | 27.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| dspm_memory-0.1.8-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 49.2 kB
Release files / dspm_memory-0.1.8.tar.gz
| Download URL | dspm_memory-0.1.8.tar.gz |
|---|---|
| Size | 27.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
f5d257d9777436892edfe65ff6905bdc29617bfbabb7984a32d095418638db3b
|
|
BLAKE2b-256 checksum How to use checksums |
64b88a1ddf2a622c1348dedb5eabef645bee28193a405454c32f26d62885d9f8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.11.15
|
Release files / dspm_memory-0.1.8-py3-none-any.whl
| Download URL | dspm_memory-0.1.8-py3-none-any.whl |
|---|---|
| Size | 21.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
3fd895e961cb2696baadf469fd0ce3153a0a94e348c4a3afad5c88a148e7d109
|
|
BLAKE2b-256 checksum How to use checksums |
cc74ac96ce87b5679876f7c2a5fb7e2f0fc65c1585f070d1eafe80a1621a51e4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.11.15
|