mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-25 17:18:11 +00:00
3dfe712384
Keep MoA reference display events off the machine-readable -Q stdout surface (platform=cli with tool_progress_mode=off) while preserving them everywhere else. Extracts the relay into module-level helpers so the policy is testable. Salvaged from #67334.
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
"""Regression coverage for machine-readable MoA quiet output."""
|
|
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
import unittest
|
|
|
|
from agent.agent_init import _relay_moa_reference_event
|
|
|
|
|
|
class MoAQuietReferenceOutputTests(unittest.TestCase):
|
|
@staticmethod
|
|
def _agent(*, platform: str, tool_progress_mode: str, quiet_mode: bool = True):
|
|
calls = []
|
|
|
|
def callback(*args, **kwargs):
|
|
calls.append((args, kwargs))
|
|
|
|
return SimpleNamespace(
|
|
platform=platform,
|
|
tool_progress_mode=tool_progress_mode,
|
|
quiet_mode=quiet_mode,
|
|
tool_progress_callback=callback,
|
|
), calls
|
|
|
|
def test_machine_readable_cli_suppresses_reference_relay(self) -> None:
|
|
agent, calls = self._agent(platform="cli", tool_progress_mode="off")
|
|
_relay_moa_reference_event(
|
|
agent,
|
|
"moa.reference",
|
|
label="local:advisor",
|
|
text="hidden",
|
|
index=1,
|
|
count=1,
|
|
)
|
|
self.assertEqual(calls, [])
|
|
|
|
def test_interactive_cli_delivers_reference_relay(self) -> None:
|
|
agent, calls = self._agent(
|
|
platform="cli",
|
|
tool_progress_mode="all",
|
|
quiet_mode=True,
|
|
)
|
|
_relay_moa_reference_event(
|
|
agent,
|
|
"moa.reference",
|
|
label="local:advisor",
|
|
text="visible",
|
|
index=1,
|
|
count=2,
|
|
)
|
|
self.assertEqual(
|
|
calls,
|
|
[
|
|
(
|
|
("moa.reference", "local:advisor", "visible", None),
|
|
{"moa_index": 1, "moa_count": 2},
|
|
)
|
|
],
|
|
)
|
|
|
|
def test_gateway_delivers_even_when_progress_mode_is_off(self) -> None:
|
|
agent, calls = self._agent(platform="discord", tool_progress_mode="off")
|
|
_relay_moa_reference_event(
|
|
agent,
|
|
"moa.aggregating",
|
|
aggregator="local:aggregator",
|
|
ref_count=2,
|
|
)
|
|
self.assertEqual(
|
|
calls,
|
|
[
|
|
(
|
|
("moa.aggregating", "local:aggregator", None, None),
|
|
{"moa_ref_count": 2},
|
|
)
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|