#!/bin/bash

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

if [ -z "$1" ]; then
    echo "使用方法: $0 YOUR_GEMINI_API_KEY"
    echo "示例: $0 AIzaSyDxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
    exit 1
fi

API_KEY=$1
BASE_URL="http://localhost:8081"

echo "Testing Gemini OpenAI Proxy..."
echo "API Key: ${API_KEY:0:10}..."
echo "Base URL: $BASE_URL"
echo

# 测试基本聊天完成
echo "=== 测试基本聊天完成 ==="
curl -s "$BASE_URL/v1/chat/completions" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $API_KEY" \
  -d '{
    "model": "gpt-3.5-turbo",
    "messages": [{"role": "user", "content": "你好！请用中文回复。"}],
    "temperature": 0.7,
    "max_tokens": 150
  }' | jq '.'

echo -e "\n"

# 测试GPT-4模型
echo "=== 测试GPT-4模型 ==="
curl -s "$BASE_URL/v1/chat/completions" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $API_KEY" \
  -d '{
    "model": "gpt-4",
    "messages": [{"role": "user", "content": "解释一下量子计算的基本原理"}],
    "temperature": 0.3,
    "max_tokens": 200
  }' | jq '.'

echo -e "\n"

# 测试流式响应
echo "=== 测试流式响应 ==="
curl -s "$BASE_URL/v1/chat/completions" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $API_KEY" \
  -d '{
    "model": "gpt-3.5-turbo",
    "messages": [{"role": "user", "content": "数数从1到10"}],
    "stream": true,
    "temperature": 0.1
  }'

echo -e "\n\n=== 测试完成 ==="