Function main

scripts/fetch-recent-chat-completion-traces.ts:342–378

()

}

async function main() {
  const args = parseArgs()
  const rows = await fetchRows(args)
  const sessions = reconstructTraceSessions(rows)
  const outDir = resolve(args.outDir)
  const indexEntries = sessions.map(buildIndexEntry)

  await mkdir(outDir, { recursive: true })
  await Promise.all(
    sessions.map((session, index) =>
      writeFile(
        join(outDir, indexEntries[index]!.file),
        JSON.stringify(buildTraceFile(session), null, 2),
      ),
    ),
  )
  await writeFile(
    join(outDir, 'index.json'),
    JSON.stringify(
      {
        generated_at: new Date().toISOString(),
        dataset: args.dataset,
        lookback_hours: args.lookbackHours,
        trace_session_id: args.traceSessionId,
        sessions: indexEntries,
      },
      null,
      2,
    ),
  )

  printSummary(args, sessions)
  console.log('')
  console.log(
    `Wrote ${sessions.length} trace file(s) and index.json to ${outDir}`,
  )
}

main().catch((error) => {
  console.error(error)