Files
hermes-agent/agent/micro_compaction.py
20acba0376 fix(sessions): proactive prune rewrites only the changed rows, in place (#124273)
* fix(sessions): proactive prune rewrites only the changed rows, in place

A prune edits the content of a few existing messages and never adds, drops or
reorders one, yet every commit went through archive_and_compact, which
archives all active rows and re-inserts the full list: state.db grew by the
whole live transcript per commit (#124102).

SessionDB.rewrite_pruned_rows resolves each changed message to its live row
(_row_id, else its tool call id or tool-call ids, else the single active row
with the same role and stored content), checks the row still holds the
original, inserts one archived twin of the pre-prune row (active=0,
compacted=1, still searchable) and updates the changed columns under the same
id. The prune tries it after the stale-history watermark check and falls back
to archive_and_compact when a row cannot be named exactly or an unwritten
message sits before durable ones.

Fixes #124102

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

* fix(sessions): in-place prune twin keeps the live row's display slot

The content UPDATE fires messages_display_identity_update, which nulls
display_identity/display_order on the live row and on its freshly inserted
twin. The lazy backfill then gives the twin its own id as display_order, so
every demoted full body renders at the END of the transcript page, after the
newest user turn (3 x 24 KB bodies below "from desktop" in the probe).

Re-stamp both rows with the pre-update identity and order inside the same
transaction: the page projection shows one row per display_order (the live
stub, in place) and the twin stays a searchable archive. This is the same
stored-identity shape #117824 proposes for the archive writer (#117750), so
the resume projection collapses the pair once that lands.

Review-fix for #124273; one assertion added to the growth invariant
(red on the PR head: 20 page rows for 17 messages).

* fix(sessions): in-place prune keeps each display slot through later compaction and backfill

Review follow-up (ehz0ah): the rewritten live row kept the archived body's
identity, so a later archive_and_compact gave its copy a fresh slot and the
full body resurfaced in the old one. The live row now takes the identity of
what it stores and keeps the slot; the twin and any earlier generation's
copies (all nulled by the identity trigger) are restamped with their own
identity and slot, and the display backfill keeps a row's stored slot when
its stored identity is still correct. The prune pre-fills slots for legacy
rows so it never carries a missing one.

* fix(agent): fail closed when an in-place prune finds the row rewritten

rewrite_pruned_rows raised one PruneRowUnresolvedError for two different
conditions: the row cannot be named (ambiguity — the full writer handles it)
and the row IS named but its stored body changed. The second means another
surface already rewrote this generation, which is what StaleHeldHistory
exists to catch; the in-place prune preserves row ids, so
_archive_watermark_for cannot see it. Returning None there handed the stale
pruned_msgs to archive_and_compact, which archives rows it never compared
and republished the held text over a newer durable edit.

Split the stale case into a PruneRowStaleError subclass and raise
StaleHeldHistory from the in-place helper so the existing handler skips the
pass instead of committing. Ambiguity keeps the fallback.

Replaces the prune-fallback regression with the two cases review asked for:
a changed stored body skips the prune and leaves the newer body active; a
genuine ambiguity over current content still commits through the full writer.

* fix(agent): make an in-place prune visible to the other lease-less writer

rewrite_pruned_rows keeps every row id and activity flag, so the id/activity
staleness gate in held_archive_watermark cannot see that a prune already
committed. A second lease-less surface (micro-compaction) holding the pre-prune
transcript passes the gate and archive_and_compact republishes the pruned-away
bodies over the winner, undoing the prune.

stale_raises now also compares the held bodies against the stored ones via the
new SessionDB.rewritten_held_row_ids, so every lease-less writer sees the
in-place generation change, not just the prune caller that owns the exception.

* fix(state): compare durable row versions, not raw content, in the in-place stale guard

Reviewers on 631c897b converged on one root defect: the stale-write guard compared
raw SQLite bytes against a loaded/normalized view, which both missed real rewrites
and invented phantom ones.

- rewrite_pruned_rows._resolve: the staleness check is now the repo's stored-row CAS
  digest (transcript_row_snapshot over a SELECT * row, covering every owned column
  including assistant tool_calls arguments) when the held dict carries DB_ROW_SNAPSHOT,
  and the loader-lens identity compare (_row_identity, the same representation
  _stamp_kept_live_prefix uses) otherwise. Raw storage bytes are never compared
  against a loaded view.
- The rewritten row's replacement dict is stamped with the row's new digest, and the
  commit runs under _execute_transcript_write so a rollback restores the stamps.
- rewritten_held_row_ids takes full held dicts and applies the same two-leg compare;
  only tool/assistant rows (the roles the in-place writer touches) are compared.
- held_archive_watermark(stale_raises=True) checks the durable prefix BEFORE the
  ids[-1] is None early return, so an unstamped trailing message caps the watermark
  but can no longer suppress version checks for the rows the caller holds exactly.

Closes the P0 stale tool_calls path, the P2 false-stale on freshly resumed
whitespace-padded rows, and the unstamped-tail blind spot (#124102).

* fix(state): gate the lease-less stale check on every role the fallback republishes

The staleness gate behind held_archive_watermark (rewritten_held_row_ids)
compared only tool/assistant held rows, but the fallback it guards —
archive_and_compact, reached whenever the in-place rewrite refuses (an
unwritten message before later durable rows, an unnamed row) — re-inserts
held rows of every model-transcript role verbatim. A user row another view
rewrote in place (set_user_message_content, the repair batch's in-place
rewrite) slipped the gate: the stale pre-edit body was republished active
and the newer edit left with zero active copies.

Compare every model-transcript role (user/assistant/tool); the per-row
comparison (DB_ROW_SNAPSHOT CAS, else the loader-lens identity) is already
role-agnostic. Bookkeeping roles stay excluded — they never reach the
model-facing list. Covers the prune fallback and micro-compaction, the two
lease-less callers of _archive_watermark_for (#124102).

Sabotage-run: reverting the role filter makes
test_a_concurrent_user_edit_stays_active_when_the_prune_falls_back fail
(stale body published); restored, the file is 22/22 green.

* style: use builtin generics in the in-place prune commit annotations

* refactor(compression): split the proactive prune and its commit into topical siblings

The code-health ratchet (CI) flags this PR's own growth: hermes_state_messages.py
(2190 > 2000), agent/context_compressor.py (5911 > its main value 5852) and
agent/conversation_compression.py (4585 > 4564). The remedy the repo prescribes:
move the PR's own code into topical siblings, keeping the facade as the seam.

- agent/context_compressor_prune.py: ProactivePruneMixin (prune_tool_results_only
  + _rewrite_pruned_rows_in_place), bound into ContextCompressor; origin-module
  symbols read through _cc() at call time (the micro_compaction.py pattern) so
  tests patching agent.context_compressor.X stay effective; logger keeps the
  agent.context_compressor name for caplog parity.
- agent/conversation_compression_watermark.py: held_archive_watermark; the origin
  module imports it and keeps _held_watermark (its only caller).
- hermes_state_prune_rewrite.py: SessionPruneRewriteMixin (rewrite_pruned_rows +
  rewritten_held_row_ids), bound into SessionDB; every SessionDB call site and
  test patch seam (db.rewrite_pruned_rows / db.rewritten_held_row_ids) unchanged.

Verified: scripts/check --only health --base origin/main -> 0 blocking; ruff clean;
prune/compaction batches green (test_proactive_prune_restart_safety 22/22,
test_micro_compaction 48/48, five proactive-prune files 25/25). The 4 failures in
test_compaction_prompt_rebuild / test_compression_review reproduce identically on
origin/main (pre-existing).

* fix(state): re-verify held rows inside the full writer's commit transaction

The lease-less stale gate (rewritten_held_row_ids) runs outside the write
transaction, so a second SessionDB connection that rewrites a covered row
AFTER the precheck but BEFORE archive_and_compact's BEGIN IMMEDIATE was
invisible to it: the commit archived the newer row and republished the stale
held text, leaving the newer edit with zero active copies (#124102, review
finding on 8c6da464).

- archive_and_compact takes held_for_cas: the caller's held dicts,
  re-verified against the live rows INSIDE the commit's transaction before
  any row moves; a mismatch raises HeldRowRewrittenError, which the two
  lease-less callers (proactive prune fallback, micro-compaction sync)
  treat exactly like StaleHeldHistory - nothing written, the pass is a
  no-op and the newer generation keeps the session. The lease path leaves
  it None: its lease rules out an overlapping rewrite.
- held_row_mismatches(conn, ...) is the compare lifted out of
  rewritten_held_row_ids so both the read-path precheck and the
  in-transaction fence share one implementation. It now chunks the id IN
  query via _id_chunks (a history past SQLite's host-parameter limit
  failed the pass instead of reclaiming) and resolves rows that have no
  _row_id but do carry the persist marker + CAS snapshot by their logical
  message_uid (idx_messages_session_uid) - the unaddressed live-replay
  loader shape, whose rows were previously exempt from the fence entirely.
- regression: test_a_user_edit_landing_after_the_precheck_stays_active_in_the_commit
  drives a second SessionDB connection's set_user_message_content at the
  archive_and_compact call boundary; sabotage (gate disabled) makes it
  fail exactly as the finding predicts.

* refactor(compression): drop the duplicate _held_watermark from the module split

The 8c6da464 sibling extraction left two identical _held_watermark
definitions back to back in agent/conversation_compression.py; Python
silently kept only the second, so editing the first had no effect (dead
code, root AGENTS.md). Remove the added copy.

* refactor(state): split the api_content sidecar writers into a topical sibling

hermes_state_messages.py crossed the FILE_LINES ratchet (2010 > 2000) after
the commit-window fence landed in it. Move set_latest_user_api_content /
set_message_api_content into hermes_state_api_content.py
(SessionApiContentMixin, bound into SessionDB in the same MRO slot): the
two row-addressed api_content sidecar UPDATEs are one topic, no behaviour
change, every call site and test patch seam unchanged.

* fix(state): flatten _archive_and_compact_on_conn onto the caller's transaction after the main merge

Main flattened this helper to run directly on the caller's write connection
(no nested _execute_transcript_write); the merge resolution had kept the old
nested shape, which opened a second BEGIN IMMEDIATE inside the live one and
deadlocked every archive_and_compact. Re-flattened while keeping the #124102
held_for_cas in-transaction CAS fence.

* fix(state): carry the held-row CAS fence through the worker RPC path

RuntimeSessionCompressionMixin.archive_and_compact rejected held_for_cas,
so a worker-side micro-compaction commit raised TypeError inside the mixin's
generic except and silently skipped the whole persist (unstamped marker,
durable transcript left un-compacted). Accept held_for_cas, project it onto
the closed handoff fields plus the persistence marker the CAS compare reads,
validate it in worker_archive, and thread it into
archive_on_connection -> _archive_and_compact_on_conn (#124102).

* test(micro): a stale held generation without row ids aborts the micro-compact

Review follow-up (ehz0ah/greptile-apps, PR #124273): ACP restore and OpenAI
session continuation load through the unaddressed live-replay projection
(get_messages_as_conversation without include_row_ids=True) — the persist
marker, durable message_uid and the stored-row CAS digest, but no _row_id.
held_row_mismatches already resolves that shape by logical message_uid, so
the lease-less fence still sees another surface's in-place prune. Pin it with
the regression both reviewers asked for: after another surface prunes in
place, micro-compaction from that stale no-row-id history is a true no-op —
the pruned stubs stay the only live generation and the pre-prune bodies are
not republished (#124102).

Sabotage: disabling the by-uid branch in held_row_mismatches fails this
test (the sync republishes the stale bodies); restored, 49/49 pass.

Co-authored-by: ehz0ah <ehz0ah@users.noreply.github.com>

* fix(sessions): the uid-resolved stale fence CASes the newest active row

held_row_mismatches resolved an id-less held dict by its logical
message_uid with a bare fetchone(): when a carried or concurrent
compaction has cloned the row back beside its archived twin, one uid
names several active physical rows, and the uid index returns them
oldest-first. The fence then CASed the oldest copy — which still held
the payload the held digest was stamped from — and cleared a held
generation as current although another writer had rewritten the newest
row; the lease-less fallback republished the stale held payload over
that newer generation (#124102).

Resolve with the same newest-active rule as the loader's
_active_logical_message_row (ORDER BY id DESC LIMIT 1). Two
regressions pin it: two active rows sharing a uid where only the
newest was rewritten in place must abort the micro-compact naming the
newest row id, and where only an older duplicate moved the held
digest must stay current.

Review follow-up to greptile-apps' 2026-10-17 P1 on PR #124273.

---------

Co-authored-by: defurniture2025 <301948806+defurniture2025@users.noreply.github.com>
Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
Co-authored-by: teknium1 <127238744+teknium1@users.noreply.github.com>
Co-authored-by: Austin Pickett <pickett.austin@gmail.com>
Co-authored-by: Austin Pickett <austinpickett@users.noreply.github.com>
Co-authored-by: ehz0ah <ehz0ah@users.noreply.github.com>
2026-10-18 19:44:32 -04:00

529 lines
30 KiB
Python

"""Micro-compaction mixin for ContextCompressor.
Rolling per-exchange summarization that folds old user/assistant exchanges into a single
summary marker between turns. OFF by default: every pass rewrites the prompt prefix and
breaks the provider prompt cache.
"""
from __future__ import annotations
import json
import logging
import time
from typing import Any, Dict, List, Optional
from agent.compaction_events import publish_micro
from agent.context_compressor_telemetry import _safe_int
from agent.message_metadata import record_absorbed_message
from agent.model_metadata import estimate_messages_tokens_rough, estimate_tokens_rough
# Log name parity with the origin module.
logger = logging.getLogger("agent.context_compressor")
def _cc():
"""The origin module, resolved lazily: avoids the import cycle and keeps tests that patch
``agent.context_compressor.X`` effective (attributes are read at call time)."""
from agent import context_compressor
return context_compressor
def _is_summary_marker(entry: Any) -> bool:
return isinstance(entry, dict) and bool(entry.get(_cc().COMPRESSED_SUMMARY_METADATA_KEY))
def _is_micro_marker(entry: Any) -> bool:
"""True for a summary marker provably absorbed into the rolling summary (micro, not batch)."""
return _is_summary_marker(entry) and bool(entry.get(_cc().MICRO_COMPACT_MARKER_KEY))
class MicroCompactionMixin:
"""Rolling micro-compaction; host must be a ``ContextCompressor``."""
def _resolve_compact_cursor(self, messages: list[dict[str, Any]], head_end: int, tail_start: int) -> int:
"""Index of the first message not yet absorbed into the rolling summary: the in-memory
cursor when valid, else just past the last summary marker."""
if head_end < self._micro_compact_cursor < tail_start:
return self._micro_compact_cursor
summaries = (i for i in range(head_end, tail_start) if self._is_context_summary_message(messages[i]))
last = max(summaries, default=-1)
cursor = last + 1 if last >= head_end else head_end
if last >= head_end:
# Resumed session: rehydrate the rolling summary from the surviving marker so the next
# pass merges, not replaces.
recovered = "" if self._micro_compact_rolling_summary.strip() else (
self._rolling_summary_from_marker(messages[last].get("content"))
)
if recovered:
self._micro_compact_rolling_summary = recovered
# Rehydration proves containment: this marker (batch or micro) becomes
# supersede/defrag-eligible; unabsorbed markers never get the key.
messages[last][_cc().MICRO_COMPACT_MARKER_KEY] = True
logger.info("Micro-compaction: recovered rolling summary from transcript (%d chars)", len(recovered))
self._micro_compact_cursor = cursor
return cursor
def _find_one_exchange(
self, messages: list[dict[str, Any]], start: int, tail_start: int,
) -> Optional[tuple[int, int]]:
"""Find the next complete exchange (full agent turn) starting at *start*; returns
``(exchange_start, exchange_end)`` or ``None``. Spans assistant+tool rows up to the next
user message; user turns are never absorbed (alternation safety, verbatim user text)."""
limit = min(tail_start, len(messages))
def _turn_row(idx: int, roles: tuple) -> bool:
return messages[idx].get("role") in roles and not self._is_context_summary_message(messages[idx])
# Skip user messages and (assistant-role) summary markers to reach a real assistant message;
# otherwise a rehydrated cursor could absorb the marker itself.
idx = start
while idx < limit and not _turn_row(idx, ("assistant",)):
idx += 1
if idx >= limit:
return None
exchange_start = idx
idx += 1
while idx < limit and _turn_row(idx, ("assistant", "tool")):
idx += 1
# Boundary must close the turn: a mid-turn stop at tail_start would put the assistant marker
# beside assistant/tool rows. Any other role is a safe splice (avoids wedging the cursor).
boundary = messages[idx] if idx < len(messages) else None
if not isinstance(boundary, dict) or boundary.get("role") in ("assistant", "tool"):
return None
return (exchange_start, idx)
def _build_micro_summary_prompt(self, existing_summary: str, exchange_text: str) -> list[dict[str, str]]:
"""Build the prompt messages for a single-exchange micro-summary."""
summary_block = existing_summary if existing_summary.strip() else "(No previous summary yet.)"
user_prompt = (
"You are a summarization agent creating a compact record of an "
"ongoing conversation. You are given a running summary and the "
"next exchange from the conversation. Merge the exchange's key "
"decisions, requirements, file paths, and open questions into the "
"summary. Preserve the summary's structure. Drop resolved details "
"that are no longer relevant. Add new decisions, file paths, and "
"open questions.\n\n"
"NEVER include API keys, tokens, passwords, secrets, credentials, "
"or connection strings in the summary \u2014 replace any that appear "
f"with [REDACTED].\n\n"
f"## Current Running Summary\n{summary_block}\n\n"
f"## Next Exchange to Merge\n{exchange_text}\n\n"
"Return ONLY the updated summary text, no preamble or explanation. "
"Do not include this instruction block in your output."
)
return [
{"role": "system", "content": "You are a conversation summarization assistant."},
{"role": "user", "content": user_prompt},
]
def _micro_summarize_one(self, exchange_text: str) -> Optional[str]:
"""Micro-summarize one exchange into the rolling summary via the aux LLM (None on failure)."""
from agent.auxiliary_client import aux_interrupt_protection, call_llm
call_kwargs = {
"task": "compression",
"messages": self._build_micro_summary_prompt(self._micro_compact_rolling_summary, exchange_text),
"max_tokens": min(1500, self.max_summary_tokens or 1500),
"temperature": 0.1,
}
self._apply_summary_route(call_kwargs)
if self.model:
call_kwargs.setdefault("main_runtime", {
"model": self.model, "provider": self.provider or "", "base_url": self.base_url or "",
"api_key": self.api_key or "", "api_mode": getattr(self, "api_mode", "") or "",
})
try:
with aux_interrupt_protection():
response = call_llm(**call_kwargs)
except Exception as exc:
logger.info("micro-summarization call failed: %s", exc)
return None
# A length stop means a partial merge; leave the exchange unabsorbed so a later pass retries.
if _cc()._response_finish_reason(response) == "length":
logger.warning(
"micro-summarization output hit the token cap (finish_reason=length) — discarding partial summary",
)
return None
message = response.choices[0].message
content = message.get("content") if isinstance(message, dict) else getattr(message, "content", message)
content = (content if isinstance(content, str) else str(content) if content else "").strip()
from agent.agent_runtime_helpers import strip_think_blocks
content = strip_think_blocks(None, content).strip()
if not content:
logger.info("micro-summarization returned empty content")
return None
if _cc()._is_refusal_response(response, content):
logger.warning("micro-summarization returned refusal content — discarding unusable summary")
return None
return content
def _needs_defrag(self) -> bool:
"""Return True when the rolling summary is large enough to defrag."""
return estimate_tokens_rough(self._micro_compact_rolling_summary) >= self._micro_compact_defrag_threshold_tokens
def _defrag_rolling_summary(self, messages: list[dict[str, Any]]) -> bool:
"""Re-summarize the rolling summary text and rewrite the marker in place.
Transcript-shape-neutral (no splice, no cursor move). Returns True when it rewrote."""
old_summary = self._micro_compact_rolling_summary
if not old_summary.strip():
return False
# Empty base turns the merge prompt into a rewrite-compactly instruction.
self._micro_compact_rolling_summary = ""
fresh_summary = self._micro_summarize_one(old_summary)
self._micro_compact_rolling_summary = fresh_summary or old_summary
if not fresh_summary:
return False
# Rewrite only the newest MICRO marker (resume rehydrates from it); a batch marker holds
# history we lack.
entry = next((e for e in reversed(messages) if _is_micro_marker(e)), None)
if entry is not None:
entry["content"] = self._render_micro_marker_content(fresh_summary)
# Content changed: clear the persisted stamp so the DB sync rewrites the row. An
# in-place pop on a live dict would be identity-skipped by the bounded flush scan;
# flag the finalizer.
entry.pop(_cc()._DB_PERSISTED_MARKER, None)
# Sibling of the finalize_turn pop site (#75170): this pop also strips the marker from a LIVE
# dict in place, so the bounded flush-scan cursor would identity-skip the rewritten marker and
# the defragged summary would never reach state.db. The compressor holds no agent reference, so
# raise a flag the finalizer consumes to invalidate agent._db_flush_scan_prefix. (The pop sites
# at module scope — fresh copies in strip-marker helpers — break identity and need no flag.)
self._flush_scan_cursor_invalidated = True
logger.info(
"Micro-compaction defrag: rolling summary re-summarized (%d -> %d chars)",
len(old_summary), len(fresh_summary),
)
return True
def _reset_micro_failure_tracking(self) -> None:
self._micro_compact_consecutive_failures = 0
self._micro_compact_last_failure_cursor = -1
def _micro_compact(
self, messages: list[dict[str, Any]], *, turn_session_id: Optional[str] = None,
) -> list[dict[str, Any]]:
"""Run one round of micro-compaction (entry point from ``finalize_turn()``). Returns the
(possibly modified) list and syncs the session DB via ``archive_and_compact`` (the
append-only flush alone would double-load on resume). ``turn_session_id`` is the session the
caller's turn started in; it parents the pass's Relay mark when this session has no scope yet."""
if not self._micro_compact_enabled:
return messages
# Cadence gate: each pass breaks prompt cache once. Counted per invocation so a no-op turn
# can't wedge it.
every_n = max(1, int(self._micro_compact_every_n_turns or 1))
if every_n > 1:
self._micro_compact_turns_since_pass += 1
if self._micro_compact_turns_since_pass < every_n:
return messages
self._micro_compact_turns_since_pass = 0
n_messages = len(messages)
exchange = self._next_exchange(messages) if n_messages >= 4 else None
if exchange is None:
return messages
exchange_start, exchange_end = exchange
# Telemetry baseline; taken only once an exchange exists so no-op turns don't pay.
_started_at = time.monotonic()
_tokens_before = estimate_messages_tokens_rough(messages)
# The history this pass rewrites, and the store's watermark, before the slow summary call: the commit
# keeps what another surface or a concurrent write added instead of archiving it unseen. A watermark
# the store could not answer for leaves the commit unbounded, so the pass does not run at all.
_held = list(messages)
_skip, _start_watermark = self._micro_start_watermark(_held)
def _telemetry(outcome: str, result: list[dict[str, Any]], **extra: Any) -> None:
self._emit_micro_compaction_telemetry(
outcome=outcome, messages_before=n_messages, messages_after=len(result),
tokens_before=_tokens_before, duration_ms=int((time.monotonic() - _started_at) * 1000),
turn_session_id=turn_session_id, **extra,
)
if _skip: # unanswerable watermark (unbounded archive) or a stale generation (no lease): skip
_telemetry(_skip, messages, tokens_after=_tokens_before)
return messages
# Pre-pass state, restored when the commit finds a compaction landed during the summary call.
_pre_cursor, _pre_summary = self._micro_compact_cursor, self._micro_compact_rolling_summary
# Defrag rewrites summary text/marker in place (no splice, no cursor move) instead of
# absorbing this turn.
if self._needs_defrag():
_marker = next((e for e in reversed(messages) if _is_micro_marker(e)), None)
_pre_marker = dict(_marker) if _marker is not None else None
defragged = self._defrag_rolling_summary(messages)
if defragged and not self._sync_micro_compact_to_db(
messages, held=_held, start_watermark=_start_watermark,
):
# Stale generation: undo the in-place rewrite so the finalizer never persists it.
self._micro_compact_rolling_summary = _pre_summary
if _marker is not None:
_marker.clear()
_marker.update(_pre_marker)
_telemetry("stale_generation", messages, tokens_after=_tokens_before)
return messages
if defragged:
self._reset_micro_failure_tracking()
outcome = "defrag" if defragged else "defrag_failed"
_telemetry(outcome, messages, tokens_after=estimate_messages_tokens_rough(messages))
return messages
# Cumulative iff it subsumes an earlier marker; captured before summarizing.
_cumulative = bool(self._micro_compact_rolling_summary.strip())
exchange_text = self._serialize_for_summary(messages[exchange_start:exchange_end])
_exchange_tokens = estimate_tokens_rough(exchange_text)
updated_summary = self._micro_summarize_one(exchange_text)
if updated_summary is None:
_outcome = self._record_micro_failure(exchange_start, exchange_end)
_telemetry(_outcome, messages, tokens_after=_tokens_before, exchange_tokens=_exchange_tokens)
return messages
self._micro_compact_rolling_summary = updated_summary
self._micro_compact_cursor = exchange_end
self._reset_micro_failure_tracking()
result = self._splice_micro_compact_result(messages, exchange_start, exchange_end, supersede=_cumulative)
self._micro_compact_cursor = self._cursor_after_splice(result, exchange_start + 1)
if not self._sync_micro_compact_to_db(result, held=_held, start_watermark=_start_watermark):
# Another compaction committed during the summary call. A true no-op: returning the spliced
# list would let finalize_turn persist its unmarked summary row beside the winning generation.
self._micro_compact_cursor, self._micro_compact_rolling_summary = _pre_cursor, _pre_summary
_telemetry("stale_generation", messages, tokens_after=_tokens_before)
return messages
_telemetry(
"absorbed", result, tokens_after=estimate_messages_tokens_rough(result), exchange_tokens=_exchange_tokens,
)
return result
def _record_micro_failure(self, exchange_start: int, exchange_end: int) -> str:
"""Count a summarize failure at this cursor; skip the exchange after too many in a row."""
# Track consecutive failures at the same cursor to avoid busy-looping every turn.
same_cursor = exchange_start == self._micro_compact_last_failure_cursor
self._micro_compact_consecutive_failures = self._micro_compact_consecutive_failures + 1 if same_cursor else 1
self._micro_compact_last_failure_cursor = exchange_start
if self._micro_compact_consecutive_failures < _cc()._MICRO_COMPACT_MAX_CONSECUTIVE_FAILURES:
return "summarize_failed"
logger.info(
"Micro-compaction: skipping exchange at cursor %d after %d consecutive failures",
exchange_start, self._micro_compact_consecutive_failures,
)
# Skip the stuck exchange; it stays in the transcript for batch compression/defrag.
self._micro_compact_cursor = exchange_end
self._reset_micro_failure_tracking()
return "exchange_skipped"
def _next_exchange(self, messages: list[dict[str, Any]]) -> Optional[tuple[int, int]]:
"""The next un-absorbed exchange inside the compressible window, or None."""
compress_start = self._align_boundary_forward(messages, self._protect_head_size(messages))
compress_end = self._find_tail_cut_by_tokens(messages, compress_start, allow_split_turn=False)
if compress_start >= compress_end:
return None
cursor = self._resolve_compact_cursor(messages, compress_start, compress_end)
return None if cursor >= compress_end else self._find_one_exchange(messages, cursor, compress_end)
@staticmethod
def _rolling_summary_from_marker(content: Any) -> str:
"""Recover the rolling-summary text from a summary marker (resume rehydration)."""
cc = _cc()
if not isinstance(content, str) or not content.strip():
return ""
# rfind: SUMMARY_PREFIX itself mentions the heading, so the first hit is in the preamble.
idx = content.rfind(cc.HISTORICAL_TASK_HEADING)
body = content[idx + len(cc.HISTORICAL_TASK_HEADING):] if idx != -1 else content
end = body.find(cc._SUMMARY_END_MARKER)
return (body[:end] if end != -1 else body).strip()
def _cursor_after_splice(self, result: list[dict[str, Any]], fallback: int) -> int:
"""Cursor position just past the summary marker in *result*. Must derive from the SPLICED
list: a splice collapses several rows into one marker (and may drop a superseded one), so
pre-splice indices land inside a later exchange and silently skip it."""
return next((idx + 1 for idx in range(len(result) - 1, -1, -1) if _is_summary_marker(result[idx])), fallback)
def _emit_micro_compaction_telemetry(
self, *, outcome: str, messages_before: int, messages_after: int, tokens_before: int | None,
tokens_after: int | None, exchange_tokens: int | None = None, duration_ms: int | None = None,
turn_session_id: str | None = None,
) -> None:
"""Emit one content-free JSON log line for a micro-compaction pass.
``tokens_delta`` < 0 means the pass shrank the transcript; ``*_total`` fields accumulate."""
try:
delta = tokens_after - tokens_before if tokens_before is not None and tokens_after is not None else None
self._micro_compact_tokens_saved_total -= delta or 0
self._micro_compact_passes += 1
# Cached reads only: the lazy properties can fire a synchronous /models probe.
# The ``threshold_tokens`` / ``context_length`` properties resolve lazily and can fire a
# synchronous /models probe on first access (#32221) — telemetry must never be the thing that
# blocks a turn. Unresolved simply reports null.
threshold = self._threshold_tokens
has_occupancy = threshold and tokens_after is not None and threshold > 0
occupancy = round(tokens_after / threshold * 100, 1) if has_occupancy else None
payload = {
"event": "micro_compaction", "session_id": getattr(self, "_session_id", "") or "", "outcome": outcome,
"messages_before": messages_before, "messages_after": messages_after,
"tokens_before": _safe_int(tokens_before), "tokens_after": _safe_int(tokens_after),
"tokens_delta": _safe_int(delta), "exchange_tokens": _safe_int(exchange_tokens),
"rolling_summary_tokens": estimate_tokens_rough(self._micro_compact_rolling_summary),
"cursor": _safe_int(self._micro_compact_cursor), "passes_total": self._micro_compact_passes,
"tokens_saved_total": self._micro_compact_tokens_saved_total, "duration_ms": _safe_int(duration_ms),
# Headroom: how full the window is being kept.
"threshold_tokens": _safe_int(threshold), "context_limit": _safe_int(self._resolved_context_length),
"occupancy_pct": occupancy, "main_model": self.model or "", "aux_model": self.summary_model or "",
}
logger.info("micro compaction telemetry: %s", json.dumps(payload, sort_keys=True, separators=(",", ":")))
publish_micro(payload, turn_session_id=turn_session_id)
except Exception as exc:
logger.debug("failed to emit micro-compaction telemetry: %s", exc)
def _micro_start_watermark(self, held: list[dict[str, Any]]) -> tuple[Optional[str], Optional[int]]:
"""``(skip_outcome, watermark)`` taken before a pass's slow step; *skip_outcome* is the telemetry
label of a pass that must not run, or None.
A store with no watermark API runs the pass with ``(None, None)``: the commit keeps today's
archive-everything behaviour. ``watermark_unavailable`` when the read itself raised — that must
NOT collapse into the same None, because None means "archive every active row", the very loss
this watermark exists to prevent. ``stale_generation`` when *held* is no longer the session's live
generation (another compaction already committed): this pass holds no compression lease, so
publishing would leave two generations live. The commit re-checks, since a compaction can also
land during the summary call.
"""
session_db, session_id = getattr(self, "_session_db", None), getattr(self, "_session_id", "")
watermark_of = getattr(session_db, "get_active_message_watermark", None)
if not session_id or not callable(watermark_of):
return None, None
try:
watermark = watermark_of(session_id)
_cc()._archive_watermark_for(session_db, session_id, held, watermark)
except _cc().StaleHeldHistory as exc:
logger.info("micro-compaction: skipping this pass, %s", exc)
return "stale_generation", None
except Exception as exc:
logger.info("micro-compaction: watermark read failed, skipping this pass: %s", exc)
return "watermark_unavailable", None
return None, watermark
def _sync_micro_compact_to_db(
self, compacted_messages: list[dict[str, Any]], *, held: Optional[list[dict[str, Any]]] = None,
start_watermark: Optional[int] = None,
) -> bool:
"""Persist the micro-compacted set to the session DB atomically and stamp rows persisted.
Without this the old exchange rows stay ``active=1`` and a resume double-loads both the
summary and the originals. Returns False only when *held* turned out to be a stale generation
(nothing written); the caller must then discard the pass. Any other outcome returns True."""
session_db, session_id = getattr(self, "_session_db", None), getattr(self, "_session_id", "")
if not session_db or not session_id:
return True
from hermes_state_errors import HeldRowRewrittenError
try:
# Micro-compaction is prefix + marker + suffix, not a contiguous tail. Identify the exact
# byte-identical originals by their persistence marker; the state transaction resolves each
# one by row id or durable identity+timestamp. In-place mutations deliberately pop
# _DB_PERSISTED_MARKER, and the fresh summary marker never has one. A positional tail_count
# can otherwise classify the summarized assistant/tool rows as rewind-only (#118481).
carried_messages = [
message for message in compacted_messages
if isinstance(message, dict) and message.get(_cc()._DB_PERSISTED_MARKER)
]
watermark = None
covered_ids = unresolved_held = None
if held is not None and start_watermark is not None:
watermark = _cc()._archive_watermark_for(session_db, session_id, held, start_watermark)
from agent.conversation_compression_archive import coverage_for_commit
covered_ids, unresolved_held = coverage_for_commit(session_db, session_id, held)
session_db.archive_and_compact(
session_id, compacted_messages, carried_messages=carried_messages, watermark=watermark,
covered_ids=covered_ids, unresolved_held=unresolved_held,
held_for_cas=held)
# Shared post-commit stamp site with batch commit and proactive prune.
# See #98450.
_cc().stamp_db_persisted_markers(compacted_messages)
except _cc().StaleHeldHistory as exc:
# Another compaction committed during the summary call. Nothing is written: the store already
# holds the winning generation. The caller discards the pass (original list, pre-pass cursor
# and summary); otherwise finalize_turn would append the unmarked summary row beside the winner.
logger.info("Micro-compaction commit skipped, %s", exc)
return False
except HeldRowRewrittenError as exc:
# The in-transaction CAS fence tripped: a held row changed between the precheck and the
# commit. Same disposition as a stale generation — nothing written, the pass is discarded.
logger.info("Micro-compaction commit skipped, %s", exc)
return False
except Exception:
logger.info(
"Micro-compaction DB sync failed — resume will double-load "
"compacted messages until the next batch compression"
)
return True
def _splice_micro_compact_result(
self, messages: list[dict[str, Any]], splice_start: int, splice_end: int, supersede: bool = True,
) -> list[dict[str, Any]]:
"""Replace *messages[splice_start:splice_end]* with an assistant-role summary marker.
Merges user turns left adjacent by a superseded marker so the result is alternation-valid."""
cc = _cc()
summary_text = self._micro_compact_rolling_summary
if not summary_text.strip():
return messages
summary_msg = {
"role": "assistant", "content": self._render_micro_marker_content(summary_text),
cc.COMPRESSED_SUMMARY_METADATA_KEY: True,
# Micro marker: eligible for supersede/defrag; batch markers never carry this key.
cc.MICRO_COMPACT_MARKER_KEY: True,
# Micro markers absorb only assistant/tool content; user turns stay in the transcript.
cc.COMPRESSED_SUMMARY_HAS_USER_TURN_KEY: False,
}
result = messages[:splice_start] + [summary_msg] + messages[splice_end:]
# Cumulative summary: keep only the newest marker. Drop an older one only if supersede AND
# it has MICRO_COMPACT_MARKER_KEY (provably absorbed); a batch marker holds MORE history.
stale = [i for i, m in enumerate(result) if _is_micro_marker(m)][:-1] if supersede else []
if stale:
result = self._merge_adjacent_user_turns([m for i, m in enumerate(result) if i not in stale])
# Deliberately no _strip_persistence_markers: micro archives in place under the same session
# id, so stamps stay accurate and a failed archive keeps the append-only flush idempotent.
return result
@staticmethod
def _render_micro_marker_content(summary_text: str) -> str:
"""Assemble the marker content wrapper around *summary_text*."""
cc = _cc()
return f"{cc.SUMMARY_PREFIX}\n\n{cc.HISTORICAL_TASK_HEADING}\n{summary_text.strip()}\n\n{cc._SUMMARY_END_MARKER}"
def _merge_adjacent_user_turns(self, result: list[dict[str, Any]]) -> list[dict[str, Any]]:
"""Merge consecutive plain-text real user turns left by a supersede. Same ``\\n\\n`` join as
``repair_message_sequence`` pass 2, done here so the marker and cursor are never collateral
damage of the downstream repair. Lists untouched."""
from agent.turn_context import drop_stale_api_content
def _plain_user(m: Any) -> bool:
return (
isinstance(m, dict) and m.get("role") == "user" and not _is_summary_marker(m)
and isinstance(m.get("content"), str)
)
merged: list[dict[str, Any]] = []
for msg in result:
prev = merged[-1] if merged else None
if _plain_user(msg) and _plain_user(prev):
prev["content"] = "\n\n".join(c for c in (prev["content"], msg["content"]) if c)
drop_stale_api_content(prev) # merged content invalidates the api_content sidecar
# The originals stay in display history as compacted rows; showing the join too
# would paint every merged input twice on resume.
prev["display_metadata"] = {**(prev.get("display_metadata") or {}),
_cc().MODEL_ONLY_DISPLAY_METADATA_KEY: True}
# The merge rewrites a live dict that may carry _db_persisted: pop the stamp
# and flag the finalizer to invalidate the bounded flush-scan cursor, or the
# merged text is identity-skipped and never reaches state.db. Same contract
# as the defrag rewrite site above.
prev.pop(_cc()._DB_PERSISTED_MARKER, None)
self._flush_scan_cursor_invalidated = True
record_absorbed_message(prev, msg) # merge witness: prev keeps its uid, records msg's
else:
merged.append(msg)
return merged