#!/bin/bash

# 最简单的测试脚本
# 使用方法: ./simple-test.sh YOUR_GEMINI_API_KEY

if [ -z "$1" ]; then
    echo "❌ 错误: 需要提供Gemini API密钥"
    echo "使用方法: $0 YOUR_GEMINI_API_KEY"
    echo ""
    echo "获取API密钥: https://ai.google.dev"
    exit 1
fi

API_KEY=$1
echo "🚀 测试简单Gemini代理..."
echo "📍 服务地址: http://localhost:8081"
echo "🔑 API Key: ${API_KEY:0:8}..."
echo ""

# 简单聊天测试
echo "💬 发送测试消息..."
response=$(curl -s http://localhost:8081/v1/chat/completions \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $API_KEY" \
  -d '{
    "model": "gpt-3.5-turbo",
    "messages": [{"role": "user", "content": "你好！请简单回复。"}],
    "max_tokens": 50
  }')

echo "📥 服务器响应:"
echo "$response"
echo ""

# 检查响应是否包含choices
if echo "$response" | grep -q '"choices"'; then
    echo "✅ 测试成功！代理服务正常工作。"
else
    echo "❌ 测试失败！可能的原因："
    echo "   - API密钥无效"
    echo "   - 网络连接问题"
    echo "   - 服务未正常启动"
fi

echo ""
echo "🔧 服务管理命令:"
echo "   docker compose ps        # 查看状态"
echo "   docker compose logs -f   # 查看日志"
echo "   docker compose restart   # 重启服务"