Function fetchRows

scripts/fetch-recent-chat-completion-traces.ts:250–301

(args: Args)

}

async function fetchRows(args: Args): Promise<TraceRow[]> {
  const bigquery = new BigQuery()
  const table = `\`${args.dataset}.chat_completion_traces\``
  const fields = `
    trace_session_id,
    agent_id,
    created_at,
    message_count,
    message_start_index,
    TO_JSON_STRING(messages) AS messages_json
  `

  const query = args.traceSessionId
    ? `
      SELECT ${fields}
      FROM ${table}
      WHERE trace_session_id = @traceSessionId
        AND trace_lineage_id = trace_session_id
        AND created_at >= TIMESTAMP_SUB(CURRENT_TIMESTAMP(), INTERVAL @lookbackHours HOUR)
      ORDER BY trace_session_id, created_at, id
    `
    : `
      WITH recent_rows AS (
        SELECT *
        FROM ${table}
        WHERE created_at >= TIMESTAMP_SUB(CURRENT_TIMESTAMP(), INTERVAL @lookbackHours HOUR)
          AND trace_lineage_id = trace_session_id
      ),
      recent_sessions AS (
        SELECT trace_session_id, MAX(created_at) AS last_created_at
        FROM recent_rows
        GROUP BY trace_session_id
        ORDER BY last_created_at DESC
        LIMIT @limit
      )
      SELECT ${fields}
      FROM recent_rows
      JOIN recent_sessions USING (trace_session_id)
      ORDER BY trace_session_id, created_at, id
    `

  const [rows] = await bigquery.query({
    query,
    params: {
      limit: args.limit,
      lookbackHours: args.lookbackHours,
      ...(args.traceSessionId ? { traceSessionId: args.traceSessionId } : {}),
    },
  })

  return rows as TraceRow[]
}

function printSummary(args: Args, sessions: TraceSession[]) {
  console.log(