Skip to content

fix(mcp_server): reset chromadb System cache on staleness reconnect (#2002) - #2026

Merged
2 commits merged into
MemPalace:developfrom
colorpanda82:fix/2002-chroma-cache-reset-on-reconnect
Aug 12, 2026
Merged

fix(mcp_server): reset chromadb System cache on staleness reconnect (#2002)#2026
2 commits merged into
MemPalace:developfrom
colorpanda82:fix/2002-chroma-cache-reset-on-reconnect

Conversation

@colorpanda82

Copy link
Copy Markdown
Contributor

Fixes #2002. Part of the #1963 concurrent-writer epic; independent of the #1976 daemon/bridge work.

Thanks @fatkobra for the go-ahead on a focused PR.

Problem

_get_client() detects a peer writer's inode/mtime change on chroma.sqlite3 and rebuilds the client via ChromaBackend.make_client(). But chromadb caches its System, and therefore the live in-memory PersistentLocalHnswSegment, keyed by path. So the rebuilt "fresh" client is handed back the same stale segment, which on its next _persist() writes its outdated index over the on-disk one, destroying records other writers had already indexed.

Observed on a live multi-writer palace: the persisted HNSW count went backwards (4 to 3). Over one day, sqlite grew by ~107 drawers while the index grew by ~40, so roughly 63% of new drawers never reached the index and vector search silently degraded to BM25.

Fix

Call the existing _force_chroma_cache_reset() on the staleness path, before make_client(), so chromadb rebuilds the segment from the on-disk state. It is guarded by the existing inode_changed or mtime_changed check, so first-open is unaffected.

_force_chroma_cache_reset() already closes the client, drops the backend's per-palace cache, and calls SharedSystemClient.clear_system_cache(). It simply was not being called on this path (it is already used in the missing-DB and repair paths). clear_system_cache() is the accepted workaround for this class of path-keyed staleness in chromadb (cf. chroma-core/chroma#2536, chroma-core/chroma#5843).

Test

test_get_client_resets_chroma_system_cache_on_reconnect asserts the reset runs before make_client() on an mtime reconnect. It fails on develop (order is ['make_client']) and passes with the fix (['reset', 'make_client']). The existing test_get_client_rearms_quarantine_on_reconnect (#1573) and the inode/mtime invalidation tests still pass.

Scope

This removes one concrete data-loss path on the staleness reconnect. It does not make multi-writer safe on its own; chromadb PersistentClient is single-process by design, which is what #1976 / #1270 address at the architecture level.

Manual multi-process reproduction

With a long-lived writer plus a genuinely separate process:

  1. long-lived writer indexes 3 records,
  2. separate process indexes 2,
  3. long-lived writer indexes 1.

Without the fix the persisted index count regresses; with the fix it ends at 6 (monotonic). Happy to add this as a @pytest.mark.slow integration test if you'd prefer it in-tree; I kept it out of the suite to keep CI deterministic.

…emPalace#2002)

_get_client() detects a peer writer's inode/mtime change and rebuilds the
client via ChromaBackend.make_client(), but chromadb caches its System (and
the live in-memory HNSW segment) keyed by path. The rebuilt client is handed
back the same stale segment, which on its next _persist() overwrites the
on-disk index, destroying records other writers had already indexed. Observed
in a live multi-writer palace: the persisted index count went backwards (4 to 3).

Call the existing _force_chroma_cache_reset() on the staleness path, before
make_client(), so chromadb rebuilds the segment from the on-disk state. The
call is guarded by the existing inode_changed/mtime_changed check, so it has
no effect on first-open.

Adds test_get_client_resets_chroma_system_cache_on_reconnect, which asserts the
reset runs before make_client on an mtime reconnect (fails without the fix).

Refs MemPalace#1963.

@fatkobra fatkobra left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reviewed the reconnect path, _force_chroma_cache_reset(), and the new regression test.

The fix is correctly placed: the full Chroma system-cache reset runs only after an inode/mtime change is detected and before ChromaBackend.make_client() reconstructs the client. First-open behavior is unchanged.

The test is also meaningful: it uses a real palace, primes the client cache, simulates a peer-write mtime change, and verifies the required reset -> make_client ordering.

This is a focused and appropriate fix for #2002. It does not claim to make multi-writer Chroma safe generally, and it complements the daemon/single-writer work in #1976 rather than duplicating it.

Approved from my side.

Non-blocking notes:

  • A future test could assert the internal SharedSystemClient.clear_system_cache() call directly.
  • ChromaBackend._client() has its own inode/mtime reopen path; it may deserve a separate investigation for long-lived non-MCP callers rather than expanding this PR.
  • GitHub flags hidden/bidirectional text in the touched files; the added hunk looks normal, but a control-character scan would be useful.

@colorpanda82

Copy link
Copy Markdown
Contributor Author

Following up on the three non-blocking notes.

Hidden/bidirectional text. Ran a byte-level scan over the full diff. No bidi controls (U+202A-202E, U+2066-2069), no zero-width or BOM (U+200B-200D, U+2060, U+FEFF), and no other invisible/format or non-ASCII control characters, in the added lines or the surrounding context. The only non-ASCII codepoint in the whole patch is a single U+2014 em dash in a test docstring (tests/test_mcp_server.py), which is neither hidden nor bidirectional, so the flag does not correspond to anything in this patch. Glad to swap that em dash for ASCII if you'd prefer the touched file stay pure ASCII.

Direct assertion of the cache reset. Kept the test at the behavioral reset -> make_client ordering level to avoid coupling to chromadb internals, but I agree an explicit assertion on SharedSystemClient.clear_system_cache() is a reasonable strengthening. Happy to add it here, or fold it into the follow-up test mentioned below, whichever you prefer.

ChromaBackend._client() reopen path. Confirmed on develop: _client() (chroma.py L2044-2064) has its own inode/mtime reopen that reconstructs PersistentClient without a system-cache reset (_force_chroma_cache_reset does not appear in chroma.py), so long-lived non-MCP callers holding a backend across a rebuild would be exposed to the same #2002 class. Filed #2028 to track it separately rather than expand this PR.

@fatkobra

Copy link
Copy Markdown
Contributor

Thanks for the thorough follow-up.

This addresses all three of my non-blocking notes:

No further changes requested from my side. My approval stands.

The MemPalace#2002 staleness reconnect makes _get_client() call
_force_chroma_cache_reset(), which clears chromadb's path-keyed
SharedSystemClient cache. Two TestDeleteBySource tests grabbed a
closets collection handle *before* calling tool_delete_by_source and
then asserted on it afterwards, by which point the reset had dropped
the Rust binding underneath the handle (AttributeError:
'RustBindingsAPI' object has no attribute 'bindings').

Re-acquire the closets collection after the tool call in both tests.
Production callers already re-acquire fresh handles per call, so this
is a test-lifetime issue, not a regression in the fix.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Gg6g5efZ1rNbBTHqGz2Tjw
@igorls igorls closed this pull request by merging all changes into MemPalace:develop in 83d8cbd Aug 12, 2026
pull Bot pushed a commit to nenyatech-mirror/mempalace that referenced this pull request Aug 12, 2026
…n inode/mtime reopen

_client() reconstructs PersistentClient on an inode/mtime change but did not drop chromadb's process-global SharedSystemClient cache first, so the rebuilt client reused the stale path-keyed System (and its in-memory HNSW segment) and could persist an outdated index over on-disk changes -- the same class as MemPalace#2002, reached via _client() instead of _get_client.

Add SharedSystemClient.clear_system_cache() to the external-change branch of _client(), mirroring mcp_server._force_chroma_cache_reset (MemPalace#2026) and repair._close_chroma_handles. Backend-level regression test asserts the reset fires on the change reopen, strictly before the reconstruct, and not on first open (chroma-core/chroma#2536, #5843).

Fixes MemPalace#2028.
pull Bot pushed a commit to nenyatech-mirror/mempalace that referenced this pull request Aug 12, 2026
offendingcommit added a commit to offendingcommit/mempalace that referenced this pull request Aug 16, 2026
* fix(mcp_server): reset chromadb System cache on staleness reconnect (MemPalace#2002)

_get_client() detects a peer writer's inode/mtime change and rebuilds the
client via ChromaBackend.make_client(), but chromadb caches its System (and
the live in-memory HNSW segment) keyed by path. The rebuilt client is handed
back the same stale segment, which on its next _persist() overwrites the
on-disk index, destroying records other writers had already indexed. Observed
in a live multi-writer palace: the persisted index count went backwards (4 to 3).

Call the existing _force_chroma_cache_reset() on the staleness path, before
make_client(), so chromadb rebuilds the segment from the on-disk state. The
call is guarded by the existing inode_changed/mtime_changed check, so it has
no effect on first-open.

Adds test_get_client_resets_chroma_system_cache_on_reconnect, which asserts the
reset runs before make_client on an mtime reconnect (fails without the fix).

Refs MemPalace#1963.

* fix(chroma): reset chromadb System cache in ChromaBackend._client() on inode/mtime reopen

_client() reconstructs PersistentClient on an inode/mtime change but did not drop chromadb's process-global SharedSystemClient cache first, so the rebuilt client reused the stale path-keyed System (and its in-memory HNSW segment) and could persist an outdated index over on-disk changes -- the same class as MemPalace#2002, reached via _client() instead of _get_client.

Add SharedSystemClient.clear_system_cache() to the external-change branch of _client(), mirroring mcp_server._force_chroma_cache_reset (MemPalace#2026) and repair._close_chroma_handles. Backend-level regression test asserts the reset fires on the change reopen, strictly before the reconstruct, and not on first open (chroma-core/chroma#2536, #5843).

Fixes MemPalace#2028.

* test(chroma): reformat test_backends.py to satisfy ruff format

Two monkeypatch.setattr calls were wrapped across lines that fit within
the line length; ruff format --check flagged them. Formatter-only, no
behavior change.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Gg6g5efZ1rNbBTHqGz2Tjw

* test(mcp_server): re-acquire closets handle after delete_by_source

The MemPalace#2002 staleness reconnect makes _get_client() call
_force_chroma_cache_reset(), which clears chromadb's path-keyed
SharedSystemClient cache. Two TestDeleteBySource tests grabbed a
closets collection handle *before* calling tool_delete_by_source and
then asserted on it afterwards, by which point the reset had dropped
the Rust binding underneath the handle (AttributeError:
'RustBindingsAPI' object has no attribute 'bindings').

Re-acquire the closets collection after the tool call in both tests.
Production callers already re-acquire fresh handles per call, so this
is a test-lifetime issue, not a regression in the fix.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Gg6g5efZ1rNbBTHqGz2Tjw

* fix(miner): close four re-mine safety gaps in process_file

MemPalace#21 (CRITICAL, data-loss): multi-batch re-mine had no completion marker.
A mid-file crash after batch 1 committed but before a later batch left
permanently silent partial data -- the surviving drawers shared the
file's unchanged on-disk mtime, so file_already_mined() treated the
file as fully mined forever. Every chunk now carries chunk_total, and
file_already_mined() verifies a matching-mtime group's drawer count
reaches chunk_total before reporting True. Drawers with no chunk_total
(legacy rows, single-shot add_drawer()) are trusted as before.

MemPalace#22 (HIGH, correctness/TOCTOU): source_mtime was captured via a fresh
os.path.getmtime() well after content was read, chunked, and
room-detected. A file appended to in that window got its new drawers
stamped with an mtime that already matched the (now newer) on-disk
state, so the appended tail was silently, permanently skipped on every
future mine. _read_text_no_follow now returns (content, mtime) from
the same fstat() that validates the file; process_file threads that
single value through instead of re-stating.

MemPalace#23 (HIGH, silent-failure): a failed stale-drawer purge was swallowed
to a debug log and mining proceeded anyway, silently producing
duplicate or orphaned drawers. A purge failure now aborts this file's
mine attempt (old drawers' stored mtime is untouched, so the next mine
still sees a mismatch and retries) and prints a visible warning,
matching every other degraded path in this module.

MemPalace#24 (LOW, data-loss): the old-drawer delete ran unconditionally, but
the closet purge+rebuild only ran when drawers_added > 0 -- a file
whose chunks all landed below min_chunk_size after boundary-splitting
lost its drawers but kept stale closets pointing at now-deleted IDs.
purge_file_closets now runs whenever the delete-and-rebuild cycle
does, regardless of the new chunk count; only the rebuild itself stays
conditional.

169 tests pass across test_miner.py/test_convo_miner*.py/test_palace.py/
test_hallways.py/test_format_miner.py/test_miner_fts5_validation.py, no
regressions. Full suite: 1 unrelated pre-existing flake in
test_mcp_server.py (module-global peer-writer-lock state leaking across
test files in full-suite ordering -- passes standalone and as a full
file; the diff here never touches mcp_server.py).

* docs(changelog): tighten 3.7.0 notes to match prior release style

Rewrite the 3.7.0 section as short, scannable bullets like 3.6.0 —
bold lead, one or two sentences per item, thematically grouped fixes —
instead of multi-paragraph issue writeups.

* fix(tests): harden hybrid search against empty Windows Chroma reads

Windows CI intermittently returns zero hybrid hits right after a fast
seed write (same class as "Nothing found on disk" on tiny collections).
Close the palace client after seeding so the next open re-reads flushed
segments, retry search once if empty, and assert non-empty results with
a clear message instead of IndexError.

* fix(ingest): never block on a non-regular file (MemPalace#2221)

`os.walk` and `glob` list a FIFO, a socket and a device node as ordinary
filenames, and MemPalace decides what to read from the suffix. Opening a
FIFO for reading parks in the kernel until a writer appears, so a named
pipe called `notes.md` in a mined directory wedged `mempalace mine`
forever — no output, no error, no progress. `mine --mode convos`,
`sweep`, `init`, `compress` and `split` blocked the same way.

Two shapes are at fault.

Four helpers already refused non-regular files with `fstat` + `S_ISREG`,
but the check sat *after* a blocking `os.open`, so it could never run.
Adding `O_NONBLOCK` to those opens makes the existing type check
reachable: the open returns immediately and the file mode decides, with
no errno guesswork. A FIFO that does have a live writer is refused just
the same. Linux open(2) states the flag has no effect on regular files;
the one exception is a write lease, where a non-blocking open fails
EAGAIN instead of waiting out lease-break-time. Leases are granted on
regular files only, so that branch re-checks the type and then opens
without the flag rather than silently dropping a file that used to be
mined.

The rest guard with `exists()`, which is true for a pipe, and then open
anyway. Those become type checks: a discovery walk drops non-regular
entries before any reader sees them, and a fixed-name read decides with
`is_file()` instead. `scan_project` and `scan_convos` already stat every
candidate for the size limit, so the type check costs no extra syscall.
Where the gate replaced an `open` that sat inside a `try`, it goes in
the same `try`: `is_file()` raises `PermissionError` on a directory
without `x`, which that handler already absorbed.

O_NONBLOCK: miner._read_text_no_follow, convo_miner._is_regular_source_file,
normalize._read_transcript_file, repair._open_regular_file_no_follow.
Type gate: miner.scan_project, miner.load_config, convo_miner.scan_convos,
sweeper.parse_claude_jsonl, sweeper.sweep_directory,
entity_detector.detect_entities, cli._gather_origin_samples,
cli._ensure_mempalace_files_gitignored, cli.cmd_compress, cli.cmd_init,
project_scanner._collect_manifest_names,
project_scanner._parse_gradle_root_project_name,
room_detector_local.detect_rooms_local, llm_refine.collect_corpus_text,
split_mega_files.main, hook_shell.count_human_messages.

* docs(changelog): note the non-regular-file ingest hang (MemPalace#2221)

* fix: 3.7.1 critical patch — re-mine honesty + SIGTERM lock release

Stack the post-3.7.0 hang and silent-skip fixes for a fast patch release:

- Keep MemPalace#2223 (non-regular file hang) and MemPalace#2088 (chunk_total / same-fstat
  mtime / purge abort / closet purge) as the base.
- On multi-batch upsert failure, delete partial drawers and closets for
  that source before re-raising so the next mine retries (MemPalace#2122, MemPalace#2151).
- Install SIGTERM/SIGHUP handlers in mcp_server.main so atexit can release
  the palace writer lease (MemPalace#2205).
- Adapt non-regular-file tests to the (content, mtime) read return type.

* fix(convo): stamp chunk_total and clean partial multi-batch mines (MemPalace#2183)

Port project-miner re-mine honesty to conversation ingest so an interrupted
transcript mine cannot permanently skip missing exchanges:

- stamp chunk_total on every convo drawer in the pass
- delete partial drawers for the source/extract_mode on upsert failure
- teach prefetch_mined_set the same completeness rule as file_already_mined

* test(repair): release SharedSystemClient after seeding for Windows rename

In-place rebuild tests archive the palace directory after _seed_palace.
backend.close() alone left chromadb's path-keyed System holding files open
on Windows (WinError 5), so rebuild_from_sqlite aborted before the mocked
upsert path and test_rebuild_from_sqlite_raises_on_upsert_failure never
raised RebuildPartialError. Clear the shared cache and GC after close.

* chore(release): 3.7.1

Bump package, plugins, lock, OpenClaw skill, and README badge to 3.7.1.
Fold Unreleased integrity notes into the 3.7.1 changelog: FIFO ingest hang,
project and convo re-mine completeness, chromadb System-cache rewind, and
SIGTERM/SIGHUP lease release.

* fix(release): preserve fork behavior after upstream sync

Keep normalized and subject-routed ingestion compatible with the 3.7.1
re-mine safeguards. Distinguish local Chroma writes from peer changes so
cache refreshes release stale clients without invalidating live handles.

* fix(ci): record HTTP status before response delivery

* fix(ci): normalize SDK HTTP status values

* test(ci): make conversation fixtures portable

---------

Co-authored-by: Cristian Deheleanu <160292664+colorpanda82@users.noreply.github.com>
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
Co-authored-by: KeilerHirsch <KeilerHirsch@users.noreply.github.com>
Co-authored-by: Igor Lins e Silva <4753812+igorls@users.noreply.github.com>
Co-authored-by: Michael Valentsev <michael@valentsev.ru>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

_get_client() reconnect leaves the stale Chroma segment live: a peer writer's persist destroys already-indexed records (index count goes backwards)

3 participants