Skip to main content

构建记忆提供者插件

记忆提供者插件为 Hermes Agent 提供持久化、跨会话的知识能力,超越内置的 MEMORY.md 和 USER.md。本指南将介绍如何构建一个。

tip

记忆提供者是两种 提供者插件 类型之一。另一种是 上下文引擎插件 ,用于替代内置的上下文压缩器。两者均遵循相同的模式:单选、配置驱动,并通过 hermes plugins 进行管理。

目录结构

每个记忆提供者都位于 plugins/memory/<name>/ 中:

plugins/memory/my-provider/
├── __init__.py # MemoryProvider implementation + register() entry point
├── plugin.yaml # Metadata (name, description, hooks)
└── README.md # Setup instructions, config reference, tools

MemoryProvider 抽象基类

你的插件需实现来自 agent/memory_provider.pyMemoryProvider 抽象基类:

from agent.memory_provider import MemoryProvider

class MyMemoryProvider(MemoryProvider):
@property
def name(self) -> str:
return "my-provider"

def is_available(self) -> bool:
"""Check if this provider can activate. NO network calls."""
return bool(os.environ.get("MY_API_KEY"))

def initialize(self, session_id: str, **kwargs) -> None:
"""Called once at agent startup.

kwargs always includes:
hermes_home (str): Active HERMES_HOME path. Use for storage.
"""
self._api_key = os.environ.get("MY_API_KEY", "")
self._session_id = session_id

# ... implement remaining methods

必需方法

核心生命周期

方法调用时机是否必须实现?
name (属性)始终调用
is_available()Agent 初始化前,激活前 — 禁止网络调用
initialize(session_id, **kwargs)Agent 启动时
get_tool_schemas()初始化后,用于工具注入
handle_tool_call(name, args)当 agent 使用你的工具时(如果你有工具)

配置

方法用途是否必须实现?
get_config_schema()声明 hermes memory setup 所需的配置字段
save_config(values, hermes_home)将非敏感配置写入本地存储位置(除非仅支持环境变量)

可选钩子

方法调用时机使用场景
system_prompt_block()系统提示词组装时提供静态提供者信息
prefetch(query)每次 API 调用前返回召回的上下文
queue_prefetch(query)每轮对话结束后预热下一轮
sync_turn(user, assistant)每轮完整对话结束后持久化对话内容
on_session_end(messages)对话结束时最终提取/刷新数据
on_pre_compress(messages)上下文压缩前保存洞察信息再丢弃
on_memory_write(action, target, content)内置记忆写入时同步到你的后端
shutdown()程序退出前清理连接

配置 Schema

get_config_schema() 返回由 hermes memory setup 使用的字段描述列表:

def get_config_schema(self):
return [
{
"key": "api_key",
"description": "My Provider API key",
"secret": True, # → written to .env
"required": True,
"env_var": "MY_API_KEY", # explicit env var name
"url": "https://my-provider.com/keys", # where to get it
},
{
"key": "region",
"description": "Server region",
"default": "us-east",
"choices": ["us-east", "eu-west", "ap-south"],
},
{
"key": "project",
"description": "Project identifier",
"default": "hermes",
},
]

带有 secret: Trueenv_var 的字段将被传入 .env。非敏感字段将传递给 save_config()

最小化 vs 完整 Schema

get_config_schema() 中的每个字段都会在 hermes memory setup 期间被提示。具有大量选项的提供者应保持 schema 尽可能简洁——仅包含用户必须配置的字段(如 API 密钥、必需凭证)。可选设置应在配置文件参考文档中说明(例如 $HERMES_HOME/myprovider.json),而不是在设置过程中全部提示。这能确保设置向导快速完成,同时仍支持高级配置。参见 Supermemory 提供者示例——它仅提示 API 密钥;其余所有选项均存在于 supermemory.json 中。

保存配置

