8.9 KiB
8.9 KiB
🔧 工具系统详解
📖 什么是工具系统
工具系统是MaiBot的信息获取能力扩展组件。如果说Action组件功能五花八门,可以拓展麦麦能做的事情,那么Tool就是在某个过程中拓宽了麦麦能够获得的信息量。
🎯 工具系统的特点
- 🔍 信息获取增强:扩展麦麦获取外部信息的能力
- 📊 数据丰富:帮助麦麦获得更多背景信息和实时数据
- 🔌 插件式架构:支持独立开发和注册新工具
- ⚡ 自动发现:工具会被系统自动识别和注册
🆚 Tool vs Action vs Command 区别
| 特征 | Action | Command | Tool |
|---|---|---|---|
| 主要用途 | 扩展麦麦行为能力 | 响应用户指令 | 扩展麦麦信息获取 |
| 触发方式 | 麦麦智能决策 | 用户主动触发 | LLM根据需要调用 |
| 目标 | 让麦麦做更多事情 | 提供具体功能 | 让麦麦知道更多信息 |
| 使用场景 | 增强交互体验 | 功能服务 | 信息查询和分析 |
🏗️ 工具基本结构
必要组件
每个工具必须继承 BaseTool 基类并实现以下属性和方法:
from src.tools.tool_can_use.base_tool import BaseTool, register_tool
class MyTool(BaseTool):
# 工具名称,必须唯一
name = "my_tool"
# 工具描述,告诉LLM这个工具的用途
description = "这个工具用于获取特定类型的信息"
# 参数定义,遵循JSONSchema格式
parameters = {
"type": "object",
"properties": {
"query": {
"type": "string",
"description": "查询参数"
},
"limit": {
"type": "integer",
"description": "结果数量限制"
}
},
"required": ["query"]
}
async def execute(self, function_args: Dict[str, Any]):
"""执行工具逻辑"""
# 实现工具功能
result = f"查询结果: {function_args.get('query')}"
return {
"name": self.name,
"content": result
}
属性说明
| 属性 | 类型 | 说明 |
|---|---|---|
name |
str | 工具的唯一标识名称 |
description |
str | 工具功能描述,帮助LLM理解用途 |
parameters |
dict | JSONSchema格式的参数定义 |
方法说明
| 方法 | 参数 | 返回值 | 说明 |
|---|---|---|---|
execute |
function_args |
dict |
执行工具核心逻辑 |
🔄 自动注册机制
工具系统采用自动发现和注册机制:
- 文件扫描:系统自动遍历
tool_can_use目录中的所有Python文件 - 类识别:寻找继承自
BaseTool的工具类 - 自动注册:只需要实现对应的类并把文件放在正确文件夹中就可自动注册
- 即用即加载:工具在需要时被实例化和调用
🎨 完整工具示例
完成一个天气查询工具
from src.tools.tool_can_use.base_tool import BaseTool, register_tool
import aiohttp
import json
class WeatherTool(BaseTool):
"""天气查询工具 - 获取指定城市的实时天气信息"""
name = "weather_query"
description = "查询指定城市的实时天气信息,包括温度、湿度、天气状况等"
parameters = {
"type": "object",
"properties": {
"city": {
"type": "string",
"description": "要查询天气的城市名称,如:北京、上海、纽约"
},
"country": {
"type": "string",
"description": "国家代码,如:CN、US,可选参数"
}
},
"required": ["city"]
}
async def execute(self, function_args, message_txt=""):
"""执行天气查询"""
try:
city = function_args.get("city")
country = function_args.get("country", "")
# 构建查询参数
location = f"{city},{country}" if country else city
# 调用天气API(示例)
weather_data = await self._fetch_weather(location)
# 格式化结果
result = self._format_weather_data(weather_data)
return {
"name": self.name,
"content": result
}
except Exception as e:
return {
"name": self.name,
"content": f"天气查询失败: {str(e)}"
}
async def _fetch_weather(self, location: str) -> dict:
"""获取天气数据"""
# 这里是示例,实际需要接入真实的天气API
api_url = f"http://api.weather.com/v1/current?q={location}"
async with aiohttp.ClientSession() as session:
async with session.get(api_url) as response:
return await response.json()
def _format_weather_data(self, data: dict) -> str:
"""格式化天气数据"""
if not data:
return "暂无天气数据"
# 提取关键信息
city = data.get("location", {}).get("name", "未知城市")
temp = data.get("current", {}).get("temp_c", "未知")
condition = data.get("current", {}).get("condition", {}).get("text", "未知")
humidity = data.get("current", {}).get("humidity", "未知")
# 格式化输出
return f"""
🌤️ {city} 实时天气
━━━━━━━━━━━━━━━━━━
🌡️ 温度: {temp}°C
☁️ 天气: {condition}
💧 湿度: {humidity}%
━━━━━━━━━━━━━━━━━━
""".strip()
📊 工具开发步骤
1. 创建工具文件
在 src/tools/tool_can_use/ 目录下创建新的Python文件:
# 例如创建 my_new_tool.py
touch src/tools/tool_can_use/my_new_tool.py
2. 实现工具类
from src.tools.tool_can_use.base_tool import BaseTool, register_tool
class MyNewTool(BaseTool):
name = "my_new_tool"
description = "新工具的功能描述"
parameters = {
"type": "object",
"properties": {
# 定义参数
},
"required": []
}
async def execute(self, function_args, message_txt=""):
# 实现工具逻辑
return {
"name": self.name,
"content": "执行结果"
}
3. 系统集成
工具创建完成后,系统会自动发现和注册,无需额外配置。
🚨 注意事项和限制
当前限制
- 独立开发:需要单独编写,暂未完全融入插件系统
- 适用范围:主要适用于信息获取场景
- 配置要求:必须开启工具处理器
开发建议
- 功能专一:每个工具专注单一功能
- 参数明确:清晰定义工具参数和用途
- 错误处理:完善的异常处理和错误反馈
- 性能考虑:避免长时间阻塞操作
- 信息准确:确保获取信息的准确性和时效性
🎯 最佳实践
1. 工具命名规范
# ✅ 好的命名
name = "weather_query" # 清晰表达功能
name = "knowledge_search" # 描述性强
name = "stock_price_check" # 功能明确
# ❌ 避免的命名
name = "tool1" # 无意义
name = "wq" # 过于简短
name = "weather_and_news" # 功能过于复杂
2. 描述规范
# ✅ 好的描述
description = "查询指定城市的实时天气信息,包括温度、湿度、天气状况"
# ❌ 避免的描述
description = "天气" # 过于简单
description = "获取信息" # 不够具体
3. 参数设计
# ✅ 合理的参数设计
parameters = {
"type": "object",
"properties": {
"city": {
"type": "string",
"description": "城市名称,如:北京、上海"
},
"unit": {
"type": "string",
"description": "温度单位:celsius(摄氏度) 或 fahrenheit(华氏度)",
"enum": ["celsius", "fahrenheit"]
}
},
"required": ["city"]
}
# ❌ 避免的参数设计
parameters = {
"type": "object",
"properties": {
"data": {
"type": "string",
"description": "数据" # 描述不清晰
}
}
}
4. 结果格式化
# ✅ 良好的结果格式
def _format_result(self, data):
return f"""
🔍 查询结果
━━━━━━━━━━━━
📊 数据: {data['value']}
📅 时间: {data['timestamp']}
📝 说明: {data['description']}
━━━━━━━━━━━━
""".strip()
# ❌ 避免的结果格式
def _format_result(self, data):
return str(data) # 直接返回原始数据
🎉 工具系统为麦麦提供了强大的信息获取能力!合理使用工具可以让麦麦变得更加智能和博学。