feat(tui): wire /rewind through command.dispatch + prefill payload (#21910)

Adds the TUI half of the /rewind feature so the Ink terminal UI gets
the same affordance as the prompt_toolkit CLI.

Python side (tui_gateway/server.py):
- /rewind added to _PENDING_INPUT_COMMANDS so slash.exec rejects it
  and the TUI falls through to command.dispatch (the only path with
  access to live session state + memory hooks).
- New command.dispatch branch for name == "rewind":
  v1 auto-picks the most recent user turn (Claude-Code-style single-
  step undo), calls SessionDB.rewind_to_message, refreshes the
  in-memory history, fires _memory_manager.on_session_switch with
  rewound=True, and returns the new "prefill" payload.
- A dedicated picker overlay (multi-step rewind) is tracked as a
  follow-up to #21910.

TS side (ui-tui/src/):
- New "prefill" variant on CommandDispatchResponse + asCommandDispatch
  validator. Mirrors "send" but does NOT auto-submit; the client drops
  the message into the composer for editing.
- createSlashHandler renders the optional notice via sys() and calls
  ctx.composer.setInput(d.message), letting the user edit-and-resubmit
  the rewound turn — the core UX promised by the issue.

Tests:
- 7 new tui_gateway tests covering prefill payload shape, in-memory
  history truncation, DB soft-delete, memory-provider notification
  (rewound=True), busy-session refusal, missing-session error, and
  registry placement in _PENDING_INPUT_COMMANDS.
- Extended asCommandDispatch vitest covering the new prefill variant
  (with + without notice, and rejection of malformed payloads).

Out of scope for v1 (tracked as #21910 follow-up):
- Dedicated picker overlay in Ink (the multi-step rewind UI). v1 auto-
  picks the most recent user turn, matching the most common case.
- Gateway platforms (Telegram, Discord, etc.) — issue scopes v1 to
  CLI + TUI only.
This commit is contained in:
SaguaroDev
2026-05-10 18:45:32 -04:00
committed by Teknium
parent 31cfa08c66
commit 243e836dce
6 changed files with 276 additions and 0 deletions

View File

@ -5496,6 +5496,7 @@ _PENDING_INPUT_COMMANDS: frozenset[str] = frozenset(
"steer",
"plan",
"goal",
"rewind",
}
)
@ -5881,6 +5882,94 @@ def _(rid, params: dict) -> dict:
{"type": "send", "notice": notice, "message": state.goal},
)
if name == "rewind":
# /rewind: pick the most-recent user message and prefill the
# composer with its text after soft-deleting everything that
# came after it in the transcript. v1 auto-picks the latest
# user turn (Claude-Code-style single-step undo); a multi-step
# picker UI is tracked as a follow-up to issue #21910.
if not session:
return _err(rid, 4001, "no active session to rewind")
if session.get("running"):
return _err(
rid, 4009, "session busy — /interrupt the current turn before /rewind"
)
db = _get_db()
if db is None:
return _db_unavailable_error(rid, code=5008)
session_key = session.get("session_key", "")
if not session_key:
return _err(rid, 4001, "no session key for rewind")
try:
recents = db.list_recent_user_messages(session_key, limit=10)
except Exception as e:
return _err(rid, 5008, f"rewind: failed to load history: {e}")
if not recents:
return _err(rid, 4018, "no user messages to rewind to")
# v1: auto-pick the most recent user turn. The Ink UI does not
# yet host a dedicated picker overlay (#21910 follow-up).
target_id = recents[0]["id"]
try:
result = db.rewind_to_message(session_key, target_id)
except ValueError as e:
return _err(rid, 4004, f"rewind: {e}")
except Exception as e:
return _err(rid, 5008, f"rewind: {e}")
# Reload the active-only transcript into the in-memory session
# history so subsequent turns see the truncated view.
try:
active = db.get_messages_as_conversation(session_key)
except Exception:
active = []
with session["history_lock"]:
session["history"] = list(active)
session["history_version"] = int(session.get("history_version", 0)) + 1
# Notify memory providers — same hook /branch fires, plus the
# rewound flag so providers caching per-turn document state
# know to invalidate. See #6672 + #21910.
agent = session.get("agent")
if agent is not None:
mm = getattr(agent, "_memory_manager", None)
if mm is not None:
try:
mm.on_session_switch(
session_key,
parent_session_id="",
reset=False,
rewound=True,
)
except Exception:
pass
if hasattr(agent, "_invalidate_system_prompt"):
try:
agent._invalidate_system_prompt()
except Exception:
pass
if hasattr(agent, "_last_flushed_db_idx"):
try:
agent._last_flushed_db_idx = len(active)
except Exception:
pass
target_msg = result.get("target_message") or {}
target_text = target_msg.get("content") or ""
if isinstance(target_text, list):
parts = [
p.get("text", "") for p in target_text
if isinstance(p, dict) and p.get("type") == "text"
]
target_text = "\n".join(t for t in parts if t)
if not isinstance(target_text, str):
target_text = ""
rewound_count = result.get("rewound_count", 0)
notice = (
f"↶ Rewound {rewound_count} message(s). "
"Edit and resubmit, or send a new message."
)
return _ok(
rid,
{"type": "prefill", "message": target_text, "notice": notice},
)
if name in {"snapshot", "snap"}:
subcommand = arg.split(maxsplit=1)[0].lower() if arg else ""
if subcommand in {"restore", "rewind"}: