fix(cli): turn-end notifications no longer paint the pet's kitty frame as base64

Symptom (Ghostty, display.pet on, display.bell_on_complete on): at the end of a turn the
input line fills with several rows of base64 and a stale copy of the status bar + pet
stays above the response panel.

Root cause: _ring_bell runs on the agent thread and terminal_notify wrote the OSC 9 /
OSC 777 sequence through its own open("/dev/tty") (or sys.stdout). The prompt_toolkit
loop thread may at that moment be mid-write of a 12 KB kitty APC pet frame, which the tty
drains ~1 KB at a time. The second writer splices into the frame; the foreign ESC aborts
the APC and the terminal paints the remainder of the payload as text at the input cursor.
The wrapped garbage scrolls the screen, so the panel that follows is printed against a
stale cursor position and the old chrome survives above it.

Change: when the CLI's Application is running, _ring_bell hands "\a" + the notification
sequence to the app loop (_run_on_app_loop -> _write_terminal_sequence), serializing it
behind the renderer and the after_render frame writer. terminal_notify.notify() keeps the
/dev/tty path for callers without a running app; the sequence builder is split out as
notification_sequence().

Verification: pty A/B with the real Application + after_render frame writer, 400 rings
vs 91 frames — base: 3 leaks (4,324 base64 chars painted); fixed: 0 leaks, 400/400
notifications delivered, 0 aborted frames.
This commit is contained in:
kshitijk4poor
2026-09-15 14:13:22 +05:30
committed by kshitij
parent 2932195c22
commit ac2359cd35
3 changed files with 72 additions and 9 deletions

View File

@@ -549,18 +549,29 @@ class CLIModalMixin:
flag = "bell_on_prompt" if prompt else "bell_on_complete"
if not getattr(self, flag, False):
return
from hermes_cli.terminal_notify import notification_sequence, notify as _terminal_notify
body = context or ("input needed" if prompt else "turn complete")
session_id = getattr(self, "session_id", "") or ""
app = getattr(self, "_app", None)
if app is not None and getattr(app, "_is_running", False):
# Agent thread. The loop thread may be mid-write of a 12 KB kitty pet frame that the tty
# drains ~1 KB at a time; a second writer on the same tty (/dev/tty, sys.stdout) splices
# in, the foreign ESC aborts the APC, and the terminal paints the rest of the payload as
# base64 at the input cursor. Serialize behind the renderer instead.
from hermes_cli.cli_terminal_mixin import _run_on_app_loop, _write_terminal_sequence
try:
seq = "\a" + notification_sequence(body, prompt=prompt, session_id=session_id, detail=detail)
_run_on_app_loop(app, lambda: _write_terminal_sequence(app, seq))
except Exception:
pass
return
try:
sys.stdout.write("\a")
sys.stdout.flush()
except Exception:
pass
try:
from hermes_cli.terminal_notify import notify as _terminal_notify
_terminal_notify(
context or ("input needed" if prompt else "turn complete"),
prompt=prompt,
session_id=getattr(self, "session_id", "") or "",
detail=detail)
_terminal_notify(body, prompt=prompt, session_id=session_id, detail=detail)
except Exception:
pass

View File

@@ -65,10 +65,16 @@ def warp_osc777(event: str, detail: str, session_id: str = "") -> str:
return f"\x1b]777;notify;warp://cli-agent;{json.dumps(payload, separators=(',', ':'))}\x07"
def notify(context: str, *, prompt: bool, session_id: str = "", detail: str = "") -> None:
"""Emit OSC 9 (plus Warp OSC 777 when supported) for a blocking prompt or turn end."""
def notification_sequence(context: str, *, prompt: bool, session_id: str = "", detail: str = "") -> str:
"""OSC 9 (plus Warp OSC 777 when supported) for a blocking prompt or turn end."""
seq = osc9(f"Hermes: {context}")
if warp_supported():
event = "permission_request" if prompt else "stop"
seq += warp_osc777(event, detail or context, session_id)
_write_tty(seq)
return seq
def notify(context: str, *, prompt: bool, session_id: str = "", detail: str = "") -> None:
"""Emit the notification straight to the tty. Only for callers that do not own a running
prompt_toolkit app; inside the CLI, ``_ring_bell`` routes it through the app output instead."""
_write_tty(notification_sequence(context, prompt=prompt, session_id=session_id, detail=detail))

View File

@@ -1,6 +1,10 @@
"""display.bell_on_prompt / bell_on_complete also drive OSC 9 + Warp OSC 777 via _ring_bell."""
import io
import json
import sys
import pytest
from cli import HermesCLI
from hermes_cli import terminal_notify
@@ -48,3 +52,45 @@ def test_warp_osc777_only_under_supported_warp_build(monkeypatch):
# Not Warp at all → OSC 9 only.
not_warp = dict(_WARP_OK, TERM_PROGRAM="ghostty")
assert prefix not in _ring(monkeypatch, flag_on=True, env=not_warp, context="approval")
def test_running_app_gets_bell_and_osc9_on_its_loop_never_a_second_tty_writer(monkeypatch):
"""With the prompt_toolkit app live, the notification must reach the tty through the app's
output ON THE APP LOOP. A parallel /dev/tty or sys.stdout write from the agent thread splices
into an in-flight kitty pet frame and the terminal paints the frame's base64 as text."""
for key in _WARP_OK:
monkeypatch.delenv(key, raising=False)
monkeypatch.setattr(terminal_notify, "_write_tty", lambda seq: pytest.fail(f"stray tty write: {seq!r}"))
fake_stdout = io.StringIO()
monkeypatch.setattr(sys, "stdout", fake_stdout)
class _Output:
raw = []
def write_raw(self, data):
self.raw.append(data)
def flush(self):
self.raw.append("<flush>")
class _Loop:
queued = []
def call_soon_threadsafe(self, fn):
self.queued.append(fn)
class _App:
_is_running = True
loop = _Loop()
output = _Output()
cli = HermesCLI.__new__(HermesCLI)
cli.bell_on_complete = True
cli.session_id = "sess-1"
cli._app = _App()
cli._ring_bell(context="turn complete")
# Nothing touched the tty from the calling thread; the write is queued for the loop.
assert _Output.raw == [] and fake_stdout.getvalue() == ""
assert len(_Loop.queued) == 1
_Loop.queued[0]()
assert _Output.raw == ["\a\x1b]9;Hermes: turn complete\x07", "<flush>"]