Function relabelUserTraces

web/src/app/api/admin/relabel-for-user/route.ts:152–191

(params: {
  userId: string
  limit: number
  promptContext: PromptContext
})

}

async function relabelUserTraces(params: {
  userId: string
  limit: number
  promptContext: PromptContext
}): Promise<RelabelResult[]> {
  const { userId, limit, promptContext } = params
  const allResults: RelabelResult[] = []

  // Run the richer relabeling in parallel
  const fullContextRelabel = relabelUsingFullFilesForUser({
    userId,
    limit,
    promptContext,
  })

  for (const model of modelsToRelabel) {
    logger.info(`Processing traces for model ${model} and user ${userId}...`)

    const traces = await getTracesWithoutRelabels(model, limit, userId)
    logger.info(
      `Found ${traces.length} traces without relabels for model ${model}`,
    )

    const modelResults = await Promise.all(
      traces.map((trace) =>
        relabelTraceWithModel({
          trace,
          model,
          promptContext,
        }),
      ),
    )

    allResults.push(...modelResults)
  }

  await fullContextRelabel

  return allResults
}

async function relabelTraceWithModel(params: {
  trace: GetRelevantFilesTrace