mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-04-25 00:51:20 +00:00
92 lines
3.2 KiB
Python
92 lines
3.2 KiB
Python
import importlib
|
|
import os
|
|
import sys
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
|
|
|
|
_cli_mod = None
|
|
|
|
|
|
def _make_cli(user_message_preview=None):
|
|
global _cli_mod
|
|
clean_config = {
|
|
"model": {
|
|
"default": "anthropic/claude-opus-4.6",
|
|
"base_url": "https://openrouter.ai/api/v1",
|
|
"provider": "auto",
|
|
},
|
|
"display": {
|
|
"compact": False,
|
|
"tool_progress": "all",
|
|
"user_message_preview": user_message_preview or {"first_lines": 2, "last_lines": 2},
|
|
},
|
|
"agent": {},
|
|
"terminal": {"env_type": "local"},
|
|
}
|
|
clean_env = {"LLM_MODEL": "", "HERMES_MAX_ITERATIONS": ""}
|
|
prompt_toolkit_stubs = {
|
|
"prompt_toolkit": MagicMock(),
|
|
"prompt_toolkit.history": MagicMock(),
|
|
"prompt_toolkit.styles": MagicMock(),
|
|
"prompt_toolkit.patch_stdout": MagicMock(),
|
|
"prompt_toolkit.application": MagicMock(),
|
|
"prompt_toolkit.layout": MagicMock(),
|
|
"prompt_toolkit.layout.processors": MagicMock(),
|
|
"prompt_toolkit.filters": MagicMock(),
|
|
"prompt_toolkit.layout.dimension": MagicMock(),
|
|
"prompt_toolkit.layout.menus": MagicMock(),
|
|
"prompt_toolkit.widgets": MagicMock(),
|
|
"prompt_toolkit.key_binding": MagicMock(),
|
|
"prompt_toolkit.completion": MagicMock(),
|
|
"prompt_toolkit.formatted_text": MagicMock(),
|
|
"prompt_toolkit.auto_suggest": MagicMock(),
|
|
}
|
|
with patch.dict(sys.modules, prompt_toolkit_stubs), patch.dict("os.environ", clean_env, clear=False):
|
|
import cli as mod
|
|
|
|
mod = importlib.reload(mod)
|
|
_cli_mod = mod
|
|
with patch.object(mod, "get_tool_definitions", return_value=[]), patch.dict(mod.__dict__, {"CLI_CONFIG": clean_config}):
|
|
return mod.HermesCLI()
|
|
|
|
|
|
class TestSubmittedUserMessagePreview:
|
|
def test_default_preview_shows_first_two_lines_and_last_two_lines(self):
|
|
cli = _make_cli()
|
|
|
|
rendered = cli._format_submitted_user_message_preview(
|
|
"line1\nline2\nline3\nline4\nline5\nline6"
|
|
)
|
|
|
|
assert "line1" in rendered
|
|
assert "line2" in rendered
|
|
assert "line5" in rendered
|
|
assert "line6" in rendered
|
|
assert "line3" not in rendered
|
|
assert "line4" not in rendered
|
|
assert "(+2 more lines)" in rendered
|
|
|
|
def test_preview_can_hide_last_lines(self):
|
|
cli = _make_cli({"first_lines": 2, "last_lines": 0})
|
|
|
|
rendered = cli._format_submitted_user_message_preview(
|
|
"line1\nline2\nline3\nline4\nline5\nline6"
|
|
)
|
|
|
|
assert "line1" in rendered
|
|
assert "line2" in rendered
|
|
assert "line5" not in rendered
|
|
assert "line6" not in rendered
|
|
assert "(+4 more lines)" in rendered
|
|
|
|
def test_invalid_first_lines_value_falls_back_to_one(self):
|
|
cli = _make_cli({"first_lines": 0, "last_lines": 2})
|
|
|
|
rendered = cli._format_submitted_user_message_preview("line1\nline2\nline3\nline4")
|
|
|
|
assert "line1" in rendered
|
|
assert "line3" in rendered
|
|
assert "line4" in rendered
|
|
assert "(+1 more line)" in rendered
|