fix: preserve transcript generations in sidebar projection
Rocky-Actor: Hermes Rocky Rocky-Mode: interactive
This commit is contained in:
1 parent
a4a87a97a3
commit
df35836b75
3 files changed
+53
-1
No files matched your search
@@ -0,0 +1,48 @@
|
||||
"""The sessions route must carry shrink metadata through to unread decisions."""
|
||||
import json
|
||||
import shutil
|
||||
|
||||
import pytest
|
||||
|
||||
from tests.test_issue4766_sidebar_source_pushdown import (
|
||||
_handle_sessions,
|
||||
_install_common_monkeypatches,
|
||||
)
|
||||
from tests.test_issue856_background_completion_unread import _run_unread_behavior
|
||||
|
||||
|
||||
@pytest.mark.skipif(shutil.which("node") is None, reason="node not on PATH")
|
||||
@pytest.mark.parametrize("retained", [0, 2])
|
||||
def test_sessions_route_preserves_post_shrink_unread(monkeypatch, retained):
|
||||
import api.routes as routes
|
||||
from api.models import Session
|
||||
from api.session_ops import truncate_session_at_keep
|
||||
|
||||
session = Session(session_id="X", messages=[
|
||||
{"role": "user", "content": str(i)} for i in range(10)
|
||||
])
|
||||
truncate_session_at_keep(session, retained)
|
||||
session.messages.append({"role": "assistant", "content": "new response"})
|
||||
row = session.compact()
|
||||
_install_common_monkeypatches(monkeypatch, [row])
|
||||
routes._session_list_cache_clear()
|
||||
try:
|
||||
# Exercise both the initial projection and the cached response.
|
||||
for _ in range(2):
|
||||
response = _handle_sessions("http://example.com/api/sessions?sidebar_source=webui")
|
||||
assert response.status == 200
|
||||
[projected] = response.json_body()["sessions"]
|
||||
result = _run_unread_behavior("""
|
||||
store[SESSION_VIEWED_COUNTS_KEY] = JSON.stringify({
|
||||
X: {message_count: 10, transcript_generation: 0},
|
||||
});
|
||||
const unread = _hasUnreadForSession(""" + json.dumps(projected) + """);
|
||||
console.log(JSON.stringify({unread, viewed: JSON.parse(store[SESSION_VIEWED_COUNTS_KEY]).X}));
|
||||
""")
|
||||
assert result == {
|
||||
"unread": True,
|
||||
"viewed": {"message_count": retained, "transcript_generation": 1},
|
||||
}
|
||||
assert "messages" not in projected
|
||||
finally:
|
||||
routes._session_list_cache_clear()
|
||||
Reference in new issue
Block a user