basant307/AI_Governance_Project
048
1/**2 * @license3 * Copyright 2025 Google LLC4 * SPDX-License-Identifier: Apache-2.05 */6 7export interface ModelPricing {8 inputPerMillionTokens?: number;9 outputPerMillionTokens?: number;10}11 12export function calculateCost(args: {13 inputTokens: number;14 outputTokens: number;15 pricing?: ModelPricing;16}): number | null {17 const { inputTokens, outputTokens, pricing } = args;18 19 if (!pricing) return null;20 21 const inputCost =22 pricing.inputPerMillionTokens != null23 ? (inputTokens / 1_000_000) * pricing.inputPerMillionTokens24 : 0;25 26 const outputCost =27 pricing.outputPerMillionTokens != null28 ? (outputTokens / 1_000_000) * pricing.outputPerMillionTokens29 : 0;30 31 const total = inputCost + outputCost;32 return total > 0 ? total : null;33}34 