Files
hermes-agent/tests/test_cli_session_store_shared_registry.py
Teknium 876e444e4e fix(cli): open the session store through the state.db registry, not a bare SessionDB()
The classic CLI froze for ~0.7-2s between the banner and the first prompt. py-spy +
strace on real PTY startups showed the main/REPL threads inside
refuse_deleted_wal_generation -> _iter_proc_fd_targets: a second full SessionDB open.

_init_session_store built a bare SessionDB(); a moment later the goal/loop/heartbeat
managers acquired the same state.db through hermes_state_registry from the REPL thread,
which is a different handle, so the whole open ran again — including the /proc-wide
deleted-WAL sidecar scan (~4.4k readlinks). Each readlink drops and re-takes the GIL
while the startup threads (plugin discovery, MCP, skill sync, banner git) are busy, so an
11ms scan stretched to 1.3s per pass, and the second pass landed exactly where the
prompt should have appeared.

Route the CLI's handle (init + the two re-open sites) through the registry so every
in-process consumer shares one writer. One scan per startup; live A/B on the same box,
interleaved x6: banner->prompt gap 0.37s mean -> 0.15s mean (plain), 1.77s -> 0.39s
under strace. The registry release path replaces close(), so /quit, /snapshot restore
and /handoff keep their semantics.
2026-09-12 05:13:50 -07:00

38 lines
1.4 KiB
Python

"""The CLI's session store must be the registry's shared handle for state.db.
``_init_session_store`` used to construct a bare ``SessionDB()``. The goal/loop/heartbeat
managers acquire the same path through ``hermes_state_registry`` from the REPL thread a
moment later, so a bare handle meant a SECOND full open — including the /proc-wide
deleted-WAL sidecar scan (~4k readlinks, each a GIL round-trip against the busy startup
threads) — which showed up as the post-banner freeze before the first prompt.
"""
from types import SimpleNamespace
import hermes_cli.goals as goals
from cli import HermesCLI
def test_cli_session_store_is_the_registry_handle_goals_reuse(monkeypatch):
import hermes_state_registry
monkeypatch.setattr(goals, "_DB_CACHE", {})
constructed = []
real_open = hermes_state_registry._open_session_db
def recording_open(path):
constructed.append(path)
return real_open(path)
monkeypatch.setattr(hermes_state_registry, "_open_session_db", recording_open)
cli = SimpleNamespace()
try:
HermesCLI._init_session_store(cli)
assert cli._session_db is not None and not cli._session_db_unavailable
# goals/loops/heartbeat go through the registry: same object, no second open.
assert goals._get_session_db() is cli._session_db
assert len(constructed) == 1
finally:
hermes_state_registry.close_all()