Teknium
3a8e3a2e88
BASE (63279301bc) exported agent.agent_init._moa_reference_output_allowed and _relay_moa_reference_event (salvaged in3dfe712384from #67334, plugin-importable, zero in-tree callers) and covered them with tests/agent/test_moa_quiet_reference_output.py. The simplification PR deleted both the helpers and the test. Restore them byte-identical to BASE. Note: the live build_moa_facade relay in agent/moa_loop.py is unchanged. A/B harness (/tmp/rf/rev/moa_quiet_ab.py) shows BASE's facade relay already fired moa.reference under platform=cli/tool_progress_mode=off — the guard only ever lived in the orphan helper. -Q is protected on both trees by cli.py nulling agent.tool_progress_callback (_configure_quiet_agent / BASE cli.py:22308).
82 lines
2.3 KiB
Python
82 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()
|