Function updateUserStatsForWand

apps/sim/app/api/wand/route.ts:92–153

(
  billingUserId: string,
  workspaceId: string | null,
  usage: {
    prompt_tokens?: number
    completion_tokens?: number
    total_tokens?: number
  },
  requestId: string,
  isBYOK = false
)

}

async function updateUserStatsForWand(
  billingUserId: string,
  workspaceId: string | null,
  usage: {
    prompt_tokens?: number
    completion_tokens?: number
    total_tokens?: number
  },
  requestId: string,
  isBYOK = false
): Promise<void> {
  if (!isBillingEnabled) {
    return
  }

  if (!usage.total_tokens || usage.total_tokens <= 0) {
    return
  }

  try {
    const promptTokens = usage.prompt_tokens || 0
    const completionTokens = usage.completion_tokens || 0

    const modelName = useWandAzure ? wandModelName : 'gpt-4o'
    let costToStore = 0

    if (!isBYOK) {
      const pricing = getModelPricing(modelName)
      const costMultiplier = getCostMultiplier()
      let modelCost = 0

      if (pricing) {
        const inputCost = (promptTokens / 1000000) * pricing.input
        const outputCost = (completionTokens / 1000000) * pricing.output
        modelCost = inputCost + outputCost
      } else {
        modelCost = (promptTokens / 1000000) * 0.005 + (completionTokens / 1000000) * 0.015
      }

      costToStore = modelCost * costMultiplier
    }

    await recordUsage({
      userId: billingUserId,
      workspaceId: workspaceId ?? undefined,
      entries: [
        {
          category: 'model',
          source: 'wand',
          description: modelName,
          cost: costToStore,
          sourceReference: `wand:${requestId}`,
          metadata: { inputTokens: promptTokens, outputTokens: completionTokens },
        },
      ],
    })

    await checkAndBillOverageThreshold(billingUserId)