Function getTracesWithoutRelabels

packages/bigquery/src/client.ts:310–355

(
  model: string,
  limit: number = 100,
  userId: string | undefined = undefined,
  dataset: string = DATASET,
)

}

export async function getTracesWithoutRelabels(
  model: string,
  limit: number = 100,
  userId: string | undefined = undefined,
  dataset: string = DATASET,
) {
  const thirtyDaysAgo = new Date(Date.now() - 30 * 24 * 60 * 60 * 1000)
    .toISOString()
    .split('T')[0]

  const query = `
    SELECT t.*
    FROM \`${dataset}.${TRACES_TABLE}\` t
    LEFT JOIN (
      SELECT r.agent_step_id, r.user_id, JSON_EXTRACT_SCALAR(r.payload, '$.user_input_id') as user_input_id
      FROM \`${dataset}.${RELABELS_TABLE}\` r
      WHERE r.model = @model
      ${userId ? `AND r.user_id = @userId` : ''}
    ) r
    ON t.agent_step_id = r.agent_step_id
       AND t.user_id = r.user_id
       AND JSON_EXTRACT_SCALAR(t.payload, '$.user_input_id') = r.user_input_id
    WHERE t.type = 'get-relevant-files'
      AND t.created_at >= @thirtyDaysAgo
      AND r.agent_step_id IS NULL
      ${userId ? `AND t.user_id = @userId` : ''}
    ORDER BY t.created_at DESC
    LIMIT @limit
  `

  const [rows] = await getClient().query({
    query,
    params: {
      model,
      thirtyDaysAgo,
      limit,
      ...(userId ? { userId } : {}),
    },
  })
  // Parse the payload as JSON if it's a string
  return rows.map((row) => ({
    ...row,
    payload:
      typeof row.payload === 'string' ? JSON.parse(row.payload) : row.payload,
  })) as GetRelevantFilesTrace[]
}

export async function getTracesWithRelabels(
  model: string,