def save_config(self, values: dict, hermes_home: str) -> None:
"""Write non-secret config to your native location."""
import json
from pathlib import Path
config_path = Path(hermes_home) / "my-provider.json"
config_path.write_text(json.dumps(values, indent=2))

对于仅支持环境变量的提供者,保留默认空操作即可。

插件入口点

def register(ctx) -> None:
"""Called by the memory plugin discovery system."""
ctx.register_memory_provider(MyMemoryProvider())

plugin.yaml

name: my-provider
version: 1.0.0
description: "Short description of what this provider does."
hooks:
- on_session_end # list hooks you implement

线程约定

sync_turn() 必须是非阻塞的。 如果你的后端存在延迟(如 API 调用、LLM 处理),请在守护线程中运行相关工作:

def sync_turn(self, user_content, assistant_content):
def _sync():
try:
self._api.ingest(user_content, assistant_content)
except Exception as e:
logger.warning("Sync failed: %s", e)

if self._sync_thread and self._sync_thread.is_alive():
self._sync_thread.join(timeout=5.0)
self._sync_thread = threading.Thread(target=_sync, daemon=True)
self._sync_thread.start()

配置文件隔离

所有存储路径 必须 使用 initialize() 中的 hermes_home 关键字参数,而非硬编码的 ~/.hermes

# CORRECT — profile-scoped
from hermes_constants import get_hermes_home
data_dir = get_hermes_home() / "my-provider"

# WRONG — shared across all profiles
data_dir = Path("~/.hermes/my-provider").expanduser()

测试

参见 tests/agent/test_memory_plugin_e2e.py 以获取使用真实 SQLite 提供者的完整端到端测试流程。

from agent.memory_manager import MemoryManager

mgr = MemoryManager()
mgr.add_provider(my_provider)
mgr.initialize_all(session_id="test-1", platform="cli")

# Test tool routing
result = mgr.handle_tool_call("my_tool", {"action": "add", "content": "test"})

# Test lifecycle
mgr.sync_all("user msg", "assistant msg")
mgr.on_session_end([])
mgr.shutdown_all()

添加 CLI 命令

记忆提供者插件可以注册自己的 CLI 子命令树(例如 hermes my-provider statushermes my-provider config)。该功能基于约定发现机制,无需修改核心文件。

工作原理

  1. 在插件目录中添加一个 cli.py 文件
  2. 定义一个 register_cli(subparser) 函数来构建 argparse 命令树
  3. 记忆插件系统通过 discover_plugin_cli_commands() 在启动时发现该函数
  4. 你的命令将出现在 hermes <provider-name> <subcommand>

当前激活提供者控制: 你的 CLI 命令仅在你的提供者是配置中激活的 memory.provider 时才会显示。如果用户未配置你的提供者,这些命令不会出现在 hermes --help 中。

示例

# plugins/memory/my-provider/cli.py

def my_command(args):
"""Handler dispatched by argparse."""
sub = getattr(args, "my_command", None)
if sub == "status":
print("Provider is active and connected.")
elif sub == "config":
print("Showing config...")
else:
print("Usage: hermes my-provider <status|config>")

def register_cli(subparser) -> None:
"""Build the hermes my-provider argparse tree.

Called by discover_plugin_cli_commands() at argparse setup time.
"""
subs = subparser.add_subparsers(dest="my_command")
subs.add_parser("status", help="Show provider status")
subs.add_parser("config", help="Show provider config")
subparser.set_defaults(func=my_command)

参考实现

参见 plugins/memory/honcho/cli.py 获取完整示例,包含 13 个子命令、跨配置文件管理(--target-profile)以及配置读写功能。

包含 CLI 的目录结构

plugins/memory/my-provider/
├── __init__.py # MemoryProvider implementation + register()
├── plugin.yaml # Metadata
├── cli.py # register_cli(subparser) — CLI commands
└── README.md # Setup instructions

单提供者规则

同一时间只能激活一个外部记忆提供者。如果用户尝试注册第二个,MemoryManager 会拒绝并发出警告。这可防止工具模式膨胀和后端冲突。