Skip to main content

DSPM Memory

Compress multi-turn LLM conversations by 80%+ while guaranteeing every constraint and decision survives.

PyPI version License: MIT Python 3.9+


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.


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
)

memory = DSPMMemory(budget=250, llm_client=llm, model="gpt-4o-mini")

memory.add_turn("user", "Build a REST API. Must use PostgreSQL, JWT auth, deadline is Friday.")
memory.add_turn("assistant", "PostgreSQL with SQLAlchemy, JWT via python-jose. Access tokens 15 min.")
memory.add_turn("user", "All PII must be encrypted at rest with AES-256. No exceptions.")
memory.add_turn("assistant", "AES-256 at rest for all PII fields, keys in AWS KMS with quarterly rotation.")

context = memory.get_context(query="What are the hard requirements?")
print(context)
print(memory.stats)

Output:

[CON] Stack: PostgreSQL, JWT auth. Deadline Friday.
[CON] Access tokens 15 minutes.
[CON] All PII must be encrypted at rest with AES-256. No exceptions.
[CON] AES-256 at rest; keys in AWS KMS, quarterly rotation.

Every constraint is present. Every time.


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 — thresholds, versions, and units survive longest
  • When a constraint is revised mid-conversation (e.g. TTL 60s → 300s), the new value supersedes the old
  • A critical is only dropped as a last resort: every critical already at its 2-word floor and budget still cannot hold them

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.


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.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.6

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for dspm-memory 0.1.6
File Size Uploaded
dspm_memory-0.1.6.tar.gz 24.4 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for dspm-memory 0.1.6
File Interpreter ABI Platform
dspm_memory-0.1.6-py3-none-any.whl Python 3 none any Details

Total release size: 44.1 kB

Release files / dspm_memory-0.1.6.tar.gz

Download URL dspm_memory-0.1.6.tar.gz
Size 24.4 kB
Tags Source
SHA-256 checksum
How to use checksums
a8bd012d0ff2626ae6eb098bcf3fc16150161244cc40aeeee6846d7b0a0ba8f4
BLAKE2b-256 checksum
How to use checksums
3e58ac3da7f2e1d6b5bed227248b340ec6d7ad07143e0b240ac091b65964b640
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.6-py3-none-any.whl

Download URL dspm_memory-0.1.6-py3-none-any.whl
Size 19.7 kB
Tags Python 3
SHA-256 checksum
How to use checksums
2e98ddae8d36c143be39c42454485d359ec9875807053ea17fd298605b051b3a
BLAKE2b-256 checksum
How to use checksums
fdd6525089d406bdc74003f9e1f23d06e723eb4abc961679f746b62bf201dedb
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.15

Release history Release notifications | RSS feed

0.1.8

2 release files

0.1.7

2 release files

This release

0.1.6 This release

2 release files

0.1.5

2 release files

0.1.4

2 release files

0.1.3

2 release files

0.1.2

2 release files

0.1.1

2 release files

0.1.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page