File size: 1,358 Bytes
115f0b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
#!/bin/bash
# 快速启动批处理（仅准备和提交，不等待完成）
set -e
echo "🚀 快速启动 Batch API 任务"
echo "========================================"
# 配置
MIN_SCORE=60
MODEL="gpt-4o-mini"
INPUT_FILE="function_dataset_v2.csv"
BATCH_REQUESTS_FILE="batch_requests_full.jsonl"
# 步骤1: 准备批量请求
echo "📋 准备批量请求..."
python3 generate_problems_batch.py prepare \
--input $INPUT_FILE \
--output $BATCH_REQUESTS_FILE \
--min-score $MIN_SCORE \
--model $MODEL
REQUEST_COUNT=$(wc -l < $BATCH_REQUESTS_FILE)
echo ""
echo "✅ 已准备 $REQUEST_COUNT 个请求"
# 步骤2: 提交
echo ""
echo "🚀 提交批处理任务..."
python3 generate_problems_batch.py submit \
--input $BATCH_REQUESTS_FILE \
--model $MODEL \
--description "Scientific computing problems - $REQUEST_COUNT samples" \
| tee batch_submit_output.txt
# 提取 Batch ID
BATCH_ID=$(grep -oP 'batch_[a-zA-Z0-9_]+' batch_submit_output.txt | head -1)
if [ -n "$BATCH_ID" ]; then
echo "$BATCH_ID" > batch_id.txt
echo ""
echo "✅ 批处理已提交！"
echo "📝 Batch ID: $BATCH_ID (已保存到 batch_id.txt)"
echo ""
echo "监控命令:"
echo " python3 generate_problems_batch.py status $BATCH_ID"
echo ""
echo "或运行监控脚本:"
echo " ./monitor_batch.sh"
fi
|