rlmness-cli 0.2.0__py3-none-any.whl

rlmagent/tui/replay.py

@@ -0,0 +1,141 @@
+"""Saved runs read back: step cards for a past session, and per-agent totals."""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.widgets import Footer, Static
+
+from rlmagent.tui.steps import StepBook, short_id
+
+
+def events_path(sessions: Path, session_id: str) -> Path:
+    return sessions / f"{session_id}.backend.events.jsonl"
+
+
+def load_events(path: Path) -> list[dict]:
+    """The run events a session saved, skipping lines that do not parse."""
+    events = []
+    for line in path.read_text(encoding="utf-8").splitlines():
+        try:
+            event = json.loads(line)
+        except json.JSONDecodeError:
+            continue
+        if isinstance(event, dict):
+            events.append(event)
+    return events
+
+
+def runs_of(events: list[dict]) -> list[StepBook]:
+    """One book per run: a root run_started begins the next one."""
+    books: list[StepBook] = []
+    for event in events:
+        if event.get("kind") == "run_started" and not event.get("parent_run_id"):
+            books.append(StepBook())
+        if not books:
+            books.append(StepBook())
+        books[-1].feed(event)
+    return books
+
+
+def agent_rows(books: list[StepBook]) -> list[dict]:
+    """Per-agent steps, tokens, cost and seconds over every run of a session."""
+    rows = []
+    for run, book in enumerate(books, 1):
+        for agent in book.agents.values():
+            steps = [book.steps[(agent.run_id, n)] for n in agent.steps]
+            costs = [s.cost for s in steps if s.cost is not None]
+            seconds = [s.seconds for s in steps if s.seconds is not None]
+            rows.append({
+                "run": run,
+                "agent": agent.run_id,
+                "parent": agent.parent,
+                "depth": agent.depth,
+                "status": agent.status,
+                "steps": len(steps),
+                "failed_steps": sum(1 for s in steps if s.state == "failed"),
+                "tokens": sum(s.tokens for s in steps),
+                "cost": round(sum(costs), 6) if costs else None,
+                "seconds": round(sum(seconds), 2) if seconds else None,
+            })
+    return rows
+
+
+def format_rows(rows: list[dict]) -> str:
+    if not rows:
+        return "No agent steps were saved for this session."
+    lines = [f"{'run':>3}  {'agent':<8} {'depth':>5} {'status':<9} {'steps':>5} "
+             f"{'tokens':>8} {'cost':>9} {'seconds':>8}"]
+    for row in rows:
+        cost = "?" if row["cost"] is None else f"${row['cost']:.4f}"
+        seconds = "?" if row["seconds"] is None else f"{row['seconds']:.1f}"
+        lines.append(f"{row['run']:>3}  {short_id(row['agent']):<8} {row['depth']:>5} "
+                     f"{row['status']:<9} {row['steps']:>5} {row['tokens']:>8} "
+                     f"{cost:>9} {seconds:>8}")
+    tokens = sum(r["tokens"] for r in rows)
+    costs = [r["cost"] for r in rows if r["cost"] is not None]
+    total = f"${sum(costs):.4f}" if costs else "unknown"
+    lines.append(f"total: {len(rows)} agents, {sum(r['steps'] for r in rows)} steps, "
+                 f"{tokens} tokens, cost {total}")
+    return "\n".join(lines)
+
+
+class ReplayApp(App[None]):
+    """A saved session's steps as cards, read only."""
+
+    CSS_PATH = Path(__file__).with_name("rlm_agent.tcss")
+    TITLE = "rlm-agent replay"
+    BINDINGS = [Binding("q", "quit", "Quit"), Binding("f2", "agent_tree", "Agents"),
+                Binding("f3", "timeline", "Timeline")]
+
+    def __init__(self, session_id: str, books: list[StepBook]) -> None:
+        super().__init__()
+        self.session_id = session_id
+        self.books = books
+
+    def compose(self) -> ComposeResult:
+        from rlmagent.tui.transcript import TranscriptView
+
+        yield Static(f"session {self.session_id} · {len(self.books)} runs", classes="step-foot")
+        yield TranscriptView()
+        yield Footer()
+
+    def on_mount(self) -> None:
+        from rlmagent.tui.transcript import TranscriptView
+
+        view = self.query_one(TranscriptView)
+        for run, book in enumerate(self.books, 1):
+            view.add_notice(f"run {run}")
+            view.new_run()
+            for key in book.order:
+                view.show_step(book.steps[key])
+
+    def on_step_card_inspect(self, event) -> None:
+        from rlmagent.tui.inspect import StepScreen
+
+        self.push_screen(StepScreen(event.step))
+
+    def _merged(self) -> StepBook:
+        merged = StepBook()
+        for book in self.books:
+            merged.agents.update(book.agents)
+            merged.steps.update(book.steps)
+            merged.child_names.update(book.child_names)
+            merged.child_notes.update(book.child_notes)
+        return merged
+
+    def action_agent_tree(self) -> None:
+        from rlmagent.tui.inspect import AgentTreeScreen
+
+        self.push_screen(AgentTreeScreen(self._merged()))
+
+    def action_timeline(self) -> None:
+        from rlmagent.tui.inspect import TimelineScreen
+
+        self.push_screen(TimelineScreen(self.books[-1] if self.books else StepBook()))
+
+
+__all__ = ["ReplayApp", "agent_rows", "events_path", "format_rows", "load_events", "runs_of"]

