feat(memory): improve OpenViking setup UX

This commit is contained in:
Hao Zhe 2026-06-17 01:00:48 +08:00
parent 3c76dac4fd
commit 2c2ca0443b
8 changed files with 1930 additions and 756 deletions

View file

@ -1153,6 +1153,9 @@ def init_agent(
"hermes_home": str(get_hermes_home()),
"agent_context": "primary",
}
if _init_kwargs["platform"] == "cli":
_init_kwargs["warning_callback"] = agent._emit_warning
_init_kwargs["status_callback"] = agent._emit_status
# Thread session title for memory provider scoping
# (e.g. honcho uses this to derive chat-scoped session keys)
if agent._session_db:

View file

@ -44,7 +44,9 @@ def _curses_select(
f"{label} - {desc}" if desc else label
for label, desc in items
]
return curses_radiolist(title, display_items, selected=default, cancel_returns=cancel_returns)
result = curses_radiolist(title, display_items, selected=default, cancel_returns=cancel_returns)
_clear_interactive_transition()
return result
def _print_cancelled_setup() -> None:
@ -229,6 +231,8 @@ def cmd_setup_provider(provider_name: str) -> None:
name, _, provider = match
_clear_interactive_transition()
_install_dependencies(name)
config = load_config()
@ -439,43 +443,53 @@ def cmd_status(args) -> None:
print(f" Built-in: always active")
print(f" Provider: {provider_name or '(none — built-in only)'}")
providers = _get_available_providers()
provider = None
for pname, _, candidate in providers:
if pname == provider_name:
provider = candidate
break
if provider_name:
provider_config = mem_config.get(provider_name, {})
if provider_config:
display_config = provider_config
if provider and hasattr(provider, "get_status_config"):
try:
display_config = provider.get_status_config(provider_config)
except Exception as e:
display_config = dict(provider_config) if isinstance(provider_config, dict) else provider_config
if isinstance(display_config, dict):
display_config["status_config_error"] = str(e)
if display_config:
print(f"\n {provider_name} config:")
for key, val in provider_config.items():
for key, val in display_config.items():
print(f" {key}: {val}")
providers = _get_available_providers()
found = any(name == provider_name for name, _, _ in providers)
if found:
if provider:
print(f"\n Plugin: installed ✓")
for pname, _, p in providers:
if pname == provider_name:
if p.is_available():
print(f" Status: available ✓")
else:
print(f" Status: not available ✗")
schema = p.get_config_schema() if hasattr(p, "get_config_schema") else []
# Check all fields that have env_var (both secret and non-secret)
required_fields = [f for f in schema if f.get("env_var")]
if required_fields:
print(f" Missing:")
for f in required_fields:
env_var = f.get("env_var", "")
url = f.get("url", "")
is_set = bool(os.environ.get(env_var))
mark = "" if is_set else ""
line = f" {mark} {env_var}"
if url and not is_set:
line += f"{url}"
print(line)
break
if provider.is_available():
print(f" Status: available ✓")
else:
print(f" Status: not available ✗")
schema = provider.get_config_schema() if hasattr(provider, "get_config_schema") else []
# Check all fields that have env_var (both secret and non-secret)
required_fields = [f for f in schema if f.get("env_var")]
if required_fields:
print(f" Missing:")
for f in required_fields:
env_var = f.get("env_var", "")
url = f.get("url", "")
is_set = bool(os.environ.get(env_var))
mark = "" if is_set else ""
line = f" {mark} {env_var}"
if url and not is_set:
line += f"{url}"
print(line)
else:
print(f"\n Plugin: NOT installed ✗")
print(f" Install the '{provider_name}' memory plugin to ~/.hermes/plugins/")
providers = _get_available_providers()
if providers:
print(f"\n Installed plugins:")
for pname, desc, _ in providers:

View file

