Function relabelWithRelace

web/src/app/api/admin/relabel-for-user/route.ts:333–393

(params: {
  trace: GetRelevantFilesTrace
  fileBlobs: GetExpandedFileContextForTrainingBlobTrace
  promptContext: PromptContext
})

}

async function relabelWithRelace(params: {
  trace: GetRelevantFilesTrace
  fileBlobs: GetExpandedFileContextForTrainingBlobTrace
  promptContext: PromptContext
}) {
  const { trace, fileBlobs, promptContext } = params
  logger.info(`Relabeling ${trace.id} with LLM reranker`)

  const filesWithPath = Object.entries(fileBlobs.payload.files).map(
    ([path, file]) => ({
      path,
      content: file.content,
    }),
  )

  const query = extractQueryFromMessages(trace.payload.messages)
  const prompt = [
    `A developer asked: "${query}".`,
    'Rank the following files from most relevant to least relevant for answering the request.',
    'Return only the file paths, one per line, and only include files from the list below.',
    filesWithPath.map((file) => `- ${file.path}`).join('\n'),
  ].join('\n\n')

  const ranked = unwrapPromptResult(
    await promptAiSdk({
      ...promptContext,
      model: models.openrouter_claude_sonnet_4,
      messages: [userMessage(prompt)],
      includeCacheControl: false,
    }),
  )

  const rankedFiles =
    ranked
      .split('\n')
      .map((line: string) => line.trim())
      .filter((line: string) => line.length > 0) || []

  const output =
    rankedFiles.length > 0
      ? rankedFiles.join('\n')
      : filesWithPath.map((file) => file.path).join('\n')

  const relabel: Relabel = {
    id: generateCompactId(),
    agent_step_id: trace.agent_step_id,
    user_id: trace.user_id,
    created_at: new Date(),
    model: 'relace-ranker',
    payload: {
      user_input_id: trace.payload.user_input_id,
      client_session_id: trace.payload.client_session_id,
      fingerprint_id: trace.payload.fingerprint_id,
      output,
    },
  }

  await insertRelabel({ relabel, logger })