mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-27 17:58:07 +00:00
e769560c76
Append a "⊙ goal 3/20" segment (turns used / turn budget) to the CLI status bar whenever a standing /goal is active. Mirrors the desktop composer goal indicator: active-goal-only — paused/done goals stay out of the bar since they already print their own glyph lines in-thread. - Snapshot: goal_active / goal_turns_used / goal_max_turns from the cached GoalManager (in-memory attribute read, no DB hit per repaint). - Rendered in all three width tiers of both _build_status_bar_text and _get_status_bar_fragments, and it respects the /statusbar toggle for free (the toggle gates _get_status_bar_fragments as a whole). - Tests: segment composition, active-only contract, all width tiers. Status-bar goal indicator concept from #43020. Co-authored-by: Akshan Krithick <akshankrithick305@gmail.com> Assisted-by: Claude Fable 5 via Hermes Agent
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
"""Status-bar goal segment (⊙ goal N/M) — active-goal-only rendering.
|
|
|
|
The segment mirrors the desktop composer goal indicator: it appears only
|
|
while a /goal is ACTIVE, shows turns used vs the turn budget, and stays out
|
|
of the bar entirely for paused/done/absent goals (those already print their
|
|
own glyph lines in the conversation thread).
|
|
"""
|
|
|
|
from datetime import datetime, timedelta
|
|
from types import SimpleNamespace
|
|
|
|
from cli import HermesCLI
|
|
|
|
|
|
def _make_cli(model: str = "anthropic/claude-sonnet-4-20250514"):
|
|
cli_obj = HermesCLI.__new__(HermesCLI)
|
|
cli_obj.model = model
|
|
cli_obj.session_start = datetime.now() - timedelta(minutes=14, seconds=32)
|
|
cli_obj.conversation_history = [{"role": "user", "content": "hi"}]
|
|
cli_obj.agent = None
|
|
return cli_obj
|
|
|
|
|
|
def _attach_goal(cli_obj, *, active: bool, turns_used: int = 3, max_turns: int = 20):
|
|
"""Bind a fake GoalManager the way _get_goal_manager caches one."""
|
|
cli_obj.session_id = "sess-goal-test"
|
|
cli_obj._goal_manager = SimpleNamespace(
|
|
session_id="sess-goal-test",
|
|
is_active=lambda: active,
|
|
state=SimpleNamespace(turns_used=turns_used, max_turns=max_turns),
|
|
)
|
|
return cli_obj
|
|
|
|
|
|
class TestStatusBarGoalSegment:
|
|
def test_goal_segment_composition(self):
|
|
cli_obj = _attach_goal(_make_cli(), active=True, turns_used=3, max_turns=20)
|
|
|
|
snapshot = cli_obj._get_status_bar_snapshot()
|
|
|
|
assert snapshot["goal_active"] is True
|
|
assert snapshot["goal_turns_used"] == 3
|
|
assert snapshot["goal_max_turns"] == 20
|
|
assert cli_obj._status_bar_goal_segment(snapshot) == "⊙ goal 3/20"
|
|
|
|
def test_goal_segment_absent_without_goal(self):
|
|
cli_obj = _make_cli() # no session_id → no goal manager
|
|
|
|
snapshot = cli_obj._get_status_bar_snapshot()
|
|
|
|
assert snapshot["goal_active"] is False
|
|
assert cli_obj._status_bar_goal_segment(snapshot) == ""
|
|
|
|
def test_goal_segment_absent_when_paused(self):
|
|
# Paused goals must NOT occupy the status bar (active-only contract).
|
|
cli_obj = _attach_goal(_make_cli(), active=False)
|
|
|
|
snapshot = cli_obj._get_status_bar_snapshot()
|
|
|
|
assert snapshot["goal_active"] is False
|
|
assert cli_obj._status_bar_goal_segment(snapshot) == ""
|
|
|
|
def test_goal_segment_without_budget_omits_counter(self):
|
|
segment = HermesCLI._status_bar_goal_segment(
|
|
{"goal_active": True, "goal_turns_used": 0, "goal_max_turns": 0}
|
|
)
|
|
|
|
assert segment == "⊙ goal"
|
|
|
|
def test_active_goal_rendered_in_wide_status_bar(self):
|
|
cli_obj = _attach_goal(_make_cli(), active=True, turns_used=5, max_turns=20)
|
|
|
|
text = cli_obj._build_status_bar_text(width=120)
|
|
|
|
assert "⊙ goal 5/20" in text
|
|
|
|
def test_active_goal_rendered_in_medium_status_bar(self):
|
|
cli_obj = _attach_goal(_make_cli(), active=True, turns_used=1, max_turns=20)
|
|
|
|
text = cli_obj._build_status_bar_text(width=60)
|
|
|
|
assert "⊙ goal 1/20" in text
|
|
|
|
def test_active_goal_rendered_in_narrow_status_bar(self):
|
|
cli_obj = _attach_goal(_make_cli(), active=True, turns_used=2, max_turns=20)
|
|
|
|
text = cli_obj._build_status_bar_text(width=50)
|
|
|
|
assert "⊙ goal" in text
|
|
|
|
def test_no_goal_segment_in_status_bar_without_goal(self):
|
|
cli_obj = _make_cli()
|
|
|
|
text = cli_obj._build_status_bar_text(width=120)
|
|
|
|
assert "⊙ goal" not in text
|