Function main

scripts/export-deepseek-session-traces.ts:482–570

()

}

async function main() {
  const args = parseArgs()
  const outputPath = resolve(args.outPath)
  const generatedAt = new Date().toISOString()

  console.log(
    [
      `Querying ${args.dataset}.message`,
      args.startDate || args.endDate
        ? `window: ${args.startDate ?? '-infinity'} to ${args.endDate ?? 'now'}`
        : `window: ${args.lookbackDays}d ending ${args.beforeDays}d ago`,
      args.totalSessions
        ? `total sessions: ${args.totalSessions} (${args.sampleMode})`
        : `sessions per agent: ${args.sessionsPerAgent}`,
      `min representative messages: ${args.minMessages}`,
      args.clientId ? `client_id: ${args.clientId}` : undefined,
    ]
      .filter(Boolean)
      .join('\n'),
  )
  console.log('')

  const candidates = await fetchCandidateSessions(args)
  const representativeRows = await fetchRepresentativeRows(
    args,
    candidates.map((candidate) => candidate.representative_message_id),
  )
  const traces = candidates.map((candidate) =>
    buildTrace(candidate, representativeRows),
  )

  if (!args.skipAggregate) {
    await mkdir(dirname(outputPath), { recursive: true })
    await Bun.write(
      outputPath,
      JSON.stringify(
        {
          generated_at: generatedAt,
          dataset: OUTPUT_DATASET_NAME,
          lookback_days: args.lookbackDays,
          before_days: args.beforeDays,
          start_date: args.startDate,
          end_date: args.endDate,
          sample_mode: args.sampleMode,
          trace_count: traces.length,
          traces,
        },
        null,
        2,
      ),
    )
  }
  const traceFiles = await writeTraceFiles({
    outputPath,
    traceDir: args.traceDir,
    dataset: OUTPUT_DATASET_NAME,
    generatedAt,
    traces,