Function extractUsageAndCost

web/src/llm-api/openai.ts:94–120

(
  usage: OpenAIUsage,
  modelShortName: string,
)

}

function extractUsageAndCost(
  usage: OpenAIUsage,
  modelShortName: string,
): UsageData {
  const inputTokenCost =
    INPUT_TOKEN_COSTS[modelShortName] ?? DEFAULT_INPUT_COST
  const cachedInputTokenCost =
    CACHED_INPUT_TOKEN_COSTS[modelShortName] ?? DEFAULT_CACHED_INPUT_COST
  const outputTokenCost =
    OUTPUT_TOKEN_COSTS[modelShortName] ?? DEFAULT_OUTPUT_COST

  const inTokens = usage.prompt_tokens ?? 0
  const cachedInTokens = usage.prompt_tokens_details?.cached_tokens ?? 0
  const outTokens = usage.completion_tokens ?? 0
  const cost =
    (inTokens / 1_000_000) * inputTokenCost +
    (cachedInTokens / 1_000_000) * cachedInputTokenCost +
    (outTokens / 1_000_000) * outputTokenCost

  return {
    inputTokens: inTokens,
    outputTokens: outTokens,
    cacheReadInputTokens: cachedInTokens,
    reasoningTokens: usage.completion_tokens_details?.reasoning_tokens ?? 0,
    cost,
  }
}

function extractShortModelName(model: string): string {
  return model.startsWith('openai/') ? model.slice('openai/'.length) : model