100d1a06eb
- 后端 API（Flask + Gunicorn） - RAG 引擎（混合检索 + 云端 Reranker + 引用溯源） - 文档解析（MinerU + 多格式支持） - Docker 生产部署配置 - 排除前端项目、敏感配置、模型文件
210 lines
6.1 KiB
Bash
210 lines
6.1 KiB
Bash
#!/bin/bash
|
|
# RAG API 完整测试脚本
|
|
# 基于 docs/curl测试手册.md
|
|
|
|
set -e
|
|
BASE_URL="http://localhost:5001"
|
|
PASS=0
|
|
FAIL=0
|
|
RESULTS=""
|
|
|
|
log_result() {
|
|
local name="$1"
|
|
local status="$2"
|
|
if [ "$status" = "PASS" ]; then
|
|
PASS=$((PASS + 1))
|
|
RESULTS="${RESULTS}✅ ${name}\n"
|
|
else
|
|
FAIL=$((FAIL + 1))
|
|
RESULTS="${RESULTS}❌ ${name}\n"
|
|
fi
|
|
}
|
|
|
|
echo "=========================================="
|
|
echo "RAG API 完整测试"
|
|
echo "=========================================="
|
|
|
|
# 1. 健康检查
|
|
echo -e "\n--- 1. 健康检查 ---"
|
|
result=$(curl -s "$BASE_URL/health")
|
|
if echo "$result" | grep -q '"status":"ok"'; then
|
|
log_result "GET /health" "PASS"
|
|
else
|
|
log_result "GET /health" "FAIL"
|
|
fi
|
|
|
|
# 2. 问答接口
|
|
echo -e "\n--- 2. 问答接口 ---"
|
|
|
|
# POST /rag (简化测试)
|
|
echo '{"message":"三峡工程","chat_history":[]}' > /tmp/rag.json
|
|
result=$(curl -s -X POST "$BASE_URL/rag" -H "Content-Type: application/json" -d @/tmp/rag.json 2>&1)
|
|
if echo "$result" | grep -q '"type":"finish"'; then
|
|
log_result "POST /rag" "PASS"
|
|
else
|
|
log_result "POST /rag" "FAIL"
|
|
fi
|
|
|
|
# POST /chat
|
|
echo '{"message":"你好","chat_history":[]}' > /tmp/chat.json
|
|
result=$(curl -s -X POST "$BASE_URL/chat" -H "Content-Type: application/json" -d @/tmp/chat.json)
|
|
if echo "$result" | grep -q '"answer"'; then
|
|
log_result "POST /chat" "PASS"
|
|
else
|
|
log_result "POST /chat" "FAIL"
|
|
fi
|
|
|
|
# 3. 检索接口
|
|
echo -e "\n--- 3. 检索接口 ---"
|
|
echo '{"query":"三峡工程","top_k":3}' > /tmp/search.json
|
|
result=$(curl -s -X POST "$BASE_URL/search" -H "Content-Type: application/json" -d @/tmp/search.json)
|
|
if echo "$result" | grep -q '"contexts"'; then
|
|
log_result "POST /search" "PASS"
|
|
else
|
|
log_result "POST /search" "FAIL"
|
|
fi
|
|
|
|
# 4. 向量库管理
|
|
echo -e "\n--- 4. 向量库管理 ---"
|
|
|
|
# GET /collections
|
|
result=$(curl -s "$BASE_URL/collections")
|
|
if echo "$result" | grep -q '"collections"'; then
|
|
log_result "GET /collections" "PASS"
|
|
else
|
|
log_result "GET /collections" "FAIL"
|
|
fi
|
|
|
|
# 创建测试向量库
|
|
echo '{"name":"test_kb_auto","display_name":"自动测试库"}' > /tmp/create_kb.json
|
|
result=$(curl -s -X POST "$BASE_URL/collections" -H "Content-Type: application/json" -d @/tmp/create_kb.json)
|
|
if echo "$result" | grep -q '"success":true'; then
|
|
log_result "POST /collections (创建)" "PASS"
|
|
else
|
|
log_result "POST /collections (创建)" "FAIL"
|
|
fi
|
|
|
|
# 修改向量库
|
|
echo '{"display_name":"自动测试库-已修改"}' > /tmp/update_kb.json
|
|
result=$(curl -s -X PUT "$BASE_URL/collections/test_kb_auto" -H "Content-Type: application/json" -d @/tmp/update_kb.json)
|
|
if echo "$result" | grep -q '"success"'; then
|
|
log_result "PUT /collections/test_kb_auto" "PASS"
|
|
else
|
|
log_result "PUT /collections/test_kb_auto" "FAIL"
|
|
fi
|
|
|
|
# 删除测试向量库
|
|
result=$(curl -s -X DELETE "$BASE_URL/collections/test_kb_auto")
|
|
if echo "$result" | grep -q '"success"'; then
|
|
log_result "DELETE /collections/test_kb_auto" "PASS"
|
|
else
|
|
log_result "DELETE /collections/test_kb_auto" "FAIL"
|
|
fi
|
|
|
|
# GET /collections/<name>/documents
|
|
result=$(curl -s "$BASE_URL/collections/public_kb/documents")
|
|
if echo "$result" | grep -q '"documents"'; then
|
|
log_result "GET /collections/public_kb/documents" "PASS"
|
|
else
|
|
log_result "GET /collections/public_kb/documents" "FAIL"
|
|
fi
|
|
|
|
# 5. 文档管理
|
|
echo -e "\n--- 5. 文档管理 ---"
|
|
result=$(curl -s "$BASE_URL/documents/list?collection=public_kb&page=1&page_size=5")
|
|
if echo "$result" | grep -q '"documents"'; then
|
|
log_result "GET /documents/list" "PASS"
|
|
else
|
|
log_result "GET /documents/list" "FAIL"
|
|
fi
|
|
|
|
# 6. 同步服务
|
|
echo -e "\n--- 6. 同步服务 ---"
|
|
result=$(curl -s "$BASE_URL/sync/status")
|
|
if echo "$result" | grep -q '"enabled"'; then
|
|
log_result "GET /sync/status" "PASS"
|
|
else
|
|
log_result "GET /sync/status" "FAIL"
|
|
fi
|
|
|
|
# 7. 反馈系统
|
|
echo -e "\n--- 7. 反馈系统 ---"
|
|
result=$(curl -s "$BASE_URL/feedback/stats")
|
|
if echo "$result" | grep -q '"stats"'; then
|
|
log_result "GET /feedback/stats" "PASS"
|
|
else
|
|
log_result "GET /feedback/stats" "FAIL"
|
|
fi
|
|
|
|
# 8. FAQ 管理
|
|
echo -e "\n--- 8. FAQ 管理 ---"
|
|
result=$(curl -s "$BASE_URL/faq?page=1&page_size=5")
|
|
if echo "$result" | grep -q '"faqs"'; then
|
|
log_result "GET /faq" "PASS"
|
|
else
|
|
log_result "GET /faq" "FAIL"
|
|
fi
|
|
|
|
# 9. 出题系统
|
|
echo -e "\n--- 9. 出题系统 ---"
|
|
result=$(curl -s "$BASE_URL/exam/health")
|
|
if echo "$result" | grep -q '"status":"ok"'; then
|
|
log_result "GET /exam/health" "PASS"
|
|
else
|
|
log_result "GET /exam/health" "FAIL"
|
|
fi
|
|
|
|
# 10. 图片服务
|
|
echo -e "\n--- 10. 图片服务 ---"
|
|
result=$(curl -s "$BASE_URL/images/list?limit=5")
|
|
if echo "$result" | grep -q '"images"'; then
|
|
log_result "GET /images/list" "PASS"
|
|
else
|
|
log_result "GET /images/list" "FAIL"
|
|
fi
|
|
|
|
# 获取第一张图片ID并测试
|
|
image_id=$(curl -s "$BASE_URL/images/list?limit=1" | grep -o '"image_id":"[^"]*"' | head -1 | cut -d'"' -f4)
|
|
if [ -n "$image_id" ]; then
|
|
http_code=$(curl -s -o /dev/null -w "%{http_code}" "$BASE_URL/images/$image_id")
|
|
if [ "$http_code" = "200" ]; then
|
|
log_result "GET /images/<id>" "PASS"
|
|
else
|
|
log_result "GET /images/<id>" "FAIL"
|
|
fi
|
|
else
|
|
log_result "GET /images/<id>" "FAIL (无图片)"
|
|
fi
|
|
|
|
# 11. 报告服务
|
|
echo -e "\n--- 11. 报告服务 ---"
|
|
result=$(curl -s "$BASE_URL/reports/weekly")
|
|
if echo "$result" | grep -q '"report"'; then
|
|
log_result "GET /reports/weekly" "PASS"
|
|
else
|
|
log_result "GET /reports/weekly" "FAIL"
|
|
fi
|
|
|
|
# 12. 知识库路由
|
|
echo -e "\n--- 12. 知识库路由 ---"
|
|
echo '{"query":"三峡工程"}' > /tmp/route.json
|
|
result=$(curl -s -X POST "$BASE_URL/kb/route" -H "Content-Type: application/json" -d @/tmp/route.json)
|
|
if echo "$result" | grep -q '"target_collections"'; then
|
|
log_result "POST /kb/route" "PASS"
|
|
else
|
|
log_result "POST /kb/route" "FAIL"
|
|
fi
|
|
|
|
# 输出结果
|
|
echo -e "\n=========================================="
|
|
echo "测试结果汇总"
|
|
echo "=========================================="
|
|
echo -e "$RESULTS"
|
|
echo "------------------------------------------"
|
|
echo "总计: $PASS 通过, $FAIL 失败"
|
|
echo "=========================================="
|
|
|
|
# 清理临时文件
|
|
rm -f /tmp/rag.json /tmp/chat.json /tmp/search.json /tmp/create_kb.json /tmp/update_kb.json /tmp/route.json
|
|
|
|
exit $FAIL |