60 lines
1.5 KiB
Bash
Executable File
60 lines
1.5 KiB
Bash
Executable File
#!/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=== 测试完成 ===" |