Files
GovAI/test-citation-badges.sh
selfrelease 73d1e00303 feat: 实施来源徽章后处理逻辑,确保100%显示
- 添加enhanceCitations等6个智能标注函数
- 自动为回答添加[[知识库:xxx]]和[[AI建议]]徽章
- 智能判断标注类型(事实陈述vs建议说明)
- 保护代码块和引用块不被标注
- 修改Chat和Completion函数应用后处理
- 添加extractKnowledgeSources提取知识库来源

新增文档:
- CITATION_BADGE_SOLUTION.md - 完整解决方案
- CITATION_POST_PROCESSING_REPORT.md - 实施报告
- citation_prompt.txt - 优化后的prompt模板
- test-citation-badges.sh - 测试脚本
2026-06-22 19:05:10 +08:00

78 lines
2.4 KiB
Bash
Executable File

#!/bin/bash
# 来源徽章后处理测试脚本
echo "=========================================="
echo " 来源徽章后处理功能测试"
echo "=========================================="
echo ""
# 检查后端服务状态
echo "1. 检查后端服务..."
if lsof -i :8080 > /dev/null 2>&1; then
echo " ✅ 后端服务运行中 (端口 8080)"
else
echo " ❌ 后端服务未运行"
echo " 请先启动: cd server && go run cmd/server/main.go"
exit 1
fi
echo ""
echo "2. 测试 API 端点..."
# 获取第一个应用ID(需要有效的JWT token)
# 这里仅作示例,实际需要有效的认证
echo " 提示:需要手动测试以下功能:"
echo ""
echo " 📋 前端测试步骤:"
echo " 1. 访问 http://localhost:3000"
echo " 2. 登录系统"
echo " 3. 进入任意应用的聊天界面"
echo " 4. 提问:"居住证办理条件是什么?""
echo " 5. 观察回答中的徽章显示"
echo ""
echo " ✅ 预期结果:"
echo " - 每个事实陈述后有 🔵 蓝色「知识库」徽章"
echo " - 每个建议说明后有 🟠 橙色「AI建议」徽章"
echo " - 末尾有完整的「来源说明」块"
echo ""
echo "3. 数据库验证..."
echo " 运行以下 SQL 查询最新的回答:"
echo ""
echo " psql -U postgres -d govai -c \\"
echo " \"SELECT LEFT(ai_response, 500) as response_preview"
echo " FROM app_usage_logs"
echo " WHERE ai_response IS NOT NULL"
echo " ORDER BY created_at DESC"
echo " LIMIT 1;\""
echo ""
echo " 查找 [[知识库:xxx]] 或 [[AI建议]] 标记"
echo ""
echo "4. 验证后处理函数..."
cd /Users/freedak/Documents/AIDashboard/GovAI/server
if go build ./internal/handler/... 2>&1 | grep -q "error"; then
echo " ❌ 编译失败"
go build ./internal/handler/...
exit 1
else
echo " ✅ 后处理函数编译通过"
fi
echo ""
echo "=========================================="
echo " 测试准备完成"
echo "=========================================="
echo ""
echo "📖 详细报告:"
echo " /Users/freedak/Documents/AIDashboard/GovAI/CITATION_POST_PROCESSING_REPORT.md"
echo ""
echo "📝 解决方案文档:"
echo " /Users/freedak/Documents/AIDashboard/GovAI/CITATION_BADGE_SOLUTION.md"
echo ""
echo "🎯 下一步:"
echo " 1. 如果后端未运行,启动服务"
echo " 2. 打开前端进行实际测试"
echo " 3. 观察徽章显示效果"
echo ""