Add Memory Gateway agent plugin

This commit is contained in:
2026-05-06 16:10:04 +08:00
parent e65731a273
commit c44af407d4
48 changed files with 3111 additions and 0 deletions

View File

@ -0,0 +1,28 @@
from __future__ import annotations
from test_hermes_register_tools import FakeHermesContext, load_plugin_module
def test_register_registers_expected_hooks():
module = load_plugin_module()
ctx = FakeHermesContext()
module.register(ctx)
assert [item[0] for item in ctx.registered_hooks] == [
"on_session_start",
"pre_llm_call",
"post_llm_call",
"on_session_end",
]
assert all(callable(item[1]) for item in ctx.registered_hooks)
def test_hook_callbacks_accept_kwargs():
module = load_plugin_module()
assert isinstance(module.on_session_start(session_id="s", extra="x"), dict)
assert isinstance(module.pre_llm_call(user_message="", session_id="s", extra="x"), dict)
assert module.post_llm_call(user_message="hi", assistant_response="hello", extra="x") is None
assert module.on_session_end(session_id="s", extra="x") is None