feat(memory): improve OpenViking setup UX

This commit is contained in:
Hao Zhe
2026-06-17 01:04:26 +08:00
parent 3c76dac4fd
commit 2c2ca0443b
8 changed files with 1930 additions and 756 deletions
@@ -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