rlmagent/tui/rlm_agent.tcss

@@ -0,0 +1,405 @@
+/* rlm-agent TUI — black, minimal, no chrome colour.
+   Geometry and palette live here so widget modules stay presentation-free. */
+
+Screen {
+    layout: vertical;
+    background: #000000;
+    color: #c8c8c8;
+}
+
+Header {
+    background: #000000;
+    color: #8a8a8a;
+    text-style: none;
+}
+
+Footer {
+    background: #000000;
+    color: #6a6a6a;
+}
+
+FooterKey {
+    background: #000000;
+    color: #6a6a6a;
+}
+
+FooterKey > .footer-key--key {
+    background: #000000;
+    color: #9a9a9a;
+}
+
+/* Row between header and footer: sidebar | main column. */
+#body {
+    height: 1fr;
+    width: 100%;
+    background: #000000;
+}
+
+Sidebar {
+    width: 28;
+    min-width: 18;
+    padding: 1 1;
+    background: #000000;
+    border-right: solid #1e1e1e;
+}
+
+#sidebar-title {
+    padding: 0 1 1 1;
+    color: #5a5a5a;
+    text-style: none;
+}
+
+#session-list {
+    height: 1fr;
+    background: #000000;
+    border: none;
+    scrollbar-size: 0 0;
+}
+
+#session-list > ListItem {
+    padding: 0 1;
+    background: #000000;
+    color: #7a7a7a;
+}
+
+#session-list > ListItem.active-session {
+    color: #e8e8e8;
+}
+
+#session-list > ListItem.-highlight {
+    background: #000000;
+    color: #c9a4ff;
+    text-style: none;
+}
+
+#main {
+    width: 1fr;
+    height: 100%;
+    background: #000000;
+}
+
+/* Transcript flexes; the composer below it is fixed height. */
+TranscriptView {
+    height: 1fr;
+    padding: 1 2;
+    background: #000000;
+    scrollbar-size: 0 0;
+}
+
+/* ── composer ─────────────────────────────────────────────────────── */
+
+Composer {
+    height: 3;
+    background: #000000;
+    border: solid #1e1e1e;
+}
+
+Composer:focus-within {
+    border: solid #333333;
+}
+
+#prompt-mark {
+    width: 3;
+    padding: 0 1;
+    color: #c9a4ff;
+    background: #000000;
+}
+
+PromptInput {
+    width: 1fr;
+    height: 1fr;
+    background: #000000;
+    border: none;
+    padding: 0;
+    scrollbar-size: 0 0;
+}
+
+#model-badge {
+    width: auto;
+    padding: 0 1;
+    color: #5a5a5a;
+    background: #000000;
+    content-align: right middle;
+}
+
+/* ── autocomplete ─────────────────────────────────────────────────── */
+
+#completion-bar {
+    height: auto;
+    max-height: 8;
+    background: #0a0a0a;
+    border: solid #1e1e1e;
+    padding: 0 1;
+}
+
+#completion-body {
+    color: #7a7a7a;
+    background: #0a0a0a;
+}
+
+/* ── model picker modal ───────────────────────────────────────────── */
+
+ModelScreen {
+    align: center middle;
+    background: #000000 70%;
+}
+
+#model-dialog {
+    width: 52;
+    height: auto;
+    max-height: 16;
+    padding: 1 2;
+    background: #0a0a0a;
+    border: solid #333333;
+}
+
+#model-title {
+    padding: 0 0 1 0;
+    color: #e8e8e8;
+}
+
+#model-list {
+    height: auto;
+    max-height: 8;
+    background: #0a0a0a;
+    border: none;
+    scrollbar-size: 0 0;
+}
+
+#model-list > ListItem {
+    padding: 0 1;
+    background: #0a0a0a;
+    color: #9a9a9a;
+}
+
+#model-list > ListItem.-highlight {
+    background: #0a0a0a;
+    color: #c9a4ff;
+    text-style: none;
+}
+
+#model-key {
+    background: #0a0a0a;
+    color: #e8e8e8;
+    border: solid #1e1e1e;
+}
+
+#model-key:focus {
+    border: solid #333333;
+}
+
+#model-hint {
+    padding: 1 0 0 0;
+    color: #5a5a5a;
+}
+
+/* ── status ───────────────────────────────────────────────────────── */
+
+#status-bar {
+    height: 1;
+    padding: 0 1;
+    background: #000000;
+    color: #5a5a5a;
+}
+
+/* Visible while a generation is running and can be stopped. */
+#status-bar.busy {
+    color: #c9a4ff;
+}
+
+/* ── transcript blocks ────────────────────────────────────────────── */
+
+.user-block {
+    padding: 1 0 0 0;
+    color: #e8e8e8;
+}
+
+.assistant-block {
+    padding: 0 0 1 0;
+    color: #b4b4b4;
+}
+
+.tool-code {
+    padding: 0 0 0 4;
+    height: auto;
+}
+
+.tool-line {
+    padding: 0 0 0 2;
+    color: #6a6a6a;
+}
+
+.tool-error {
+    color: #a05050;
+}
+
+.tool-output {
+    padding: 0 0 0 2;
+    height: auto;
+    color: #6a6a6a;
+}
+
+.notice-block {
+    padding: 0 0 1 0;
+    color: #5a5a5a;
+}
+
+#approval-dialog {
+    width: 80;
+    height: auto;
+    max-height: 30;
+    padding: 1 2;
+    background: #0a0a0a;
+    border: solid #333333;
+}
+
+#approval-body {
+    max-height: 18;
+    overflow-y: auto;
+    margin: 1 0;
+}
+
+/* Step cards: one per agent step, indented by depth in code. */
+.step-card {
+    height: auto;
+    margin-bottom: 1;
+    padding: 0 1;
+    border-left: solid #2a2a2a;
+}
+
+.step-card.step-live {
+    border-left: solid #c9a4ff;
+}
+
+.step-card.step-failed {
+    border-left: solid #d36a6a;
+}
+
+.step-head {
+    color: #8a8a8a;
+}
+
+.step-card.step-live > .step-head {
+    color: #c9a4ff;
+}
+
+.step-output {
+    color: #9a9a9a;
+}
+
+.step-foot {
+    color: #5a5a5a;
+}
+
+.step-card:focus {
+    background: #0d0d0d;
+}
+
+.inspect-screen {
+    align: center middle;
+    background: #000000 80%;
+}
+
+.inspect-body {
+    width: 90%;
+    height: 90%;
+    padding: 1 2;
+    background: #000000;
+    border: solid #2a2a2a;
+}
+
+#task-panel {
+    height: auto;
+    max-height: 50%;
+    padding: 1 1 0 1;
+    color: #8a8a8a;
+    border-top: solid #1e1e1e;
+}
+
+/* ── folded errors, shell box, echoes, side answers, queue ───────── */
+
+.error-block {
+    height: auto;
+    padding: 0 0 1 0;
+}
+
+.shell-block {
+    height: auto;
+    margin: 0 0 1 2;
+    padding: 0 1;
+    color: #9a9a9a;
+    border: round #333333;
+    border-title-color: #c9a4ff;
+    border-subtitle-color: #5a5a5a;
+}
+
+.shell-block.shell-failed {
+    border: round #a05050;
+}
+
+.command-echo {
+    padding: 1 0 0 0;
+    color: #9a9a9a;
+}
+
+.side-answer {
+    height: auto;
+    margin: 0 0 1 2;
+    padding: 0 1;
+    color: #9a9a9a;
+    border-left: thick #c9a4ff;
+}
+
+.kind-notice {
+    padding: 0 0 1 0;
+    color: #6a6a6a;
+}
+
+#queued-strip {
+    height: auto;
+    max-height: 6;
+    padding: 0 1;
+    color: #8a8a8a;
+    background: #0a0a0a;
+    border-top: solid #1e1e1e;
+}
+
+/* ── session search and sections ─────────────────────────────────── */
+
+#session-search {
+    height: 3;
+    margin: 0 0 1 0;
+    background: #0a0a0a;
+    border: solid #1e1e1e;
+    color: #c8c8c8;
+}
+
+#session-search:focus {
+    border: solid #333333;
+}
+
+#session-list > ListItem.session-heading {
+    padding: 1 1 0 1;
+    color: #5a5a5a;
+    background: #000000;
+}
+
+/* ── session search and sections ─────────────────────────────────── */
+
+#session-search {
+    height: 3;
+    margin: 0 0 1 0;
+    background: #0a0a0a;
+    border: solid #1e1e1e;
+    color: #c8c8c8;
+}
+
+#session-search:focus {
+    border: solid #333333;
+}
+
+#session-list > ListItem.session-heading {
+    padding: 1 1 0 1;
+    color: #5a5a5a;
+    background: #000000;
+}

