Skip to main content

CogniKernel — persistent, structured project memory for Claude Code and Codex

Project description

CogniKernel

Persistent, structured project memory for Claude Code — and Codex. CogniKernel watches a coding session through its hook surfaces, extracts the decisions, constraints, and abandoned approaches worth keeping, consolidates them into an event-sourced store, and injects them back as a compact context block the next time you work — so the agent stops re-deciding what you already decided. The store is keyed on the project path, so memory made in one tool travels to the other.

It is not a vector-database wrapper. It is an event-sourced log of typed memory with lexical-primary retrieval, write-time consolidation, and a fail-open reliability spine designed never to break your session.

And there is no LLM in the loop. Most memory tools work by sending your transcripts to a generative model to "summarize what mattered" — another API key, per-session token cost, added latency, and your session content leaving the machine. CogniKernel treats extraction as classification, not generation: a deterministic sanitize → classify → consolidate pipeline, with two small fine-tuned encoder models (~130 MB ONNX, run locally on CPU in milliseconds) scoring salience and detecting when a new decision supersedes an old one. No API calls, no tokens billed, nothing leaves your machine. The only LLM involved is the coding agent you already run — CogniKernel makes it remember.


The fine-tuned models

Two small encoder models do the actual thinking behind CogniKernel — not a keyword list, not an LLM call:

Model What it decides Why it's an encoder, not an LLM
salience_v2 For every sentence in a session transcript: is this worth keeping, and what type — DECISION, CONSTRAINT_HARD/SOFT, APPROACH_ABANDONED_DO_NOT_RETRY, or noise? This is the model that decides what gets remembered. A fine-tuned SetFit classifier (bge-small backbone, ~130 MB ONNX) runs this on CPU in milliseconds. An LLM doing the same job costs an API call and real latency on every session — CogniKernel spends neither.
supersession_xenc Does this new fact replace an existing one (a decision changed, a constraint was relaxed), or does it just restate what's already stored? This is the model behind latest-wins consolidation. Same constraint: this fires on every capture. A cross-encoder scores it locally in milliseconds instead of round-tripping to a model API.

Validated on held-out data, not vibes. Each model is scored against a frozen evaluation set mined from real captured project stores that the training data never sees — retraining on the eval set is a standing rule violation, not a shortcut we occasionally allow. Because the underlying task (is this sentence a decision, or just description?) is itself genuinely ambiguous at the margin, we also measure how much two independent human labelers agree with each other on the same sentences, and hold the model to that ceiling rather than to a naive 100%. The model's minority classes (rare event types with few historical examples) are tracked and grown deliberately, because a thin eval slice can hide the exact class that most needs work.

