#!/bin/bash
# WellAPI 图像生成测试

API_URL="https://api.wellapi.ai/v1/images/generations"

echo "🎨 测试 WellAPI 图像生成..."
echo ""

# 测试 1: 标准图像生成端点
echo "测试 1: /v1/images/generations"
curl -s -X POST "$API_URL" \
  -H "Authorization: Bearer $WELL_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "dall-e-3",
    "prompt": "a cute sun rising, watercolor style, children book illustration",
    "size": "1024x1024",
    "n": 1
  }' | head -c 500

echo ""
echo ""

# 测试 2: chat completions 带图像
echo "测试 2: /v1/chat/completions"
curl -s -X POST "https://api.wellapi.ai/v1/chat/completions" \
  -H "Authorization: Bearer $WELL_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-4o-image",
    "messages": [
      {"role": "user", "content": "Generate an image: a cute sun rising"}
    ]
  }' | head -c 500

echo ""
echo ""
echo "完成!"