mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-27 11:22:03 +00:00
83aa84ae3b
Render the reactive pet pane in the classic CLI (steady redraw, right-aligned) and wire the /pet command to list and switch pets, plus an enable/disable toggle. Backed by hermes_cli/pets.py and the CLI commands mixin, registered in the central command registry. Covered by the CLI pet pane and toggle tests.
136 lines
4.5 KiB
Python
136 lines
4.5 KiB
Python
"""The base-CLI petdex pane: reactive half-block sprite above the prompt.
|
|
|
|
Mirrors the TUI's PetPane. The methods are tested in isolation via __new__ so
|
|
we don't pay the full HermesCLI.__init__ cost; a synthetic spritesheet exercises
|
|
the real engine decode + half-block fragment building.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import threading
|
|
|
|
import pytest
|
|
|
|
from agent.pet import store
|
|
from agent.pet.constants import FRAME_H, FRAME_W
|
|
from agent.pet.render import PetRenderer
|
|
from cli import HermesCLI
|
|
|
|
|
|
@pytest.fixture
|
|
def boba_like(tmp_path, monkeypatch):
|
|
"""Install a synthetic pet into a temp HERMES_HOME and return its slug."""
|
|
from PIL import Image
|
|
|
|
home = tmp_path / ".hermes"
|
|
home.mkdir()
|
|
monkeypatch.setenv("HERMES_HOME", str(home))
|
|
|
|
cols, rows = 8, 9
|
|
sheet = Image.new("RGBA", (FRAME_W * cols, FRAME_H * rows), (0, 0, 0, 0))
|
|
for r in range(rows):
|
|
color = (20 + r * 25, 60, 120, 255)
|
|
for c in range(cols):
|
|
block = Image.new("RGBA", (FRAME_W, FRAME_H), color)
|
|
sheet.paste(block, (c * FRAME_W, r * FRAME_H))
|
|
|
|
pet_dir = store.pets_dir() / "boba"
|
|
pet_dir.mkdir(parents=True, exist_ok=True)
|
|
sheet.save(pet_dir / "spritesheet.webp")
|
|
(pet_dir / "pet.json").write_text(
|
|
'{"id":"boba","displayName":"Boba","description":"d","spritesheetPath":"spritesheet.webp"}'
|
|
)
|
|
return "boba"
|
|
|
|
|
|
def _make_cli():
|
|
cli_obj = HermesCLI.__new__(HermesCLI)
|
|
cli_obj._app = None
|
|
cli_obj._pet_lock = threading.Lock()
|
|
cli_obj._pet_enabled = False
|
|
cli_obj._pet_renderer = None
|
|
cli_obj._pet_slug = ""
|
|
cli_obj._pet_cols = 18
|
|
cli_obj._pet_scale = 0.7
|
|
cli_obj._pet_frames_cache = {}
|
|
cli_obj._pet_frame_idx = 0
|
|
cli_obj._agent_running = False
|
|
# Transient-beat + reasoning state (set by HermesCLI.__init__ in production).
|
|
cli_obj._pet_event = ""
|
|
cli_obj._pet_event_until = 0.0
|
|
cli_obj._pet_reasoning = False
|
|
# Blocking-modal state — a live one maps the pet to `waiting`.
|
|
cli_obj._approval_state = None
|
|
cli_obj._clarify_state = None
|
|
cli_obj._sudo_state = None
|
|
cli_obj._secret_state = None
|
|
cli_obj._slash_confirm_state = None
|
|
return cli_obj
|
|
|
|
|
|
def test_pet_state_tracks_agent_running():
|
|
cli_obj = _make_cli()
|
|
assert cli_obj._derive_pet_state() == "idle"
|
|
cli_obj._agent_running = True
|
|
assert cli_obj._derive_pet_state() == "run"
|
|
|
|
|
|
def test_pet_state_waits_on_a_blocking_modal():
|
|
# A live clarify/approval pauses the agent on the user → `waiting`, even
|
|
# while the turn is technically still running.
|
|
cli_obj = _make_cli()
|
|
cli_obj._agent_running = True
|
|
cli_obj._clarify_state = {"question": "?"}
|
|
assert cli_obj._derive_pet_state() == "waiting"
|
|
|
|
|
|
def test_pet_pane_collapsed_when_disabled():
|
|
# No renderer resolved → the window reports zero height and no fragments,
|
|
# so it's invisible for users without a pet.
|
|
cli_obj = _make_cli()
|
|
assert cli_obj._pet_widget_height() == 0
|
|
assert cli_obj._pet_fragments() == []
|
|
|
|
|
|
def test_pet_fragments_render_half_blocks(boba_like):
|
|
cli_obj = _make_cli()
|
|
cli_obj._pet_renderer = PetRenderer(
|
|
str(store.load_pet("boba").spritesheet), mode="unicode", scale=0.4, unicode_cols=14
|
|
)
|
|
cli_obj._pet_cols = 14
|
|
cli_obj._pet_enabled = True
|
|
|
|
height = cli_obj._pet_widget_height()
|
|
assert height > 0
|
|
|
|
frags = cli_obj._pet_fragments()
|
|
assert frags, "expected fragments for an enabled pet"
|
|
# Each fragment is a (style, text) pair; glyphs are half-blocks or blanks.
|
|
glyphs = {text for _, text in frags}
|
|
assert glyphs <= {"▀", "▄", " ", "\n"}
|
|
# Opaque cells carry a truecolor foreground style.
|
|
assert any(text == "▀" and "fg:#" in style for style, text in frags)
|
|
# Row count in the fragment stream matches the reported window height.
|
|
assert sum(1 for _, text in frags if text == "\n") == height - 1
|
|
|
|
|
|
def test_pet_resolve_config_enables_and_disables(boba_like):
|
|
from hermes_cli.config import load_config, save_config
|
|
|
|
cli_obj = _make_cli()
|
|
|
|
cfg = load_config()
|
|
cfg.setdefault("display", {}).setdefault("pet", {})
|
|
cfg["display"]["pet"].update({"enabled": True, "slug": "boba"})
|
|
save_config(cfg)
|
|
|
|
cli_obj._pet_resolve_config()
|
|
assert cli_obj._pet_enabled is True
|
|
assert cli_obj._pet_renderer is not None
|
|
assert cli_obj._pet_slug == "boba"
|
|
|
|
cfg["display"]["pet"]["enabled"] = False
|
|
save_config(cfg)
|
|
cli_obj._pet_resolve_config()
|
|
assert cli_obj._pet_enabled is False
|
|
assert cli_obj._pet_renderer is None
|