问题: - 用户询问'我什么时候签到的'时,AI 说'系统只记录是否签到,没有时间戳' - 实际上数据库 signs 表有完整的 sign_time 和 sign_text 字段 解决方案: - 修改 executeCheck 方法,改用 ListSigns 查询今天的签到记录 - 返回完整的签到详情:签到时间(HH:MM:SS格式)+ 签到内容 - 未签到时仍返回简单提示 改动内容: - executeCheck 使用 ListSigns 替代 HasSignedOnDay - 构建 ListOptions 查询今天的签到记录(过滤 NodeID + 时间范围) - 返回格式:'XXX 今天已经签到过了。\n签到时间:10:30:45\n签到内容:上海-Kevin-GAT562签到' - 更新测试验证返回内容包含时间和签到文本 - 更新 mockSignStore.ListSigns 支持 NodeID 和 Limit 过滤 使用效果: - 用户:'我今天签到了吗?' → 返回是否签到 + 时间 + 内容 - 用户:'我什么时候签到的?' → 返回签到时间:10:30:45 测试: - ✅ 未签到场景测试通过 - ✅ 已签到场景测试通过(验证时间和内容) - ✅ 所有签到工具测试通过 - ✅ 项目编译成功 Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
4.9 KiB
4.9 KiB
签到工具查询功能增强
问题描述
原签到工具只支持签到操作,不支持查询。当用户询问"今天有多少人签到"或"最近几天的签到情况"时,AI 无法调用工具获取准确数据,导致:
- 数据不准确:AI 只能根据上下文猜测或给出模糊答案
- 无法查询历史:询问超过一天的数据时,AI 一问三不知
- 功能不完整:数据库层已有完善的查询能力(
CountSigns、CountSignsByDay),但工具层未暴露给 AI - 状态判断不准:用户问"我今天签到了吗"时,AI 基于对话记忆判断而非查询数据库,导致误判
解决方案
为签到工具添加查询和检查功能,支持以下三种操作模式:
1. 签到操作 (action=sign)
保持原有功能不变:
- 记录节点今日签到信息
- 每个节点每天只能签到一次
- 必填参数:地区、名字、设备
2. 查询操作 (action=query)
查询签到统计功能:
- 查询指定日期或日期范围的签到统计
- 返回总人次和按天分组的统计数据
- 可选参数:
date: 查询日期(格式:YYYY-MM-DD),默认今天days: 查询最近 N 天,默认只查询 date 指定的那一天
3. 检查操作 (action=check) 新增
检查当前节点今天是否已签到:
- 用于回答"我今天签到了吗"、"我什么时候签到的"之类的问题
- 直接查询数据库,不依赖对话历史
- 返回明确的签到状态,包括签到时间和签到内容
修改内容
1. 扩展 SignStore 接口
type SignStore interface {
CreateSign(nodeID string, longName, shortName *string, signText string, signTime time.Time) (*storepkg.SignRecord, error)
HasSignedOnDay(nodeID string, day time.Time) (bool, error)
GetNodeInfo(nodeID string) (*storepkg.NodeInfoRecord, error)
// 新增查询方法
CountSigns(opts storepkg.ListOptions) (int64, error)
CountSignsByDay(opts storepkg.ListOptions) ([]storepkg.SignDayCount, error)
ListSigns(opts storepkg.ListOptions) ([]storepkg.SignRecord, error)
}
2. 更新工具定义
工具描述中明确说明支持两种操作:
- action=sign:签到
- action=query:查询统计
3. 实现查询逻辑
executeSign(): 原签到逻辑executeQuery(): 新增查询逻辑- 解析日期参数
- 构建查询条件(Since/Until)
- 调用 store 查询接口
- 格式化返回结果
4. 扩展参数结构
type signParams struct {
Action string `json:"action"` // sign 或 query
// 签到参数
Region string `json:"region"`
Name string `json:"name"`
Device string `json:"device"`
TxPower string `json:"tx_power"`
AntennaLength string `json:"antenna_length"`
Altitude string `json:"altitude"`
RawText string `json:"raw_text"`
// 查询参数
Date string `json:"date"` // YYYY-MM-DD
Days int `json:"days"` // 最近 N 天
}
使用示例
AI 调用示例
检查今天是否签到(check 操作)
用户:"我今天签到了吗?"
AI 调用:
{
"action": "check"
}
返回(未签到):
Test Node 今天还没有签到。
返回(已签到):
Test Node 今天已经签到过了。
签到时间:10:30:45
签到内容:上海闵行-Kevin-GAT562签到
用户:"我什么时候签到的?"
AI 同样调用 check 操作,返回包含签到时间和内容的完整信息。
查询今天的签到情况
{
"action": "query",
"date": "2024-06-23"
}
返回:
2024-06-23 的签到统计:
总计:5 人次
按天统计:
- 2024-06-23: 5 人
查询最近 7 天的签到情况
{
"action": "query",
"date": "2024-06-23",
"days": 7
}
返回:
最近 7 天的签到统计:
总计:32 人次
按天统计:
- 2024-06-23: 5 人
- 2024-06-22: 6 人
- 2024-06-21: 4 人
- 2024-06-20: 7 人
- 2024-06-19: 3 人
- 2024-06-18: 4 人
- 2024-06-17: 3 人
签到(保持原有功能)
{
"action": "sign",
"region": "上海闵行",
"name": "Kevin",
"device": "GAT562"
}
测试
创建了完整的单元测试 sign_test.go,覆盖:
- ✅ 查询今天的签到统计
- ✅ 查询最近 N 天的签到统计
- ✅ 查询指定日期的签到统计
- ✅ 签到功能(回归测试)
所有测试通过。
影响范围
- ✅ 向后兼容:原有签到功能完全保持不变
- ✅ 数据库无需修改:查询接口已存在
- ✅ 新增功能:AI 现在可以准确回答签到统计问题
- ✅ 编译通过:整个项目构建成功
后续建议
可以考虑进一步增强:
- 支持按节点查询:某个特定节点的签到历史
- 支持按地区统计:哪些地区签到最活跃
- 支持导出详细列表:不仅是统计,还能看到每条签到的详细内容