Function insertChatCompletionTraceBigquery

packages/bigquery/src/client.ts:117–162

({
  row,
  dataset,
  logger,
}: {
  row: ChatCompletionTraceRow
  dataset?: string
  logger: Logger
})

}

export async function insertChatCompletionTraceBigquery({
  row,
  dataset,
  logger,
}: {
  row: ChatCompletionTraceRow
  dataset?: string
  logger: Logger
}) {
  const resolvedDataset = dataset ?? DATASET
  try {
    await getClient()
      .dataset(resolvedDataset)
      .table(CHAT_COMPLETION_TRACES_TABLE)
      .insert({
        ...row,
        request: JSON.stringify(row.request),
        messages: JSON.stringify(row.messages),
        delta_message_hashes: JSON.stringify(row.delta_message_hashes),
        tools: row.tools ? JSON.stringify(row.tools) : null,
      })

    logger.debug(
      {
        traceId: row.id,
        userId: row.user_id,
        clientId: row.client_id,
        traceSessionId: row.trace_session_id,
        traceLineageId: row.trace_lineage_id,
        runId: row.run_id,
        messageStartIndex: row.message_start_index,
        messageDeltaCount: row.message_delta_count,
        fullSnapshot: row.full_snapshot,
      },
      'Inserted chat completion trace into BigQuery',
    )
    return true
  } catch (error) {
    logger.error(
      { error: getErrorObject(error), traceId: row.id },
      'Failed to insert chat completion trace into BigQuery',
    )

    return false
  }
}

export async function insertMessageBigquery({
  row,