ce9aa869fc
Salvaged from PR #3243 by @Mibayy, reimplemented against current main (the original diff targeted a removed gateway/run.py handler). - /compact is now a first-class alias of /compress (CLI, gateway, Telegram/Slack/Discord command lists, autocomplete) — also fixes the dangling '/compact' references in gateway error messages (gateway/run.py context-exhausted banners). - --preview / --dry-run: report what WOULD be compressed (message counts, token estimate, 'here [N]' boundary) without touching the transcript. Flags coexist with the existing 'here [N]' / focus-topic args on both the CLI and gateway surfaces via shared pure helpers in hermes_cli/partial_compress.py. - --aggressive (LLM-free hard truncation) is intentionally NOT implemented: it would need its own transcript-persistence branch outside the guarded _compress_context rotation machinery (#44794 data-loss class). The flag is recognized and returns an explanatory message pointing at '/compress here [N]' and /undo instead of being mis-parsed as a focus topic. - locales: gateway.compress.aggressive_unsupported added to all 16 catalogs (parity test enforced). - release.py: AUTHOR_MAP entry for contributor credit.
120 lines
4 KiB
Python
120 lines
4 KiB
Python
"""Tests for gateway /compress --preview/--dry-run/--aggressive flags
|
|
(PR #3243 salvage).
|
|
|
|
The preview path must return a report WITHOUT building an agent or
|
|
touching the transcript; --aggressive must return an explanatory
|
|
message rather than being mis-parsed as a focus topic.
|
|
"""
|
|
|
|
from datetime import datetime
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from gateway.config import GatewayConfig, Platform, PlatformConfig
|
|
from gateway.platforms.base import MessageEvent
|
|
from gateway.session import SessionEntry, SessionSource, build_session_key
|
|
|
|
|
|
def _make_source() -> SessionSource:
|
|
return SessionSource(
|
|
platform=Platform.TELEGRAM,
|
|
user_id="u1",
|
|
chat_id="c1",
|
|
user_name="tester",
|
|
chat_type="dm",
|
|
)
|
|
|
|
|
|
def _make_event(text: str) -> MessageEvent:
|
|
return MessageEvent(text=text, source=_make_source(), message_id="m1")
|
|
|
|
|
|
def _make_history(n_pairs: int = 3) -> list[dict[str, str]]:
|
|
h: list[dict[str, str]] = []
|
|
for i in range(n_pairs):
|
|
h.append({"role": "user", "content": f"u{i}"})
|
|
h.append({"role": "assistant", "content": f"a{i}"})
|
|
return h
|
|
|
|
|
|
def _make_runner(history: list[dict[str, str]]):
|
|
from gateway.run import GatewayRunner
|
|
|
|
runner = object.__new__(GatewayRunner)
|
|
runner.config = GatewayConfig(
|
|
platforms={Platform.TELEGRAM: PlatformConfig(enabled=True, token="***")}
|
|
)
|
|
session_entry = SessionEntry(
|
|
session_key=build_session_key(_make_source()),
|
|
session_id="sess-1",
|
|
created_at=datetime.now(),
|
|
updated_at=datetime.now(),
|
|
platform=Platform.TELEGRAM,
|
|
chat_type="dm",
|
|
)
|
|
runner.session_store = MagicMock()
|
|
runner.session_store.get_or_create_session.return_value = session_entry
|
|
runner.session_store.load_transcript.return_value = history
|
|
runner.session_store.rewrite_transcript = MagicMock()
|
|
runner.session_store.update_session = MagicMock()
|
|
runner.session_store._save = MagicMock()
|
|
runner._session_db = None
|
|
return runner
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_preview_reports_without_mutating():
|
|
runner = _make_runner(_make_history(3))
|
|
result = await runner._handle_compress_command(_make_event("/compress --preview"))
|
|
assert "no changes made" in result.lower()
|
|
assert "6 of 6" in result
|
|
runner.session_store.rewrite_transcript.assert_not_called()
|
|
runner.session_store.update_session.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_dry_run_alias_matches_preview():
|
|
runner = _make_runner(_make_history(3))
|
|
result = await runner._handle_compress_command(_make_event("/compress --dry-run"))
|
|
assert "no changes made" in result.lower()
|
|
runner.session_store.rewrite_transcript.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_preview_with_here_boundary():
|
|
runner = _make_runner(_make_history(4))
|
|
result = await runner._handle_compress_command(
|
|
_make_event("/compress --preview here 2")
|
|
)
|
|
assert "last 2 exchange" in result
|
|
assert "4 of 8" in result
|
|
runner.session_store.rewrite_transcript.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_aggressive_returns_unsupported_note_without_mutating():
|
|
runner = _make_runner(_make_history(3))
|
|
result = await runner._handle_compress_command(
|
|
_make_event("/compress --aggressive")
|
|
)
|
|
assert "--aggressive is not supported" in result
|
|
runner.session_store.rewrite_transcript.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_aggressive_dry_run_shows_preview_plus_note():
|
|
runner = _make_runner(_make_history(3))
|
|
result = await runner._handle_compress_command(
|
|
_make_event("/compress --aggressive --dry-run")
|
|
)
|
|
assert "no changes made" in result.lower()
|
|
assert "--aggressive is not supported" in result
|
|
runner.session_store.rewrite_transcript.assert_not_called()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_preview_still_requires_enough_history():
|
|
runner = _make_runner(_make_history(1)) # only 2 messages
|
|
result = await runner._handle_compress_command(_make_event("/compress --preview"))
|
|
assert "not enough" in result.lower()
|