They're optional but on by default — here's the fail-open contract. cognikernel init writes a per-project config that already selects the fine-tuned path (extractor = "v2-broad", cross_encoder_supersession = true). What's missing after init is the ~270 MB of model weights themselves — that's what cognikernel install-heads fetches. Until you run it, CogniKernel works, but degrades both decisions to a deterministic keyword/lexical fallback (weaker: it can miss decisions phrased outside its trigger vocabulary, and it can't catch a paraphrased correction lexical matching misses) — cognikernel doctor names this state explicitly rather than leaving it invisible (see below). Nothing breaks either way — that's the fail-open design — but the fine-tuned path is the better-quality one, and the one new projects are configured to use by default. See Quickstart to install them.


The memory loop

Everything CogniKernel does is one loop: observe → extract → consolidate → store → retrieve → assemble → inject. The left rail captures; the right rail recalls; the spine underneath keeps both honest.

   +========================= CLAUDE CODE SESSION =========================+
   | working memory  -  the context window the agent reasons in            |
   +--------------+-----------------------------------------+--------------+
                  | Stop hook captures transcript           | inject block
                  v                                         ^
   +--------------+---------------+         +---------------+--------------+
   | [1] EXTRACTION PIPELINE      |         | [6] COMPRESSION + INJECTION  |
   |   sanitize -> classify ->    |         |   authority-weighted budget, |
   |   salience (ONNX) ->         |         |   drop-to-fit (keep every    |
   |   decision-key + contracts   |         |   constraint) -> block       |
   +--------------+---------------+         +---------------+--------------+
                  | enqueue                                 | rank + fit
                  v                                         ^
   +--------------+---------------+         +---------------+--------------+
   | [2] WORKER + CONSOLIDATION   |         | [5] RETRIEVAL                |
   |   claim -> delta-merge ->    |         |   FTS5 BM25 + dense -> RRF   |
   |   supersede (latest-wins) -> |         |   prohibition_search (K1)    |
   |   project   (idempotent)     |         |   skeleton graph (PageRank)  |
   +--------------+---------------+         +---------------+--------------+
                  | persist (atomic)                        | recall
                  v                                         ^
   +--------------+-----------------------------------------+--------------+
   | [3] EVENT-SOURCED STORE   *   SQLite (WAL)                            |
   | typed events | evidence | provenance | FTS5 | embeddings | ledger     |
   +----------------------------------------------------------------------+
   | [4] RELIABILITY SPINE   atomic migrations | idempotent replay |       |
   | doctor --strict | fail-open hooks | import-linter | CI gate           |
   +----------------------------------------------------------------------+

The four hook surfaces

CogniKernel attaches to a session at four points. Each is fail-open — if memory is unavailable or errors, the hook logs at WARNING, returns cleanly, and the session continues. salience_v2 and supersession_xenc (above) do the classification behind Capture and Session block; nothing here is a hardcoded keyword list unless the fallback path is active.

Surface Hook Authority What it does Why you care
Session block SessionStart advisory injects the canonical decisions/constraints/skeleton block a new session already knows what the last one decided — no "let me re-read the codebase to remember where we were"
CK-1 recall UserPromptSubmit advisory surfaces prompt-relevant memory, dual-evidence gated, dedup'd via render ledger ask about a subsystem and the relevant prior decision rides in with your prompt, unasked
Read/Edit gate PreToolUse hard / JIT read-efficiency gate on Read/Grep; just-in-time prohibition surfacing on Write/Edit (K2) the agent gets warned at the moment it's about to violate a past decision, not three files later when you notice
Capture Stop side-effect extracts and persists decisions — you never write memory to CLAUDE.md by hand you never write down what you decided; the next session already has it

What gets remembered

Memory is typed, not free-text chunks. The type drives ranking, rendering, and supersession:

  • DECISION — a choice that was made ("use Redis for the rate limiter")
  • CONSTRAINT_HARD / CONSTRAINT_SOFT — rules, graded by deontic force
  • APPROACH_ABANDONED_DO_NOT_RETRY — a dead end, kept in the graveyard so the agent doesn't re-attempt it
  • conventions, config facts, schema decisions (canonical role keys)

A decision key lets a later restatement supersede an earlier one (latest-wins), so the store self-consolidates instead of accumulating contradictions. An optional cross-encoder adds semantic supersession when the encoder backend is installed.


Retrieval

Lexical-primary, with dense as a fused signal — never pure vector:

  • Hybrid core — FTS5 BM25 ∪ optional dense embeddings → Reciprocal Rank Fusion
  • prohibition_search — a type-restricted lexical pool so a "do not do X" rule can't be crowded out by topically-similar prose at the moment you're about to do X
  • Skeleton — an AST symbol graph ranked by PageRank; find_related unions semantic (embedding) neighbours with import-graph-adjacent events to surface what a change touches (the semantic axis needs the embedding extra)
  • Golden-record consolidation at read — latest-wins reconciliation so recall returns one coherent answer, not a pile of revisions

What it saves you

Benchmarked in a three-arm comparison — CogniKernel vs flat curated notes vs no memory — with real agent sessions across four multi-session projects:

  • File reads: the universal win. The CogniKernel arm made the fewest file reads in every project — typically 2–4× fewer (23 vs 63, 16 vs 47/53, 40 vs 89/83), and in the best case 3 reads vs 29 because the injected block + AST skeleton carried the whole repo's shape. Fewer reads means fewer tool round-trips and more of the context window left for actual work — your session gets longer before compaction, not just cheaper.
  • Tokens: up to ~20% cheaper where memory matters. Price-weighted token cost (cache-write 1.25×, cache-read 0.1×, output 5×) came out 18–23% lower on projects with evolving decisions and cross-file dependencies — and roughly a wash on small implementation-heavy projects where the code itself is cheap to re-read. We publish the honest number, not the raw-token one (raw sums look ~30–40% better, but ~95% of any session's bill is discounted cache-read).
  • Recall instead of re-derivation. Where memory earns its keep is projects whose state is too large, too evolving, or too long-lived to re-derive cheaply: the agent starts already knowing the decisions, constraints, and dead ends, instead of spending the first quarter of the session rediscovering them.

Reliability — the spine

The system is designed to degrade legibly, never silently:

  • Atomic migrations — each numbered migration applies its body + version bump in one transaction; safe to crash mid-script
  • Idempotent replay — a re-run worker job can't double-count or drift decay (evidence-provenance guard)
  • Fail-open hooks — every surface swallows its own failure and logs at WARNING; silence never reads as success
  • cognikernel doctor --strict — per-subsystem health (schema, FTS5, embeddings, symbols, worker queue); non-zero exit when degraded
  • Architecture enforcementimport-linter layered contracts, guarded by a meta-test so a typo can't silently disable them
  • CI promotion gate — lint + full suite (incl. tests/reliability/ failure-injection) on every PR; see CONTRIBUTING.md

Cross-platform (Codex)

The store is platform-neutral — one SQLite DB per logical project, so Claude Code and Codex working in the same directory share one memory. Project resolution is alias-aware: C:\repo and /mnt/c/repo resolve to the same store, so memory follows the checkout across Windows, WSL, and native mounts; for genuinely different checkout paths, an opt-in project_identity key in .cognikernel/config.toml pins them to one shared store. Codex reads memory through the registered MCP server (get_session_state / recall); the capture direction is pull-based, because Codex has no Stop-hook equivalent:

  • cognikernel codex-sync <project> scans ~/.codex/sessions for rollouts whose recorded cwd maps to the project and captures the delta through the same extraction pipeline (a rollout→transcript adapter is the only Codex-specific code; delta/dedup/idempotency are shared and unchanged).
  • Automatic at the handoff — Claude's SessionStart drains pending Codex rollouts before building the block, and the MCP server's queue drainer pulls new rollouts each cycle, so a live Claude session picks up Codex-side decisions without waiting for the next session; on the Codex side, init writes an AGENTS.md instruction + a ck-sync skill so Codex pulls at session start.
  • init provisions both.mcp.json (Claude) and .codex/config.toml (with the server's cwd + project env pinned) + AGENTS.md (Codex), idempotently and without clobbering existing settings.
  • cognikernel doctor reports a codex health line (sessions dir + rollout count, or "nothing to sync" — Codex is optional, so its absence is healthy).

A decision made in Codex reaches the next Claude session's block, and vice versa. The action-point surfaces (CK-1, PreToolUse gate) are Claude-only — Codex has no per-prompt/per-tool hook — so on Codex the loop degrades to the shared block + MCP recall.


Interfaces

MCP tools (the session block is injected automatically; these are for targeted use): recall · find_related · skeleton · get_session_state

CLI:

  • cognikernel init <project> — register the project and install the session hooks
  • cognikernel doctor [--strict] <project> — subsystem health report
  • cognikernel codex-sync <project> — capture Codex CLI sessions for this project
  • cognikernel install-heads — install the trained encoder artifacts (salience + cross-encoder ONNX bodies): downloaded from the heads-v1 release and sha256-verified, or copied from a local models/ export when present
  • cognikernel show <project> / cognikernel reset <project> — inspect / clear stored memory

Quickstart

# 1. Install the package
uv sync                      # core (lexical-only)
uv sync --extra embedding    # + dense retrieval (fastembed + numpy) — recommended

# 2. Register the project (writes .mcp.json, hooks, and a per-project config that
#    already selects the fine-tuned extraction path — see below)
uv run cognikernel init .

# 3. Install the fine-tuned encoder heads — the models described above
uv run cognikernel install-heads     # ~270 MB, one-time, sha256-verified

# 4. Confirm everything is wired up
uv run cognikernel doctor .

Step 3 is the one easiest to skip and most worth not skipping: cognikernel init already configured this project to use salience_v2 and supersession_xenc (extractor = "v2-broad", cross_encoder_supersession = true in .cognikernel/config.toml) — install-heads is what actually supplies the model weights those settings call for. Skip it and CogniKernel still runs, just on the weaker deterministic fallback (fail-open, not a hard error). The download pulls from the heads-v1 release by default; pass --source <dir> to install from a local models/ export instead, or --no-download to skip the network fetch and rely on the fallback deliberately.

Step 4 actually tells you which path is active — cognikernel doctor's subsystem health block reports salience_head and supersession_head by name:

-- subsystem health -----------------------------------------
  [OK] salience_head    : installed, loads from ~/.cognikernel/models/salience_v2
  [OK] supersession_head: installed, loads from ~/.cognikernel/models/supersession_xenc

If you skipped install-heads (or ran uv sync without --extra embedding, which is where the ONNX runtime itself comes from), the same lines say not installed / onnxruntime/tokenizers not installed and name the exact command to fix it — this is always informational, never a doctor --strict failure, since the legacy fallback is a fully supported mode.

Then start a Claude Code session in the project — the memory block appears automatically at session start, and decisions are captured when the session ends.


Project layout

src/cognikernel/
  integration/   hooks, CLI, MCP server, session/worker orchestration
  extraction/    sanitize -> classify -> salience -> decision-key pipeline
  delta/         delta-merge + supersession (latest-wins; cross-encoder optional)
  retrieval/     hybrid BM25 + dense -> RRF
  storage/       event-sourced SQLite, FTS5, migrations, render ledger
  embedding/     optional dense vectors (fastembed)
  symbols/       AST skeleton + PageRank graph
  compression/   authority-weighted drop-to-fit budget
  injection/     block template assembly
  model.py       Event — the dependency-free domain primitive
tests/
  unit/          per-subsystem
  reliability/   crash-replay · worker-contention · corrupt-input injection

Status

Schema v18 (includes the Codex cross-platform capture). Architecture contracts: 3 kept / 0 broken. CI gate: lint + full suite on Ubuntu (3.11/3.12) and Windows. See CONTRIBUTING.md for the Definition of Done that gates every change.

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

cognikernel-0.1.0.tar.gz (838.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

cognikernel-0.1.0-py3-none-any.whl (296.6 kB view details)

Uploaded Python 3

File details

Details for the file cognikernel-0.1.0.tar.gz.

File metadata

  • Download URL: cognikernel-0.1.0.tar.gz
  • Upload date:
  • Size: 838.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.11.21 {"installer":{"name":"uv","version":"0.11.21","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":null,"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for cognikernel-0.1.0.tar.gz
Algorithm Hash digest
SHA256 6c6b93f03ecdefe581adcc6292b778ed98b20d5f8d91b0e0fde2ff7973a512b1
MD5 36be9a4048e14b9049083921c17f4330
BLAKE2b-256 5b25f5315d1398ca2883799de52db9e5b02e1474c87a073a796a4f1101bc3855

See more details on using hashes here.

File details

Details for the file cognikernel-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: cognikernel-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 296.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.11.21 {"installer":{"name":"uv","version":"0.11.21","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":null,"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for cognikernel-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 d17a4fa05b7efdabd66ba2d2db1e22d9e7937961912f23d40d1522901f4b8fd6
MD5 add1b82655b056dc3b0599d1d1663ad9
BLAKE2b-256 237a3d975bad39c64d9fcb82004c44f90e1dad6ac3818c2481f14e3e1226d422

See more details on using hashes here.

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