Materialize CLI/TUI sessions before regenerating title (#4183)

This commit is contained in:
Rod Boev committed 2026-06-14 13:21:32 -04:00
1 parent d4d4b1e041
commit e0dbbaec82
2 files changed
+37 -3

No files matched your search

+2 -3
View File
@@ -7750,11 +7750,10 @@ def handle_post(handler, parsed) -> bool:
sid = body["session_id"]
prefer_latest = bool(body.get("prefer_latest", False))
try:
s = get_session(sid)
s = _ensure_full_session_before_mutation(sid, s)
s = _get_or_materialize_session(sid)
except KeyError:
return bad(handler, "Session not found", 404)
if getattr(s, "read_only", False):
except PermissionError:
return bad(handler, "Read-only imported sessions cannot be renamed", 403)
next_title, reason, raw_preview = generate_session_title_for_session(s, prefer_latest=prefer_latest)
if not next_title:
@@ -0,0 +1,35 @@
"""Issue #4183: regenerate-title must materialize CLI/TUI sessions from state.db.
The /api/session/title/regenerate endpoint must use _get_or_materialize_session
so sessions that only exist in state.db (no sidecar JSON) can have their titles
regenerated instead of returning "Session not found".
"""
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
ROUTES_PY = (ROOT / "api" / "routes.py").read_text(encoding="utf-8")
def test_regenerate_endpoint_uses_materialize_fallback():
start = ROUTES_PY.index('"/api/session/title/regenerate"')
end = ROUTES_PY.index('"/api/personality/set"', start)
block = ROUTES_PY[start:end]
assert "_get_or_materialize_session(sid)" in block, (
"regenerate handler must use _get_or_materialize_session to find "
"sessions that only exist in state.db (CLI/TUI sessions)"
)
assert "get_session(sid)" not in block, (
"regenerate handler must not use bare get_session — it misses "
"CLI/TUI sessions that lack a sidecar JSON file"
)
def test_regenerate_endpoint_catches_permission_error():
start = ROUTES_PY.index('"/api/session/title/regenerate"')
end = ROUTES_PY.index('"/api/personality/set"', start)
block = ROUTES_PY[start:end]
assert "except PermissionError:" in block, (
"regenerate handler must catch PermissionError from "
"_get_or_materialize_session for read-only imported sessions"
)