Function recordSearchEmbeddingUsage

apps/sim/lib/knowledge/embeddings.ts:418–451

(params: {
  userId: string
  workspaceId?: string | null
  embeddingModel: string
  query: string
  isBYOK: boolean
  sourceReference: string
})

 * there is no workspace to attribute to. Best-effort: never throws.
 */
export async function recordSearchEmbeddingUsage(params: {
  userId: string
  workspaceId?: string | null
  embeddingModel: string
  query: string
  isBYOK: boolean
  sourceReference: string
}): Promise<void> {
  const { userId, workspaceId, embeddingModel, query, isBYOK, sourceReference } = params
  if (isBYOK || !workspaceId) return
  try {
    const { count } = estimateTokenCount(
      query,
      getEmbeddingModelInfo(embeddingModel).tokenizerProvider
    )
    const cost = calculateCost(embeddingModel, count, 0, false)
    if (!cost || cost.total <= 0) return
    await recordUsage({
      userId,
      workspaceId,
      entries: [
        {
          category: 'model',
          source: 'knowledge-base',
          description: embeddingModel,
          cost: cost.total,
          sourceReference,
        },
      ],
    })
  } catch (error) {
    logger.warn('Failed to record search embedding usage', { error: getErrorMessage(error) })
  }
}