rlmagent/tui/search_box.py

@@ -0,0 +1,91 @@
+"""A one-line search field with undo and Emacs-style kill and yank.
+
+Ctrl+Z undoes the last change. Ctrl+K cuts to the end, Ctrl+U to the start,
+Ctrl+W the word before the cursor; Ctrl+Y puts the last cut back at the
+cursor. Cuts in a row join into one, as a terminal user would expect.
+"""
+
+from __future__ import annotations
+
+from textual.binding import Binding
+from textual.widgets import Input
+
+#: How many earlier values Ctrl+Z can step back through.
+UNDO_DEPTH = 50
+
+
+class SearchBox(Input):
+    """``Input`` plus undo history and a single-slot kill ring."""
+
+    BINDINGS = [
+        Binding("ctrl+z", "undo", "Undo", show=False),
+        Binding("ctrl+k", "kill_to_end", "Cut to end", show=False),
+        Binding("ctrl+u", "kill_to_start", "Cut to start", show=False),
+        Binding("ctrl+w", "kill_word", "Cut word", show=False),
+        Binding("ctrl+y", "yank", "Paste cut", show=False),
+    ]
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+        self._history: list[str] = []
+        self._last = ""
+        self._undoing = False
+        self.killed = ""
+        self._killing = False
+
+    def on_input_changed(self, event: Input.Changed) -> None:
+        # Record the value before each change so Ctrl+Z can return to it.
+        if event.input is self and not self._undoing and event.value != self._last:
+            self._history.append(self._last)
+            del self._history[:-UNDO_DEPTH]
+        self._last = event.value
+        self._undoing = False
+
+    def action_undo(self) -> None:
+        if self._history:
+            self._undoing = True
+            self.value = self._history.pop()
+            self.cursor_position = len(self.value)
+
+    def _cut(self, start: int, end: int, *, before: bool) -> None:
+        piece = self.value[start:end]
+        if not piece:
+            return
+        # Consecutive cuts build one piece, in reading order.
+        if self._killing:
+            self.killed = piece + self.killed if before else self.killed + piece
+        else:
+            self.killed = piece
+        self.value = self.value[:start] + self.value[end:]
+        self.cursor_position = start
+        self._killing = True
+
+    def action_kill_to_end(self) -> None:
+        self._cut(self.cursor_position, len(self.value), before=False)
+
+    def action_kill_to_start(self) -> None:
+        self._cut(0, self.cursor_position, before=True)
+
+    def action_kill_word(self) -> None:
+        end = self.cursor_position
+        start = end
+        while start and self.value[start - 1] == " ":
+            start -= 1
+        while start and self.value[start - 1] != " ":
+            start -= 1
+        self._cut(start, end, before=True)
+
+    def action_yank(self) -> None:
+        if self.killed:
+            at = self.cursor_position
+            self.value = self.value[:at] + self.killed + self.value[at:]
+            self.cursor_position = at + len(self.killed)
+        self._killing = False
+
+    def on_key(self, event) -> None:
+        # Any other key ends a run of cuts.
+        if event.key not in ("ctrl+k", "ctrl+u", "ctrl+w"):
+            self._killing = False
+
+
+__all__ = ["SearchBox", "UNDO_DEPTH"]