forked from zhanggangyong/agent_management
95 lines
2.8 KiB
Bash
Executable File
95 lines
2.8 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Docker测试脚本
|
|
|
|
set -e
|
|
|
|
echo "=========================================="
|
|
echo "Facebook搜索Agent Docker测试"
|
|
echo "=========================================="
|
|
|
|
# 颜色定义
|
|
GREEN='\033[0;32m'
|
|
RED='\033[0;31m'
|
|
YELLOW='\033[1;33m'
|
|
NC='\033[0m' # No Color
|
|
|
|
# 检查Docker是否安装
|
|
if ! command -v docker &> /dev/null; then
|
|
echo -e "${RED}错误: Docker未安装${NC}"
|
|
exit 1
|
|
fi
|
|
|
|
# 检查docker-compose是否安装
|
|
if ! command -v docker-compose &> /dev/null && ! docker compose version &> /dev/null; then
|
|
echo -e "${RED}错误: docker-compose未安装${NC}"
|
|
exit 1
|
|
fi
|
|
|
|
# 使用docker compose或docker-compose
|
|
if docker compose version &> /dev/null; then
|
|
DOCKER_COMPOSE="docker compose"
|
|
else
|
|
DOCKER_COMPOSE="docker-compose"
|
|
fi
|
|
|
|
echo -e "${YELLOW}1. 构建Docker镜像...${NC}"
|
|
cd /home/taiji/tools/aks_agent/facebook_agent
|
|
$DOCKER_COMPOSE build
|
|
|
|
echo -e "${YELLOW}2. 启动服务...${NC}"
|
|
$DOCKER_COMPOSE up -d
|
|
|
|
echo -e "${YELLOW}3. 等待服务启动...${NC}"
|
|
sleep 5
|
|
|
|
echo -e "${YELLOW}4. 检查服务状态...${NC}"
|
|
$DOCKER_COMPOSE ps
|
|
|
|
echo -e "${YELLOW}5. 测试API服务 (端口18000)...${NC}"
|
|
echo "测试根路径:"
|
|
curl -s http://localhost:18000/ | python3 -m json.tool || echo -e "${RED}API服务未响应${NC}"
|
|
|
|
echo -e "\n测试健康检查:"
|
|
curl -s http://localhost:18000/health | python3 -m json.tool || echo -e "${RED}健康检查失败${NC}"
|
|
|
|
echo -e "\n${YELLOW}6. 测试搜索功能...${NC}"
|
|
SEARCH_RESULT=$(curl -s -X POST http://localhost:18000/search \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"query": "technology news", "limit": 3}')
|
|
|
|
if [ $? -eq 0 ]; then
|
|
echo "$SEARCH_RESULT" | python3 -m json.tool | head -50
|
|
echo -e "${GREEN}搜索测试完成${NC}"
|
|
else
|
|
echo -e "${RED}搜索测试失败${NC}"
|
|
fi
|
|
|
|
echo -e "\n${YELLOW}7. 测试MCP服务 (端口18001)...${NC}"
|
|
echo "测试MCP健康检查:"
|
|
curl -s http://localhost:18001/health | python3 -m json.tool || echo -e "${RED}MCP服务未响应${NC}"
|
|
|
|
echo -e "\n测试MCP工具列表:"
|
|
curl -s -X POST http://localhost:18001/mcp/call \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"method": "tools/list", "params": {"id": "test"}}' | python3 -m json.tool || echo -e "${RED}MCP工具列表获取失败${NC}"
|
|
|
|
echo -e "\n${YELLOW}8. 查看服务日志...${NC}"
|
|
echo "API服务日志 (最后10行):"
|
|
docker logs facebook-api --tail 10 2>&1 || echo "无法获取日志"
|
|
|
|
echo -e "\nMCP服务日志 (最后10行):"
|
|
docker logs facebook-mcp --tail 10 2>&1 || echo "无法获取日志"
|
|
|
|
echo -e "\n${GREEN}=========================================="
|
|
echo "测试完成！"
|
|
echo "==========================================${NC}"
|
|
echo ""
|
|
echo "服务地址:"
|
|
echo " - API服务: http://20.6.9.191:18000"
|
|
echo " - MCP服务: http://20.6.9.191:18001"
|
|
echo ""
|
|
echo "查看日志: docker logs facebook-api 或 docker logs facebook-mcp"
|
|
echo "停止服务: docker-compose down"
|
|
|