import { setCurrentStrategy } from '../strategies/selector';
import { langGraphGenerator } from '../index';

const BOOK_ID = '198';

async function main() {
  setCurrentStrategy('per-chapter');
  console.log(`[test] 生成书${BOOK_ID}...`);
  const start = Date.now();
  await langGraphGenerator.generate(BOOK_ID, '牛顿发现万有引力的故事', '1000');
  console.log(`[test] 完成, 耗时: ${((Date.now() - start) / 1000).toFixed(1)}s`);
}

main().catch(e => console.error('[test] 失败:', e));
