Skip to content

工具系统

Yuxi 的工具系统基于注册机制,支持多种工具类型的动态组装。

工具注册机制

Yuxi 的工具系统采用 @tool 装饰器注册机制,核心位于 backend/package/yuxi/agents/toolkits/registry.py

@tool 装饰器

python
from yuxi.agents.toolkits.registry import tool

@tool(category="buildin", tags=["示例"], display_name="示例工具")
def example_tool(text: str) -> str:
    """示例工具:返回处理后的文本"""
    ...

装饰器参数:

  • category: 工具分类,用于分组(buildinmysqldebug
  • tags: 标签列表,用于前端展示
  • display_name: 显示名称(给人看的名字)
  • icon: 图标名称(可选)

自动发现

导入 toolkits 包时会自动触发注册:

python
from yuxi.agents.toolkits import buildin, mysql  # 触发 @tool 装饰器执行

toolkits/__init__.py 中已包含 buildinmysqldebug 模块的导入,这些模块加载时会自动注册所有带 @tool 装饰器的函数。

工具分类

内置工具 (buildin)

工具说明
ask_user_question向用户发起交互式提问
present_artifacts展示 Agent 沙盒 outputs 目录下的产物文件
install_skill从沙盒路径或 Git 来源安装当前用户私有 Skill,并激活当前主智能体会话;子智能体禁用
tavily_searchTavily 网页搜索(需配置 TAVILY_API_KEY

Qwen-Image 生成能力已迁移为内置 Skill image-gen。模型调用与图片下载在 Agent 沙盒中完成,生成后的图片保存到 /home/gem/user-data/outputs/,再通过 present_artifacts 展示。

MySQL 工具 (mysql)

工具说明
mysql_list_tables列出数据库中所有表
mysql_describe_table获取表结构信息
mysql_query执行只读 SQL 查询

知识库工具 (kbs)

知识库工具使用 @tool(category="knowledge") 注册,并通过内置 knowledge-base Skill 的 tool_dependencies 按需加载。get_common_kb_tools() 仍可用于直接获取完整工具列表:

python
from yuxi.agents.toolkits.kbs import get_common_kb_tools

kb_tools = get_common_kb_tools()
# 返回: [list_kbs, get_mindmap, query_kb, find_kb_document, open_kb_document]
工具说明
list_kbs列出用户可访问的知识库
get_mindmap获取知识库的思维导图结构
query_kb在指定知识库中检索内容,返回结构化的 resource_id(即 kb_id)/file_id/chunk
find_kb_document在已知文件内按关键词或正则定位内容
open_kb_documentfile_id 分段打开知识库文档(默认窗口 1800 行)

工具组装

工具组装在 Graph 创建阶段完成。内置 Agent 会先调用 prepare_agent_runtime_context 过滤当前用户可用资源,再调用 resolve_configured_runtime_tools(context) 加载已配置工具:

  1. 基础工具:从 context.tools 中按名称筛选
  2. MCP 工具:根据 context.mcps 加载 MCP 服务器工具
  3. Skill 依赖工具:由 SkillsMiddleware 在 Skill 激活后按需追加,包括 knowledge-base 绑定的知识库工具
python
from yuxi.agents.context import prepare_agent_runtime_context
from yuxi.agents.toolkits.service import resolve_configured_runtime_tools

context = await prepare_agent_runtime_context(context, user=current_user, db=db)
tools = await resolve_configured_runtime_tools(context)

Skills 集成

Skills 与工具是两种不同的扩展机制。工具是具体的功能实现,而 Skills 是包含提示词、工具依赖和元数据的完整技能包。通过 context.skills 配置 Skills 时,对应的技能文件会被挂载到沙盒的 /home/gem/skills/<slug>/...,智能体可以通过读取 SKILL.md 来了解如何使用这些技能。

关于 Skills 的详细机制,请参阅 Skills 管理

本项目基于 MIT License 开源,欢迎使用和贡献。