7674bfb449
Some checks failed
Release & Docker / release (push) Has been cancelled
#3835 (#3800): preserve full compaction summaries. #3061: pending-intent RFC (docs). Full suite 8273, Codex SAFE, Opus SHIP, CI 11/11. #3837 dropped (red CI). Co-authored-by: rodboev <rodboev@users.noreply.github.com> Co-authored-by: franksong2702 <franksong2702@users.noreply.github.com>
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""
|
|
Regression coverage for compaction summary normalization (#3800).
|
|
"""
|
|
|
|
import sys
|
|
import types
|
|
|
|
from api.routes import _handle_session_compress, get_session
|
|
from api.streaming import _compact_summary_text
|
|
from tests.test_sprint46 import (
|
|
_FakeAgent,
|
|
_FakeHandler,
|
|
_install_fake_compression_runtime,
|
|
_make_session,
|
|
)
|
|
|
|
|
|
def _install_manual_summary(monkeypatch, summary_text):
|
|
agent_module = types.ModuleType("agent")
|
|
agent_module.__path__ = []
|
|
feedback_module = types.ModuleType("agent.manual_compression_feedback")
|
|
feedback_module.summarize_manual_compression = (
|
|
lambda original_messages, compressed_messages, before_count, after_count: {
|
|
"reference_message": summary_text,
|
|
}
|
|
)
|
|
monkeypatch.setitem(sys.modules, "agent", agent_module)
|
|
monkeypatch.setitem(sys.modules, "agent.manual_compression_feedback", feedback_module)
|
|
|
|
|
|
def _run_manual_compaction_summary(monkeypatch, cleanup_test_sessions, summary_text):
|
|
sid = _make_session()
|
|
cleanup_test_sessions.append(sid)
|
|
|
|
_install_fake_compression_runtime(monkeypatch, _FakeAgent)
|
|
_install_manual_summary(monkeypatch, summary_text)
|
|
|
|
handler = _FakeHandler()
|
|
_handle_session_compress(handler, {"session_id": sid})
|
|
|
|
assert handler.status == 200
|
|
payload = handler.payload()
|
|
return payload["session"]["compression_anchor_summary"], get_session(sid)
|
|
|
|
|
|
def test_manual_and_streaming_compaction_preserve_long_summaries(
|
|
monkeypatch, cleanup_test_sessions
|
|
):
|
|
long_summary = ("Alpha summary line.\n" + "Long detail " * 40 + "tail").strip()
|
|
|
|
route_summary, stored_session = _run_manual_compaction_summary(
|
|
monkeypatch, cleanup_test_sessions, long_summary
|
|
)
|
|
|
|
assert route_summary == _compact_summary_text(long_summary)
|
|
assert route_summary == stored_session.compression_anchor_summary
|
|
assert route_summary is not None
|
|
assert len(route_summary) > 320
|
|
|
|
|
|
def test_manual_and_streaming_compaction_normalize_blank_summaries(
|
|
monkeypatch, cleanup_test_sessions
|
|
):
|
|
blank_summary = " \n\t "
|
|
|
|
route_summary, stored_session = _run_manual_compaction_summary(
|
|
monkeypatch, cleanup_test_sessions, blank_summary
|
|
)
|
|
|
|
assert route_summary is None
|
|
assert route_summary == _compact_summary_text(blank_summary)
|
|
assert stored_session.compression_anchor_summary is None
|