Разработка плагина провайдера памяти

Плагины провайдера памяти дают Hermes Agent постоянные знания, которые сохраняются между сессиями, — поверх встроенных MEMORY.md и USER.md. Ниже разберём, как собрать такой плагин.

совет

Провайдеры памяти — один из двух типов плагинов-провайдеров. Второй тип — плагины движка контекста, которые заменяют встроенный компрессор контекста. Оба следуют одной схеме: один активный провайдер, конфигурация через файл настроек, управление через hermes plugins.

Структура директорий

Каждый провайдер памяти размещается в plugins/memory/<name>/:

plugins/memory/my-provider/
├── __init__.py      # Реализация MemoryProvider + точка входа register()
├── plugin.yaml      # Метаданные (name, description, hooks)
└── README.md        # Инструкции по настройке, описание конфига и инструментов

Абстрактный класс MemoryProvider

Ваш плагин реализует абстрактный базовый класс MemoryProvider из agent/memory_provider.py:

from agent.memory_provider import MemoryProvider

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

    def is_available(self) -> bool:
        """Проверить, может ли провайдер активироваться. Сетевые вызовы ЗАПРЕЩЕНЫ."""
        return bool(os.environ.get("MY_API_KEY"))

    def initialize(self, session_id: str, **kwargs) -> None:
        """Вызывается один раз при запуске агента.

        kwargs всегда содержит:
          hermes_home (str): Активный путь HERMES_HOME. Используй для хранения данных.
        """
        self._api_key = os.environ.get("MY_API_KEY", "")
        self._session_id = session_id

    # ... реализуй оставшиеся методы

Обязательные методы

Основной жизненный цикл

МетодКогда вызываетсяОбязателен?
name (property)ВсегдаДа
is_available()Инициализация агента, до активацииДа — без сетевых вызовов
initialize(session_id, **kwargs)Запуск агентаДа
get_tool_schemas()После инициализации, при инъекции инструментовДа
handle_tool_call(tool_name, args, **kwargs)При использовании агентом ваших инструментовДа (если есть инструменты)

Конфигурация

МетодНазначениеОбязателен?
get_config_schema()Объявить поля конфига для hermes memory setupДа
save_config(values, hermes_home)Записать несекретный конфиг в собственное место храненияДа (если не только через переменные окружения)

Опциональные хуки

МетодКогда вызываетсяПрименение
system_prompt_block()Сборка системного промптаСтатическая информация о провайдере
prefetch(query, *, session_id="")Перед каждым API-вызовомВернуть извлечённый из памяти контекст
queue_prefetch(query)После каждого ходаПредзагрузка для следующего хода
sync_turn(user, assistant, *, session_id="")После каждого завершённого ходаСохранить диалог
on_session_end(messages)Завершение разговораФинальная выгрузка/сброс данных
on_pre_compress(messages)Перед сжатием контекстаСохранить инсайты до удаления
on_memory_write(action, target, content)Запись во встроенную памятьЗеркалирование во внешний бэкенд
shutdown()Завершение процессаЗакрыть соединения

Схема конфигурации

get_config_schema() возвращает список дескрипторов полей, которые использует hermes memory setup:

def get_config_schema(self):
    return [
        {
            "key": "api_key",
            "description": "API-ключ провайдера",
            "secret": True,           # → записывается в .env
            "required": True,
            "env_var": "MY_API_KEY",   # явное имя переменной окружения
            "url": "https://my-provider.com/keys",  # где получить ключ
        },
        {
            "key": "region",
            "description": "Регион сервера",
            "default": "us-east",
            "choices": ["us-east", "eu-west", "ap-south"],
        },
        {
            "key": "project",
            "description": "Идентификатор проекта",
            "default": "hermes",
        },
    ]

Поля с secret: True и env_var попадают в .env. Несекретные поля передаются в save_config().

совет
Минимальная vs полная схема

