first commit

scripts/07_smoke_test.sh

@@ -0,0 +1,183 @@
+#!/usr/bin/env bash
+# ══════════════════════════════════════════════════
+# 07_smoke_test.sh
+# 端到端冒烟测试：验证三条业务闭环
+# 用法：bash scripts/07_smoke_test.sh
+# ══════════════════════════════════════════════════
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
+cd "$PROJECT_DIR"
+
+RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; BLUE='\033[0;34m'; NC='\033[0m'
+info()  { echo -e "${BLUE}[INFO]${NC} $*"; }
+ok()    { echo -e "${GREEN}[✓]${NC} $*"; }
+fail()  { echo -e "${RED}[✗]${NC} $*"; FAILED=$((FAILED+1)); }
+warn()  { echo -e "${YELLOW}[~]${NC} $*"; }
+
+FAILED=0
+API_BASE="http://localhost"
+
+echo ""
+echo -e "${BLUE}══════════════════════════════════════════${NC}"
+echo -e "${BLUE}  AI合规智能中枢 端到端冒烟测试${NC}"
+echo -e "${BLUE}══════════════════════════════════════════${NC}"
+echo ""
+
+# ── 基础健康检查 ────────────────────────────────
+info "=== 基础设施健康检查 ==="
+
+check_service() {
+    local name=$1; local url=$2
+    if curl -sf "$url" > /dev/null 2>&1; then
+        ok "$name"
+    else
+        fail "$name（$url 不可达）"
+    fi
+}
+
+check_service "API 网关 (Nginx)"           "http://localhost/health"
+check_service "业务后端 (FastAPI)"         "http://localhost:8000/health"
+check_service "嵌入服务 (BGE-M3)"          "http://localhost:8010/health"
+check_service "解析服务 (MinerU)"           "http://localhost:8011/health"
+check_service "Milvus HTTP"               "http://localhost:9091/healthz"
+check_service "Neo4j Browser"             "http://localhost:7474"
+echo ""
+
+# ── 嵌入服务测试 ────────────────────────────────
+info "=== 嵌入服务测试 ==="
+EMBED_RESP=$(curl -sf -X POST http://localhost:8010/embed \
+    -H "Content-Type: application/json" \
+    -d '{"texts": ["GB 18384 电动汽车碰撞安全要求"], "batch_size": 1}' 2>/dev/null || echo "{}")
+
+if echo "$EMBED_RESP" | python3 -c "import sys,json; d=json.load(sys.stdin); assert len(d.get('dense',[])[0])==1024" 2>/dev/null; then
+    ok "BGE-M3 嵌入：返回 1024 维向量"
+else
+    fail "BGE-M3 嵌入失败，响应：${EMBED_RESP:0:200}"
+fi
+echo ""
+
+# ── 创建测试 PDF ────────────────────────────────
+info "=== 创建测试文档 ==="
+TEST_PDF="$PROJECT_DIR/data/uploads/test_regulation.txt"
+cat > "$TEST_PDF" << 'EOF'
+GB 18384-2020 电动汽车安全要求
+
+第一章 总则
+本标准规定了电动汽车的安全要求，适用于M1类纯电动汽车。
+
+第二章 电气安全
+2.1 绝缘电阻要求
+    直流电路绝缘电阻不得低于100Ω/V。
+2.2 碰撞安全
+    车辆碰撞后，高压电系统应自动断电。
+    碰撞后5秒内，高压系统电压应降至60V以下。
+
+第三章 防水要求
+高压系统防护等级应达到IP67。
+EOF
+ok "测试文档创建：$TEST_PDF"
+echo ""
+
+# ── 闭环①：文件上传 → 向量化 → 问答 ───────────
+info "=== 闭环①：法规入库 → 检索问答 ==="
+
+# 创建工作空间
+WORKSPACE_RESP=$(curl -sf -X POST "$API_BASE/api/kb/workspaces" \
+    -H "Content-Type: application/json" \
+    -d '{"name": "测试法规库", "domain": "vehicle_safety"}' 2>/dev/null || echo "{}")
+WS_ID=$(echo "$WORKSPACE_RESP" | python3 -c "import sys,json; print(json.load(sys.stdin).get('id',''))" 2>/dev/null || echo "")
+
+if [[ -n "$WS_ID" ]]; then
+    ok "工作空间创建：$WS_ID"
+else
+    warn "工作空间创建失败（可能接口未完全实现），跳过后续上传测试"
+    WS_ID="test-workspace"
+fi
+
+# 上传文件
+UPLOAD_RESP=$(curl -sf -X POST "$API_BASE/api/kb/files/upload" \
+    -F "file=@$TEST_PDF" \
+    -F "workspace_id=$WS_ID" 2>/dev/null || echo "{}")
+TASK_ID=$(echo "$UPLOAD_RESP" | python3 -c "import sys,json; print(json.load(sys.stdin).get('task_id',''))" 2>/dev/null || echo "")
+
+if [[ -n "$TASK_ID" ]]; then
+    ok "文件上传任务已创建：$TASK_ID"
+
+    # 轮询任务状态（最多等待120秒）
+    info "等待向量化完成..."
+    for i in {1..24}; do
+        TASK_STATUS=$(curl -sf "$API_BASE/api/kb/tasks/$TASK_ID" 2>/dev/null | \
+            python3 -c "import sys,json; print(json.load(sys.stdin).get('status','unknown'))" 2>/dev/null || echo "unknown")
+        if [[ "$TASK_STATUS" == "completed" ]]; then
+            ok "向量化完成（${i}×5s）"
+            break
+        elif [[ "$TASK_STATUS" == "failed" ]]; then
+            fail "向量化失败"
+            break
+        fi
+        echo -n "."
+        sleep 5
+    done
+    echo ""
+
+    # 检索问答
+    QA_RESP=$(curl -sf -X POST "$API_BASE/api/kb/qa" \
+        -H "Content-Type: application/json" \
+        -d "{\"query\": \"碰撞后高压系统电压要求\", \"workspace_id\": \"$WS_ID\", \"top_k\": 3}" 2>/dev/null || echo "{}")
+    ANSWER=$(echo "$QA_RESP" | python3 -c "import sys,json; print(json.load(sys.stdin).get('answer','')[:100])" 2>/dev/null || echo "")
+
+    if [[ -n "$ANSWER" ]]; then
+        ok "问答成功：${ANSWER}..."
+    else
+        warn "问答返回空（LLM API 可能未配置或响应缓慢）"
+    fi
+else
+    warn "文件上传失败（接口可能未实现）"
+fi
+echo ""
+
+# ── 闭环②：合规审查 ────────────────────────────
+info "=== 闭环②：文档上传 → 合规审查 ==="
+
+CHECK_RESP=$(curl -sf -X POST "$API_BASE/api/compliance/check" \
+    -H "Content-Type: application/json" \
+    -d '{"query": "供应商文件是否符合GB 18384碰撞安全要求", "domains": ["vehicle_safety"]}' 2>/dev/null || echo "{}")
+RISK=$(echo "$CHECK_RESP" | python3 -c "import sys,json; print(json.load(sys.stdin).get('risk_level','unknown'))" 2>/dev/null || echo "unknown")
+
+if [[ "$RISK" != "unknown" && -n "$RISK" ]]; then
+    ok "合规审查完成，风险等级：$RISK"
+else
+    warn "合规审查接口返回空（功能可能未完全实现）"
+fi
+echo ""
+
+# ── 闭环③：法规监控 ────────────────────────────
+info "=== 闭环③：法规监控源配置 ==="
+
+SOURCE_RESP=$(curl -sf -X POST "$API_BASE/api/regulation/sources" \
+    -H "Content-Type: application/json" \
+    -d '{"name": "测试监控源", "url": "https://std.samr.gov.cn", "domain": "vehicle_safety"}' 2>/dev/null || echo "{}")
+SOURCE_ID=$(echo "$SOURCE_RESP" | python3 -c "import sys,json; print(json.load(sys.stdin).get('id',''))" 2>/dev/null || echo "")
+
+if [[ -n "$SOURCE_ID" ]]; then
+    ok "监控源配置成功：$SOURCE_ID"
+else
+    warn "监控源配置返回空（功能可能未完全实现）"
+fi
+echo ""
+
+# ── 汇总 ────────────────────────────────────────
+echo ""
+echo -e "${BLUE}══════════════════════════════════════════${NC}"
+if [[ $FAILED -eq 0 ]]; then
+    echo -e "${GREEN}  全部检查通过！${NC}"
+else
+    echo -e "${YELLOW}  完成，${FAILED} 项失败${NC}（部分功能可能尚未实现）"
+fi
+echo -e "${BLUE}══════════════════════════════════════════${NC}"
+echo ""
+echo "查看服务日志："
+echo "  docker compose logs -f compliance-backend"
+echo "  docker compose logs -f celery-worker"