Function extractRequestMetadata

web/src/llm-api/helpers.ts:30–62

(params: {
  body: unknown
  logger: Logger
})

}

export function extractRequestMetadata(params: {
  body: unknown
  logger: Logger
}) {
  const { body, logger } = params

  const typedBody = body as ChatCompletionRequestBody | undefined
  const metadata = typedBody?.codebuff_metadata

  const rawClientId = metadata?.client_id
  const clientId = typeof rawClientId === 'string' ? rawClientId : null
  if (!clientId) {
    logger.warn(
      { request: createRequestAuditRecord(body) },
      'Received request without client_id',
    )
  }

  const rawRunId = metadata?.run_id
  const clientRequestId: string | null =
    typeof rawRunId === 'string' ? rawRunId : null
  if (!clientRequestId) {
    logger.warn(
      { request: createRequestAuditRecord(body) },
      'Received request without run_id',
    )
  }

  const n = metadata?.n
  const rawCostMode = metadata?.cost_mode
  const costMode = typeof rawCostMode === 'string' ? rawCostMode : undefined
  return { clientId, clientRequestId, costMode, ...(n && { n }) }
}

export async function insertMessageToBigQuery(params: {
  messageId: string