红尘
d73ff9b0fb
Deploy Site / deploy-vercel (push) Has been cancelled
Deploy Site / deploy-docs (push) Has been cancelled
Docker / shell lint / Lint Dockerfile (hadolint) (push) Has been cancelled
Docker / shell lint / Lint docker/ shell scripts (shellcheck) (push) Has been cancelled
Docker Build and Publish / build-amd64 (push) Has been cancelled
Docker Build and Publish / build-arm64 (push) Has been cancelled
Lint (ruff + ty) / ruff + ty diff (push) Has been cancelled
Lint (ruff + ty) / ruff enforcement (blocking) (push) Has been cancelled
Lint (ruff + ty) / Windows footguns (blocking) (push) Has been cancelled
Nix Lockfile Fix / auto-fix-main (push) Has been cancelled
Nix Lockfile Fix / fix (push) Has been cancelled
Nix / nix (macos-latest) (push) Has been cancelled
Nix / nix (ubuntu-latest) (push) Has been cancelled
OSV-Scanner / Scan lockfiles (push) Has been cancelled
Build Skills Index / build-index (push) Has been cancelled
Tests / test (1) (push) Has been cancelled
Tests / test (2) (push) Has been cancelled
Tests / test (3) (push) Has been cancelled
Tests / test (4) (push) Has been cancelled
Tests / test (5) (push) Has been cancelled
Tests / test (6) (push) Has been cancelled
Tests / e2e (push) Has been cancelled
uv.lock check / uv lock --check (push) Has been cancelled
Docker Build and Publish / merge (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
Tests / save-durations (push) Has been cancelled
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
"""Regression tests for memory provider selection during AIAgent init."""
|
|
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
|
|
class RecordingMemoryProvider:
|
|
name = "recording"
|
|
|
|
def __init__(self):
|
|
self.init_kwargs = None
|
|
self.init_session_id = None
|
|
|
|
def is_available(self):
|
|
return True
|
|
|
|
def initialize(self, session_id, **kwargs):
|
|
self.init_session_id = session_id
|
|
self.init_kwargs = dict(kwargs)
|
|
|
|
def get_tool_schemas(self):
|
|
return []
|
|
|
|
def shutdown(self):
|
|
pass
|
|
|
|
|
|
def test_blank_memory_provider_does_not_auto_enable_honcho():
|
|
"""Blank memory.provider should remain opt-out even if Honcho fallback looks configured."""
|
|
cfg = {"memory": {"provider": ""}, "agent": {}}
|
|
honcho_cfg = SimpleNamespace(enabled=True, api_key="stale-key", base_url=None)
|
|
|
|
with (
|
|
patch("hermes_cli.config.load_config", return_value=cfg),
|
|
patch("hermes_cli.config.save_config") as save_config,
|
|
patch(
|
|
"plugins.memory.honcho.client.HonchoClientConfig.from_global_config",
|
|
return_value=honcho_cfg,
|
|
) as from_global_config,
|
|
patch("plugins.memory.load_memory_provider") as load_memory_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,
|
|
)
|
|
|
|
assert agent._memory_manager is None
|
|
from_global_config.assert_not_called()
|
|
load_memory_provider.assert_not_called()
|
|
save_config.assert_not_called()
|
|
|
|
|
|
def test_aiagent_forwards_user_id_alt_to_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-alt",
|
|
platform="feishu",
|
|
user_id="open-id",
|
|
user_id_alt="union-id",
|
|
)
|
|
|
|
assert agent._memory_manager is not None
|
|
assert provider.init_session_id == "sess-alt"
|
|
assert provider.init_kwargs["user_id"] == "open-id"
|
|
assert provider.init_kwargs["user_id_alt"] == "union-id"
|
|
assert provider.init_kwargs["platform"] == "feishu"
|