Function getTracesAndAllDataForUser

packages/bigquery/src/client.ts:491–595

(
  userId?: string,
  limit = 50,
  pageCursor?: string,
  dataset = DATASET,
)

  relabels: Relabel[]
}
export async function getTracesAndAllDataForUser(
  userId?: string,
  limit = 50,
  pageCursor?: string,
  dataset = DATASET,
): Promise<TraceBundle[]> {
  const EXTRA_TRACE_TYPES = [
    'get-expanded-file-context-for-training',
    'get-expanded-file-context-for-training-blobs',
    'grade-run',
  ] as const

  /* prettier-ignore */
  const sql = `
  /*──────────────── base (latest N get-relevant-files rows) ─────────────*/
  WITH base AS (
    SELECT id, agent_step_id, user_id, created_at, type, payload
    FROM   \`${dataset}.${TRACES_TABLE}\`
    WHERE  ${userId ? 'user_id = @userId AND' : ''} type = 'get-relevant-files' AND JSON_EXTRACT_SCALAR(payload, '$.request_type') = 'Key'
           ${pageCursor ? 'AND created_at < @pageCursor' : ''}
    ORDER  BY created_at DESC
    LIMIT  @limit
  ),

  /*───────────── extra traces that share agent_step_id,user_id ──────────*/
  filtered AS (
    SELECT t.id, t.agent_step_id, t.user_id, t.created_at, t.type, t.payload
    FROM   \`${dataset}.${TRACES_TABLE}\` AS t
    JOIN   base                         AS b
           USING (agent_step_id, user_id)
    WHERE  t.type IN (${EXTRA_TRACE_TYPES.map(t => `'${t}'`).join(', ')})
  ),

  all_rows AS (
    SELECT * FROM base
    UNION ALL
    SELECT * FROM filtered
  )

  /*────────────────────────── final aggregation ─────────────────────────*/
  SELECT
    ANY_VALUE(IF(type='get-relevant-files', t.id,         NULL)) AS id,
    t.agent_step_id,
    ANY_VALUE(IF(type='get-relevant-files', t.user_id,    NULL)) AS user_id,
    ANY_VALUE(IF(type='get-relevant-files', t.created_at, NULL)) AS created_at,
    ANY_VALUE(IF(type='get-relevant-files', t.payload,    NULL)) AS payload,

    ARRAY_AGG(
      CASE
        WHEN type <> 'get-relevant-files'
        THEN STRUCT(t.id, t.created_at, t.type, t.payload)
      END
      IGNORE NULLS
      ORDER BY t.created_at
    ) AS related_traces,

    ARRAY_AGG(r IGNORE NULLS) AS relabels
  FROM all_rows AS t