Каждое поле в get_config_schema() выводится как отдельный запрос во время hermes memory setup. Провайдерам с большим числом параметров стоит держать схему минимальной — включайте только то, что пользователь обязан настроить (API-ключ, обязательные учётные данные). Дополнительные настройки лучше документировать через конфиг-файл (например, $HERMES_HOME/myprovider.json), не перегружая мастер установки. Так первичная настройка остаётся быстрой, а гибкость для продвинутых сценариев сохраняется. Пример такого подхода — провайдер Supermemory: он запрашивает только API-ключ, а все остальные параметры живут в supermemory.json.

Сохранение конфигурации

def save_config(self, values: dict, hermes_home: str) -> None:
    """Записать несекретный конфиг в собственное место хранения."""
    import json
    from pathlib import Path
    config_path = Path(hermes_home) / "my-provider.json"
    config_path.write_text(json.dumps(values, indent=2))

Если провайдер работает только через переменные окружения, оставьте реализацию по умолчанию (no-op).

Точка входа плагина

def register(ctx) -> None:
    """Вызывается системой обнаружения плагинов памяти."""
    ctx.register_memory_provider(MyMemoryProvider())

plugin.yaml

name: my-provider
version: 1.0.0
description: "Краткое описание того, что делает этот провайдер."
hooks:
  - on_session_end    # перечислите реализуемые хуки

Контракт потоков выполнения

sync_turn() ДОЛЖЕН быть неблокирующим. Если у бэкенда есть задержка (API-вызовы, обработка LLM), выносите работу в поток-демон:

def sync_turn(self, user_content, assistant_content, *, session_id="", messages=None):
    def _sync():
        try:
            self._api.ingest(user_content, assistant_content, session_id=session_id, messages=messages)
        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()

messages — опциональный контекст диалога в формате OpenAI на момент завершения хода. Если он передан, в него входят сообщения пользователя и ассистента, вызовы инструментов ассистентом и результаты этих вызовов. Провайдеры, которым необработанный контекст хода не нужен, могут не объявлять параметр messages — Hermes продолжит вызывать их со старой сигнатурой.

Облачные провайдеры должны документировать, какие части messages уходят за пределы устройства. Вызовы инструментов и их результаты могут содержать пути к файлам, вывод команд и другие данные рабочего пространства.

Изоляция профилей

Все пути к данным обязаны использовать kwarg hermes_home из initialize(), а не захардкоженный ~/.hermes:

# ПРАВИЛЬНО — путь в рамках профиля
from hermes_constants import get_hermes_home
data_dir = get_hermes_home() / "my-provider"

# НЕПРАВИЛЬНО — общий путь для всех профилей
data_dir = Path("~/.hermes/my-provider").expanduser()

Тестирование

Сквозные паттерны тестирования смотрите в tests/agent/test_memory_provider.py и соседних файлах: tests/agent/test_memory_session_switch.py, tests/agent/test_memory_user_id.py, tests/run_agent/test_memory_provider_init.py.

from agent.memory_manager import MemoryManager

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

# Тест маршрутизации инструментов
result = mgr.handle_tool_call("my_tool", {"action": "add", "content": "test"})

# Тест жизненного цикла
mgr.sync_all("user msg", "assistant msg")
mgr.on_session_end([])
mgr.shutdown_all()

Добавление CLI-команд

Плагины провайдера памяти могут регистрировать собственное дерево CLI-подкоманд (например, hermes my-provider status, hermes 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):
    """Обработчик, вызываемый 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:
    """Построить дерево argparse для hermes my-provider.

    Вызывается discover_plugin_cli_commands() во время настройки argparse.
    """
    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)

Эталонная реализация

Полный пример с 13 подкомандами, управлением несколькими профилями (--target-profile) и чтением/записью конфига смотрите в plugins/memory/honcho/cli.py.

Структура директорий с CLI

plugins/memory/my-provider/
├── __init__.py      # Реализация MemoryProvider + register()
├── plugin.yaml      # Метаданные
├── cli.py           # register_cli(subparser) — CLI-команды
└── README.md        # Инструкции по настройке

Правило единственного провайдера

Одновременно может быть активен только один внешний провайдер памяти. При попытке зарегистрировать второй MemoryManager отклонит его с предупреждением. Так схема инструментов не раздувается, а бэкенды не конфликтуют между собой.

ESC