162 lines
4.8 KiB
Bash
Executable File
162 lines
4.8 KiB
Bash
Executable File
#!/bin/bash
|
||
# 记忆系统快速查询脚本
|
||
|
||
MEMORY_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
||
CLI_SCRIPT="$MEMORY_ROOT/memory_cli.py"
|
||
|
||
# 帮助信息
|
||
show_help() {
|
||
cat << EOF
|
||
记忆系统快速查询工具
|
||
|
||
用法: $0 [命令] [选项]
|
||
|
||
命令:
|
||
today 查看今天的事件
|
||
week 查看本周的事件
|
||
recent <天数> 查看最近N天的事件
|
||
search <关键词> 搜索标题包含关键词的事件
|
||
types 统计事件类型分布
|
||
tags 统计标签使用情况
|
||
stats 显示系统统计信息
|
||
help 显示此帮助信息
|
||
|
||
示例:
|
||
$0 today
|
||
$0 week
|
||
$0 recent 3
|
||
$0 search "认证"
|
||
$0 types
|
||
$0 tags
|
||
$0 stats
|
||
|
||
EOF
|
||
}
|
||
|
||
# 检查Python环境
|
||
check_python() {
|
||
if ! command -v python3 &> /dev/null; then
|
||
echo "错误: 未找到python3，请确保Python 3已安装"
|
||
exit 1
|
||
fi
|
||
|
||
# 检查必要的Python模块
|
||
python3 -c "import sys; sys.exit(0)" 2>/dev/null || {
|
||
echo "错误: Python环境异常"
|
||
exit 1
|
||
}
|
||
}
|
||
|
||
# 检查CLI脚本是否存在并可执行
|
||
check_cli() {
|
||
if [ ! -f "$CLI_SCRIPT" ]; then
|
||
echo "错误: 找不到 memory_cli.py 脚本 ($CLI_SCRIPT)"
|
||
exit 1
|
||
fi
|
||
|
||
if [ ! -x "$CLI_SCRIPT" ] && [ ! -x "$(which python3)" ]; then
|
||
echo "错误: 无法执行Python脚本，请检查权限"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 执行查询并处理错误
|
||
run_query() {
|
||
if ! python3 "$CLI_SCRIPT" query "$@" 2>&1; then
|
||
echo "查询执行失败，可能的原因:"
|
||
echo "1. 记忆系统尚未初始化"
|
||
echo "2. 查询语法错误"
|
||
echo "3. 文件权限问题"
|
||
echo "请检查 memory_cli.py 是否正常工作"
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
# 主逻辑
|
||
main() {
|
||
check_python
|
||
check_cli
|
||
|
||
case "${1:-help}" in
|
||
"today")
|
||
echo "=== 今天的事件 ==="
|
||
run_query --where "date='$(date +%Y-%m-%d)'" --order-by "timestamp desc"
|
||
;;
|
||
"week")
|
||
echo "=== 本周的事件 ==="
|
||
run_query --where "date>='$(date -d 'last monday' +%Y-%m-%d)'" --order-by "timestamp desc"
|
||
;;
|
||
"recent")
|
||
days="${2:-7}"
|
||
echo "=== 最近 $days 天的事件 ==="
|
||
run_query --where "timestamp >= '$(date -d "$days days ago" +%Y-%m-%dT%H:%M:%S)'" --order-by "timestamp desc"
|
||
;;
|
||
"search")
|
||
keyword="${2:-}"
|
||
if [ -z "$keyword" ]; then
|
||
echo "错误: 请提供搜索关键词"
|
||
exit 1
|
||
fi
|
||
echo "=== 搜索包含 '$keyword' 的事件 ==="
|
||
# 注意: 当前CLI不支持LIKE操作，这里用简单的方式
|
||
run_query --where "title LIKE '%$keyword%'" 2>/dev/null || run_query | grep -i "$keyword"
|
||
;;
|
||
"types")
|
||
echo "=== 事件类型统计 ==="
|
||
if run_query --select "type" --format json 2>/dev/null | python3 -c "
|
||
import sys, json
|
||
try:
|
||
data = json.load(sys.stdin)
|
||
types = {}
|
||
for item in data:
|
||
t = item.get('type', 'unknown')
|
||
types[t] = types.get(t, 0) + 1
|
||
if not types:
|
||
print('暂无事件数据')
|
||
else:
|
||
for t, count in sorted(types.items(), key=lambda x: x[1], reverse=True):
|
||
print(f'{t}: {count}')
|
||
except Exception as e:
|
||
print(f'统计失败: {e}')
|
||
print('请检查Python环境和JSON数据格式')
|
||
" 2>/dev/null; then
|
||
echo "统计功能需要Python 3和JSON支持"
|
||
fi
|
||
;;
|
||
"tags")
|
||
echo "=== 标签使用统计 ==="
|
||
if run_query --select "tags" --format json 2>/dev/null | python3 -c "
|
||
import sys, json
|
||
try:
|
||
data = json.load(sys.stdin)
|
||
tags = {}
|
||
for item in data:
|
||
for tag in item.get('tags', []):
|
||
tags[tag] = tags.get(tag, 0) + 1
|
||
if not tags:
|
||
print('暂无标签数据')
|
||
else:
|
||
for tag, count in sorted(tags.items(), key=lambda x: x[1], reverse=True)[:20]:
|
||
print(f'{tag}: {count}')
|
||
except Exception as e:
|
||
print(f'统计失败: {e}')
|
||
print('请检查Python环境和JSON数据格式')
|
||
" 2>/dev/null; then
|
||
echo "统计功能需要Python 3和JSON支持"
|
||
fi
|
||
;;
|
||
"stats")
|
||
echo "=== 记忆系统统计 ==="
|
||
echo "Episodic事件数量: $(find "$MEMORY_ROOT/episodic" -name "*.md" 2>/dev/null | wc -l)"
|
||
echo "Long-term文档数量: $(find "$MEMORY_ROOT/long-term" -name "*.md" 2>/dev/null | wc -l)"
|
||
echo "Short-term文件数量: $(find "$MEMORY_ROOT/short-term" -name "*.md" 2>/dev/null | wc -l)"
|
||
echo "总计: $(find "$MEMORY_ROOT" -name "*.md" 2>/dev/null | wc -l)"
|
||
;;
|
||
"help"|*)
|
||
show_help
|
||
;;
|
||
esac
|
||
}
|
||
|
||
main "$@"
|