@ -27,16 +27,16 @@ def _collect_masked_input(
while True:
ch = read_char()
if ch == "":
write("\n")
write("\r\n")
raise EOFError
if ch in _ENTER_CHARS:
write("\n")
write("\r\n")
return "".join(value)
if ch == "\x03":
write("\n")
write("\r\n")
raise KeyboardInterrupt
if ch in _EOF_CHARS:
write("\n")
write("\r\n")
raise EOFError
if ch in _BACKSPACE_CHARS:
if value:

File diff suppressed because it is too large Load diff

View file

@ -45,6 +45,22 @@ def test_curses_select_accepts_explicit_cancel_value(monkeypatch):
assert captured["cancel_returns"] == _CANCELLED
def test_curses_select_clears_after_picker_returns(monkeypatch):
events = []
def fake_radiolist(title, items, selected=0, *, cancel_returns=None):
events.append("picker")
return selected
monkeypatch.setattr("hermes_cli.curses_ui.curses_radiolist", fake_radiolist)
monkeypatch.setattr(memory_setup, "_clear_interactive_transition", lambda: events.append("clear"))
result = _curses_select("Pick one", [("first", "")], default=0)
assert result == 0
assert events == ["picker", "clear"]
def test_cmd_setup_top_level_cancel_writes_nothing(monkeypatch):
save_config = MagicMock()
load_config = MagicMock(side_effect=AssertionError("cancel should not load config"))
@ -95,6 +111,60 @@ def test_cmd_setup_clears_interactive_picker_before_provider_post_setup(monkeypa
assert events == ["select", "clear", "install", "post_setup"]
def test_cmd_setup_provider_clears_before_provider_post_setup(monkeypatch):
events = []
class PostSetupProvider:
def post_setup(self, hermes_home, config):
events.append("post_setup")
monkeypatch.setattr(memory_setup, "_get_available_providers", lambda: [("openviking", "local", PostSetupProvider())])
monkeypatch.setattr(memory_setup, "_clear_interactive_transition", lambda: events.append("clear"), raising=False)
monkeypatch.setattr(memory_setup, "_install_dependencies", lambda name: events.append("install"))
monkeypatch.setattr(memory_setup, "get_hermes_home", lambda: "/tmp/hermes-test")
monkeypatch.setattr("hermes_cli.config.load_config", lambda: {"memory": {}})
memory_setup.cmd_setup_provider("openviking")
assert events == ["clear", "install", "post_setup"]
def test_cmd_status_prefers_provider_status_config(monkeypatch, capsys):
class StatusProvider:
def get_status_config(self, provider_config):
assert provider_config["endpoint"] == "http://stale.local"
return {
"use_ovcli_config": True,
"ovcli_config_path": "/tmp/ovcli.conf.VPS_ROOT",
"endpoint": "https://vps.example",
"account": "acct",
"user": "alice",
"agent": "hermes",
}
def is_available(self):
return True
config = {
"memory": {
"provider": "openviking",
"openviking": {
"use_ovcli_config": True,
"ovcli_config_path": "/tmp/ovcli.conf.VPS_ROOT",
"endpoint": "http://stale.local",
},
}
}
monkeypatch.setattr("hermes_cli.config.load_config", lambda: config)
monkeypatch.setattr(memory_setup, "_get_available_providers", lambda: [("openviking", "API key / local", StatusProvider())])
memory_setup.cmd_status(SimpleNamespace())
output = capsys.readouterr().out
assert "endpoint: https://vps.example" in output
assert "http://stale.local" not in output
def test_cmd_setup_generic_choice_cancel_writes_nothing(tmp_path, monkeypatch):
class ChoiceProvider:
def __init__(self):

View file

@ -25,7 +25,7 @@ def test_collect_masked_input_shows_feedback_without_echoing_secret():
value, output = _run_collect("secret\n")
assert value == "secret"
assert output == "API key: ******\n"
assert output == "API key: ******\r\n"
assert "secret" not in output
@ -33,7 +33,7 @@ def test_collect_masked_input_handles_backspace():
value, output = _run_collect("sec\x7fret\r")
assert value == "seret"
assert output == "API key: ***\b \b***\n"
assert output == "API key: ***\b \b***\r\n"
assert "secret" not in output
@ -47,7 +47,7 @@ def test_collect_masked_input_raises_keyboard_interrupt():
"API key: ",
)
assert "".join(output) == "API key: \n"
assert "".join(output) == "API key: \r\n"
def test_masked_secret_prompt_falls_back_to_getpass_for_non_tty(monkeypatch):

File diff suppressed because it is too large Load diff

View file

@ -90,6 +90,8 @@ def test_aiagent_forwards_user_id_alt_to_memory_provider():
assert provider.init_kwargs["user_id"] == "open-id"
assert provider.init_kwargs["user_id_alt"] == "union-id"
assert provider.init_kwargs["platform"] == "feishu"
assert "warning_callback" not in provider.init_kwargs
assert "status_callback" not in provider.init_kwargs
class CoreShadowProvider:
@ -132,3 +134,34 @@ def test_core_tool_names_rejected_from_memory_routing_table():
assert "clarify" not in schema_names
assert "delegate_task" not in schema_names
assert "honcho_search" in schema_names
def test_aiagent_forwards_warning_callback_to_cli_memory_provider():
provider = RecordingMemoryProvider()
cfg = {"memory": {"provider": "recording"}, "agent": {}}
with (
patch("hermes_cli.config.load_config", return_value=cfg),
patch("plugins.memory.load_memory_provider", return_value=provider),
patch("agent.model_metadata.get_model_context_length", return_value=204_800),
patch("run_agent.get_tool_definitions", return_value=[]),
patch("run_agent.check_toolset_requirements", return_value={}),
patch("run_agent.OpenAI"),
):
from run_agent import AIAgent
agent = AIAgent(
api_key="test-key-1234567890",
base_url="https://openrouter.ai/api/v1",
quiet_mode=True,
skip_context_files=True,
skip_memory=False,
session_id="sess-cli",
platform="cli",
)
assert agent._memory_manager is not None
assert provider.init_session_id == "sess-cli"
assert provider.init_kwargs["platform"] == "cli"
assert provider.init_kwargs["warning_callback"] == agent._emit_warning
assert provider.init_kwargs["status_callback"] == agent._emit_status