#!/bin/bash

echo "======================================"
echo "  Chroma 服务测试脚本"
echo "======================================"
echo ""

CHROMA_URL="http://localhost:8000"
API_BASE="http://localhost:8080"

echo "[1/4] 测试 Chroma 健康状态..."
HEALTH=$(curl -s -o /dev/null -w "%{http_code}" $CHROMA_URL/api/v1/heartbeat)
if [ "$HEALTH" -eq 200 ]; then
    echo "✓ Chroma 服务正常"
else
    echo "✗ Chroma 服务异常 (HTTP $HEALTH)"
    exit 1
fi

echo ""
echo "[2/4] 测试知识库检索..."
SEARCH_RESULT=$(curl -s -X POST $API_BASE/api/knowledge/search \
  -H "Content-Type: application/json" \
  -d '{
    "tenantId": "1",
    "collectionName": "workflow_knowledge_base",
    "question": "产品功能",
    "topK": 5
  }')

if [ -n "$SEARCH_RESULT" ]; then
    echo "✓ 知识库检索成功"
    echo "结果: $SEARCH_RESULT"
else
    echo "✗ 知识库检索失败"
fi

echo ""
echo "[3/4] 测试对话接口..."
CHAT_RESULT=$(curl -s -X POST $API_BASE/api/chat \
  -H "Content-Type: application/json" \
  -d '{
    "question": "你们的产品有什么功能？",
    "companyId": 1
  }')

if [ -n "$CHAT_RESULT" ]; then
    echo "✓ 对话接口正常"
    echo "响应: $CHAT_RESULT"
else
    echo "✗ 对话接口异常"
fi

echo ""
echo "[4/4] 测试数据迁移..."
echo "请手动执行迁移测试: mvn test -Dtest=ChromaMigrationTest"

echo ""
echo "======================================"
echo "  测试完成！"
echo "======================================"