Function insertMessageToBigQuery

web/src/llm-api/helpers.ts:64–113

(params: {
  messageId: string
  userId: string
  startTime: Date
  request: unknown
  reasoningText: string
  responseText: string
  usageData: UsageData
  logger: Logger
  insertMessageBigquery: InsertMessageBigqueryFn
})

}

export async function insertMessageToBigQuery(params: {
  messageId: string
  userId: string
  startTime: Date
  request: unknown
  reasoningText: string
  responseText: string
  usageData: UsageData
  logger: Logger
  insertMessageBigquery: InsertMessageBigqueryFn
}) {
  const {
    messageId,
    userId,
    startTime,
    request,
    reasoningText,
    responseText,
    usageData,
    logger,
    insertMessageBigquery,
  } = params

  await setupBigQuery({ logger })
  const success = await insertMessageBigquery({
    row: {
      id: messageId,
      user_id: userId,
      finished_at: new Date(),
      created_at: startTime,
      request,
      reasoning_text: reasoningText,
      response: responseText,
      output_tokens: usageData.outputTokens,
      reasoning_tokens:
        usageData.reasoningTokens > 0 ? usageData.reasoningTokens : undefined,
      cost: usageData.cost,
      upstream_inference_cost: undefined,
      input_tokens: usageData.inputTokens,
      cache_read_input_tokens:
        usageData.cacheReadInputTokens > 0
          ? usageData.cacheReadInputTokens
          : undefined,
    },
    logger,
  })
  if (!success) {
    logger.error({ request }, 'Failed to insert message into BigQuery')
  }
}

export async function consumeCreditsForMessage(params: {
  messageId: string