跳到主要内容

添加工具

在编写工具之前,问问自己:这是否应该是一个 技能 呢?

当能力可以表示为指令 + Shell 命令 + 现有工具(arXiv 搜索、git 工作流、Docker 管理、PDF 处理)时,将其设为技能 (Skill)

当它需要与 API 密钥、自定义处理逻辑、二进制数据处理或流式传输进行端到端集成(浏览器自动化、TTS、视觉分析)时,将其设为工具 (Tool)

概述

添加一个工具涉及修改2 个文件

  1. tools/your_tool.py — 处理器、模式、检查函数、registry.register() 调用
  2. toolsets.py — 在 _HERMES_CORE_TOOLS(或特定的工具集)中添加工具名称

任何包含顶级 registry.register() 调用的 tools/*.py 文件在启动时都会自动发现——无需手动导入列表。

步骤 1:创建工具文件

每个工具文件都遵循相同的结构:

# tools/weather_tool.py
"""天气工具 -- 查询指定位置的当前天气。"""

import json
import os
import logging

logger = logging.getLogger(__name__)


# --- 可用性检查 ---

def check_weather_requirements() -> bool:
"""如果工具的依赖项可用,则返回 True。"""
return bool(os.getenv("WEATHER_API_KEY"))


# --- 处理器 (Handler) ---

def weather_tool(location: str, units: str = "metric") -> str:
"""获取指定位置的天气。返回 JSON 字符串。"""
api_key = os.getenv("WEATHER_API_KEY")
if not api_key:
return json.dumps({"error": "WEATHER_API_KEY 未配置"})
try:
# ... 调用天气 API ...
return json.dumps({"location": location, "temp": 22, "units": units})
except Exception as e:
return json.dumps({"error": str(e)})


# --- 模式 (Schema) ---

WEATHER_SCHEMA = {
"name": "weather",
"description": "获取指定位置的当前天气。",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "城市名称或坐标(例如 'London' 或 '51.5,-0.1')"
},
"units": {
"type": "string",
"enum": ["metric", "imperial"],
"description": "温度单位(默认:metric)",
"default": "metric"
}
},
"required": ["location"]
}
}


# --- 注册 (Registration) ---

from tools.registry import registry

registry.register(
name="weather",
toolset="weather",
schema=WEATHER_SCHEMA,
handler=lambda args, **kw: weather_tool(
location=args.get("location", ""),
units=args.get("units", "metric")),
check_fn=check_weather_requirements,
requires_env=["WEATHER_API_KEY"],
)

关键规则

重要
  • 处理器 必须 返回 JSON 字符串(通过 json.dumps()),绝不能返回原始字典
  • 错误 必须 作为 {"error": "message"} 返回,绝不能抛出异常
  • check_fn 在构建工具定义时调用——如果它返回 False,则该工具将被静默排除
  • handler 接收 (args: dict, **kwargs),其中 args 是 LLM 的工具调用参数

步骤 2:添加到工具集

toolsets.py 中,添加工具名称:

# 如果它应该在所有平台(CLI + 消息传递)上可用:
_HERMES_CORE_TOOLS = [
...
"weather", # <-- 在这里添加
]

# 或者创建一个新的独立工具集:
"weather": {
"description": "天气查询工具",
"tools": ["weather"],
"includes": []
},

步骤 3:添加发现导入 (不再需要)

带有顶级 registry.register() 调用的工具模块会被 tools/registry.py 中的 discover_builtin_tools() 自动发现。无需维护手动导入列表——只需在 tools/ 中创建文件,它就会在启动时被加载。

异步处理器 (Async Handlers)

如果你的处理器需要异步代码,请使用 is_async=True 标记它:

async def weather_tool_async(location: str) -> str:
async with aiohttp.ClientSession() as session:
...
return json.dumps(result)

registry.register(
name="weather",
toolset="weather",
schema=WEATHER_SCHEMA,
handler=lambda args, **kw: weather_tool_async(args.get("location", "")),
check_fn=check_weather_requirements,
is_async=True, # registry 会自动调用 _run_async()
)

注册表会透明地处理异步桥接——你永远不需要自己调用 asyncio.run()

需要 task_id 的处理器

管理每个会话状态的工具会通过 **kwargs 接收到 task_id

def _handle_weather(args, **kw):
task_id = kw.get("task_id")
return weather_tool(args.get("location", ""), task_id=task_id)

registry.register(
name="weather",
...
handler=_handle_weather,
)

Agent-Loop 拦截工具

某些工具(todomemorysession_searchdelegate_task)需要访问每个会话的 Agent 状态。这些工具在到达注册表之前,会被 run_agent.py 拦截。注册表仍然保留它们的模式,但如果拦截被绕过,dispatch() 将返回一个回退错误。

可选:设置向导集成

如果你的工具需要 API 密钥,请将其添加到 hermes_cli/config.py

OPTIONAL_ENV_VARS = {
...
"WEATHER_API_KEY": {
"description": "天气查询的 API 密钥",
"prompt": "天气 API 密钥",
"url": "https://weatherapi.com/",
"tools": ["weather"],
"password": True,
},
}

检查清单

  • 使用处理器、模式、检查函数和注册信息创建工具文件
  • 已添加到 toolsets.py 的适当工具集中
  • 已将发现导入添加到 model_tools.py
  • 处理器返回 JSON 字符串,错误返回 {"error": "..."}
  • 可选:已将 API 密钥添加到 hermes_cli/config.pyOPTIONAL_ENV_VARS
  • 可选:已添加到 toolset_distributions.py 以进行批量处理
  • 使用 hermes chat -q "为伦敦使用天气工具" 进行测试