跳到主要内容

扩展 CLI

Hermes 在 HermesCLI 上暴露了受保护的扩展钩子，因此包装器 CLI 可以在不覆盖包含 1000 多个行的 run() 方法的情况下，添加小部件、按键绑定和布局自定义功能。这确保了您的扩展与内部更改解耦。

扩展点​

目前有五个可用的扩展切口：

Hook用途何时覆盖
_get_extra_tui_widgets()将小部件注入布局当您需要持久的 UI 元素（面板、状态栏、迷你播放器）时
_register_extra_tui_keybindings(kb, *, input_area)添加键盘快捷键当您需要热键（切换面板、传输控制、模态快捷键）时
_build_tui_layout_children(**widgets)完全控制小部件的顺序当您需要重新排序或包装现有小部件时（罕见）
process_command()添加自定义斜杠命令当您需要处理 /mycommand 时（预先存在的钩子）
_build_tui_style_dict()自定义 prompt_toolkit 样式当您需要自定义颜色或样式时（预先存在的钩子）

前三个是新的受保护钩子。后两个已经存在。

快速入门：包装器 CLI​

#!/usr/bin/env python3
"""my_cli.py — 示例包装器 CLI，用于扩展 Hermes。"""

from cli import HermesCLI
from prompt_toolkit.layout import FormattedTextControl, Window
from prompt_toolkit.filters import Condition


class MyCLI(HermesCLI):

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self._panel_visible = False

    def _get_extra_tui_widgets(self):
        """在状态栏上方添加一个可切换的信息面板。"""
        cli_ref = self
        return [
            Window(
                FormattedTextControl(lambda: "📊 我的自定义面板内容"),
                height=1,
                filter=Condition(lambda: cli_ref._panel_visible),
            ),
        ]

    def _register_extra_tui_keybindings(self, kb, *, input_area):
        """F2 切换自定义面板。"""
        cli_ref = self

        @kb.add("f2")
        def _toggle_panel(event):
            cli_ref._panel_visible = not cli_ref._panel_visible

    def process_command(self, cmd: str) -> bool:
        """添加一个 /panel 斜杠命令。"""
        if cmd.strip().lower() == "/panel":
            self._panel_visible = not self._panel_visible
            state = "visible" if self._panel_visible else "hidden"
            print(f"面板现在是 {state}")
            return True
        return super().process_command(cmd)


if __name__ == "__main__":
    cli = MyCLI()
    cli.run()

运行它：

cd ~/.hermes/hermes-agent
source .venv/bin/activate
python my_cli.py

钩子参考​

_get_extra_tui_widgets()​

返回一个包含要插入 TUI 布局的 prompt_toolkit 小部件的列表。小部件出现在分隔符和状态栏之间——位于输入区域上方，主输出下方。

def _get_extra_tui_widgets(self) -> list:
    return []  # 默认：没有额外的小部件

每个小部件都应该是一个 prompt_toolkit 容器（例如 Window、ConditionalContainer、HSplit）。使用 ConditionalContainer 或 filter=Condition(...) 使小部件可切换。

from prompt_toolkit.layout import ConditionalContainer, Window, FormattedTextControl
from prompt_toolkit.filters import Condition

def _get_extra_tui_widgets(self):
    return [
        ConditionalContainer(
            Window(FormattedTextControl("状态：已连接"), height=1),
            filter=Condition(lambda: self._show_status),
        ),
    ]

_register_extra_tui_keybindings(kb, *, input_area)​

在 Hermes 注册其自身按键绑定之后，但在布局构建之前调用。将您的按键绑定添加到 kb 中。

def _register_extra_tui_keybindings(self, kb, *, input_area):
    pass  # 默认：没有额外的按键绑定

参数：

  • kb — prompt_toolkit 应用程序的 KeyBindings 实例
  • input_area — 主 TextArea 小部件，如果您需要读取或操作用户输入
def _register_extra_tui_keybindings(self, kb, *, input_area):
    cli_ref = self

    @kb.add("f3")
    def _clear_input(event):
        input_area.text = ""

    @kb.add("f4")
    def _insert_template(event):
        input_area.text = "/search "

避免与内置按键绑定冲突：Enter（提交）、Escape Enter（换行）、Ctrl-C（中断）、Ctrl-D（退出）、Tab（自动建议接受）。功能键 F2+ 和 Ctrl 组合通常是安全的。

_build_tui_layout_children(**widgets)​

只有当您需要完全控制小部件顺序时才覆盖此方法。大多数扩展应该使用 _get_extra_tui_widgets() 代替。

def _build_tui_layout_children(self, *, sudo_widget, secret_widget,
    approval_widget, clarify_widget, spinner_widget, spacer,
    status_bar, input_rule_top, image_bar, input_area,
    input_rule_bot, voice_status_bar, completions_menu) -> list:

默认实现返回：

[
    Window(height=0),       # anchor
    sudo_widget,            # sudo 密码提示（条件）
    secret_widget,          # secret 输入提示（条件）
    approval_widget,        # 危险命令批准（条件）
    clarify_widget,         # 澄清问题 UI（条件）
    spinner_widget,         # 思考加载指示器（条件）
    spacer,                 # 填充剩余垂直空间
    *self._get_extra_tui_widgets(),  # 您的小部件放在这里
    status_bar,             # 模型/令牌/上下文状态行
    input_rule_top,         # ─── 输入框上方的边框
    image_bar,              # 附加图像指示器
    input_area,             # 用户文本输入
    input_rule_bot,         # ─── 输入框下方的边框
    voice_status_bar,       # 语音模式状态（条件）
    completions_menu,       # 自动完成下拉菜单
]

布局图​

从上到下的默认布局：

  1. 输出区域 — 滚动对话历史记录
  2. 分隔符
  3. 额外小部件 — 来自 _get_extra_tui_widgets()
  4. 状态栏 — 模型、上下文%、已用时间
  5. 图像栏 — 附加图像计数
  6. 输入区域 — 用户提示
  7. 语音状态 — 录音指示器
  8. 自动完成菜单 — 自动完成建议

提示​

  • 在状态更改后使显示失效：调用 self._invalidate() 来触发 prompt_toolkit 重绘。
  • 访问智能体状态：self.agent、self.model、self.conversation_history 均可访问。
  • 自定义样式：覆盖 _build_tui_style_dict() 并为您的自定义样式类添加条目。
  • 斜杠命令：覆盖 process_command()，处理您的命令，并为所有其他命令调用 super().process_command(cmd)。
  • 除非绝对必要，否则不要覆盖 run() — 扩展钩子专门存在就是为了避免这种耦合。