2026-03-10 13:55:19 +08:00
|
|
|
|
# 工具系统
|
|
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
Yuxi 的工具系统基于注册机制,支持多种工具类型的动态组装。
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
## 工具注册机制
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
Yuxi 的工具系统采用 `@tool` 装饰器注册机制,核心位于 `backend/package/yuxi/agents/toolkits/registry.py`。
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
### @tool 装饰器
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
|
|
|
|
|
```python
|
2026-03-25 00:36:42 +08:00
|
|
|
|
from yuxi.agents.toolkits.registry import tool
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-06-04 23:17:48 +08:00
|
|
|
|
@tool(category="buildin", tags=["示例"], display_name="示例工具")
|
|
|
|
|
|
def example_tool(text: str) -> str:
|
|
|
|
|
|
"""示例工具:返回处理后的文本"""
|
2026-03-25 00:36:42 +08:00
|
|
|
|
...
|
2026-03-10 13:55:19 +08:00
|
|
|
|
```
|
|
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
装饰器参数:
|
|
|
|
|
|
- **category**: 工具分类,用于分组(`buildin`、`mysql`、`debug`)
|
|
|
|
|
|
- **tags**: 标签列表,用于前端展示
|
|
|
|
|
|
- **display_name**: 显示名称(给人看的名字)
|
|
|
|
|
|
- **icon**: 图标名称(可选)
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
### 自动发现
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
导入 `toolkits` 包时会自动触发注册:
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
|
|
|
|
|
```python
|
2026-03-25 00:36:42 +08:00
|
|
|
|
from yuxi.agents.toolkits import buildin, mysql # 触发 @tool 装饰器执行
|
2026-03-10 13:55:19 +08:00
|
|
|
|
```
|
|
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
`toolkits/__init__.py` 中已包含 `buildin`、`mysql`、`debug` 模块的导入,这些模块加载时会自动注册所有带 `@tool` 装饰器的函数。
|
|
|
|
|
|
|
|
|
|
|
|
## 工具分类
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
### 内置工具 (buildin)
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
| 工具 | 说明 |
|
|
|
|
|
|
|------|------|
|
|
|
|
|
|
| `ask_user_question` | 向用户发起交互式提问 |
|
2026-06-02 20:04:33 +08:00
|
|
|
|
| `present_artifacts` | 展示 Agent 沙盒 outputs 目录下的产物文件 |
|
2026-06-04 23:17:48 +08:00
|
|
|
|
| `install_skill` | 从沙盒路径或 Git 来源安装当前用户私有 Skill,并激活当前主智能体会话;子智能体禁用 |
|
2026-03-25 00:36:42 +08:00
|
|
|
|
| `tavily_search` | Tavily 网页搜索(需配置 `TAVILY_API_KEY`) |
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-06-02 20:04:33 +08:00
|
|
|
|
Qwen-Image 生成能力已迁移为内置 Skill `image-gen`。模型调用与图片下载在 Agent 沙盒中完成,生成后的图片保存到 `/home/gem/user-data/outputs/`,再通过 `present_artifacts` 展示。
|
|
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
### MySQL 工具 (mysql)
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
| 工具 | 说明 |
|
|
|
|
|
|
|------|------|
|
|
|
|
|
|
| `mysql_list_tables` | 列出数据库中所有表 |
|
|
|
|
|
|
| `mysql_describe_table` | 获取表结构信息 |
|
|
|
|
|
|
| `mysql_query` | 执行只读 SQL 查询 |
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
### 知识库工具 (kbs)
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
知识库工具通过 `get_common_kb_tools()` 获取,不通过 `@tool` 装饰器注册:
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
|
|
|
|
|
```python
|
2026-03-20 12:08:00 +08:00
|
|
|
|
from yuxi.agents.toolkits.kbs import get_common_kb_tools
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
2026-03-25 00:36:42 +08:00
|
|
|
|
kb_tools = get_common_kb_tools()
|
2026-06-03 17:55:10 +08:00
|
|
|
|
# 返回: [list_kbs, get_mindmap, query_kb, find_kb_document, open_kb_document]
|
2026-03-25 00:36:42 +08:00
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
|
|
| 工具 | 说明 |
|
|
|
|
|
|
|------|------|
|
|
|
|
|
|
| `list_kbs` | 列出用户可访问的知识库 |
|
|
|
|
|
|
| `get_mindmap` | 获取知识库的思维导图结构 |
|
2026-06-03 17:55:10 +08:00
|
|
|
|
| `query_kb` | 在指定知识库中检索内容,返回结构化的 `resource_id`(即 `kb_id`)/`file_id`/`chunk` |
|
|
|
|
|
|
| `find_kb_document` | 在已知文件内按关键词或正则定位内容 |
|
|
|
|
|
|
| `open_kb_document` | 按 `file_id` 分段打开知识库文档(默认窗口 1800 行) |
|
2026-03-25 00:36:42 +08:00
|
|
|
|
|
|
|
|
|
|
## 工具组装
|
|
|
|
|
|
|
2026-06-04 23:17:48 +08:00
|
|
|
|
工具组装在 Graph 创建阶段完成。内置 Agent 会先调用 `prepare_agent_runtime_context` 过滤当前用户可用资源,再调用 `resolve_configured_runtime_tools(context)` 加载已配置工具:
|
2026-03-25 00:36:42 +08:00
|
|
|
|
|
|
|
|
|
|
1. **基础工具**:从 `context.tools` 中按名称筛选
|
|
|
|
|
|
2. **MCP 工具**:根据 `context.mcps` 加载 MCP 服务器工具
|
|
|
|
|
|
3. **知识库工具**:由 `KnowledgeBaseMiddleware` 独立处理
|
2026-06-04 23:17:48 +08:00
|
|
|
|
4. **Skill 依赖工具**:由 `SkillsMiddleware` 在 Skill 激活后按需追加
|
2026-03-25 00:36:42 +08:00
|
|
|
|
|
|
|
|
|
|
```python
|
2026-06-04 23:17:48 +08:00
|
|
|
|
from yuxi.agents.context import prepare_agent_runtime_context
|
|
|
|
|
|
from yuxi.agents.toolkits.service import resolve_configured_runtime_tools
|
2026-03-25 00:36:42 +08:00
|
|
|
|
|
2026-06-04 23:17:48 +08:00
|
|
|
|
context = await prepare_agent_runtime_context(context, user=current_user, db=db)
|
|
|
|
|
|
tools = await resolve_configured_runtime_tools(context)
|
2026-03-10 13:55:19 +08:00
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
|
|
## Skills 集成
|
|
|
|
|
|
|
2026-06-03 17:55:10 +08:00
|
|
|
|
Skills 与工具是两种不同的扩展机制。工具是具体的功能实现,而 Skills 是包含提示词、工具依赖和元数据的完整技能包。通过 `context.skills` 配置 Skills 时,对应的技能文件会被挂载到沙盒的 `/home/gem/skills/<slug>/...`,智能体可以通过读取 SKILL.md 来了解如何使用这些技能。
|
2026-03-10 13:55:19 +08:00
|
|
|
|
|
|
|
|
|
|
关于 Skills 的详细机制,请参阅 [Skills 管理](./skills-management.md)。
|