mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-10-23 05:33:43 +00:00
26 lines
1,021 B
TypeScript
26 lines
1,021 B
TypeScript
import { ModelInfo } from "../shared/api"
|
|
|
|
export function calculateApiCost(
|
|
modelInfo: ModelInfo,
|
|
inputTokens: number,
|
|
outputTokens: number,
|
|
cacheCreationInputTokens?: number,
|
|
cacheReadInputTokens?: number,
|
|
): number {
|
|
const modelCacheWritesPrice = modelInfo.cacheWritesPrice
|
|
let cacheWritesCost = 0
|
|
if (cacheCreationInputTokens && modelCacheWritesPrice) {
|
|
cacheWritesCost = (modelCacheWritesPrice / 1_000_000) * cacheCreationInputTokens
|
|
}
|
|
const modelCacheReadsPrice = modelInfo.cacheReadsPrice
|
|
let cacheReadsCost = 0
|
|
if (cacheReadInputTokens && modelCacheReadsPrice) {
|
|
cacheReadsCost = (modelCacheReadsPrice / 1_000_000) * cacheReadInputTokens
|
|
}
|
|
const baseInputCost = ((modelInfo.inputPrice || 0) / 1_000_000) * inputTokens
|
|
const outputCost = ((modelInfo.outputPrice || 0) / 1_000_000) * outputTokens
|
|
const totalCost = cacheWritesCost + cacheReadsCost + baseInputCost + outputCost
|
|
return totalCost
|
|
}
|
|
|
|
export const parseApiPrice = (price: any) => (price ? parseFloat(price) * 1_000_000 : undefined)
|