fix(session): follow compression continuations for transcript reads
This commit is contained in:
@@ -121,6 +121,26 @@ async def test_session_crud_and_message_history(adapter, session_db):
|
||||
assert session_db.get_session(session_id) is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_session_messages_follow_compression_tip(adapter, session_db):
|
||||
source_id = session_db.create_session("source-session", "api_server")
|
||||
session_db.append_message(source_id, "user", "before compression")
|
||||
session_db.end_session(source_id, "compression")
|
||||
session_db.create_session("tip-session", "api_server", parent_session_id=source_id)
|
||||
session_db.replace_messages(source_id, [])
|
||||
session_db.append_message("tip-session", "user", "after compression")
|
||||
|
||||
app = _create_session_app(adapter)
|
||||
async with TestClient(TestServer(app)) as cli:
|
||||
messages_resp = await cli.get(f"/api/sessions/{source_id}/messages")
|
||||
assert messages_resp.status == 200
|
||||
messages = await messages_resp.json()
|
||||
|
||||
assert messages["object"] == "list"
|
||||
assert messages["session_id"] == "tip-session"
|
||||
assert [m["content"] for m in messages["data"]] == ["after compression"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_session_fork_uses_current_sessiondb_branch_primitives(adapter, session_db):
|
||||
source_id = session_db.create_session("source-session", "api_server", model="test-model")
|
||||
|
||||
@@ -638,6 +638,37 @@ class TestWebServerEndpoints:
|
||||
for r in results
|
||||
)
|
||||
|
||||
def test_get_session_messages_follows_compression_tip(self):
|
||||
"""Reading a compressed session by its old id should hydrate from the
|
||||
live continuation, matching /resume behavior."""
|
||||
import time as _time
|
||||
|
||||
from hermes_state import SessionDB
|
||||
|
||||
db = SessionDB()
|
||||
try:
|
||||
db.create_session(session_id="desktop-root", source="cli")
|
||||
db.append_message(session_id="desktop-root", role="user", content="before compression")
|
||||
db.end_session("desktop-root", "compression")
|
||||
now = _time.time()
|
||||
db._conn.execute(
|
||||
"UPDATE sessions SET started_at = ?, ended_at = ? WHERE id = ?",
|
||||
(now - 10, now - 5, "desktop-root"),
|
||||
)
|
||||
db.create_session(session_id="desktop-tip", source="cli", parent_session_id="desktop-root")
|
||||
db._conn.execute("UPDATE sessions SET started_at = ? WHERE id = ?", (now - 4, "desktop-tip"))
|
||||
db.replace_messages("desktop-root", [])
|
||||
db.append_message(session_id="desktop-tip", role="user", content="after compression")
|
||||
db._conn.commit()
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
resp = self.client.get("/api/sessions/desktop-root/messages")
|
||||
assert resp.status_code == 200
|
||||
payload = resp.json()
|
||||
assert payload["session_id"] == "desktop-tip"
|
||||
assert [m["content"] for m in payload["messages"]] == ["after compression"]
|
||||
|
||||
def test_get_sessions_archived_is_boolean(self):
|
||||
from hermes_state import SessionDB
|
||||
|
||||
|
||||
Reference in New Issue
Block a user