#!/usr/bin/env bash
# run_similar_batches.sh v2 — 12题/批全量相似问法（修复 tmp 冲突与 _i 注入）
cd "/home/zyw/Downloads/dl-hub/01-AI课程设计项目/知识库问答库"
mkdir -p 相似问法分片
BATCH=12
N=1299
TOTAL=$(( (N + BATCH - 1) / BATCH ))
WORKERS=$1
for w in $(seq 0 $((WORKERS-1))); do
  (
    for b in $(seq $w $WORKERS $((TOTAL-1))); do
      start=$(( b * BATCH ))
      out="相似问法分片/sim-$b.json"
      [ -f "$out" ] && [ -s "$out" ] && continue
      echo "[w$w] 批次 $b start=$start"
      timeout 1200 python3 gen_similar.py "$start" "$BATCH" "$out" >> "相似问法分片/run-$w.log" 2>&1
    done
  ) &
done
wait
echo "ALL WORKERS DONE"
ls 相似问法分片/sim-*.json 2>/dev/null | wc -l
