#!/bin/bash
# 测试脚本 - 生成少量样本以验证功能

# 检查是否设置了API密钥
if [ -z "$OPENAI_API_KEY" ]; then
    echo "❌ 错误: 未设置 OPENAI_API_KEY 环境变量"
    echo "请运行: export OPENAI_API_KEY='your-api-key'"
    exit 1
fi

# 激活虚拟环境
source .venv/bin/activate

echo "🧪 测试运行 - 生成2个高质量样本"
echo "================================"
echo ""

# 运行脚本生成2个样本
python3 generate_problems_openai.py \
    --max-samples 2 \
    --min-score 90 \
    --model gpt-4o-mini \
    --output test_problems.jsonl

echo ""
echo "================================"
echo "✅ 测试完成！"
echo ""
echo "查看结果："
echo "  cat test_problems.jsonl | python3 -m json.tool | head -n 50"
echo ""
echo "查看生成的问题数量："
echo "  wc -l test_problems.jsonl"